summaryrefslogtreecommitdiff
path: root/github-prs-widget/init.lua
diff options
context:
space:
mode:
Diffstat (limited to 'github-prs-widget/init.lua')
-rw-r--r--github-prs-widget/init.lua434
1 files changed, 434 insertions, 0 deletions
diff --git a/github-prs-widget/init.lua b/github-prs-widget/init.lua
new file mode 100644
index 0000000..0ff76a9
--- /dev/null
+++ b/github-prs-widget/init.lua
@@ -0,0 +1,434 @@
+-------------------------------------------------
+-- GitHub Widget for Awesome Window Manager
+-- Shows the number of currently assigned merge requests
+-- and information about them
+-- More details could be found here:
+-- https://github.com/streetturtle/awesome-wm-widgets/tree/master/github-prs-widget
+
+-- @author Pavel Makhov
+-- @copyright 2021 Pavel Makhov
+-------------------------------------------------
+
+local awful = require("awful")
+local wibox = require("wibox")
+local watch = require("awful.widget.watch")
+local json = require("json")
+local spawn = require("awful.spawn")
+local naughty = require("naughty")
+local gears = require("gears")
+local beautiful = require("beautiful")
+local gfs = require("gears.filesystem")
+local color = require("gears.color")
+
+local HOME_DIR = os.getenv("HOME")
+local WIDGET_DIR = HOME_DIR .. '/.config/awesome/awesome-wm-widgets/github-prs-widget/'
+local ICONS_DIR = WIDGET_DIR .. 'icons/'
+
+local AVATARS_DIR = HOME_DIR .. '/.cache/awmw/github-widget/avatars/'
+local DOWNLOAD_AVATAR_CMD = [[sh -c "curl -L --create-dirs -o ''\\]] .. AVATARS_DIR .. [[%s %s"]]
+
+local GET_PRS_CMD = "gh api -X GET search/issues "
+ .. "-f 'q=review-requested:%s is:unmerged is:open' "
+ .. "-f per_page=30 "
+ .. "--jq '[.items[] | {url,repository_url,title,html_url,comments,assignees,user,created_at,draft}]'"
+
+local github_widget = wibox.widget {
+ {
+ {
+ {
+ {
+ {
+ id = 'icon',
+ widget = wibox.widget.imagebox
+ },
+ {
+ id = 'error_marker',
+ draw = function(_, _, cr, width, height)
+ cr:set_source(color(_config.fg_urgent))
+ cr:arc(width - height / 6, height / 6, height / 6, 0, math.pi * 2)
+ cr:fill()
+ end,
+ visible = false,
+ layout = wibox.widget.base.make_widget,
+ },
+ layout = wibox.layout.stack
+ },
+ margins = 4,
+ layout = wibox.container.margin
+ },
+ {
+ id = "txt",
+ widget = wibox.widget.textbox
+ },
+ {
+ id = "new_pr",
+ widget = wibox.widget.textbox
+ },
+ spacing = 4,
+ layout = wibox.layout.fixed.horizontal,
+ },
+ left = 4,
+ right = 4,
+ widget = wibox.container.margin
+ },
+ shape = function(cr, width, height)
+ gears.shape.rounded_rect(cr, width, height, 4)
+ end,
+ widget = wibox.container.background,
+ set_text = function(self, new_value)
+ self:get_children_by_id('txt')[1]:set_text(new_value)
+ end,
+ set_icon = function(self, new_value)
+ self:get_children_by_id('icon')[1]:set_image(new_value)
+ end,
+ is_everything_ok = function(self, is_ok)
+ if is_ok then
+ self:get_children_by_id('error_marker')[1]:set_visible(false)
+ self:get_children_by_id('icon')[1]:set_opacity(1)
+ self:get_children_by_id('icon')[1]:emit_signal('widget:redraw_needed')
+ else
+ self.txt:set_text('')
+ self:get_children_by_id('error_marker')[1]:set_visible(true)
+ self:get_children_by_id('icon')[1]:set_opacity(0.2)
+ self:get_children_by_id('icon')[1]:emit_signal('widget:redraw_needed')
+ end
+ end
+}
+
+local function show_warning(message)
+ naughty.notify{
+ preset = naughty.config.presets.critical,
+ title = 'GitHub PRs Widget',
+ text = message}
+end
+
+local popup = awful.popup{
+ ontop = true,
+ visible = false,
+ shape = gears.shape.rounded_rect,
+ border_width = 1,
+ maximum_width = 400,
+ offset = { y = 5 },
+ widget = {}
+}
+
+--- Converts string representation of date (2020-06-02T11:25:27Z) to date
+local function parse_date(date_str)
+ local pattern = "(%d+)%-(%d+)%-(%d+)T(%d+):(%d+):(%d+)%Z"
+ local y, m, d, h, min, sec, _ = date_str:match(pattern)
+
+ return os.time{year = y, month = m, day = d, hour = h, min = min, sec = sec}
+end
+
+--- Converts seconds to "time ago" represenation, like '1 hour ago'
+local function to_time_ago(seconds)
+ local days = seconds / 86400
+ if days > 1 then
+ days = math.floor(days + 0.5)
+ return days .. (days == 1 and ' day' or ' days') .. ' ago'
+ end
+
+ local hours = (seconds % 86400) / 3600
+ if hours > 1 then
+ hours = math.floor(hours + 0.5)
+ return hours .. (hours == 1 and ' hour' or ' hours') .. ' ago'
+ end
+
+ local minutes = ((seconds % 86400) % 3600) / 60
+ if minutes > 1 then
+ minutes = math.floor(minutes + 0.5)
+ return minutes .. (minutes == 1 and ' minute' or ' minutes') .. ' ago'
+ end
+end
+
+local function ellipsize(text, length)
+ return (text:len() > length and length > 0)
+ and text:sub(0, length - 3) .. '...'
+ or text
+end
+
+local warning_shown = false
+local tooltip = awful.tooltip {
+ mode = 'outside',
+ preferred_positions = {'bottom'},
+}
+
+local config = {}
+
+config.reviewer = nil
+
+config.bg_normal = '#aaaaaa'
+config.bg_focus = '#ffffff'
+
+
+local function worker(user_args)
+
+ local args = user_args or {}
+
+ -- Setup config for the widget instance.
+ -- The `_config` table will keep the first existing value after checking
+ -- in this order: user parameter > beautiful > module default
+ local _config = {}
+ for prop, value in pairs(config) do
+ _config[prop] = args[prop] or beautiful[prop] or value
+ end
+
+ local icon = args.icon or ICONS_DIR .. 'git-pull-request.svg'
+ local reviewer = args.reviewer
+ local timeout = args.timeout or 60
+
+ local current_number_of_prs
+
+ local to_review_rows = {layout = wibox.layout.fixed.vertical}
+ local rows = {layout = wibox.layout.fixed.vertical}
+
+ github_widget:set_icon(icon)
+
+ local update_widget = function(widget, stdout, stderr, _, _)
+
+ if stderr ~= '' then
+ if not warning_shown then
+ show_warning(stderr)
+ warning_shown = true
+ widget:is_everything_ok(false)
+ tooltip:add_to_object(widget)
+
+ widget:connect_signal('mouse::enter', function()
+ tooltip.text = stderr
+ end)
+ end
+ return
+ end
+
+ warning_shown = false
+ tooltip:remove_from_object(widget)
+ widget:is_everything_ok(true)
+
+ local prs = json.decode(stdout)
+
+ current_number_of_prs = #prs
+
+ if current_number_of_prs == 0 then
+ widget:set_visible(false)
+ return
+ end
+
+ widget:set_visible(true)
+ widget:set_text(current_number_of_prs)
+
+ for i = 0, #rows do rows[i]=nil end
+
+ for i = 0, #to_review_rows do to_review_rows[i]=nil end
+ table.insert(to_review_rows, {
+ {
+ markup = '<span size="large" color="#ffffff">PRs to review</span>',
+ align = 'center',
+ forced_height = 20,
+ widget = wibox.widget.textbox
+ },
+ bg = _config.bg_normal,
+ widget = wibox.container.background
+ })
+
+ local current_time = os.time(os.date("!*t"))
+
+ for _, pr in ipairs(prs) do
+ local path_to_avatar = AVATARS_DIR .. pr.user.id
+ local index = string.find(pr.repository_url, "/[^/]*$")
+ local repo = string.sub(pr.repository_url, index + 1)
+
+ local row = wibox.widget {
+ {
+ {
+ {
+ {
+ resize = true,
+ image = path_to_avatar,
+ forced_width = 40,
+ forced_height = 40,
+ widget = wibox.widget.imagebox
+ },
+ id = 'avatar',
+ margins = 4,
+ layout = wibox.container.margin
+ },
+ {
+ {
+ id = 'title',
+ markup = '<b>' .. ellipsize(pr.title, 60) .. '</b>',
+ widget = wibox.widget.textbox,
+ forced_width = 400
+ },
+ {
+ {
+ {
+ {
+ image = ICONS_DIR .. 'book.svg',
+ forced_width = 12,
+ forced_height = 12,
+ resize = true,
+ widget = wibox.widget.imagebox
+ },
+ {
+ text = repo,
+ widget = wibox.widget.textbox
+ },
+ spacing = 4,
+ expand = 'none',
+ layout = wibox.layout.fixed.horizontal
+ },
+ {
+ {
+ image = ICONS_DIR .. 'user.svg',
+ forced_width = 12,
+ forced_height = 12,
+ resize = true,
+ widget = wibox.widget.imagebox
+ },
+ {
+ text = pr.user.login,
+ widget = wibox.widget.textbox
+ },
+ spacing = 4,
+ expand = 'none',
+ layout = wibox.layout.fixed.horizontal
+ },
+ spacing = 8,
+ expand = 'none',
+ layout = wibox.layout.fixed.horizontal
+ },
+ {
+ {
+ {
+ image = ICONS_DIR .. 'user.svg',
+ forced_width = 12,
+ forced_height = 12,
+ resize = true,
+ widget = wibox.widget.imagebox
+ },
+ {
+ text = to_time_ago(os.difftime(current_time, parse_date(pr.created_at))),
+ widget = wibox.widget.textbox
+ },
+ spacing = 4,
+ expand = 'none',
+ layout = wibox.layout.fixed.horizontal
+
+ },
+ {
+ {
+ image = ICONS_DIR .. 'message-square.svg',
+ forced_width = 12,
+ forced_height = 12,
+ resize = true,
+ widget = wibox.widget.imagebox
+ },
+ {
+ text = pr.comments,
+ widget = wibox.widget.textbox
+ },
+ spacing = 4,
+ expand = 'none',
+ layout = wibox.layout.fixed.horizontal
+
+ },
+ spacing = 8,
+ layout = wibox.layout.fixed.horizontal
+ },
+ layout = wibox.layout.fixed.vertical
+ },
+ spacing = 4,
+ layout = wibox.layout.fixed.vertical
+ },
+ spacing = 8,
+ layout = wibox.layout.fixed.horizontal
+ },
+ margins = 8,
+ layout = wibox.container.margin
+ },
+ bg = _config.bg_normal,
+ widget = wibox.container.background
+ }
+
+ if not gfs.file_readable(path_to_avatar) then
+ spawn.easy_async(string.format(
+ DOWNLOAD_AVATAR_CMD,
+ pr.user.id,
+ pr.user.avatar_url), function()
+ row:get_children_by_id('avatar')[1]:set_image(path_to_avatar)
+ end)
+ end
+
+ row:connect_signal("mouse::enter", function(c) c:set_bg(_config.bg_focus) end)
+ row:connect_signal("mouse::leave", function(c) c:set_bg(_config.bg_normal) end)
+
+ row:get_children_by_id('title')[1]:buttons(
+ awful.util.table.join(
+ awful.button({}, 1, function()
+ spawn.with_shell("xdg-open " .. pr.html_url)
+ popup.visible = false
+ end)
+ )
+ )
+ row:get_children_by_id('avatar')[1]:buttons(
+ awful.util.table.join(
+ awful.button({}, 1, function()
+ spawn.with_shell("xdg-open " .. pr.user.html_url)
+ popup.visible = false
+ end)
+ )
+ )
+
+ local old_cursor, old_wibox
+ row:get_children_by_id('title')[1]:connect_signal("mouse::enter", function()
+ local wb = mouse.current_wibox
+ old_cursor, old_wibox = wb.cursor, wb
+ wb.cursor = "hand1"
+ end)
+ row:get_children_by_id('title')[1]:connect_signal("mouse::leave", function()
+ if old_wibox then
+ old_wibox.cursor = old_cursor
+ old_wibox = nil
+ end
+ end)
+
+ row:get_children_by_id('avatar')[1]:connect_signal("mouse::enter", function()
+ local wb = mouse.current_wibox
+ old_cursor, old_wibox = wb.cursor, wb
+ wb.cursor = "hand1"
+ end)
+ row:get_children_by_id('avatar')[1]:connect_signal("mouse::leave", function()
+ if old_wibox then
+ old_wibox.cursor = old_cursor
+ old_wibox = nil
+ end
+ end)
+
+ table.insert(to_review_rows, row)
+ end
+
+ table.insert(rows, to_review_rows)
+ popup:setup(rows)
+ end
+
+ github_widget:buttons(
+ awful.util.table.join(
+ awful.button({}, 1, function()
+ if popup.visible then
+ popup.visible = not popup.visible
+ github_widget:set_bg('#00000000')
+ else
+ github_widget:set_bg(beautiful.bg_focus)
+ popup:move_next_to(mouse.current_widget_geometry)
+ end
+ end)
+ )
+ )
+
+ watch(string.format(GET_PRS_CMD, reviewer),
+ timeout, update_widget, github_widget)
+
+ return github_widget
+end
+
+return setmetatable(github_widget, { __call = function(_, ...) return worker(...) end })