# homepage: https://github.com/AstroNvim/astroui
{ helpers, icons, lib, pkgs, ... }:

{
  extra = {
    packages = [
      (import ./package { inherit lib pkgs; })
    ];

    config = ''
      local sign_handlers = {}
      -- gitsigns handlers
      local gitsigns_handler = function(_)
        local gitsigns_avail, gitsigns = pcall(require, "gitsigns")
        if gitsigns_avail then vim.schedule(gitsigns.preview_hunk) end
      end
      for _, sign in ipairs { "Topdelete", "Untracked", "Add", "Change", "Changedelete", "Delete" } do
        local name = "GitSigns" .. sign
        if not sign_handlers[name] then sign_handlers[name] = gitsigns_handler end
      end
      sign_handlers["gitsigns_extmark_signs_"] = gitsigns_handler
      -- diagnostic handlers
      local diagnostics_handler = function(args)
        if args.mods:find "c" then
          vim.schedule(vim.lsp.buf.code_action)
        else
          vim.schedule(vim.diagnostic.open_float)
        end
      end
      for _, sign in ipairs { "Error", "Hint", "Info", "Warn" } do
        local name = "DiagnosticSign" .. sign
        if not sign_handlers[name] then sign_handlers[name] = diagnostics_handler end
      end
      -- DAP handlers
      local dap_breakpoint_handler = function(_)
        local dap_avail, dap = pcall(require, "dap")
        if dap_avail then vim.schedule(dap.toggle_breakpoint) end
      end
      for _, sign in ipairs { "", "Rejected", "Condition" } do
        local name = "DapBreakpoint" .. sign
        if not sign_handlers[name] then sign_handlers[name] = dap_breakpoint_handler end
      end

      require('astroui').setup({
        icons = ${helpers.toLuaObject icons},
        status = {
          fallback_colors = {
            none = "NONE",
            fg = "#abb2bf",
            bg = "#1e222a",
            dark_bg = "#2c323c",
            blue = "#61afef",
            green = "#98c379",
            grey = "#5c6370",
            bright_grey = "#777d86",
            dark_grey = "#5c5c5c",
            orange = "#ff9640",
            purple = "#c678dd",
            bright_purple = "#a9a1e1",
            red = "#e06c75",
            bright_red = "#ec5f67",
            white = "#c9c9c9",
            yellow = "#e5c07b",
            bright_yellow = "#ebae34",
          },
          modes = {
            ["n"] = { "NORMAL", "normal" },
            ["no"] = { "OP", "normal" },
            ["nov"] = { "OP", "normal" },
            ["noV"] = { "OP", "normal" },
            ["no␖"] = { "OP", "normal" },
            ["niI"] = { "NORMAL", "normal" },
            ["niR"] = { "NORMAL", "normal" },
            ["niV"] = { "NORMAL", "normal" },
            ["i"] = { "INSERT", "insert" },
            ["ic"] = { "INSERT", "insert" },
            ["ix"] = { "INSERT", "insert" },
            ["t"] = { "TERM", "terminal" },
            ["nt"] = { "TERM", "terminal" },
            ["v"] = { "VISUAL", "visual" },
            ["vs"] = { "VISUAL", "visual" },
            ["V"] = { "LINES", "visual" },
            ["Vs"] = { "LINES", "visual" },
            ["␖"] = { "BLOCK", "visual" },
            ["␖s"] = { "BLOCK", "visual" },
            ["R"] = { "REPLACE", "replace" },
            ["Rc"] = { "REPLACE", "replace" },
            ["Rx"] = { "REPLACE", "replace" },
            ["Rv"] = { "V-REPLACE", "replace" },
            ["s"] = { "SELECT", "visual" },
            ["S"] = { "SELECT", "visual" },
            ["␓"] = { "BLOCK", "visual" },
            ["c"] = { "COMMAND", "command" },
            ["cv"] = { "COMMAND", "command" },
            ["ce"] = { "COMMAND", "command" },
            ["r"] = { "PROMPT", "inactive" },
            ["rm"] = { "MORE", "inactive" },
            ["r?"] = { "CONFIRM", "inactive" },
            ["!"] = { "SHELL", "inactive" },
            ["null"] = { "null", "inactive" },
          },
          separators = {
            none = { "", "" },
            left = { "", "  " },
            right = { "  ", "" },
            center = { "  ", "  " },
            tab = { "", " " },
            breadcrumbs = "  ",
            path = "  ",
          },
          attributes = {
            buffer_active = { bold = true, italic = true },
            buffer_picker = { bold = true },
            macro_recording = { bold = true },
            git_branch = { bold = true },
            git_diff = { bold = true },
            virtual_env = { bold = true },
          },
          icon_highlights = {
            file_icon = {
              tabline = function(self) return self.is_active or self.is_visible end,
              statusline = true,
            },
          },
          sign_handlers = sign_handlers,
          setup_colors = function()
            local astroui = require "astroui"
            ---@type AstroUIStatusOpts
            local status_opts = astroui.config.status
            local color = assert(status_opts.fallback_colors)
            local get_hlgroup = astroui.get_hlgroup
            local lualine_mode = require("astroui.status.hl").lualine_mode
            local function resolve_lualine(orig, ...) return (not orig or orig == "NONE") and lualine_mode(...) or orig end

            local Normal = get_hlgroup("Normal", { fg = color.fg, bg = color.bg })
            local Comment = get_hlgroup("Comment", { fg = color.bright_grey, bg = color.bg })
            local Error = get_hlgroup("Error", { fg = color.red, bg = color.bg })
            local StatusLine = get_hlgroup("StatusLine", { fg = color.fg, bg = color.dark_bg })
            local TabLine = get_hlgroup("TabLine", { fg = color.grey, bg = color.none })
            local TabLineFill = get_hlgroup("TabLineFill", { fg = color.fg, bg = color.dark_bg })
            local TabLineSel = get_hlgroup("TabLineSel", { fg = color.fg, bg = color.none })
            local WinBar = get_hlgroup("WinBar", { fg = color.bright_grey, bg = color.bg })
            local WinBarNC = get_hlgroup("WinBarNC", { fg = color.grey, bg = color.bg })
            local Conditional = get_hlgroup("Conditional", { fg = color.bright_purple, bg = color.dark_bg })
            local String = get_hlgroup("String", { fg = color.green, bg = color.dark_bg })
            local TypeDef = get_hlgroup("TypeDef", { fg = color.yellow, bg = color.dark_bg })
            local NvimEnvironmentName = get_hlgroup("NvimEnvironmentName", { fg = color.yellow, bg = color.dark_bg })
            local GitSignsAdd = get_hlgroup("GitSignsAdd", { fg = color.green, bg = color.dark_bg })
            local GitSignsChange = get_hlgroup("GitSignsChange", { fg = color.orange, bg = color.dark_bg })
            local GitSignsDelete = get_hlgroup("GitSignsDelete", { fg = color.bright_red, bg = color.dark_bg })
            local DiagnosticError = get_hlgroup("DiagnosticError", { fg = color.bright_red, bg = color.dark_bg })
            local DiagnosticWarn = get_hlgroup("DiagnosticWarn", { fg = color.orange, bg = color.dark_bg })
            local DiagnosticInfo = get_hlgroup("DiagnosticInfo", { fg = color.white, bg = color.dark_bg })
            local DiagnosticHint = get_hlgroup("DiagnosticHint", { fg = color.bright_yellow, bg = color.dark_bg })
            local HeirlineInactive =
              resolve_lualine(get_hlgroup("HeirlineInactive", { bg = nil }).bg, "inactive", color.dark_grey)
            local HeirlineNormal = resolve_lualine(get_hlgroup("HeirlineNormal", { bg = nil }).bg, "normal", color.blue)
            local HeirlineInsert = resolve_lualine(get_hlgroup("HeirlineInsert", { bg = nil }).bg, "insert", color.green)
            local HeirlineVisual = resolve_lualine(get_hlgroup("HeirlineVisual", { bg = nil }).bg, "visual", color.purple)
            local HeirlineReplace =
              resolve_lualine(get_hlgroup("HeirlineReplace", { bg = nil }).bg, "replace", color.bright_red)
            local HeirlineCommand =
              resolve_lualine(get_hlgroup("HeirlineCommand", { bg = nil }).bg, "command", color.bright_yellow)
            local HeirlineTerminal =
              resolve_lualine(get_hlgroup("HeirlineTerminal", { bg = nil }).bg, "insert", HeirlineInsert)

            local colors = {
              close_fg = Error.fg,
              fg = StatusLine.fg,
              bg = StatusLine.bg,
              section_fg = StatusLine.fg,
              section_bg = StatusLine.bg,
              git_branch_fg = Conditional.fg,
              mode_fg = StatusLine.bg,
              treesitter_fg = String.fg,
              virtual_env_fg = NvimEnvironmentName.fg,
              scrollbar = TypeDef.fg,
              git_added = GitSignsAdd.fg,
              git_changed = GitSignsChange.fg,
              git_removed = GitSignsDelete.fg,
              diag_ERROR = DiagnosticError.fg,
              diag_WARN = DiagnosticWarn.fg,
              diag_INFO = DiagnosticInfo.fg,
              diag_HINT = DiagnosticHint.fg,
              winbar_fg = WinBar.fg,
              winbar_bg = WinBar.bg,
              winbarnc_fg = WinBarNC.fg,
              winbarnc_bg = WinBarNC.bg,
              tabline_bg = TabLineFill.bg,
              tabline_fg = TabLineFill.bg,
              buffer_fg = Comment.fg,
              buffer_path_fg = WinBarNC.fg,
              buffer_close_fg = Comment.fg,
              buffer_bg = TabLineFill.bg,
              buffer_active_fg = Normal.fg,
              buffer_active_path_fg = WinBarNC.fg,
              buffer_active_close_fg = Error.fg,
              buffer_active_bg = Normal.bg,
              buffer_visible_fg = Normal.fg,
              buffer_visible_path_fg = WinBarNC.fg,
              buffer_visible_close_fg = Error.fg,
              buffer_visible_bg = Normal.bg,
              buffer_overflow_fg = Comment.fg,
              buffer_overflow_bg = TabLineFill.bg,
              buffer_picker_fg = Error.fg,
              tab_close_fg = Error.fg,
              tab_close_bg = TabLineFill.bg,
              tab_fg = TabLine.fg,
              tab_bg = TabLine.bg,
              tab_active_fg = TabLineSel.fg,
              tab_active_bg = TabLineSel.bg,
              inactive = HeirlineInactive,
              normal = HeirlineNormal,
              insert = HeirlineInsert,
              visual = HeirlineVisual,
              replace = HeirlineReplace,
              command = HeirlineCommand,
              terminal = HeirlineTerminal,
            }

            local user_colors = status_opts.colors
            if type(user_colors) == "table" then
              colors = require("astrocore").extend_tbl(colors, user_colors)
            elseif type(user_colors) == "function" then
              local new_colors = user_colors(colors)
              if new_colors then colors = new_colors end
            end

            for _, section in ipairs {
              "git_branch",
              "file_info",
              "git_diff",
              "diagnostics",
              "lsp",
              "macro_recording",
              "mode",
              "cmd_info",
              "treesitter",
              "nav",
              "virtual_env",
            } do
              if not colors[section .. "_bg"] then colors[section .. "_bg"] = colors["section_bg"] end
              if not colors[section .. "_fg"] then colors[section .. "_fg"] = colors["section_fg"] end
            end

            return colors
          end,
        }
      })
    '';
  };
}