Initial commit

awesome/awesome-wm-widgets/github-contributions-widget/github-contributions-widget.lua

@@ -0,0 +1,104 @@
+-------------------------------------------------
+-- Github Contributions Widget for Awesome Window Manager
+-- Shows the contributions graph
+-- More details could be found here:
+-- https://github.com/streetturtle/awesome-wm-widgets/tree/master/github-contributions-widget
+
+-- @author Pavel Makhov
+-- @copyright 2020 Pavel Makhov
+-------------------------------------------------
+
+local awful = require("awful")
+local naughty = require("naughty")
+local wibox = require("wibox")
+local gears = require("gears")
+local widget_themes = require("awesome-wm-widgets.github-contributions-widget.themes")
+
+local GET_CONTRIBUTIONS_CMD = [[bash -c "curl -s https://github-contributions.vercel.app/api/v1/%s]]
+    .. [[ | jq -r '[.contributions[] ]]
+    .. [[ | select ( .date | strptime(\"%%Y-%%m-%%d\") | mktime < now)][:%s]| .[].intensity'"]]
+
+local github_contributions_widget = wibox.widget{
+    reflection = {
+        horizontal = true,
+        vertical = true,
+    },
+    widget = wibox.container.mirror
+}
+
+local function show_warning(message)
+    naughty.notify{
+        preset = naughty.config.presets.critical,
+        title = 'Github Contributions Widget',
+        text = message}
+end
+
+local function worker(user_args)
+
+    local args = user_args or {}
+    local username = args.username or 'streetturtle'
+    local days = args.days or 365
+    local color_of_empty_cells = args.color_of_empty_cells
+    local with_border = args.with_border
+    local margin_top = args.margin_top or 1
+    local theme = args.theme or 'standard'
+
+    if widget_themes[theme] == nil then
+        show_warning('Theme ' .. theme .. ' does not exist')
+        theme = 'standard'
+    end
+
+    if with_border == nil then with_border = true end
+
+    local function get_square(color)
+        if color_of_empty_cells ~= nil and color == widget_themes[theme][0] then
+            color = color_of_empty_cells
+        end
+
+        return wibox.widget{
+            fit = function()
+                return 3, 3
+            end,
+            draw = function(_, _, cr, _, _)
+                cr:set_source(gears.color(color))
+                cr:rectangle(0, 0, with_border and 2 or 3, with_border and 2 or 3)
+                cr:fill()
+            end,
+            layout = wibox.widget.base.make_widget
+        }
+    end
+
+    local col = {layout = wibox.layout.fixed.vertical}
+    local row = {layout = wibox.layout.fixed.horizontal}
+    local day_idx = 5 - os.date('%w')
+    for _ = 0, day_idx do
+        table.insert(col, get_square(color_of_empty_cells))
+    end
+
+    local update_widget = function(_, stdout, _, _, _)
+        for intensity in stdout:gmatch("[^\r\n]+") do
+            if day_idx %7 == 0 then
+                table.insert(row, col)
+                col = {layout = wibox.layout.fixed.vertical}
+            end
+            table.insert(col, get_square(widget_themes[theme][tonumber(intensity)]))
+            day_idx = day_idx + 1
+        end
+        github_contributions_widget:setup(
+            {
+                row,
+                top = margin_top,
+                layout = wibox.container.margin
+            }
+        )
+    end
+
+    awful.spawn.easy_async(string.format(GET_CONTRIBUTIONS_CMD, username, days),
+        function(stdout)
+            update_widget(github_contributions_widget, stdout)
+        end)
+
+    return github_contributions_widget
+end
+
+return setmetatable(github_contributions_widget, { __call = function(_, ...) return worker(...) end })