From d45f0b724a1781d1ad0458ce929c9920df5dc20d Mon Sep 17 00:00:00 2001
From: Cris Ward <cris@duodesign.co.uk>
Date: Wed, 30 Aug 2017 21:45:23 +0100
Subject: [PATCH] added imap status

---
 README.md   |  4 ++--
 src/imap.cr | 27 +++++++++++++++++++++++++++
 2 files changed, 29 insertions(+), 2 deletions(-)

diff --git a/README.md b/README.md
index f1d6d69..da7d48b 100644
--- a/README.md
+++ b/README.md
@@ -22,8 +22,8 @@ mailboxes = imap.get_mailboxes
 if mailboxes.size > 0
   mailbox = mailboxes[0]
   imap.select(mailbox)
-  message_count = imap.get_message_count
-  puts "There are #{message_count} message in #{mailbox}"
+  status = imap.status(mailbox, ["MESSAGES", "UNSEEN"])
+  puts "There are #{status["MESSAGES"]} message in #{mailbox} #{status["UNSEEN"]} unread."
 end
 imap.close
 ```
diff --git a/src/imap.cr b/src/imap.cr
index 39ffd17..e41d8f2 100644
--- a/src/imap.cr
+++ b/src/imap.cr
@@ -105,6 +105,33 @@ module Imap
       return 0
     end
 
+    # Sends a STATUS command, and returns the status of the indicated
+    # `mailbox`. `attr` is a list of one or more attributes whose
+    # statuses are to be requested.  Supported attributes include:
+    #
+    #   * MESSAGES:: the number of messages in the mailbox.
+    #   * RECENT:: the number of recent messages in the mailbox.
+    #   * UNSEEN:: the number of unseen messages in the mailbox.
+    #
+    # The return value is a hash of attributes. For example:
+    # ```
+    #   p imap.status("inbox", ["MESSAGES", "RECENT"])
+    #   #=> {"RECENT"=>0, "MESSAGES"=>44}
+    #```
+    def status(mailbox, attr : Array(String))
+      param = "(#{attr.join(" ")})"
+      res = command("STATUS", mailbox, param)
+      vals = Hash(String, Int32).new
+      counts = res[0].match(/\(([^)]+)\)/)
+      if counts && counts[1]
+        counts[1].scan(/\w+ \d+/) do |match|
+          key, value = match[0].to_s.split(" ", 2)
+          vals[key] = value.to_i
+        end
+      end
+      return vals
+    end
+
     private def process_mail_headers(res)
       ip = nil
       from = nil