local wibox = require("wibox")
local gears = require("gears")
local awful = require("awful")
local theme = require("beautiful")
local dpi = theme.xresources.apply_dpi

local module = {}

local client = client
local screen = screen

local function ori(pos)
    if pos == "left" or pos == "right" then
        return "v"
    end
    return "h"
end

local function list2map(list)
    local set = {}
    for _, l in ipairs(list) do
        set[l] = true
    end
    return set
end

local function doubleclicked(obj)
    if obj.doubleclick_timer then
        obj.doubleclick_timer:stop()
        obj.doubleclick_timer = nil
        return true
    end
    obj.doubleclick_timer = gears.timer.start_new(0.3, function()
        obj.doubleclick_timer = nil
    end)
    return false
end

local menu_selection_symbol
local menu_marker = function(condition)
    if condition then
        return menu_selection_symbol
    end
    return ""
end

local menu_move2tag = function(c, scr)
    local list = {}
    local s = scr or awful.screen.focused()
    local count = 0
    for _, t in pairs(s.tags) do
        if t ~= awful.screen.focused().selected_tag then
            count = count + 1
            local entry = {
                t.index .. ": " .. t.name .. menu_marker(t.selected) .. " ",
                function()
                    c:move_to_tag(t)
                end
            }
            table.insert(list, entry)
        end
    end
    if count > 0 then
        return list
    end
    return nil
end

local menu_move2screen = function(c)
    local list = {}
    local count = 0
    for s in screen do
        local desc = next(s.outputs) or ""
        if s.index ~= awful.screen.focused().index then
            count = count + 1
            local entry = {s.index .. ": " .. desc .. " ", menu_move2tag(c, s)}
            table.insert(list, entry)
        end
    end
    if count > 1 then
        return list
    end
    return nil
end

function module.menu_client(c)
    local list = {}

    local list_tags = menu_move2tag(c)
    if list_tags then
        table.insert(list, {"move to tag", list_tags})
    end

    local list_screens = menu_move2screen(c)
    if list_screens then
        table.insert(list, {"move to screen", list_screens})
    end

    table.insert(list, {
        "fullscreen" .. menu_marker(c.fullscreen),
        function()
            c.fullscreen = not c.fullscreen
            c:raise()
        end
    })

    table.insert(list, {
        "maximize" .. menu_marker(c.maximized),
        function()
            c.maximized = not c.maximized
            c:raise()
        end
    })

    table.insert(list, {
        "master" .. menu_marker(c == awful.client.getmaster()),
        function()
            c:swap(awful.client.getmaster())
        end
    })

    table.insert(list, {
        "sticky" .. menu_marker(c.sticky),
        function()
            c.sticky = not c.sticky
        end
    })

    table.insert(list, {
        "top" .. menu_marker(c.ontop),
        function()
            c.ontop = not c.ontop
        end
    })

    table.insert(list, {
        "minimize" .. menu_marker(c.minimized),
        function()
            if c.minimized then
                c.minimized = false
                c:raise()
            else
                c.minimized = true
            end
        end
    })

    table.insert(list, {
        "floating" .. menu_marker(c.floating),
        function()
            c.floating = not c.floating
        end
    })

    table.insert(list, {
        menu_marker(nil) .. "close",
        function()
            c:kill()
        end
    })

    return list
end

local rounded_corner_shape = function(radius, position)
    if position == "bottom" then
        return function(cr, width, height)
            gears.shape.partially_rounded_rect(cr, width, height, false, false, true, true, radius)
        end
    elseif position == "top" then
        return function(cr, width, height)
            gears.shape.partially_rounded_rect(cr, width, height, true, true, false, false, radius)
        end
    end
    return nil
end

local function new(config)
    local cfg = config or {}
    local positions = cfg.positions or {"left", "right", "top", "bottom"}
    local button_positions = cfg.button_positions or {"top"}
    local border_width = cfg.border_width or dpi(6)
    local rounded_corner = cfg.rounded_corner or nil

    local color_normal = cfg.color_normal or "#56666f"
    local color_focus = cfg.color_focus or "#a1bfcf"
    local color_hover = cfg.color_hover or nil

    local button_size = cfg.button_size or dpi(40)
    local spacing_widget = cfg.spacing_widget or nil

    local button_maximize_size = cfg.button_maximize_size or button_size
    local button_minimize_size = cfg.button_minimize_size or button_size
    local button_floating_size = cfg.button_floating_size or button_size
    local button_top_size = cfg.button_top_size or button_size
    local button_sticky_size = cfg.button_sticky_size or button_size
    local button_close_size = cfg.button_close_size or button_size

    local color_maximize_normal = cfg.color_maximize_normal or "#a9dd9d"
    local color_maximize_focus = cfg.color_maximize_focus or "#a9dd9d"
    local color_maximize_hover = cfg.color_maximize_hover or "#c3f7b7"

    local color_minimize_normal = cfg.color_minimize_normal or "#f0eaaa"
    local color_minimize_focus = cfg.color_minimize_focus or "#f0eaaa"
    local color_minimize_hover = cfg.color_minimize_hover or "#f6ffea"

    local color_close_normal = cfg.color_close_normal or "#fd8489"
    local color_close_focus = cfg.color_close_focus or "#fd8489"
    local color_close_hover = cfg.color_close_hover or "#ff9ea3"

    local color_floating_normal = cfg.color_floating_normal or "#ddace7"
    local color_floating_focus = cfg.color_floating_focus or "#ddace7"
    local color_floating_hover = cfg.color_floating_hover or "#f7c6ff"

    local color_sticky_normal = cfg.color_sticky_normal or "#fb8965"
    local color_sticky_focus = cfg.color_sticky_focus or "#fb8965"
    local color_sticky_hover = cfg.color_sticky_hover or "#ffa37f"

    local color_top_normal = cfg.color_top_normal or "#7fc1ca"
    local color_top_focus = cfg.color_top_focus or "#7fc1ca"
    local color_top_hover = cfg.color_top_hover or "#99dbe4"

    local stealth = cfg.stealth or false

    local show_button_tooltips = cfg.show_button_tooltips or false -- tooltip might intercept mouseclicks; not recommended!
    local show_title_tooltip = cfg.show_title_tooltip or false -- might fuck up sloppy mouse focus; not recommended!
    menu_selection_symbol = cfg.menu_selection_symbol or " ✔"

    local layout = cfg.layout or "fixed" -- "fixed" | "ratio"
    local button_ratio = cfg.button_ratio or 0.2

    local align_horizontal = cfg.align_horizontal or "right" -- "left" | "center" | "right"
    local align_vertical = cfg.align_vertical or "center" -- "top" | "center" | "bottom"
    local buttons = cfg.buttons or {"floating", "minimize", "maximize", "close"}

    local button_left_click = cfg.button_left_click or function(c)
        if c.maximized then
            c.maximized = false
        end
        c:emit_signal("request::activate", "mouse_click", {raise = true})
        awful.mouse.client.move(c)
    end
    local button_double_click = cfg.button_double_click or function(c)
        c.maximized = not c.maximized
    end
    local button_middle_click = cfg.middle_click or function(c)
        c:emit_signal("request::activate", "mouse_click", {raise = true})
        awful.mouse.client.resize(c)
    end
    local button_right_click = cfg.right_click or function(c)
        if c.client_menu then
            c.client_menu:hide()
        end
        c.client_menu = awful.menu(module.menu_client(c))
        c.client_menu:toggle()
    end

    local resize_factor = cfg.resize_factor or 0.01
    local button_wheel_up = cfg.button_wheel_up or function(_)
        awful.client.incwfact(resize_factor)
    end
    local button_wheel_down = cfg.button_wheel_down or function(_)
        awful.client.incwfact(-1 * resize_factor)
    end
    local button_back = cfg.button_back or function(_)
        awful.client.swap.byidx(-1)
    end
    local button_forward = cfg.button_forward or function(_)
        awful.client.swap.byidx(1)
    end

    local button_funcs = {}
    button_funcs[1] = function(c)
        if doubleclicked(c) then
            button_double_click(c)
        else
            button_left_click(c)
        end
    end
    button_funcs[2] = function(c)
        button_middle_click(c)
    end
    button_funcs[3] = function(c)
        button_right_click(c)
    end
    button_funcs[4] = function(c)
        button_wheel_up(c)
    end
    button_funcs[5] = function(c)
        button_wheel_down(c)
    end
    button_funcs[8] = function(c)
        button_back(c)
    end
    button_funcs[9] = function(c)
        button_forward(c)
    end
    local function handle_button_press(c, button)
        local func = button_funcs[button]
        if func then
            func(c)
        end
    end

    local button_definitions = {}
    button_definitions["maximize"] = {
        name = "maximize",
        color_normal = color_maximize_normal,
        color_focus = color_maximize_focus,
        color_hover = color_maximize_hover,
        button_size = button_maximize_size,
        action = function(cl)
            cl.maximized = not cl.maximized
        end
    }

    button_definitions["minimize"] = {
        name = "minimize",
        color_normal = color_minimize_normal,
        color_focus = color_minimize_focus,
        color_hover = color_minimize_hover,
        button_size = button_minimize_size,
        action = function(cl)
            -- for whatever reason setting minimized does not work without wrapping it.
            awful.spawn.easy_async_with_shell("sleep 0", function()
                cl.minimized = true
            end)
        end
    }

    button_definitions["floating"] = {
        name = "floating",
        color_normal = color_floating_normal,
        color_focus = color_floating_focus,
        color_hover = color_floating_hover,
        button_size = button_floating_size,
        action = function(cl)
            cl.floating = not cl.floating
        end
    }

    button_definitions["close"] = {
        name = "close",
        color_normal = color_close_normal,
        color_focus = color_close_focus,
        color_hover = color_close_hover,
        button_size = button_close_size,
        action = function(cl)
            cl:kill()
        end
    }

    button_definitions["sticky"] = {
        name = "sticky",
        color_normal = color_sticky_normal,
        color_focus = color_sticky_focus,
        color_hover = color_sticky_hover,
        button_size = button_sticky_size,
        action = function(cl)
            cl.sticky = not cl.sticky
        end
    }

    button_definitions["top"] = {
        name = "top",
        color_normal = color_top_normal,
        color_focus = color_top_focus,
        color_hover = color_top_hover,
        button_size = button_top_size,
        action = function(cl)
            cl.top = not cl.top
        end
    }

    if layout ~= "fixed" and layout ~= "ratio" then
        layout = "fixed"
    end

    if type(button_positions) == "string" then
        button_positions = {button_positions}
    end

    local smart_border_titlebars = function(c)
        local button_widgets = {}

        local border_bg = wibox.widget.base.make_widget_declarative(
            {id = "border_bg", bg = color_normal, widget = wibox.container.background})

        border_bg:connect_signal("button::press", function(_, _, _, button)
            handle_button_press(c, button)
        end)

        if color_hover then
            border_bg:connect_signal("mouse::enter", function()
                border_bg.bg = color_hover
            end)
            border_bg:connect_signal("mouse::leave", function()
                if client.focus == c then
                    border_bg.bg = color_focus
                else
                    border_bg.bg = color_normal
                end
            end)
        end

        local border_expander, border_expander_center

        if layout == "fixed" then
            border_expander_center = wibox.widget.base.make_widget_declarative(
                {
                    fill_vertical = true,
                    fill_horizontal = true,
                    content_fill_vertical = true,
                    content_fill_horizontal = true,
                    border_bg,
                    widget = wibox.container.place
                })
            border_expander = wibox.widget.base.make_widget_declarative(
                {
                    {layout = wibox.layout.fixed.horizontal},
                    border_bg,
                    {layout = wibox.layout.fixed.horizontal},
                    widget = wibox.layout.align.horizontal
                })
        end

        local _button_positions = list2map(button_positions)

        for _, pos in pairs(positions) do
            local tb = awful.titlebar(c, {size = border_width, position = pos, bg = "#00000000"})

            local btn_layout
            if layout == "fixed" then
                btn_layout = ori(pos) == "v" and wibox.layout.fixed.vertical or wibox.layout.fixed.horizontal
            end
            if layout == "ratio" then
                btn_layout = ori(pos) == "v" and wibox.layout.ratio.vertical or wibox.layout.ratio.horizontal
            end

            if _button_positions[pos] then
                -- border with buttons
                local button_layout = wibox.widget.base.make_widget_declarative(
                    {id = "button_layout", spacing_widget = spacing_widget, layout = btn_layout})

                local titlebar_widget

                if layout == "fixed" then
                    if ori(pos) == "v" then
                        local expander = align_vertical == "center" and border_expander_center or border_expander
                        titlebar_widget = wibox.widget.base.make_widget_declarative(
                            {
                                align_vertical == "top" and button_layout or expander,
                                align_vertical == "center" and button_layout or expander,
                                align_vertical == "bottom" and button_layout or expander,
                                expand = align_vertical == "center" and "none" or "inside",
                                layout = wibox.layout.align.vertical
                            })
                    else
                        local expander = align_horizontal == "center" and border_expander_center or border_expander
                        titlebar_widget = wibox.widget.base.make_widget_declarative(
                            {
                                align_horizontal == "left" and button_layout or expander,
                                align_horizontal == "center" and button_layout or expander,
                                align_horizontal == "right" and button_layout or expander,
                                expand = align_horizontal == "center" and "none" or "inside",
                                layout = wibox.layout.align.horizontal
                            })
                    end
                end

                if layout == "ratio" then
                    titlebar_widget = wibox.widget.base.make_widget_declarative(
                        {button_layout, id = "titlebar_widget", bg = color_normal, widget = wibox.container.background})
                end

                tb:setup{
                    titlebar_widget,
                    bg = "#00000000",
                    shape = rounded_corner and rounded_corner_shape(rounded_corner, pos) or nil,
                    widget = wibox.container.background()
                }

                local ratio_button_layout = wibox.widget.base.make_widget_declarative(
                    {
                        homogeneous = layout == "ratio" and true or false,
                        expand = true,
                        layout = ori(pos) == "h" and wibox.layout.grid.horizontal or wibox.layout.grid.vertical
                    })

                local list_of_buttons = {}
                for _, btn in pairs(buttons) do
                    local b = button_definitions[btn]

                    if not b then
                        -- custom button
                        b = {}
                        b.name = cfg["button_" .. btn .. "_name"] or btn
                        b.button_size = cfg["button_" .. btn .. "_size"] or button_size
                        b.color_focus = cfg["color_" .. btn .. "_focus"] or "#ff00ff"
                        b.color_normal = cfg["color_" .. btn .. "_normal"] or "#ff00ff"
                        b.color_hover = cfg["color_" .. btn .. "_normal"] or "#ff1aff"
                        b.action = cfg["button_" .. btn .. "_function"] or nil
                    end

                    local button_widget = wibox.widget.base.make_widget_declarative(
                        {
                            id = b.name,
                            forced_width = ori(pos) == "h" and b.button_size or nil,
                            forced_height = ori(pos) == "v" and b.button_size or nil,
                            bg = b.color_normal,
                            widget = wibox.container.background
                        })

                    if show_button_tooltips then
                        awful.tooltip {
                            objects = {button_widget},
                            timer_function = function()
                                return b.name
                            end
                        }
                    end

                    button_widget:connect_signal("mouse::enter", function()
                        button_widget.bg = b.color_hover
                    end)

                    button_widget:connect_signal("mouse::leave", function()
                        if stealth then
                            if c == client.focus then
                                button_widget.bg = color_focus
                            else
                                button_widget.bg = color_normal
                            end
                        else
                            if c == client.focus then
                                button_widget.bg = b.color_focus
                            else
                                button_widget.bg = b.color_normal
                            end
                        end
                    end)

                    button_widget:connect_signal("button::press", function(_, _, _, button)
                        if button == 1 then
                            if b.action then
                                b.action(c)
                            end
                        else
                            handle_button_press(c, button)
                        end
                    end)

                    table.insert(list_of_buttons, button_widget)

                    button_widgets[b.name] = button_widget
                end

                if layout == "ratio" then
                    ratio_button_layout:set_children(list_of_buttons)
                    local ratio_children = {}
                    table.insert(ratio_children, border_bg)
                    table.insert(ratio_children, ratio_button_layout)
                    table.insert(ratio_children, border_bg)
                    button_layout:set_children(ratio_children)

                    if (ori(pos) == "h" and align_horizontal == "left") or (ori(pos) == "v" and align_vertical == "top") then
                        button_layout:ajust_ratio(2, 0, button_ratio, 1.0 - button_ratio)
                    end
                    if (ori(pos) == "h" and align_horizontal == "right") or
                        (ori(pos) == "v" and align_vertical == "bottom") then
                        button_layout:ajust_ratio(2, 1.0 - button_ratio, button_ratio, 0)
                    end
                    if (ori(pos) == "h" and align_horizontal == "center") or
                        (ori(pos) == "v" and align_vertical == "center") then
                        local side_ratio = (1.0 - button_ratio) / 2
                        button_layout:ajust_ratio(2, side_ratio, button_ratio, side_ratio)
                    end
                end

                if layout == "fixed" then
                    button_layout:set_children(list_of_buttons)
                end
            else
                tb:setup{
                    border_bg,
                    bg = "#00000000",
                    shape = rounded_corner and rounded_corner_shape(rounded_corner, pos) or nil,
                    widget = wibox.container.background
                }
            end
        end

        -- show client title tooltip on border hover
        if show_title_tooltip then
            awful.tooltip {
                objects = {border_bg},
                timer_function = function()
                    return c.name
                end
            }
        end

        -- focus
        c:connect_signal("focus", function(_)
            for k, btn in pairs(button_widgets) do
                local b = button_definitions[k]
                if b then
                    btn.bg = stealth and color_focus or b.color_focus
                end
            end
            border_bg.bg = color_focus
        end)

        -- unfocus
        c:connect_signal("unfocus", function(_)
            for k, btn in pairs(button_widgets) do
                local b = button_definitions[k]
                if b then
                    btn.bg = stealth and color_normal or b.color_normal
                end
            end
            border_bg.bg = color_normal
        end)
    end

    client.connect_signal("request::titlebars", smart_border_titlebars)
end

return setmetatable(module, {
    __call = function(_, ...)
        new(...)
        return module
    end
})