-------------------------------------------------
-- APT Widget for Awesome Window Manager
-- Lists containers and allows to manage them
-- More details could be found here:
-- https://github.com/streetturtle/awesome-wm-widgets/tree/master/apt-widget
-- @author Pavel Makhov
-- @copyright 2021 Pavel Makhov
-------------------------------------------------
local awful = require("awful")
local wibox = require("wibox")
local spawn = require("awful.spawn")
local naughty = require("naughty")
local gears = require("gears")
local beautiful = require("beautiful")
--- Utility Function to handle alternative paths
local function script_path()
local str = debug.getinfo(2, "S").source:sub(2)
return str:match("(.*/)")
end
local WIDGET_DIR = script_path()
local ICONS_DIR = WIDGET_DIR .. "/icons/"
local LIST_PACKAGES = [[sh -c "LC_ALL=c apt list --upgradable 2>/dev/null"]]
--- Utility function to show warning messages
local function show_warning(message)
naughty.notify({
preset = naughty.config.presets.critical,
title = "Docker Widget",
text = message,
})
end
local wibox_popup = wibox({
ontop = true,
visible = false,
shape = function(cr, width, height)
gears.shape.rounded_rect(cr, width, height, 4)
end,
border_width = 1,
border_color = beautiful.bg_focus,
max_widget_size = 500,
height = 500,
width = 300,
})
local apt_widget = wibox.widget({
{
{
id = "icon",
widget = wibox.widget.imagebox,
},
margins = 4,
layout = wibox.container.margin,
},
layout = wibox.layout.fixed.horizontal,
set_icon = function(self, new_icon)
self:get_children_by_id("icon")[1].image = new_icon
end,
})
local apt_widget_button = wibox.widget({
{
apt_widget,
widget = wibox.container.margin,
},
widget = clickable_container,
})
--- Parses the line and creates the package table out of it
--- yaru-theme-sound/focal-updates,focal-updates 20.04.10.1 all [upgradable from: 20.04.8]
local parse_package = function(line)
local name, _, nv, type, ov = line:match("(.*)%/(.*)%s(.*)%s(.*)%s%[upgradable from: (.*)]")
if name == nil then
return nil
end
local package = {
name = name,
new_version = nv,
type = type,
old_version = ov,
}
return package
end
local function worker(user_args)
local args = user_args or {}
local icon = args.icon or ICONS_DIR .. "white-black.svg"
apt_widget:set_icon(icon)
local pointer = 0
local min_widgets = 5
local carousel = false
local function rebuild_widget(containers, errors, _, _)
local to_update = {}
if errors ~= "" then
show_warning(errors)
return
end
local rows = wibox.layout.fixed.vertical()
rows:connect_signal("button::press", function(_, _, _, button)
if carousel then
if button == 4 then -- up scrolling
local cnt = #rows.children
local first_widget = rows.children[1]
rows:insert(cnt + 1, first_widget)
rows:remove(1)
elseif button == 5 then -- down scrolling
local cnt = #rows.children
local last_widget = rows.children[cnt]
rows:insert(1, last_widget)
rows:remove(cnt + 1)
end
else
if button == 5 then -- up scrolling
if pointer < #rows.children and ((#rows.children - pointer) >= min_widgets) then
pointer = pointer + 1
rows.children[pointer].visible = false
end
elseif button == 4 then -- down scrolling
if pointer > 0 then
rows.children[pointer].visible = true
pointer = pointer - 1
end
end
end
end)
for line in containers:gmatch("[^\r\n]+") do
local package = parse_package(line)
if package ~= nil then
local refresh_button = wibox.widget({
{
{
id = "icon",
image = ICONS_DIR .. "refresh-cw.svg",
resize = false,
widget = wibox.widget.imagebox,
},
margins = 4,
widget = wibox.container.margin,
},
shape = gears.shape.circle,
opacity = 0.5,
widget = wibox.container.background,
})
local old_cursor, old_wibox
refresh_button:connect_signal("mouse::enter", function(c)
c:set_opacity(1)
c:emit_signal("widget::redraw_needed")
local wb = mouse.current_wibox
old_cursor, old_wibox = wb.cursor, wb
wb.cursor = "hand1"
end)
refresh_button:connect_signal("mouse::leave", function(c)
c:set_opacity(0.5)
c:emit_signal("widget::redraw_needed")
if old_wibox then
old_wibox.cursor = old_cursor
old_wibox = nil
end
end)
local row = wibox.widget({
{
{
{
{
id = "checkbox",
checked = false,
color = beautiful.bg_normal,
paddings = 2,
shape = gears.shape.circle,
forced_width = 20,
forced_height = 20,
check_color = beautiful.fg_urgent,
border_color = beautiful.bg_urgent,
border_width = 1,
widget = wibox.widget.checkbox,
},
valign = "center",
layout = wibox.container.place,
},
{
{
id = "name",
markup = "" .. package["name"] .. "",
widget = wibox.widget.textbox,
},
halign = "left",
layout = wibox.container.place,
},
{
refresh_button,
halign = "right",
valign = "center",
fill_horizontal = true,
layout = wibox.container.place,
},
spacing = 8,
layout = wibox.layout.fixed.horizontal,
},
margins = 8,
layout = wibox.container.margin,
},
id = "row",
bg = beautiful.bg_normal,
widget = wibox.container.background,
click = function(self, checked)
local a = self:get_children_by_id("checkbox")[1]
if checked == nil then
a:set_checked(not a.checked)
else
a:set_checked(checked)
end
if a.checked then
to_update[ package["name"] ] = self
else
to_update[ package["name"] ] = false
end
end,
update = function(self)
refresh_button:get_children_by_id("icon")[1]:set_image(ICONS_DIR .. "watch.svg")
self:get_children_by_id("name")[1]:set_opacity(0.4)
self:get_children_by_id("name")[1]:emit_signal("widget::redraw_needed")
spawn.easy_async(
string.format([[sh -c 'yes | aptdcon --hide-terminal -u %s']], package["name"]),
function(stdout, stderr) -- luacheck:ignore 212
rows:remove_widgets(self)
end
)
end,
})
row:connect_signal("mouse::enter", function(c)
c:set_bg(beautiful.bg_focus)
end)
row:connect_signal("mouse::leave", function(c)
c:set_bg(beautiful.bg_normal)
end)
row:connect_signal("button::press", function(c, _, _, button)
if button == 1 then
c:click()
end
end)
refresh_button:buttons(gears.table.join(awful.button({}, 1, function()
row:update()
end)))
rows:add(row)
end
end
local header_checkbox = wibox.widget({
checked = false,
color = beautiful.bg_normal,
paddings = 2,
shape = gears.shape.circle,
forced_width = 20,
forced_height = 20,
check_color = beautiful.fg_urgent,
border_color = beautiful.bg_urgent,
border_width = 1,
widget = wibox.widget.checkbox,
})
header_checkbox:connect_signal("button::press", function(c)
c:set_checked(not c.checked)
local cbs = rows.children
for _, v in ipairs(cbs) do
v:click(c.checked)
end
end)
local header_refresh_icon = wibox.widget({
image = ICONS_DIR .. "refresh-cw.svg",
resize = false,
widget = wibox.widget.imagebox,
})
header_refresh_icon:buttons(gears.table.join(awful.button({}, 1, function()
print(#to_update)
for _, v in pairs(to_update) do
if v ~= nil then
v:update()
end
end
end)))
local header_row = wibox.widget({
{
{
{
header_checkbox,
valign = "center",
layout = wibox.container.place,
},
{
{
id = "name",
markup = "" .. #rows.children .. " packages to update",
widget = wibox.widget.textbox,
},
halign = "center",
layout = wibox.container.place,
},
{
header_refresh_icon,
halign = "right",
valign = "center",
layout = wibox.container.place,
},
layout = wibox.layout.align.horizontal,
},
margins = 8,
layout = wibox.container.margin,
},
bg = beautiful.bg_normal,
widget = wibox.container.background,
})
wibox_popup:setup({
header_row,
rows,
layout = wibox.layout.fixed.vertical,
})
end
apt_widget:buttons(gears.table.join(awful.button({}, 1, function()
if wibox_popup.visible then
wibox_popup.visible = not wibox_popup.visible
else
spawn.easy_async(LIST_PACKAGES, function(stdout, stderr)
rebuild_widget(stdout, stderr)
wibox_popup.visible = true
awful.placement.top(wibox_popup, { margins = { top = 20 }, parent = mouse })
end)
end
end)))
wibox_popup:connect_signal("mouse::leave", function()
if wibox_popup.visible then
wibox_popup.visible = false
end
end)
return apt_widget_button
end
return setmetatable(apt_widget, {
__call = function(_, ...)
return worker(...)
end,
})