-- contrib/batproc_linux.lua
-- Copyright (C) 2010  Adrian C. <anrxc@sysphere.org>
-- Copyright (C) 2017  Jörg Thalheim <joerg@higgsboson.tk>
--
-- This file is part of Vicious.
--
-- Vicious is free software: you can redistribute it and/or modify
-- it under the terms of the GNU General Public License as
-- published by the Free Software Foundation, either version 2 of the
-- License, or (at your option) any later version.
--
-- Vicious is distributed in the hope that it will be useful,
-- but WITHOUT ANY WARRANTY; without even the implied warranty of
-- MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
-- GNU General Public License for more details.
--
-- You should have received a copy of the GNU General Public License
-- along with Vicious.  If not, see <https://www.gnu.org/licenses/>.

-- {{{ Grab environment
local tonumber = tonumber
local io = { open = io.open }
local setmetatable = setmetatable
local math = {
    min = math.min,
    floor = math.floor
}
local string = {
    find = string.find,
    match = string.match,
    format = string.format
}
-- }}}


-- Batproc: provides state, charge, and remaining time for a requested battery using procfs
-- vicious.contrib.batproc
local batproc_linux = {}


-- {{{ Battery widget type
local function worker(format, batid)
    local battery_state = {
        ["full"] = "↯",
        ["unknown"] = "⌁",
        ["charged"] = "↯",
        ["charging"] = "+",
        ["discharging"] = "-"
    }

    -- Get /proc/acpi/battery info
    local f = io.open("/proc/acpi/battery/"..batid.."/info")
    -- Handler for incompetent users
    if not f then return {battery_state["unknown"], 0, "N/A"} end
    local infofile = f:read("*all")
    f:close()

    -- Check if the battery is present
    if infofile == nil or string.find(infofile, "present:[%s]+no") then
        return {battery_state["unknown"], 0, "N/A"}
    end

    -- Get capacity information
    local capacity = string.match(infofile, "last full capacity:[%s]+([%d]+).*")


    -- Get /proc/acpi/battery state
    local f = io.open("/proc/acpi/battery/"..batid.."/state")
    local statefile = f:read("*all")
    f:close()

    -- Get state information
    local state = string.match(statefile, "charging state:[%s]+([%a]+).*")
    local state = battery_state[state] or battery_state["unknown"]

    -- Get charge information
    local rate = string.match(statefile, "present rate:[%s]+([%d]+).*")
    local remaining = string.match(statefile, "remaining capacity:[%s]+([%d]+).*")


    -- Calculate percentage (but work around broken BAT/ACPI implementations)
    local percent = math.min(math.floor(remaining / capacity * 100), 100)

    -- Calculate remaining (charging or discharging) time
    if state == "+" then
        timeleft = (tonumber(capacity) - tonumber(remaining)) / tonumber(rate)
    elseif state == "-" then
        timeleft = tonumber(remaining) / tonumber(rate)
    else
        return {state, percent, "N/A"}
    end
    local hoursleft = math.floor(timeleft)
    local minutesleft = math.floor((timeleft - hoursleft) * 60 )
    local time = string.format("%02d:%02d", hoursleft, minutesleft)

    return {state, percent, time}
end
-- }}}

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