From 7c6e3658c752c99382bcf0268719e8c2d3f49979 Mon Sep 17 00:00:00 2001
From: Manos Pitsidianakis <el13635@mail.ntua.gr>
Date: Sat, 12 Sep 2020 21:33:25 +0300
Subject: [PATCH] melib/imap: try NOOPing in connect()

---
 melib/src/backends/imap/connection.rs | 20 ++++++++++++++++++--
 1 file changed, 18 insertions(+), 2 deletions(-)

diff --git a/melib/src/backends/imap/connection.rs b/melib/src/backends/imap/connection.rs
index 877347dce..64c5364f4 100644
--- a/melib/src/backends/imap/connection.rs
+++ b/melib/src/backends/imap/connection.rs
@@ -526,8 +526,23 @@ impl ImapConnection {
                     self.stream = Err(err);
                 }
             }
-            if self.stream.is_ok() {
-                return Ok(());
+            if debug!(self.stream.is_ok()) {
+                let mut ret = String::new();
+                if let Err(err) = try_await(async {
+                    self.send_command(b"NOOP").await?;
+                    self.read_response(&mut ret, RequiredResponses::empty())
+                        .await
+                })
+                .await
+                {
+                    debug!("connect(): connection is probably dead: {:?}", &err);
+                } else {
+                    debug!(
+                        "connect(): connection is probably alive, NOOP returned {:?}",
+                        &ret
+                    );
+                    return Ok(());
+                }
             }
             let new_stream = debug!(ImapStream::new_connection(&self.server_conf).await);
             if let Err(err) = new_stream.as_ref() {
@@ -714,6 +729,7 @@ impl ImapConnection {
             }
             Err(err)
         } else {
+            *self.uid_store.is_online.lock().unwrap() = (Instant::now(), Ok(()));
             Ok(())
         }
     }