X-Git-Url: https://git.madduck.net/etc/awesome.git/blobdiff_plain/0ef78642276735127d2206be184c1e071ad6b0ed..c02fcdb2e9480927f7b0b21f38d20dd203b7d8bd:/widget/imap.lua

diff --git a/widget/imap.lua b/widget/imap.lua
index 84a1cd0..7e83d36 100644
--- a/widget/imap.lua
+++ b/widget/imap.lua
@@ -1,16 +1,15 @@
-
 --[[
-                                                  
-     Licensed under GNU General Public License v2 
-      * (c) 2013, Luke Bonham                     
-                                                  
+
+     Licensed under GNU General Public License v2
+      * (c) 2013, Luca CPZ
+
 --]]
 
 local helpers  = require("lain.helpers")
 local naughty  = require("naughty")
 local wibox    = require("wibox")
-local string   = { format = string.format,
-                   gsub   = string.gsub }
+local awful    = require("awful")
+local string   = string
 local type     = type
 local tonumber = tonumber
 
@@ -55,23 +54,18 @@ local function factory(args)
             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 = string.gsub(f, "%d+", "")
-            _ = nil
-
+            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
-                naughty.notify({ preset = mail_notification_preset, text = nt })
+            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
 
             helpers.set_map(mail, mailcount)