X-Git-Url: https://git.madduck.net/etc/awesome.git/blobdiff_plain/ad1c15b7c11d779eec233edc5df8a1e1f874fd63..4b634be781379140002b191b04e167a4460ae5f4:/widgets/imap.lua

diff --git a/widgets/imap.lua b/widgets/imap.lua
index febff23..c404032 100644
--- a/widgets/imap.lua
+++ b/widgets/imap.lua
@@ -20,9 +20,9 @@ local setmetatable = setmetatable
 
 -- Mail IMAP check
 -- lain.widgets.imap
-local imap = {}
 
 local function worker(args)
+    local imap     = {}
     local args     = args or {}
 
     local server   = args.server
@@ -74,14 +74,18 @@ local function worker(args)
             else
                 nt = mail .. " has <b>" .. mailcount .. "</b> new messages"
             end
-            naughty.notify({ preset = mail_notification_preset, text = nt })
+            naughty.notify({
+                preset = mail_notification_preset,
+                text = nt,
+                screen = client.focus and client.focus.screen or 1
+            })
         end
 
         helpers.set_map(mail, mailcount)
     end
 
     helpers.newtimer(mail, timeout, update, true)
-    return imap.widget
+    return setmetatable(imap, { __index = imap.widget })
 end
 
-return setmetatable(imap, { __call = function(_, ...) return worker(...) end })
+return setmetatable({}, { __call = function(_, ...) return worker(...) end })