awesome/lib/lain/widget/alsabar.lua

--[[

     Licensed under GNU General Public License v2
      * (c) 2013, Luca CPZ
      * (c) 2013, Rman

--]]

local helpers = require('lain.helpers')
local awful = require('awful')
local naughty = require('naughty')
local wibox = require('wibox')
local math = math
local string = string
local type = type
local tonumber = tonumber

-- ALSA volume bar
-- lain.widget.alsabar

local function factory(args)
	local alsabar = {
		colors = {
			background = '#000000',
			mute = '#EB8F8F',
			unmute = '#A4CE8A',
		},

		_current_level = 0,
		_playback = 'off',
	}

	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 margins = args.margins or 1
	local ticks = args.ticks or false
	local ticks_size = args.ticks_size or 7
	local tick = args.tick or '|'
	local tick_pre = args.tick_pre or '['
	local tick_post = args.tick_post or ']'
	local tick_none = args.tick_none or ' '

	alsabar.cmd = args.cmd or 'amixer'
	alsabar.channel = args.channel or 'Master'
	alsabar.togglechannel = args.togglechannel
	alsabar.colors = args.colors or alsabar.colors
	alsabar.followtag = args.followtag or false
	alsabar.notification_preset = args.notification_preset

	if not alsabar.notification_preset then
		alsabar.notification_preset = { font = 'Monospace 10' }
	end

	local format_cmd = string.format('%s get %s', alsabar.cmd, alsabar.channel)

	if alsabar.togglechannel then
		format_cmd = {
			awful.util.shell,
			'-c',
			string.format('%s get %s; %s get %s', alsabar.cmd, alsabar.channel, alsabar.cmd, alsabar.togglechannel),
		}
	end

	alsabar.bar = wibox.widget({
		color = alsabar.colors.unmute,
		background_color = alsabar.colors.background,
		forced_height = height,
		forced_width = width,
		margins = margins,
		paddings = margins,
		ticks = ticks,
		ticks_size = ticks_size,
		widget = wibox.widget.progressbar,
	})

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

	function alsabar.update(callback)
		helpers.async(format_cmd, function(mixer)
			local vol, playback = string.match(mixer, '([%d]+)%%.*%[([%l]*)')

			if not vol or not playback then
				return
			end

			if vol ~= alsabar._current_level or playback ~= alsabar._playback then
				alsabar._current_level = tonumber(vol)
				alsabar.bar:set_value(alsabar._current_level / 100)
				if alsabar._current_level == 0 or playback == 'off' then
					alsabar._playback = playback
					alsabar.tooltip:set_text('[Muted]')
					alsabar.bar.color = alsabar.colors.mute
				else
					alsabar._playback = 'on'
					alsabar.tooltip:set_text(string.format('%s: %s', alsabar.channel, vol))
					alsabar.bar.color = alsabar.colors.unmute
				end

				volume_now = {
					level = alsabar._current_level,
					status = alsabar._playback,
				}

				settings()

				if type(callback) == 'function' then
					callback()
				end
			end
		end)
	end

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

			preset.title = string.format('%s - %s%%', alsabar.channel, alsabar._current_level)

			if alsabar._playback == 'off' then
				preset.title = preset.title .. ' Muted'
			end

			-- tot is the maximum number of ticks to display in the notification
			local tot = alsabar.notification_preset.max_ticks

			if not tot then
				local wib = awful.screen.focused().mywibox
				-- if we can grab mywibox, tot is defined as its height if
				-- horizontal, or width otherwise
				if wib then
					if wib.position == 'left' or wib.position == 'right' then
						tot = wib.width
					else
						tot = wib.height
					end
				-- fallback: default horizontal wibox height
				else
					tot = 20
				end
			end

			local int = math.modf((alsabar._current_level / 100) * tot)
			preset.text =
				string.format('%s%s%s%s', tick_pre, string.rep(tick, int), string.rep(tick_none, tot - int), tick_post)

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

			if not alsabar.notification then
				alsabar.notification = naughty.notify({
					preset = preset,
					destroy = function()
						alsabar.notification = nil
					end,
				})
			else
				naughty.replace_text(alsabar.notification, preset.title, preset.text)
			end
		end)
	end

	helpers.newtimer(string.format('alsabar-%s-%s', alsabar.cmd, alsabar.channel), timeout, alsabar.update)

	return alsabar
end

return factory