--[[
                                                  
     Licensed under GNU General Public License v2 
      * (c) 2013, Luke Bonham                     
      * (c) 2013, Rman                            
                                                  
--]]

local helpers      = require("lain.helpers")
local awful        = require("awful")
local naughty      = require("naughty")
local wibox        = require("wibox")
local math         = { modf   = math.modf }
local string       = { format = string.format,
                       match  = string.match,
                       rep    = string.rep }
local tonumber     = tonumber
local setmetatable = setmetatable

-- ALSA volume bar
-- lain.widgets.alsabar
local alsabar = {
    colors = {
        background = "#000000",
        mute       = "#EB8F8F",
        unmute     = "#A4CE8A"
    },

    _current_level = 0,
    _muted         = false
}

local function worker(args)
    local args       = args or {}
    local timeout    = args.timeout or 5
    local settings   = args.settings or function() end
    local width      = args.width or 63
    local height     = args.height or 1
    local ticks      = args.ticks or false
    local ticks_size = args.ticks_size or 7
    local vertical   = args.vertical or false

    alsabar.cmd                 = args.cmd or "amixer"
    alsabar.channel             = args.channel or "Master"
    alsabar.colors              = args.colors or alsabar.colors
    alsabar.followtag           = args.followtag or false
    alsabar._notify             = args.notify or "on"
    alsabar.notification_preset = args.notification_preset

    if not alsabar.notification_preset then
        alsabar.notification_preset      = naughty.config.defaults
        alsabar.notification_preset.font = "Monospace 11"
    end

    alsabar.bar = wibox.widget {
        forced_height    = height,
        forced_width     = width,
        color            = alsabar.colors.unmute,
        background_color = alsabar.colors.background,
        margins          = 1,
        paddings         = 1,
        ticks            = ticks,
        ticks_size       = ticks_size,
        widget           = wibox.widget.progressbar,
        layout           = vertical and wibox.container.rotate
    }

    alsabar.tooltip = awful.tooltip({ objects = { alsabar.bar } })

    function alsabar.update()
        helpers.async(alsabar.cmd, function(mixer)
            local volu,mute = string.match(mixer, "([%d]+)%%.*%[([%l]*)")
            if (volu and tonumber(volu) ~= alsabar._current_level) or (mute and string.match(mute, "on") ~= alsabar._muted) then
                alsabar._current_level = tonumber(volu) or alsabar._current_level
                alsabar.bar:set_value(alsabar._current_level / 100)
                if (not mute and tonumber(volu) == 0) or mute == "off" then
                    alsabar._muted = true
                    alsabar.tooltip:set_text ("[Muted]")
                    alsabar.bar.color = alsabar.colors.mute
                else
                    alsabar._muted = false
                    alsabar.tooltip:set_text(string.format("%s: %s", alsabar.channel, volu))
                    alsabar.bar.color = alsabar.colors.unmute
                end

                volume_now = {}
                volume_now.level = tonumber(volu)
                volume_now.status = mute

                settings()
            end
        end)
    end

    function alsabar.notify()
        alsabar.update(function()
            local preset = alsabar.notification_preset

            if alsabar._muted then
                preset.title = string.format("%s - Muted", alsabar.channel)
            else
                preset.title = string.format("%s - %s%%", alsabar.channel, alsabar._current_level)
            end

            int = math.modf((alsabar._current_level / 100) * awful.screen.focused().mywibox.height)
            preset.text = string.format("[%s%s]", string.rep("|", int),
                          string.rep(" ", awful.screen.focused().mywibox.height - int))

            if alsabar.followtag then preset.screen = awful.screen.focused() end

            alsabar.id = naughty.notify ({
                replaces_id = alsabar.id,
                preset      = preset
            }).id
        end)
    end

    timer_id = string.format("alsabar-%s-%s", alsabar.cmd, alsabar.channel)

    helpers.newtimer(timer_id, timeout, alsabar.update)

    return alsabar
end

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