X-Git-Url: https://git.madduck.net/etc/awesome.git/blobdiff_plain/eb8cec907a50f3fd6df4b14ae10b910444017f92..5d892e0ceb9d4dece6d2d61b2e5c4046bbc91e9f:/widgets/maildir.lua?ds=inline

diff --git a/widgets/maildir.lua b/widgets/maildir.lua
index 00ab771..7d4ca28 100644
--- a/widgets/maildir.lua
+++ b/widgets/maildir.lua
@@ -7,89 +7,87 @@
                                                   
 --]]
 
-local newtimer        = require("lain.helpers").newtimer
+local spawn            = require("awful").spawn
+local wibox            = require("wibox")
 
-local wibox           = require("wibox")
+local element_in_table = require("lain.helpers").element_in_table
+local newtimer         = require("lain.helpers").newtimer
+local read_pipe        = require("lain.helpers").read_pipe
+local spairs           = require("lain.helpers").spairs
 
-local io              = io
-local os              = { getenv = os.getenv }
-local pairs           = pairs
-local string          = { len    = string.len,
-                          match  = string.match }
-local table           = { sort   = table.sort }
+local io               = { popen  = io.popen }
+local os               = { getenv = os.getenv }
+local pairs            = pairs
+local string           = { format = string.format,
+                           len    = string.len,
+                           match  = string.match }
 
-local setmetatable    = setmetatable
+local setmetatable     = setmetatable
 
 -- Maildir check
 -- lain.widgets.maildir
 local maildir = {}
 
-function worker(args)
+local function worker(args)
     local args         = args or {}
     local timeout      = args.timeout or 60
     local mailpath     = args.mailpath or os.getenv("HOME") .. "/Mail"
     local ignore_boxes = args.ignore_boxes or {}
     local settings     = args.settings or function() end
+    local ext_mail_cmd = args.external_mail_cmd
 
-    widget = wibox.widget.textbox('')
+    maildir.widget = wibox.widget.textbox('')
 
     function update()
+        if ext_mail_cmd then awful.spawn(ext_mail_cmd) end
+
         -- Find pathes to mailboxes.
-        local p = io.popen("find " .. mailpath ..
-                           " -mindepth 1 -maxdepth 1 -type d" ..
-                           " -not -name .git")
+        local p = io.popen(string.format("find %s -mindepth 1 -maxdepth 2 -type d -not -name .git", mailpath))
         local boxes = {}
         repeat
             line = p:read("*l")
-            if line ~= nil
-            then
+            if line then
                 -- Find all files in the "new" subdirectory. For each
                 -- file, print a single character (no newline). Don't
                 -- match files that begin with a dot.
                 -- Afterwards the length of this string is the number of
                 -- new mails in that box.
-                local np = io.popen("find " .. line ..
-                                    "/new -mindepth 1 -type f " ..
-                                    "-not -name '.*' -printf a")
-                local mailstring = np:read("*all")
+                local mailstring = read_pipe(string.format("find %s /new -mindepth 1 -type f -not -name '.*' -printf a", line))
 
                 -- Strip off leading mailpath.
-                local box = string.match(line, mailpath .. "/*([^/]+)")
+                local box      = string.match(line, mailpath .. "/(.*)")
                 local nummails = string.len(mailstring)
-                if nummails > 0
-                then
+
+                if nummails > 0 then
                     boxes[box] = nummails
                 end
             end
-        until line == nil
+        until not line
+        p:close()
 
-        table.sort(boxes)
+        local newmail = "no mail"
+        local total = 0
 
-        newmail = "no mail"
-
-        local count = 0
-        for box, number in pairs(boxes)
-        do
-            count = count + 1
+        for box, number in spairs(boxes) do
             -- Add this box only if it's not to be ignored.
-            if not util.element_in_table(box, ignore_boxes)
-            then
-                if newmail == ""
-                then
-                    newmail = box .. "(" .. number .. ")"
+            if not util.element_in_table(box, ignore_boxes) then
+                total = total + number
+                if newmail == "no mail" then
+                    newmail = string.format("%s(%s)", box, number)
                 else
-                    newmail = newmail .. ", " ..
-                              box .. "(" .. number .. ")"
+                    newmail = string.format("%s, %s(%s)", newmail, box, number)
                 end
             end
         end
 
+        widget = maildir.widget
+
         settings()
     end
 
     newtimer(mailpath, timeout, update, true)
 
-    return widget
+    return setmetatable(maildir, { __index = maildir.widget })
 end
 
 return setmetatable(maildir, { __call = function(_, ...) return worker(...) end })