widget.imap: ensure mailcount is not nil before notifying; closes #400

diff --git a/widget/imap.lua b/widget/imap.lua
index 93aced95495997b0272825ddedc2e62f4fb80471..7e83d36e46d798debfcfe30dcc69fd19e1c3cfde 100644 (file)
--- a/widget/imap.lua
+++ b/widget/imap.lua
@@ -54,20 +54,17 @@ local function factory(args)
             mail_notification_preset.screen = awful.screen.focused()
         end
 
             mail_notification_preset.screen = awful.screen.focused()
         end
 

-        curl = string.format("%s --url imaps://%s:%s/INBOX -u %s:%q %s -k",
-               head_command, server, port, mail, password, request)
+        local curl = string.format("%s --url imaps://%s:%s/INBOX -u %s:%q %s -k",
+                     head_command, server, port, mail, password, request)

 
         helpers.async(curl, function(f)
             mailcount = tonumber(f:match("UNSEEN (%d+)"))
             widget = imap.widget
             settings()
 
 
         helpers.async(curl, function(f)
             mailcount = tonumber(f:match("UNSEEN (%d+)"))
             widget = imap.widget
             settings()
 

-            if notify == "on" and mailcount >= 1 and mailcount > helpers.get_map(mail) then
-                if mailcount == 1 then
-                    nt = mail .. " has one new message"
-                else
-                    nt = mail .. " has <b>" .. mailcount .. "</b> new messages"
-                end
+            if notify == "on" and mailcount and mailcount >= 1 and mailcount > helpers.get_map(mail) then
+                local nt = mail .. " has <b>" .. mailcount .. "</b> new message"
+                if mailcount >= 1 then nt = nt .. "s" end

                 naughty.notify { preset = mail_notification_preset, text = nt }
             end
 
                 naughty.notify { preset = mail_notification_preset, text = nt }
             end