[gitlab] new widget \o/

4 files changed, 356 insertions, 0 deletions

diff --git a/gitlab-widget/gitlab.lua b/gitlab-widget/gitlab.lua
new file mode 100644
index 0000000..d7a8e35
--- /dev/null
+++ b/gitlab-widget/gitlab.lua
@@ -0,0 +1,353 @@
+-------------------------------------------------
+-- Gitlab Widget for Awesome Window Manager
+-- Shows the number of currently assigned pull requests
+-- More details could be found here:
+-- https://github.com/streetturtle/awesome-wm-widgets/tree/master/gitlab-widget
+
+-- @author Pavel Makhov
+-- @copyright 2020 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 HOME_DIR = os.getenv("HOME")
+local WIDGET_DIR = HOME_DIR .. '/.config/awesome/awesome-wm-widgets/gitlab-widget/'
+local GET_PRS_CMD= [[bash -c "curl -s --show-error --header 'PRIVATE-TOKEN: %s' '%s/api/v4/merge_requests'"]]
+local DOWNLOAD_AVATAR_CMD = [[bash -c "curl -L -n --create-dirs -o %s/.cache/awmw/gitlab-widget/avatars/%s %s"]]
+
+local gitlab_widget = wibox.widget {
+    {
+        {
+            id = 'icon',
+            widget = wibox.widget.imagebox
+        },
+        margins = 4,
+        layout = wibox.container.margin
+    },
+    {
+        id = "txt",
+        widget = wibox.widget.textbox
+    },
+    {
+        id = "new_pr",
+        widget = wibox.widget.textbox
+    },
+    layout = wibox.layout.fixed.horizontal,
+    set_text = function(self, new_value)
+        self.txt.text = new_value
+    end,
+    set_icon = function(self, new_value)
+        self:get_children_by_id('icon')[1]:set_image(new_value)
+    end
+}
+
+local function show_warning(message)
+    naughty.notify{
+        preset = naughty.config.presets.critical,
+        title = 'Gitlab Widget',
+        text = message}
+end
+
+local popup = awful.popup{
+    ontop = true,
+    visible = false,
+    shape = gears.shape.rounded_rect,
+    border_width = 1,
+    border_color = beautiful.bg_focus,
+    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, mil = 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 function worker(args)
+
+    local args = args or {}
+
+    local icon = args.icon or WIDGET_DIR .. '/icons/gitlab-icon.svg'
+    local api_token = args.api_token or show_warning('API Token is not set')
+    local host = args.host or show_warning('Gitlab host is not set')
+    local timeout = args.timeout or 60
+
+    local current_number_of_prs
+
+    local to_review_rows = {layout = wibox.layout.fixed.vertical}
+    local my_review_rows = {layout = wibox.layout.fixed.vertical}
+    local rows = {layout = wibox.layout.fixed.vertical}
+
+    gitlab_widget:set_icon(icon)
+
+    local update_widget = function(widget, stdout, stderr, _, _)
+
+        if stderr ~= '' then
+            show_warning(stderr)
+            return
+        end
+
+        local result = json.decode(stdout)
+
+        current_number_of_prs = rawlen(result)
+
+        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 = beautiful.bg_normal,
+            widget = wibox.container.background
+        })
+
+        for i = 0, #my_review_rows do my_review_rows[i]=nil end
+        table.insert(my_review_rows, {
+            {
+                markup = '<span size="large" color="#ffffff">My PRs</span>',
+                align = 'center',
+                forced_height = 20,
+                widget = wibox.widget.textbox
+            },
+            bg = beautiful.bg_normal,
+            widget = wibox.container.background
+        })
+        local current_time = os.time(os.date("!*t"))
+
+        for _, pr in ipairs(result) do
+            local path_to_avatar = os.getenv("HOME") ..'/.cache/awmw/gitlab-widget/avatars/' .. pr.author.id
+
+            local row = wibox.widget {
+                {
+                    {
+                        {
+                                {
+                                    resize = true,
+                                    image = path_to_avatar,
+                                    forced_width = 40,
+                                    forced_height = 40,
+                                    widget = wibox.widget.imagebox
+                                },
+                                id = 'avatar',
+                                margins = 8,
+                                layout = wibox.container.margin
+                        },
+                        {
+                            {
+                                id = 'title',
+                                markup = '<b>' .. ellipsize(pr.title, 50) .. '</b>',
+                                widget = wibox.widget.textbox,
+                                forced_width = 400
+                            },
+                            {
+                                {
+                                    {
+                                        {
+                                            text = pr.source_branch,
+                                            widget = wibox.widget.textbox
+                                        },
+                                        {
+                                            text = '->',
+                                            widget = wibox.widget.textbox
+                                        },
+                                        {
+                                            text = pr.target_branch,
+                                            widget = wibox.widget.textbox
+                                        },
+                                        spacing = 8,
+                                        layout = wibox.layout.fixed.horizontal
+                                    },
+                                    {
+                                        {
+                                            text = pr.author.name,
+                                            widget = wibox.widget.textbox
+                                        },
+                                        {
+                                            text = to_time_ago(os.difftime(current_time, parse_date(pr.created_at))),
+                                            widget = wibox.widget.textbox
+                                        },
+                                        spacing = 8,
+                                        expand = 'none',
+                                        layout = wibox.layout.fixed.horizontal
+                                    },
+                                    forced_width = 285,
+                                    layout = wibox.layout.fixed.vertical
+                                },
+                                {
+                                    {
+                                        {
+                                            -- image = number_of_approves > 0 and WIDGET_DIR .. '/check.svg' or '',
+                                            image = WIDGET_DIR .. '/icons/check.svg',
+                                            resize = false,
+                                            widget = wibox.widget.imagebox
+                                        },
+                                        {
+                                            text = pr.upvotes,
+                                            widget = wibox.widget.textbox
+                                        },
+                                        layout = wibox.layout.fixed.horizontal
+                                    },
+                                    {
+                                        {
+                                            image = WIDGET_DIR .. '/icons/message-circle.svg',
+                                            resize = false,
+                                            widget = wibox.widget.imagebox
+                                        },
+                                        {
+                                            text = pr.user_notes_count,
+                                            widget = wibox.widget.textbox
+                                        },
+                                        layout = wibox.layout.fixed.horizontal
+                                    },
+                                    layout = wibox.layout.fixed.vertical
+                                },
+                                layout = wibox.layout.fixed.horizontal
+                            },
+
+                            spacing = 8,
+                            layout = wibox.layout.fixed.vertical
+                        },
+                        spacing = 8,
+                        layout = wibox.layout.fixed.horizontal
+                    },
+                    margins = 8,
+                    layout = wibox.container.margin
+                },
+                bg = beautiful.bg_normal,
+                widget = wibox.container.background
+            }
+
+            if not gfs.file_readable(path_to_avatar) then
+                spawn.easy_async(string.format(
+                        DOWNLOAD_AVATAR_CMD,
+                        HOME_DIR,
+                        pr.author.id,
+                        pr.author.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(beautiful.bg_focus) end)
+            row:connect_signal("mouse::leave", function(c) c:set_bg(beautiful.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.web_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.author.web_url)
+                                popup.visible = false
+                            end)
+                    )
+            )
+
+            local old_cursor, old_wibox
+            row:get_children_by_id('title')[1]:connect_signal("mouse::enter", function(c)
+                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(c)
+                if old_wibox then
+                    old_wibox.cursor = old_cursor
+                    old_wibox = nil
+                end
+            end)
+
+            local old_cursor, old_wibox
+            row:get_children_by_id('avatar')[1]:connect_signal("mouse::enter", function(c)
+                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(c)
+                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)
+        if (#my_review_rows > 1) then
+            table.insert(rows, my_review_rows)
+        end
+        popup:setup(rows)
+    end
+
+    gitlab_widget:buttons(
+            awful.util.table.join(
+                    awful.button({}, 1, function()
+                        if popup.visible then
+                            popup.visible = not popup.visible
+                        else
+                            popup:move_next_to(mouse.current_widget_geometry)
+                        end
+                    end)
+            )
+    )
+
+    watch(string.format(GET_PRS_CMD, api_token, host),
+        -- string.format(GET_PRS_CMD, host, workspace, repo_slug, uuid, uuid),
+            timeout, update_widget, gitlab_widget)
+    return gitlab_widget
+end
+
+return setmetatable(gitlab_widget, { __call = function(_, ...) return worker(...) end })
diff --git a/gitlab-widget/icons/check.svg b/gitlab-widget/icons/check.svg
new file mode 100644
index 0000000..e9e44ac
--- /dev/null
+++ b/gitlab-widget/icons/check.svg
@@ -0,0 +1 @@
+<svg xmlns="http://www.w3.org/2000/svg" width="16" height="16" viewBox="0 0 24 24" fill="none" stroke="#A3BE8C" stroke-width="2" stroke-linecap="round" stroke-linejoin="round" class="feather feather-check"><polyline points="20 6 9 17 4 12"></polyline></svg>
\ No newline at end of file
diff --git a/gitlab-widget/icons/gitlab-icon.svg b/gitlab-widget/icons/gitlab-icon.svg
new file mode 100644
index 0000000..abe3f37
--- /dev/null
+++ b/gitlab-widget/icons/gitlab-icon.svg
@@ -0,0 +1 @@
+<svg xmlns="http://www.w3.org/2000/svg" width="64" height="64" fill-rule="evenodd"><path d="M32 61.477L43.784 25.2H20.216z" fill="#e24329"/><path d="M32 61.477L20.216 25.2H3.7z" fill="#fc6d26"/><path d="M3.7 25.2L.12 36.23a2.44 2.44 0 0 0 .886 2.728L32 61.477z" fill="#fca326"/><path d="M3.7 25.2h16.515L13.118 3.366c-.365-1.124-1.955-1.124-2.32 0z" fill="#e24329"/><path d="M32 61.477L43.784 25.2H60.3z" fill="#fc6d26"/><path d="M60.3 25.2l3.58 11.02a2.44 2.44 0 0 1-.886 2.728L32 61.477z" fill="#fca326"/><path d="M60.3 25.2H43.784l7.098-21.844c.365-1.124 1.955-1.124 2.32 0z" fill="#e24329"/></svg>
\ No newline at end of file
diff --git a/gitlab-widget/icons/message-circle.svg b/gitlab-widget/icons/message-circle.svg
new file mode 100644
index 0000000..43eacbb
--- /dev/null
+++ b/gitlab-widget/icons/message-circle.svg
@@ -0,0 +1 @@
+<svg xmlns="http://www.w3.org/2000/svg" width="16" height="16" viewBox="0 0 24 24" fill="none" stroke="#ECEFF4" stroke-width="2" stroke-linecap="round" stroke-linejoin="round" class="feather feather-message-circle"><path d="M21 11.5a8.38 8.38 0 0 1-.9 3.8 8.5 8.5 0 0 1-7.6 4.7 8.38 8.38 0 0 1-3.8-.9L3 21l1.9-5.7a8.38 8.38 0 0 1-.9-3.8 8.5 8.5 0 0 1 4.7-7.6 8.38 8.38 0 0 1 3.8-.9h.5a8.48 8.48 0 0 1 8 8v.5z"></path></svg>
\ No newline at end of file