systray_hints/init.lua

234 lines
7.1 KiB
Lua

-- S Y S T R A Y H I N T S
-- rts/oarion7 - ryanthomas.org
-- Control the awesomewm systray from the keyboard using vimium-like
-- number hints. Developed and tested on awesome v4.3.
local awful = require("awful")
local gears = require("gears")
local b = require("beautiful")
local wibox = require("wibox")
local s
awful.screen.connect_for_each_screen(function(screen)
if screen.systray then s = screen end
end)
if s == nil then return nil end
local systray_hints = {
font = b.systray_hints_font or b.taglist_font or b.font,
bgcolor = b.systray_hints_bg or b.taglist_bg_occupied or "#55465a",
highlight = b.systray_hints_bg_highlight or "#aa53aa",
highlight_alt = b.systray_hints_bg_highlight_alt or "#426f5a",
color = b.systray_hints_fg or b.taglist_fg_occupied or "#fdf6e3",
bordercolor = b.systray_hints_border or "#fdf6e333",
spacing = 1,
mouse_buttons = { "Left", "Up", "Right" },
default_button = 3,
popup = popup,
systray = s.systray,
wibox = s.mywibox, --wibox in which to locate the system tray
run = run,
}
local total
local was_hidden
local icon_width
local icons_x
local icons_y
local function delay(time, cmd)
gears.timer( { timeout = time, autostart = true, single_shot = true,
callback = function () cmd () end, } )
end
local function execute(choice, mouse_button)
local saved
local factor
local target
saved = mouse.coords({x = x, y = y})
if choice == 1 then factor = 0 else factor = choice - 1 end
target = icons_x + ( icon_width * factor )
mouse.coords { x = target , y = icons_y }
if mouse_button ~= 2 then
root.fake_input("button_press" , tostring(mouse_button))
root.fake_input("button_release", tostring(mouse_button))
delay(0.05, function () mouse.coords({x = saved.x, y = saved.y}, true) end)
end
if systray_hints.popup then systray_hints.popup.visible = false end
end
local function highlight_multidigits(total)
local color
for i = 9, total do
color = systray_hints.highlight_alt
if i == 9 then
i = 1
color = systray_hints.highlight
end
systray_hints.popup.widget:get_children()[1]:get_children()[i].widget:set_bg(color)
end
end
local function get_key_input(total)
local grabber
local mouse_button
local function conc(n) return tonumber( 1 .. n ) end
mouse_button = systray_hints.default_button
grabber = awful.keygrabber {
mask_modkeys = true,
autostart = true,
keypressed_callback = function(self, mod, key, cmd)
if key == '1' and total > 9 then
if systray_hints.popup then
highlight_multidigits(total)
end
grabber.keypressed_callback = function(self, mod, key, cmd)
if key == "Return" then
execute(1, mouse_button)
grabber:stop()
elseif not key:match("%D") and conc(key) <= total then
execute(conc(key), mouse_button)
grabber:stop()
else
grabber:stop()
if was_hidden then systray_hints.systray.visible = false end
if systray_hints.popup then systray_hints.popup.visible = false end
end
end
elseif key == systray_hints.mouse_buttons[1] then mouse_button = 1
elseif key == systray_hints.mouse_buttons[2] then mouse_button = 2
elseif key == systray_hints.mouse_buttons[3] then mouse_button = 3
elseif not key:match("%D") and tonumber(key) <= total then
execute(tonumber(key), mouse_button)
grabber:stop()
else
grabber:stop()
if was_hidden then systray_hints.systray.visible = false end
if systray_hints.popup then systray_hints.popup.visible = false end
end
end,
}
end
local function show_popup(x, y, w, h)
local hints = {}
local hint_width
local x = math.floor(x)
local y = math.floor(y)
total = math.floor( ( w - ( w % h ) ) / h + 1 )
icon_width = math.floor(w / total )
icons_x = math.floor( x + icon_width / 2)
icons_y = math.floor( y + icon_width / 2)
hint_width = icon_width - ( systray_hints.spacing * 2 )
--Decide whether hints should display above or below systray icons.
if y >= 100 then y = y - hint_width else y = y + hint_width end
--Hide if already displayed
if systray_hints.popup then systray_hints.popup.visible = false end
local widget_shape = function(cr, width, height)
gears.shape.rounded_rect(cr, width, height, 5)
end
for i = 1, total do
local text
local placement = {}
local background = {}
local margins = {}
text = wibox.widget.textbox(tostring(i))
text.font = systray_hints.font
text.markup = '<span color="' .. systray_hints.color .. '">' ..
i .. '</span>'
table.insert(placement, text)
placement.widget = wibox.container.place
table.insert(background, placement)
background.widget = wibox.container.background
background.bg = systray_hints.bgcolor
background.forced_width = hint_width
background.shape = widget_shape
background.shape_border_width = 2
background.shape_border_color = systray_hints.bordercolor
table.insert(margins, background)
margins.widget = wibox.container.margin
margins.right = systray_hints.spacing
margins.left = systray_hints.spacing
table.insert(hints, margins)
end
hints.layout = wibox.layout.fixed.horizontal
systray_hints.popup = awful.popup {
widget = {
screen = s,
hints,
layout = wibox.layout.fixed.horizontal,
},
x = x,
y = y,
visible = true,
ontop = true,
bg = "#00000000",
}
end
local function get_geometry(hi)
local g = gears.matrix.transform_rectangle
return g(hi:get_matrix_to_device(), 0, 0, hi:get_size())
end
local function find_widget(wb, wdg)
local hierarchy
local function traverse(hi)
if hi:get_widget() == wdg then
hierarchy = hi
end
for _, child in ipairs(hi:get_children()) do
traverse(child)
end
end
traverse(wb._drawable._widget_hierarchy)
return hierarchy
end
systray_hints.run = function ()
local function show_hints()
show_popup( get_geometry(find_widget(systray_hints.wibox, systray_hints.systray)) )
get_key_input(total)
end
if not systray_hints.systray.visible then
was_hidden = true
systray_hints.systray.visible = true
delay(0.05, function ()
show_hints()
end)
else
show_hints()
end
end
return systray_hints