local conditions = require "core.lualine.conditions"
local colors = require "core.lualine.colors"

return {
  mode = {
    function()
      local alias = {
        n = "NORMAL",
        i = "INSERT",
        c = "COMMAND",
        V = "VISUAL",
        [""] = "VISUAL",
        v = "VISUAL",
        R = "REPLACE",
      }
      local alias_mode = alias[vim.fn.mode()]
      if alias_mode == nil then
        alias_mode = vim.fn.mode()
      end
      return alias_mode .. " "
      -- return " "
    end,
    left_padding = 0,
    right_padding = 0,
    condition = function()
      return true
    end,
    color = { fg = colors.fg, bg = colors.bg },
  },
  branch = {
    "branch",
    icon = " ",
    condition = function()
      return conditions.hide_in_width() and conditions.check_git_workspace()
    end,
    color = { gui = "bold", fg = colors.fg, bg = colors.bg },
  },
  filename = {
    "filename",
    condition = function()
      return true
    end,
    color = { fg = colors.fg, bg = colors.bg },
  },
  diff = {
    "diff",
    symbols = { added = "  ", modified = "柳", removed = " " },
    color_added = { fg = colors.green },
    color_modified = { fg = colors.yellow },
    color_removed = { fg = colors.red },
    condition = conditions.hide_in_width,
    color = {},
  },
  python_env = {
    function()
      local utils = require "core.lualine.utils"
      if vim.bo.filetype == "python" then
        local venv = os.getenv "CONDA_DEFAULT_ENV"
        if venv then
          return string.format("  (%s)", utils.env_cleanup(venv))
        end
        venv = os.getenv "VIRTUAL_ENV"
        if venv then
          return string.format("  (%s)", utils.env_cleanup(venv))
        end
        return ""
      end
      return ""
    end,
    condition = conditions.hide_in_width,
    color = { fg = colors.green },
  },
  diagnostics = {
    "diagnostics",
    sources = { "nvim_lsp" },
    symbols = { error = " ", warn = " ", info = " ", hint = " " },
    condition = conditions.hide_in_width,
    color = {},
  },
  treesitter = {
    function()
      if next(vim.treesitter.highlighter.active) then
        return "  "
      end
      return ""
    end,
    condition = conditions.hide_in_width,
    color = { fg = colors.green },
  },
  lsp = {
    function(msg)
      msg = msg or "LSP Inactive"
      local buf_clients = vim.lsp.buf_get_clients()
      if next(buf_clients) == nil then
        return msg
      end
      local buf_ft = vim.bo.filetype
      local buf_client_names = {}

      -- add client
      local utils = require "lsp.utils"
      local active_client = utils.get_active_client_by_ft(buf_ft)
      for _, client in pairs(buf_clients) do
        if client.name ~= "null-ls" then
          table.insert(buf_client_names, client.name)
        end
      end
      vim.list_extend(buf_client_names, active_client or {})

      -- add formatter
      local formatters = require "lsp.null-ls.formatters"
      local supported_formatters = formatters.list_supported_names(buf_ft)
      vim.list_extend(buf_client_names, supported_formatters)

      -- add linter
      local linters = require "lsp.null-ls.linters"
      local supported_linters = linters.list_supported_names(buf_ft)
      vim.list_extend(buf_client_names, supported_linters)

      return table.concat(buf_client_names, ", ")
    end,
    icon = " ",
    condition = conditions.hide_in_width,
    color = { fg = colors.fg, bg = colors.bg },
  },
  location = {
    "location",
    condition = conditions.hide_in_width,
    color = { fg = colors.fg, bg = colors.bg },
  },
  progress = {
    "progress",
    condition = conditions.hide_in_width,
    color = { fg = colors.fg, bg = colors.bg },
  },
  spaces = {
    function()
      local label = "Spaces: "
      if not vim.api.nvim_buf_get_option(0, "expandtab") then
        label = "Tab size: "
      end
      return label .. vim.api.nvim_buf_get_option(0, "shiftwidth") .. " "
    end,
    condition = conditions.hide_in_width,
    color = {},
  },
  encoding = {
    "o:encoding",
    upper = true,
    condition = conditions.hide_in_width,
    color = { fg = colors.fg, bg = colors.bg },
  },
  filetype = { "filetype", condition = conditions.hide_in_width, color = {} },
  scrollbar = {
    function()
      local current_line = vim.fn.line "."
      local total_lines = vim.fn.line "$"
      local chars = {
        "__",
        "▁▁",
        "▂▂",
        "▃▃",
        "▄▄",
        "▅▅",
        "▆▆",
        "▇▇",
        "██",
      }
      local line_ratio = current_line / total_lines
      local index = math.ceil(line_ratio * #chars)
      return chars[index]
    end,
    left_padding = 0,
    right_padding = 0,
    condition = function()
      return true
    end,
    color = { fg = colors.yellow, bg = colors.bg },
  },
}