-- Helper functions

function tr(key)
    return noctalia.tr(key)
end

function cfg(key)
    return noctalia.getConfig(key)
end

function getState(key)
    return noctalia.state.get(key)
end

function setState(key, val)
    noctalia.state.set(key, val)
end

function watchState(key, cb)
    noctalia.state.watch(key, cb)
end

local function getBackgroundColor()
    local color = cfg("panel_card_color")
    local opacity = cfg("panel_card_opacity") / 100
    return `{color}/{opacity}`
end

local function labelBox(subtext, text, glyph)
    return ui.column({
        fill = getBackgroundColor(),
        flexGrow = 1,
        height = 100,
        radius = 8,
        align = "center",
        padding = 12,
        justify = "space_between"
    }, {
        ui.glyph({
            name = glyph,
            size = 30
        }),
        ui.label({
            text = text,
            fontWeight = "bold"
        }),
        ui.label({
            text = subtext,
            fontSize = 12,
            fontWeight = "medium"
        }),
    })
end

local function statLabel(text, glyph: string?)
    local items = {}
    if glyph ~= nil then
        table.insert(items, ui.glyph({
            name = glyph,
            size = 16
        }))
    end

    table.insert(items, ui.label({
        text = text,
        fontSize = 12,
        fontWeight = "semibold"
    }))

    return ui.row({
        gap = 8,
        align = "center"
    }, items)
end

local function render()
    local buttons = {}

    if getState("isRemoteCheckRunning") then
        table.insert(buttons, ui.button({
            text = tr("panel.cancel"),
            glyph = "cancel",
            flexGrow = 1,
            variant = "destructive",
            onClick = "onCancelClicked",
        }))
    else
        if getState("isUpdateAvailable") and not cfg("hide_update_button") then
            table.insert(buttons, ui.button({
                text = tr("panel.update"),
                glyph = "progress-down",
                flexGrow = 1,
                variant = "primary",
                onClick = "onUpdateClicked",
            }))
        else
            table.insert(buttons, ui.button({
                text = tr("panel.check"),
                glyph = "search",
                flexGrow = 1,
                variant = "outline",
                onClick = "onCheckClicked",
            }))
        end
    end

    if not cfg("hide_clean_button") then
        table.insert(buttons, ui.button({
            text = tr("panel.clean"),
            glyph = "trash",
            flexGrow = 1,
            variant = "destructive",
            onClick = "onCleanClicked",
        }))
    end

    local lastCheckTimeStr = getState("lastRemoteCheckTimeStr")
    local generationItems = {
        statLabel(`SYS: {getState("nixosGenerations")}`, "device-desktop"),
        statLabel(`{tr("panel.current")}: {getState("nixosCurrentGen")}`, "device-desktop-check")
    }

    if getState("hasHomeManager") and getState("hmGenerations") ~= ""
        and getState("hmCurrentGen") ~= "" then
        table.insert(generationItems,
            statLabel(`HM: {getState("hmGenerations")}`, "home")
        )

        table.insert(generationItems,
            statLabel(`{tr("panel.current")}: {getState("hmCurrentGen")}`, "home-check")
        )
    end

    panel.render(
        ui.column({
            gap = 8,
        }, {
            ui.row({
                justify = "space_between",
                align = "center"
            }, {
                ui.label({
                    text = tr("panel.title"),
                    fontSize = 16,
                }),

                ui.label({
                    text = cfg("branch"),
                    fontSize = 12,
                }),
            }),
            ui.separator({
                thickness = 2,
                orientation = "horizontal",
                spacing = 4
            }),
            ui.row({
                gap = 8,
                align = "center",
                justify = "space_between"
            }, {
                labelBox(tr("panel.local"), getState("localHash"), "devices-pc"),
                labelBox(tr("panel.remote"), getState("remoteHash"), "world"),
            }),
            ui.separator({
                thickness = 2,
                orientation = "horizontal",
                spacing = 4
            }),
            ui.row({
                fill = getBackgroundColor(),
                align = "center",
                justify = "space_between",
                flexGrow = 1,
                paddingV = 4,
                paddingH = 8,
                radius = 8,
            }, generationItems),
            ui.row({
                fill = getBackgroundColor(),
                align = "center",
                justify = "center",
                flexGrow = 1,
                paddingV = 4,
                paddingH = 8,
                radius = 8,
            }, {
                statLabel(`{tr("panel.store_size")}: {getState("storeSize")}`, "database"),
                ui.spacer({width = 16}),
                statLabel(`{tr("panel.closure_size")}: {getState("closureSize")}`, "server"),
            }),
            ui.separator({
                thickness = 2,
                orientation = "horizontal",
                spacing = 4
            }),
            ui.label({
                text = `{tr("panel.last_checked")}: {lastCheckTimeStr}`,
                fontSize = 12
            }),
            ui.row({
                gap = 8,
                align = "center",
                justify = "space_between"
            }, buttons)
        })
    )
end

-- State watcher
watchState("isRemoteCheckRunning", render)
watchState("lastRemoteCheckTimeStr", render)
watchState("localHash", render)
watchState("remoteHash", render)
watchState("storeSize", render)
watchState("closureSize", render)

-- Hooks
function onOpen(_ctx)
    render()
end

function update()
    render()
end

-- Click handler
function onCheckClicked()
    setState("command", "check")
end

function onCancelClicked()
    setState("command", "cancel")
end

function onUpdateClicked()
    local command = cfg("update_command")
    if command == "" then
        noctalia.notifyError(tr("notification.update_command_is_empty"))
        return
    end

    if cfg("close_on_enter") then
        command = `{command} && echo && read -p "{tr("terminal.press_enter_to_exit")}"`
    end

    panel.close()
    noctalia.runInTerminal(command)
end

function onCleanClicked()
    local command = cfg("clean_command")
    if command == "" then
        noctalia.notifyError(tr("notification.clean_command_is_empty"))
        return
    end

    if cfg("close_on_enter") then
        command = `{command} && echo && read -p "{tr("terminal.press_enter_to_exit")}"`
    end

    panel.close()
    noctalia.runInTerminal(command)
end