prompt-style.lua

---prompt style.
local os = require "os"
local string = require "string"
local table = require "table"
local lfs = require "lfs"
local ansicolors = require "ansicolors"
local prompt = require "prompt"
---@diagnostic disable: deprecated
if table.unpack == nil then table.unpack = unpack end

---wakatime
---@param cmd string
---@return string
local function wakatime(cmd)
    cmd = cmd or "wakatime-cli --write --plugin=repl-lua-wakatime " ..
        "--entity-type=app --entity=lua --alternate-language=lua --project=%s &"
    local s, _ = string.find(cmd, "%s")
    if s ~= nil then
        local project = io.popen("git rev-parse --show-toplevel 2> /dev/null"):read()
        if project == nil then
            project = lfs.currentdir()
        end
        project = string.match(project, "[^/]+$")
        cmd = string.format(cmd, project)
    end
    io.popen(cmd)
    return ""
end

---get distribution
---@return string
local function get_distribution()
    local line = io.popen("lsb_release -i 2>/dev/null"):read()
    if line == nil then return "linux" end
    return string.lower(string.gsub(line, ".*:%s*", ""))
end

---get os
---@return string
local function get_os()
    if os.getenv("PREFIX") == "/data/data/com.termux/files/usr" then
        return "android"
    end
    local binary_format = string.match(package.cpath, '([^.]+)[;|$]')
    if binary_format == "so" then
        return get_distribution()
    elseif binary_format == "dll" then
        return "windows"
    elseif binary_format == "dylib" then
        return "macos"
    end
    return "unknown"
end

---get icon
---@return string
local function get_icon()
    local icons = {
        unknown = "?",
        android = "",
        arch = "",
        centos = "",
        debian = "",
        docker = "",
        gentoo = "",
        linux = "",
        macos = "",
        ubuntu = "",
        nixos = "",
        windows = ""
    }
    return icons[get_os()]
end

---get version
---@param name string
---@param logo string
---@param format string
---@return string
local function get_version(name, logo, format)
    name = name or prompt.name
    logo = logo or ""
    format = format or " %s "
    name = string.format(format, name)
    local version = string.gsub(_VERSION, ".*%s+", "")
    return logo .. name .. version
end

---get time
---@param format string
---@return string
local function get_time(format)
    format = format or "%H:%M:%S"
    local time = os.date(format, os.time())
    if type(time) == "string" then return time end
    return ""
end

---get cwd
---@return string
local function get_cwd()
    local cwd = string.gsub(lfs.currentdir(), os.getenv("HOME") or "", "~")
    cwd = string.gsub(cwd, "[^/]+$", "%%{bright}%0")
    return cwd
end

---generate ps1.
-- `sections` is an array whose element is like `{ "white", "blue",   get_cwd }`
-- and its order determine the order of prompt sections.
---@param char string
---@param sections {(1: integer, 2: integar, 3: string | function(): string) | string}[]
---@return function(): string
local function generate_ps1(char, sections)
    char = char or "❯ "
    sections = sections or {
            ---@diagnostic disable: missing-parameter
            { "",      "",       wakatime },
            { "black", "yellow", get_icon() }, { "blue", "black", get_version() },
            { "white", "blue", get_cwd }, { "black", "white", get_time }
        }
    local sep = ""
    local format = " %s "
    return function()
        local ps1 = ""
        local last_bg = ""
        for _, v in ipairs(sections) do
            if type(v) == "string" then
                if string.match(v, "%s") then
                    format = v
                else
                    sep = v
                end
            else
                local fg, bg, text = table.unpack(v)
                if type(text) == "function" then text = text() end
                if text ~= "" then
                    text = string.format(format, text)
                    if last_bg == "" then
                        ps1 = ps1 .. "%{" .. fg .. " " .. bg .. "bg}" .. text
                    else
                        ps1 = ps1 .. "%{" .. last_bg .. " " .. bg .. "bg}" .. sep ..
                            "%{" .. fg .. "}" .. text
                    end
                    last_bg = bg
                end
            end
        end
        ps1 = ps1 .. "%{reset " .. last_bg .. "}" .. sep
        return ansicolors(ps1) .. "\n" .. char
    end
end

---@export
return {
    wakatime = wakatime,
    get_distribution = get_distribution,
    get_os = get_os,
    get_icon = get_icon,
    get_version = get_version,
    get_time = get_time,
    get_cwd = get_cwd,
    generate_ps1 = generate_ps1
}