widgets/imap.lua

   1
   2 --[[
   3
   4      Licensed under GNU General Public License v2
   5       * (c) 2013, Luke Bonham
   6
   7 --]]
   8
   9 local helpers      = require("lain.helpers")
  10 local async        = require("lain.asyncshell")
  11
  12 local naughty      = require("naughty")
  13 local wibox        = require("wibox")
  14
  15 local string       = { format = string.format,
  16                        gsub   = string.gsub }
  17 local tonumber     = tonumber
  18
  19 local setmetatable = setmetatable
  20
  21 -- Mail IMAP check
  22 -- lain.widgets.imap
  23
  24 local function worker(args)
  25     local imap     = {}
  26     local args     = args or {}
  27
  28     local server   = args.server
  29     local mail     = args.mail
  30     local password = args.password
  31
  32     local port     = args.port or 993
  33     local timeout  = args.timeout or 60
  34     local is_plain = args.is_plain or false
  35     local settings = args.settings or function() end
  36
  37     local head_command  = "curl --connect-timeout 3 -fsm 3"
  38     local request = "-X 'SEARCH (UNSEEN)'"
  39
  40     helpers.set_map(mail, 0)
  41
  42     if not is_plain
  43     then
  44         local f = io.popen(password)
  45         password = f:read("*a"):gsub("\n", "")
  46         f:close()
  47     end
  48
  49     imap.widget = wibox.widget.textbox('')
  50
  51     function update()
  52         mail_notification_preset = {
  53             icon     = helpers.icons_dir .. "mail.png",
  54             position = "top_left"
  55         }
  56
  57         curl = string.format("%s --url imaps://%s:%s/INBOX -u %s:%s %s -k",
  58                head_command, server, port, mail, password, request)
  59
  60         async.request(curl, function(f)
  61             ws = f:read("*a")
  62             f:close()
  63
  64             _, mailcount = string.gsub(ws, "%d+", "")
  65             _ = nil
  66
  67             widget = imap.widget
  68             settings()
  69
  70             if mailcount >= 1 and mailcount > helpers.get_map(mail)
  71             then
  72                 if mailcount == 1 then
  73                     nt = mail .. " has one new message"
  74                 else
  75                     nt = mail .. " has <b>" .. mailcount .. "</b> new messages"
  76                 end
  77                 naughty.notify({
  78                     preset = mail_notification_preset,
  79                     text = nt,
  80                     screen = client.focus and client.focus.screen or 1
  81                 })
  82             end
  83
  84             helpers.set_map(mail, mailcount)
  85         end)
  86
  87     end
  88
  89     helpers.newtimer(mail, timeout, update, true)
  90
  91     return setmetatable(imap, { __index = imap.widget })
  92 end
  93
  94 return setmetatable({}, { __call = function(_, ...) return worker(...) end })