Initial commit

lib/lain/widget/pulsebar.lua

@@ -0,0 +1,185 @@
+--[[
+
+     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
+
+-- PulseAudio volume bar
+-- lain.widget.pulsebar
+
+local function factory(args)
+	local pulsebar = {
+		colors = {
+			background = '#000000',
+			mute_background = '#000000',
+			mute = '#EB8F8F',
+			unmute = '#A4CE8A',
+		},
+
+		_current_level = 0,
+		_mute = 'no',
+		device = 'N/A',
+	}
+
+	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 paddings = args.paddings 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 ' '
+
+	pulsebar.colors = args.colors or pulsebar.colors
+	pulsebar.followtag = args.followtag or false
+	pulsebar.notification_preset = args.notification_preset
+	pulsebar.devicetype = args.devicetype or 'sink'
+	pulsebar.cmd = args.cmd
+		or 'pacmd list-'
+			.. pulsebar.devicetype
+			.. "s | sed -n -e '/*/,$!d' -e '/index/p' -e '/base volume/d' -e '/volume:/p' -e '/muted:/p' -e '/device\\.string/p'"
+
+	if not pulsebar.notification_preset then
+		pulsebar.notification_preset = {
+			font = 'Monospace 10',
+		}
+	end
+
+	pulsebar.bar = wibox.widget({
+		color = pulsebar.colors.unmute,
+		background_color = pulsebar.colors.background,
+		forced_height = height,
+		forced_width = width,
+		margins = margins,
+		paddings = paddings,
+		ticks = ticks,
+		ticks_size = ticks_size,
+		widget = wibox.widget.progressbar,
+	})
+
+	pulsebar.tooltip = awful.tooltip({ objects = { pulsebar.bar } })
+
+	function pulsebar.update(callback)
+		helpers.async(
+			{ awful.util.shell, '-c', type(pulsebar.cmd) == 'string' and pulsebar.cmd or pulsebar.cmd() },
+			function(s)
+				volume_now = {
+					index = string.match(s, 'index: (%S+)') or 'N/A',
+					device = string.match(s, 'device.string = "(%S+)"') or 'N/A',
+					muted = string.match(s, 'muted: (%S+)') or 'N/A',
+				}
+
+				pulsebar.device = volume_now.index
+
+				local ch = 1
+				volume_now.channel = {}
+				for v in string.gmatch(s, ':.-(%d+)%%') do
+					volume_now.channel[ch] = v
+					ch = ch + 1
+				end
+
+				volume_now.left = volume_now.channel[1] or 'N/A'
+				volume_now.right = volume_now.channel[2] or 'N/A'
+
+				local volu = volume_now.left
+				local mute = volume_now.muted
+
+				if volu:match('N/A') or mute:match('N/A') then
+					return
+				end
+
+				if volu ~= pulsebar._current_level or mute ~= pulsebar._mute then
+					pulsebar._current_level = tonumber(volu)
+					pulsebar.bar:set_value(pulsebar._current_level / 100)
+					if pulsebar._current_level == 0 or mute == 'yes' then
+						pulsebar._mute = mute
+						pulsebar.tooltip:set_text('[muted]')
+						pulsebar.bar.color = pulsebar.colors.mute
+						pulsebar.bar.background_color = pulsebar.colors.mute_background
+					else
+						pulsebar._mute = 'no'
+						pulsebar.tooltip:set_text(
+							string.format('%s %s: %s', pulsebar.devicetype, pulsebar.device, volu)
+						)
+						pulsebar.bar.color = pulsebar.colors.unmute
+						pulsebar.bar.background_color = pulsebar.colors.background
+					end
+
+					settings()
+
+					if type(callback) == 'function' then
+						callback()
+					end
+				end
+			end
+		)
+	end
+
+	function pulsebar.notify()
+		pulsebar.update(function()
+			local preset = pulsebar.notification_preset
+
+			preset.title = string.format('%s %s - %s%%', pulsebar.devicetype, pulsebar.device, pulsebar._current_level)
+
+			if pulsebar._mute == 'yes' then
+				preset.title = preset.title .. ' muted'
+			end
+
+			-- tot is the maximum number of ticks to display in the notification
+			-- fallback: default horizontal wibox height
+			local wib, tot = awful.screen.focused().mywibox, 20
+
+			-- 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
+			end
+
+			local int = math.modf((pulsebar._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 pulsebar.followtag then
+				preset.screen = awful.screen.focused()
+			end
+
+			if not pulsebar.notification then
+				pulsebar.notification = naughty.notify({
+					preset = preset,
+					destroy = function()
+						pulsebar.notification = nil
+					end,
+				})
+			else
+				naughty.replace_text(pulsebar.notification, preset.title, preset.text)
+			end
+		end)
+	end
+
+	helpers.newtimer(string.format('pulsebar-%s-%s', pulsebar.devicetype, pulsebar.device), timeout, pulsebar.update)
+
+	return pulsebar
+end
+
+return factory