--        ██╗  ██╗███████╗██╗    ██╗███████╗
--        ██║ ██╔╝██╔════╝╚██╗ ██╔╝██╔════╝
--        █████╔╝ █████╗    ╚████╔╝ ███████╗
--        ██╔═██╗ ██╔══╝     ╚██╔╝  ╚════██║
--        ██║  ██╗███████╗    ██║   ███████║
--        ╚═╝  ╚═╝╚══════╝    ╚═╝   ╚══════╝


-- ===================================================================
-- Initialization
-- ===================================================================


local awful = require("awful")
local gears = require("gears")
local naughty = require("naughty")
local beautiful = require("beautiful")
local volume_widget = require("widgets.volume-widget.volume")
local dpi = beautiful.xresources.apply_dpi


local modkey = "Mod4"

-- define module table
local keys = {}


-- ===================================================================
-- Movement Functions (Called by some keybinds)
-- ===================================================================


-- Move given client to given direction
local function move_client(c, direction)
    awful.client.swap.global_bydirection(direction, c, nil)
end


-- Resize client in given direction
local floating_resize_amount = dpi(20)
local tiling_resize_factor = 0.05

local function resize_client(c, direction)
    if awful.layout.get(mouse.screen) == awful.layout.suit.floating or (c and c.floating) then
        if direction == "up" then
            c:relative_move(0, 0, 0, -floating_resize_amount)
        elseif direction == "down" then
            c:relative_move(0, 0, 0, floating_resize_amount)
        elseif direction == "left" then
            c:relative_move(0, 0, -floating_resize_amount, 0)
        elseif direction == "right" then
            c:relative_move(0, 0, floating_resize_amount, 0)
        end
    else
        if direction == "up" then
            awful.client.incwfact(-tiling_resize_factor)
        elseif direction == "down" then
            awful.client.incwfact(tiling_resize_factor)
        elseif direction == "left" then
            awful.tag.incmwfact(-tiling_resize_factor)
        elseif direction == "right" then
            awful.tag.incmwfact(tiling_resize_factor)
        end
    end
end


-- raise focused client
local function raise_client()
    if client.focus then
        client.focus:raise()
    end
end


-- ===================================================================
-- Mouse bindings
-- ===================================================================


-- Mouse buttons on the desktop
keys.desktopbuttons = gears.table.join(
    -- left click on desktop to hide notification
    awful.button({}, 1,
        function ()
            naughty.destroy_all_notifications()
        end
    ),
    awful.button({}, 7, function() volume_widget:inc(5) end),
    awful.button({}, 6, function() volume_widget:dec(5) end)
)

-- Mouse buttons on the client
keys.clientbuttons = gears.table.join(
    -- Raise client
    awful.button({}, 1,
        function(c)
            client.focus = c
            c:raise()
        end
    ),

    -- Move and Resize Client
    awful.button({modkey}, 1, awful.mouse.client.move),
    awful.button({modkey}, 3, awful.mouse.client.resize),
    awful.button({}, 7, function() volume_widget:inc(5) end),
    awful.button({}, 6, function() volume_widget:dec(5) end)
)


-- ===================================================================
-- Desktop Key bindings
-- ===================================================================


keys.globalkeys = gears.table.join(
    -- =========================================
    -- SPAWN APPLICATION KEY BINDINGS
    -- =========================================

    -- Spawn terminal
    awful.key({modkey}, "Return",
        function()
            local c = client.focus
            if c and c.class == "Alacritty" then
                awful.spawn(string.format(os.getenv("XDG_CONFIG_HOME") .. "/awesome/scripts/launch_alacritty.sh %d", c.pid))
            else
                awful.spawn(apps.terminal)
            end
        end,
        {description = "open a terminal", group = "launcher"}
    ),
    -- Spawn browser
    awful.key({modkey}, "b",
        function()
            awful.spawn("firefox")
        end,
        {description = "Open Firefox", group = "launcher"}
    ),
    -- Lock the screen
    awful.key({"Control", "Mod1"}, "l",
       function()
          awful.spawn('bash -c "i3lockr --blur 25"')
       end,
       {description = "Open lock", group = "launcher"}
    ),
    -- Generate IMEC TOTP
    awful.key({modkey}, "i",
        function()
            awful.spawn.with_shell("ykman oath accounts code \"Microsoft (IMEC)\" | cut -d ' '  -f 4 | xsel --clipboard")
        end,
        {description = "Generate IMEC TOTP", group = "launcher"}
    ),
    -- Spawn file manager
    awful.key({modkey}, "a",
        function()
            awful.spawn("nautilus")
        end,
        {description = "Open Nautilus", group = "launcher"}
    ),
    -- launch rofi
    awful.key({modkey}, "XF86Launch5",
        function()
            awful.spawn(os.getenv("XDG_CONFIG_HOME") .. "/awesome/scripts/toggle_rofi.sh")
        end,
        {description = "application launcher", group = "launcher"}
    ),
    -- launch rofi
    awful.key({modkey}, "d",
        function()
            awful.spawn(os.getenv("XDG_CONFIG_HOME") .. "/awesome/scripts/toggle_rofi.sh")
        end,
        {description = "application launcher", group = "launcher"}
    ),
    -- launch clipman
    awful.key({modkey}, "c",
        function()
            awful.spawn.with_shell('CM_LAUNCHER=rofi-script rofi -modi "clipmenu:/usr/bin/clipmenu" -show clipmenu')
        end,
        {description = "Clipboard manager", group = "launcher"}
    ),
    -- Go to sleep
    awful.key({"Control", "Mod2", "Mod4"}, "q",
        function()
            awful.spawn.with_shell('systemctl suspend')
        end,
        {description = "Clipboard manager", group = "launcher"}
    ),

    -- Quake
    awful.key({ }, "F9", function () awful.screen.focused().quake:toggle() end),

    -- =========================================
    -- FUNCTION KEYS
    -- =========================================

    -- Brightness
    awful.key({}, "XF86MonBrightnessUp",
        function()
            awful.spawn("xbacklight -inc 10", false)
        end,
        {description = "+10%", group = "hotkeys"}
    ),
    awful.key({}, "XF86MonBrightnessDown",
        function()
            awful.spawn("xbacklight -dec 10", false)
        end,
        {description = "-10%", group = "hotkeys"}
    ),

    -- ALSA volume control
    awful.key({}, "XF86AudioRaiseVolume",
        function()
             volume_widget:inc(5)
        end,
        {description = "volume up", group = "hotkeys"}
    ),
    awful.key({}, "XF86AudioLowerVolume",
        function()
             volume_widget:dec(5)
        end,
        {description = "volume down", group = "hotkeys"}
    ),
    awful.key({}, "XF86AudioMute",
        function()
             volume_widget:toggle()
        end,
        {description = "toggle mute", group = "hotkeys"}
    ),
    awful.key({}, "XF86AudioNext",
        function()
            awful.spawn("playerctl next", false)
        end,
        {description = "next music", group = "hotkeys"}
    ),
    awful.key({}, "XF86AudioPrev",
        function()
            awful.spawn("playerctl previous", false)
        end,
        {description = "previous music", group = "hotkeys"}
    ),
    awful.key({}, "XF86AudioPlay",
        function()
            awful.spawn("playerctl play-pause", false)
        end,
        {description = "play/pause music", group = "hotkeys"}
    ),

    -- Screenshot on prtscn using scrot
    awful.key({}, "Print",
        function()
            awful.util.spawn(apps.screenshot, false)
        end
    ),

    -- =========================================
    -- RELOAD / QUIT AWESOME
    -- =========================================

    -- Reload Awesome
    awful.key({modkey, "Shift"}, "r",
        awesome.restart,
        {description = "reload awesome", group = "awesome"}
    ),

    -- Quit Awesome
    awful.key({modkey}, "Escape",
        function()
            -- emit signal to show the exit screen
            awful.util.spawn("rofi -show power-menu -modi power-menu:" .. os.getenv("XDG_CONFIG_HOME") .. "/awesome/scripts/rofi-power-menu")
        end,
        {description = "toggle exit screen", group = "hotkeys"}
    ),

    awful.key({}, "XF86PowerOff",
        function()
            -- emit signal to show the exit screen
            awesome.emit_signal("show_exit_screen")
        end,
        {description = "toggle exit screen", group = "hotkeys"}
    ),

    awful.key({}, "XF86Calculator",
        function()
            awful.util.spawn("rofi -show calc -modi calc -no-show-match -no-sort -calc-command \"echo -n '{result}' | xsel --clipboard\"")
        end,
        {description = "Launch the calculator", group = "hotkeys"}
    ),

    -- =========================================
    -- CLIENT FOCUSING
    -- =========================================

    -- Focus client by direction (hjkl keys)
    -- awful.key({modkey}, "j",
    --     function()
    --         awful.client.focus.bydirection("down")
    --         raise_client()
    --     end,
    --     {description = "focus down", group = "client"}
    -- ),
    -- awful.key({modkey}, "k",
    --     function()
    --         awful.client.focus.bydirection("up")
    --         raise_client()
    --     end,
    --     {description = "focus up", group = "client"}
    -- ),
    -- awful.key({modkey}, "h",
    --     function()
    --         awful.client.focus.bydirection("left")
    --         raise_client()
    --     end,
    --     {description = "focus left", group = "client"}
    -- ),
    -- awful.key({modkey}, "l",
    --     function()
    --         awful.client.focus.bydirection("right")
    --         raise_client()
    --     end,
    --     {description = "focus right", group = "client"}
    -- ),

    -- Focus client by direction (arrow keys)
    -- awful.key({modkey}, "Down",
    --     function()
    --         awful.client.focus.bydirection("down")
    --         raise_client()
    --     end,
    --     {description = "focus down", group = "client"}
    -- ),
    -- awful.key({modkey}, "Up",
    --     function()
    --         awful.client.focus.bydirection("up")
    --         raise_client()
    --     end,
    --     {description = "focus up", group = "client"}
    -- ),
    -- awful.key({modkey}, "Left",
    --     function()
    --         awful.client.focus.bydirection("left")
    --         raise_client()
    --     end,
    --     {description = "focus left", group = "client"}
    -- ),
    -- awful.key({modkey}, "Right",
    --     function()
    --         awful.client.focus.bydirection("right")
    --         raise_client()
    --     end,
    --     {description = "focus right", group = "client"}
    -- ),

    -- Focus client by index (cycle through clients)
    awful.key({modkey}, "Tab",
        function()
            awful.client.focus.byidx(1)
        end,
        {description = "focus next by index", group = "client"}
    ),
    awful.key({modkey, "Shift"}, "Tab",
        function()
            awful.client.focus.byidx(-1)
        end,
        {description = "focus previous by index", group = "client"}
    ),

    -- =========================================
    -- SCREEN FOCUSING
    -- =========================================

    -- Focus screen by index (cycle through screens)
    awful.key({modkey}, "s",
        function()
            awful.screen.focus_relative(1)
        end
    ),

    -- =========================================
    -- CLIENT RESIZING
    -- =========================================

    awful.key({modkey, "Control"}, "Down",
        function(c)
            resize_client(client.focus, "down")
        end
    ),
    awful.key({modkey, "Control"}, "Up",
        function(c)
            resize_client(client.focus, "up")
        end
    ),
    awful.key({modkey, "Control"}, "Left",
        function(c)
            resize_client(client.focus, "left")
        end
    ),
    awful.key({modkey, "Control"}, "Right",
        function(c)
            resize_client(client.focus, "right")
        end
    ),
    awful.key({modkey, "Control"}, "j",
        function(c)
            resize_client(client.focus, "down")
        end
    ),
    awful.key({ modkey, "Control" }, "k",
        function(c)
            resize_client(client.focus, "up")
        end
    ),
    awful.key({modkey, "Control"}, "h",
        function(c)
            resize_client(client.focus, "left")
        end
    ),
    awful.key({modkey, "Control"}, "l",
        function(c)
            resize_client(client.focus, "right")
        end
    ),

    -- =========================================
    -- GAP CONTROL
    -- =========================================

    -- Gap control
    awful.key({modkey, "Shift"}, "minus",
        function()
            awful.tag.incgap(5, nil)
        end,
        {description = "increment gaps size for the current tag", group = "gaps"}
    ),
    awful.key({modkey}, "minus",
        function()
            awful.tag.incgap(-5, nil)
        end,
        {description = "decrement gap size for the current tag", group = "gaps"}
    ),

    -- =========================================
    -- LAYOUT SELECTION
    -- =========================================

    -- select next layout
    awful.key({modkey}, "space",
        function()
            awful.layout.inc(1)
        end,
        {description = "select next", group = "layout"}
    ),
    -- select previous layout
    awful.key({modkey, "Shift"}, "space",
        function()
            awful.layout.inc(-1)
        end,
        {description = "select previous", group = "layout"}
    ),

    -- =========================================
    -- CLIENT MINIMIZATION
    -- =========================================

    -- restore minimized client
    awful.key({modkey, "Shift"}, "n",
        function()
            local c = awful.client.restore()
            -- Focus restored client
            if c then
                client.focus = c
                c:raise()
            end
        end,
        {description = "restore minimized", group = "client"}
    )
)


-- ===================================================================
-- Client Key bindings
-- ===================================================================


keys.clientkeys = gears.table.join(
    -- Move to edge or swap by direction
    awful.key({modkey, "Shift"}, "Down",
        function(c)
            move_client(c, "down")
        end
    ),
    awful.key({modkey, "Shift"}, "Up",
        function(c)
            move_client(c, "up")
        end
    ),
    awful.key({modkey, "Shift"}, "Left",
        function(c)
            move_client(c, "left")
        end
    ),
    awful.key({modkey, "Shift"}, "Right",
        function(c)
            move_client(c, "right")
        end
    ),
    awful.key({modkey, "Shift"}, "j",
        function(c)
            move_client(c, "down")
        end
    ),
    awful.key({modkey, "Shift"}, "k",
        function(c)
            move_client(c, "up")
        end
    ),
    awful.key({modkey, "Shift"}, "h",
        function(c)
            move_client(c, "left")
        end
    ),
    awful.key({modkey, "Shift"}, "l",
        function(c)
            move_client(c, "right")
        end
    ),

    -- close client
    awful.key({modkey}, "q",
        function(c)
            c:kill()
            for i, c in ipairs(client.get()) do
                c.minimized = false
            end
        end,
        {description = "close", group = "client"}
    ),

    -- float client
    awful.key({modkey}, "f",
        function(c)
             awful.client.floating.toggle(c)
        end,
        {description = "float", group = "client"}
    ),

    -- Minimize
    awful.key({modkey}, "n",
        function(c)
            c.minimized = true
        end,
        {description = "minimize", group = "client"}
    ),

    -- Maximize
    awful.key({modkey}, "m",
        function(c)
            c.maximized = not c.maximized
            c:raise()
        end,
        {description = "(un)maximize", group = "client"}
    )
)

-- Bind all key numbers to tags
for i = 1, 9 do
    keys.globalkeys = gears.table.join(keys.globalkeys,
        -- Switch to tag
        awful.key({modkey}, "#" .. i + 9,
            function()
                local screen = awful.screen.focused()
                local tag = screen.tags[i]
                if tag then
                    tag:view_only()
                end
            end,
            {description = "view tag #"..i, group = "tag"}
        ),
        -- Move client to tag
        awful.key({modkey, "Shift"}, "#" .. i + 9,
            function()
                if client.focus then
                    local tag = client.focus.screen.tags[i]
                    if tag then
                        client.focus:move_to_tag(tag)
                        tag:view_only()
                    end
                end
            end,
            {description = "move focused client to tag #"..i, group = "tag"}
        )
    )
end

return keys