--[[

Powerarrow Dark Awesome WM theme
github.com/lcpz

--]]
local gears = require("gears")
local lain = require("lain")
local awful = require("awful")
local wibox = require("wibox")
local xresources = require("beautiful.xresources")
local dpi = xresources.apply_dpi
local xrdb = xresources.get_current_theme()

local naughty = require("naughty")
local nconf = naughty.config
local os = os
local my_table = awful.util.table or gears.table -- 4.{0,1} compatibility

local theme = {}
theme.dir = os.getenv("HOME") .. "/.config/awesome/themes/powerarrow-dark"
theme.wallpaper = os.getenv("HOME") .. "/Images/wp10300105.jpg"
theme.font = (xrdb.font_family or "JetBrainsMono Nerd Font") .. " " .. (xrdb.font_size or "10")
theme.fg_normal = xrdb.foreground or "#DDDDFF"
theme.fg_focus = xrdb.color4 or "#9A9FF1"
theme.fg_urgent = xrdb.background or "#011626"
theme.bg_normal = xrdb.background or "#011626"
theme.bg_focus = xrdb.color0 or "#213646"
theme.bg_widget = theme.bg_normal
theme.bg_widget_alt = theme.bg_focus
theme.bg_systray = theme.bg_widget
theme.systray_icon_spacing = dpi(4)
theme.bg_urgent = xrdb.color9 or "#EA7A7A"
theme.border_width = dpi(2)
theme.border_normal = theme.bg_normal
theme.border_focus = theme.fg_normal
theme.border_marked = xrdb.color5 or "#CC9393"
theme.notification_icon_size = dpi(128)
-- theme.jules
theme.taglist_fg_focus    = "#3992af"
theme.taglist_fg_occupied = "#164b5d"
theme.taglist_fg_urgent   = "#ED7572"
theme.taglist_fg_empty    = "#828282"
theme.taglist_spacing     = 2
theme.taglist_font        = "awesomewm-font 11"
-- theme.border_radius = dpi(10)
theme.fg_warning = xrdb.color3 or "#E3D18A"

theme.hotkeys_bg = theme.bg_normal
theme.hotkeys_fg = theme.fg_normal
theme.hotkeys_border_width = theme.border_width
theme.hotkeys_border_color = theme.border_focus
theme.hotkeys_modifiers_fg = theme.fg_focus
theme.hotkeys_shape = function(cr, width, height)
    gears.shape.rounded_rect(cr, width, height, theme.border_radius)
end

theme.hotkeys_font = string.gsub(theme.font, "\\d+", "") .. "Mono 11"
theme.hotkeys_description_font = string.gsub(theme.font, "\\d+", "") .. " 9"

theme.tasklist_bg_focus = theme.bg_widget_alt
theme.tasklist_bg_normal = theme.bg_widget

theme.taglist_bg_focus = theme.bg_widget_alt
theme.taglist_bg_normal = theme.bg_widget
theme.taglist_bg_occupied = theme.bg_widget
theme.taglist_bg_empty = theme.bg_widget

theme.titlebar_bg_focus = theme.bg_focus
theme.titlebar_bg_normal = theme.bg_normal
theme.titlebar_fg_focus = theme.fg_focus
theme.menu_height = dpi(16)
theme.menu_width = dpi(140)
theme.menu_submenu_icon = theme.dir .. "/icons/submenu.png"
theme.taglist_squares_sel = theme.dir .. "/icons/square_sel.png"
theme.taglist_squares_unsel = theme.dir .. "/icons/square_unsel.png"
theme.layout_tile = gears.color.recolor_image(theme.dir .. "/icons/tile.svg", theme.fg_normal)
theme.layout_tileleft = gears.color.recolor_image(theme.dir .. "/icons/tileleft.svg", theme.fg_normal)
theme.layout_tilebottom = gears.color.recolor_image(theme.dir .. "/icons/tilebottom.svg", theme.fg_normal)
theme.layout_tiletop = gears.color.recolor_image(theme.dir .. "/icons/tiletop.svg", theme.fg_normal)
theme.layout_fairv = gears.color.recolor_image(theme.dir .. "/icons/fairv.svg", theme.fg_normal)
theme.layout_fairh = gears.color.recolor_image(theme.dir .. "/icons/fairh.svg", theme.fg_normal)
theme.layout_spiral = gears.color.recolor_image(theme.dir .. "/icons/spiral.svg", theme.fg_normal)
theme.layout_dwindle = gears.color.recolor_image(theme.dir .. "/icons/dwindle.svg", theme.fg_normal)
theme.layout_max = gears.color.recolor_image(theme.dir .. "/icons/max.svg", theme.fg_normal)
theme.layout_fullscreen = theme.dir .. "/icons/fullscreen.png"
theme.layout_magnifier = theme.dir .. "/icons/magnifier.png"
theme.layout_floating = gears.color.recolor_image(theme.dir .. "/icons/floating.svg", theme.fg_normal)
theme.layout_centerwork = gears.color.recolor_image(theme.dir .. "/icons/centerwork.svg", theme.fg_normal)
theme.layout_termfair = gears.color.recolor_image(theme.dir .. "/icons/termfair.svg", theme.fg_normal)
theme.layout_centerfair = gears.color.recolor_image(theme.dir .. "/icons/centerfair.svg", theme.fg_normal)
theme.widget_ac = theme.dir .. "/icons/ac.png"
theme.widget_battery = theme.dir .. "/icons/battery.png"
theme.widget_battery_low = theme.dir .. "/icons/battery_low.png"
theme.widget_battery_empty = theme.dir .. "/icons/battery_empty.png"
theme.widget_mem = theme.dir .. "/icons/mem.png"
theme.widget_cpu = theme.dir .. "/icons/cpu.png"
theme.widget_temp = theme.dir .. "/icons/temp.png"
theme.widget_net = theme.dir .. "/icons/net.png"
theme.widget_hdd = theme.dir .. "/icons/hdd.png"
theme.widget_music = theme.dir .. "/icons/note.png"
theme.widget_music_on = theme.dir .. "/icons/note_on.png"
theme.widget_vol = theme.dir .. "/icons/vol.png"
theme.widget_vol_low = theme.dir .. "/icons/vol_low.png"
theme.widget_vol_no = theme.dir .. "/icons/vol_no.png"
theme.widget_vol_mute = theme.dir .. "/icons/vol_mute.png"
theme.widget_mail = theme.dir .. "/icons/mail.png"
theme.widget_mail_on = theme.dir .. "/icons/mail_on.png"
theme.tasklist_plain_task_name = true
theme.tasklist_disable_icon = true
theme.useless_gap = dpi(3)
theme.titlebar_close_button_focus = theme.dir .. "/icons/titlebar/close_focus.png"
theme.titlebar_close_button_normal = theme.dir .. "/icons/titlebar/close_normal.png"
theme.titlebar_ontop_button_focus_active = theme.dir .. "/icons/titlebar/ontop_focus_active.png"
theme.titlebar_ontop_button_normal_active = theme.dir .. "/icons/titlebar/ontop_normal_active.png"
theme.titlebar_ontop_button_focus_inactive = theme.dir .. "/icons/titlebar/ontop_focus_inactive.png"
theme.titlebar_ontop_button_normal_inactive = theme.dir .. "/icons/titlebar/ontop_normal_inactive.png"
theme.titlebar_sticky_button_focus_active = theme.dir .. "/icons/titlebar/sticky_focus_active.png"
theme.titlebar_sticky_button_normal_active = theme.dir .. "/icons/titlebar/sticky_normal_active.png"
theme.titlebar_sticky_button_focus_inactive = theme.dir .. "/icons/titlebar/sticky_focus_inactive.png"
theme.titlebar_sticky_button_normal_inactive = theme.dir .. "/icons/titlebar/sticky_normal_inactive.png"
theme.titlebar_floating_button_focus_active = theme.dir .. "/icons/titlebar/floating_focus_active.png"
theme.titlebar_floating_button_normal_active = theme.dir .. "/icons/titlebar/floating_normal_active.png"
theme.titlebar_floating_button_focus_inactive = theme.dir .. "/icons/titlebar/floating_focus_inactive.png"
theme.titlebar_floating_button_normal_inactive = theme.dir .. "/icons/titlebar/floating_normal_inactive.png"
theme.titlebar_maximized_button_focus_active = theme.dir .. "/icons/titlebar/maximized_focus_active.png"
theme.titlebar_maximized_button_normal_active = theme.dir .. "/icons/titlebar/maximized_normal_active.png"
theme.titlebar_maximized_button_focus_inactive = theme.dir .. "/icons/titlebar/maximized_focus_inactive.png"
theme.titlebar_maximized_button_normal_inactive = theme.dir .. "/icons/titlebar/maximized_normal_inactive.png"

nconf.defaults.border_width = 0
nconf.defaults.border_color = theme.border_focus
nconf.defaults.margin = dpi(16)
-- nconf.defaults.shape = function(cr, width, height)
--     gears.shape.rounded_rect(cr, width, height, theme.border_radius)
-- end
nconf.defaults.timeout = 5
nconf.padding = dpi(16)
nconf.presets.critical.bg = theme.bg_urgent
nconf.presets.critical.fg = theme.fg_urgent
nconf.presets.low.border_color = theme.border_normal
nconf.presets.low.bg = theme.bg_normal
nconf.presets.normal.bg = theme.bg_focus
nconf.defaults.icon_size = 64
nconf.spacing = 8

theme.notification_max_width = dpi(500)

local markup = lain.util.markup
local separators = lain.util.separators

local keyboardlayout = awful.widget.keyboardlayout:new()
keyboardlayout.widget.font = theme.font

-- Textclock
local clock = awful.widget.watch("date +'%a %d %b %R'", 60, function(widget, stdout)
    widget:set_markup(" " .. markup.font(theme.font, stdout))
end)

-- Calendar
theme.cal = lain.widget.cal({
    attach_to = { clock },
    followtag = true,
    notification_preset = {
        font = theme.font,
        fg = theme.fg_normal,
        bg = theme.bg_widget,
        icon_size = dpi(128),
        margin = dpi(10),
        shape = function(cr, width, height)
            gears.shape.rounded_rect(cr, width, height, theme.border_radius)
        end,
    },
})

local colorize = function(color, text)
    return "<span foreground='" .. color .. "'>" .. text .. "</span>"
end

local centigrade = "<span font_size='small' font_stretch='condensed'>°C </span>"

-- Return first entry in the table which is higher than the given value
local get_first_higher = function(value, arr)
    for _, pair in ipairs(arr) do
        if value >= pair[1] then
            return pair[2]
        end
    end
    return nil
end

-- MEM
local mem = lain.widget.mem({
    settings = function()
        local total = tonumber(mem_now.total)
        local color = get_first_higher(tonumber(mem_now.used), {
            { total * 0.8, theme.bg_urgent },
            { total * 0.6, theme.fg_warning },
        }) or theme.fg_normal

        local used_gb = tonumber(mem_now.used) / 1024.0
        widget:set_markup(
            markup.font(
                theme.font,
                colorize(color, string.format(" 󰍛 %.1f<span font_stretch='condensed'>GB </span>", used_gb))
            )
        )
    end,
})

-- CPU
local cpu = lain.widget.cpu({
    settings = function()
        local color = get_first_higher(tonumber(cpu_now.usage), {
            { 95, theme.bg_urgent },
            { 75, theme.fg_warning },
        }) or theme.fg_normal
        widget:set_markup(markup.font(theme.font, colorize(color, "  " .. cpu_now.usage .. "% ")))
    end,
})

-- Coretemp
local tempfile = "/sys/devices/virtual/thermal/thermal_zone0/temp"
for file in io.popen("ls /sys/devices/virtual/thermal/"):lines() do
    local type = io.open("/sys/devices/virtual/thermal/" .. file .. "/type", "r"):read()
    if type == "x86_pkg_temp" then
        tempfile = "/sys/devices/virtual/thermal/" .. file .. "/temp"
    end
end

local temp = lain.widget.temp({
    tempfile = tempfile,
    settings = function()
        local color = get_first_higher(tonumber(coretemp_now), {
            { 90.0, theme.bg_urgent },
            { 75.0, theme.fg_warning },
        }) or theme.fg_normal
        widget:set_markup(markup.font(theme.font, colorize(color, "  " .. coretemp_now:sub(1, -3) .. centigrade)))
    end,
})

-- weather
local weather = lain.widget.weather({
    APPID = "7bb02484397fc49b0dcffe9d53744616",
    lat = 43.60426,
    lon = 1.44367,
    settings = function()
        local units = math.floor(weather_now["main"]["temp"])
        widget:set_markup(markup.font(theme.font, units .. centigrade))
    end,
    notification_preset = {
        font = theme.font,
        fg = theme.fg_normal,
        bg = theme.bg_widget_alt,
        icon_size = dpi(128),
        margin = dpi(10),
        shape = function(cr, width, height)
            gears.shape.rounded_rect(cr, width, height, theme.border_radius)
        end,
    },
})

-- Battery
local baticon = wibox.widget.imagebox("#FF0000")

function baticon:set_image_colored(image)
    self:set_image(gears.color.recolor_image(image, theme.fg_normal))
end

local bat = lain.widget.bat({
    settings = function()
        if bat_now.status and bat_now.status ~= "N/A" then
            if bat_now.ac_status == 1 then
                baticon:set_image_colored(theme.widget_ac)
            elseif not bat_now.perc and tonumber(bat_now.perc) <= 5 then
                baticon:set_image_colored(theme.widget_battery_empty)
            elseif not bat_now.perc and tonumber(bat_now.perc) <= 15 then
                baticon:set_image_colored(theme.widget_battery_low)
            else
                baticon:set_image_colored(theme.widget_battery)
            end
            widget:set_markup(markup.font(theme.font, " " .. bat_now.perc .. "% "))
        else
            widget:set_markup(markup.font(theme.font, " AC "))
            baticon:set_image_colored(theme.widget_ac)
        end
    end,
})

-- Do not disturb mode (disabled notifications)
-- TODO:
--  - Customize the icon
--  - Add a tooltip
--  - Add buttons to disable/enable the mode
theme.dnd = wibox.widget({
    visible = naughty.is_suspended(),
    widget = wibox.widget.textbox("🔕"),

    update = function()
        theme.dnd.visible = naughty.is_suspended()
    end,
})

-- ALSA volume
theme.volume = lain.widget.alsa({
    settings = function()
        local vol_level = tonumber(volume_now.level)
        if vol_level == nil then
            vol_level = ""
        end
        local volicon
        if volume_now.status == "off" or vol_level == nil then
            volicon = "󰖁"
        elseif vol_level == 0 then
            volicon = "󰕿"
        elseif vol_level <= 50 then
            volicon = "󰖀"
        else
            volicon = "󰕾"
        end

        widget:set_markup(markup.font(theme.font, " " .. volicon .. " " .. vol_level .. "% "))
    end,
})
theme.volume.widget:buttons(awful.util.table.join(
    awful.button({}, 1, function()
        awful.spawn.easy_async("amixer set Master toggle", function()
            theme.volume.update()
        end)
    end),
    awful.button({}, 3, function()
        awful.spawn.easy_async("switch-audio", function(stdout)
            require("naughty").notify({ text = stdout:match("(.+)\n"), ignore_suspend = true })
            theme.volume.update()
        end)
    end),
    awful.button({}, 4, function()
        awful.spawn.easy_async("amixer set Master 1%+", function()
            theme.volume.update()
        end)
    end),
    awful.button({}, 5, function()
        awful.spawn.easy_async("amixer set Master 1%-", function()
            theme.volume.update()
        end)
    end)
))

-- Separators
local spr = wibox.widget.textbox(" ")
-- local arrd = separators.arrow_right(theme.bg_widget_alt, "alpha")
local arrd = spr

local function build_bar(widgets, layout)
    if layout == nil then
        layout = wibox.layout.fixed.horizontal
    end
    local bar = { layout = layout }

    local colors = { theme.bg_widget_alt, theme.bg_widget }
    -- local arr = spr
    -- local arr = separators.arrow_left("alpha", colors[2])

    for i, widget_list in ipairs(widgets) do
        -- table.insert(bar, arr)
        if widget_list[1] == nil then
            widget_list = { widget_list }
        end
        for _, widget in ipairs(widget_list) do
            table.insert(bar, wibox.container.background(widget, colors[(i % 2) + 1]))
        end
        -- arr = spr
        -- arr = separators.arrow_left(colors[(i % 2) + 1], colors[((i + 1) % 2) + 1])
    end

    return bar
end

function theme.at_screen_connect(s)
    -- Quake application
    s.quake = lain.util.quake({
        app = awful.util.terminal,
        argname = "--name %s",
        name = "quake",
        followtag = true,
        width = 0.99,
        horiz = "center",
    })

    -- Tags
    awful.tag(awful.util.tagnames, s, awful.layout.layouts[1])

    -- Create a promptbox for each screen
    s.mypromptbox = awful.widget.prompt()
    -- We need one layoutbox per screen.
    s.mylayoutbox = awful.widget.layoutbox(s)
    s.mylayoutbox:buttons(my_table.join(
        awful.button({}, 1, function()
            awful.layout.inc(1)
        end),
        awful.button({}, 2, function()
            awful.layout.set(awful.layout.layouts[1])
        end),
        awful.button({}, 3, function()
            awful.layout.inc(-1)
        end),
        awful.button({}, 4, function()
            awful.layout.inc(1)
        end),
        awful.button({}, 5, function()
            awful.layout.inc(-1)
        end)
    ))
    -- Create a taglist widget
    s.mytaglist = awful.widget.taglist(s, awful.widget.taglist.filter.all, awful.util.taglist_buttons)

    -- Create a tasklist widget
    -- s.mytasklist = awful.widget.tasklist(s, awful.widget.tasklist.filter.currenttags, awful.util.tasklist_buttons)
    s.mytasklist = awful.widget.tasklist({
        screen = s,
        filter = awful.widget.tasklist.filter.currenttags,
        buttons = awful.util.tasklist_buttons,
        layout = {
            spacing_widget = {
                {
                    forced_width = 5,
                    thickness = 1,
                    color = theme.bg_focus,
                    widget = wibox.widget.separator,
                },
                valign = "center",
                halign = "center",
                widget = wibox.container.place,
            },
            spacing = 1,
            layout = wibox.layout.fixed.horizontal,
        },
        -- Notice that there is *NO* wibox.wibox prefix, it is a template,
        -- not a widget instance.
        widget_template = {
            -- {
            --     wibox.widget.base.make_widget(),
            --     forced_height = 5,
            --     id            = 'background_role',
            --     widget        = wibox.container.background,
            -- },
            {
                {
                    id = "clienticon",
                    widget = awful.widget.clienticon,
                },
                -- margins = 1,
                id = "background_role",
                widget = wibox.container.background,
            },
            create_callback = function(self, c, index, objects) --luacheck: no unused args
                self:get_children_by_id("clienticon")[1].client = c
            end,
            layout = wibox.layout.align.vertical,
        },
    })

    -- Create the wibox
   --  s.mywibox = awful.wibar({ position = "top", screen = s, height = dpi(26, s), bg = "alpha", fg = theme.fg_normal })
    s.mywibox = awful.wibar({ position = "top", screen = s, height = dpi(26, s), bg = "#011626", fg = theme.fg_normal })

    -- Add widgets to the wibox
    s.mywibox:setup({
        layout = wibox.layout.align.horizontal,
        {
            -- Left widgets
            layout = wibox.layout.fixed.horizontal,
            --spr,
            s.mytaglist,
            s.mypromptbox,
        },
        {
            -- Middle widget
            layout = wibox.layout.fixed.horizontal,
            s.mytasklist,
            arrd,
        },
        build_bar({ -- Right widgets
            {
                wibox.widget.systray(),
                theme.dnd,
            },
            theme.volume.widget,
            mem.widget,
            cpu.widget,
            temp.widget,
            { baticon, bat.widget },
            {
                spr,
                weather.icon,
                weather.widget,
            },
            { spr, clock, spr },
            { wibox.container.margin(s.mylayoutbox, dpi(4), dpi(4), dpi(4), dpi(4)) },
        }),
    })
end

return theme