---------------------------------------------------
-- Licensed under the GNU General Public License v2
--  * (c) 2010, Adrian C. <anrxc@sysphere.org>
--  * (c) org-awesome, Damien Leone
---------------------------------------------------

-- {{{ Grab environment
local io = { lines = io.lines }
local setmetatable = setmetatable
local string = { find = string.find }
local os = {
    time = os.time,
    date = os.date
}
-- }}}


-- Org: provides agenda statistics for Emacs org-mode
-- vicious.widgets.org
local org_all = {}


-- {{{ OrgMode widget type
local function worker(format, warg)
    if not warg then return end

    -- Compute delays
    local today  = os.time{ year=os.date("%Y"), month=os.date("%m"), day=os.date("%d") }
    local soon   = today + 24 * 3600 * 3 -- 3 days ahead is close
    local future = today + 24 * 3600 * 7 -- 7 days ahead is maximum

    -- Initialize counters
    local count = { past = 0, today = 0, soon = 0, future = 0 }

    -- Get data from agenda files
    for i=1, #warg do
       for line in io.lines(warg[i]) do
          local scheduled = string.find(line, "SCHEDULED:")
          local closed    = string.find(line, "CLOSED:")
          local deadline  = string.find(line, "DEADLINE:")

          if (scheduled and not closed) or (deadline and not closed) then
             local b, e, y, m, d = string.find(line, "(%d%d%d%d)-(%d%d)-(%d%d)")

             if b then
                local  t = os.time{ year = y, month = m, day = d }

                if     t <  today  then count.past   = count.past   + 1
                elseif t == today  then count.today  = count.today  + 1
                elseif t <= soon   then count.soon   = count.soon   + 1
                elseif t <= future then count.future = count.future + 1
                end
             end
          end
       end
    end

    return {count.past, count.today, count.soon, count.future}
end
-- }}}

return setmetatable(org_all, { __call = function(_, ...) return worker(...) end })