local empty = require('lualine.component'):extend()
function empty:draw(default_highlight)
  self.status = ''
  self.applied_separator = ''
  self:apply_highlights(default_highlight)
  self:apply_section_separators()
  return self.status
end

-- Put proper separators and gaps between components in sections
local function process_sections(sections)
  local bg = vim.g.srcery_xgray1

  for name, section in pairs(sections) do
    local left = name:sub(9, 10) < 'x'
    for pos = 1, name ~= 'lualine_z' and #section or #section - 1 do
      table.insert(section, pos * 2, { empty, color = { fg = bg, bg = bg } })
    end
    for id, comp in ipairs(section) do
      if type(comp) ~= 'table' then
        comp = { comp }
        section[id] = comp
      end
      comp.separator = left and { right = '' } or { left = '' }
    end
  end
  return sections
end

local function search_result()
  local last_search = vim.fn.getreg('/')
  if not last_search or last_search == '' then
    return ''
  end
  local searchcount = vim.fn.searchcount { maxcount = 9999 }
  if searchcount.total == 0 then
    return ''
  end
  return last_search .. '(' .. searchcount.current .. '/' .. searchcount.total .. ')'
end

local function recording_macro()
  local reg = vim.api.nvim_call_function("reg_recording", {})
  if reg ~= "" then
    return "@" .. reg
  else
    return ""
  end
end

local function modified()
  if vim.bo.modified then
    return '+'
  elseif vim.bo.modifiable == false or vim.bo.readonly == true then
    return '-'
  end
  return ''
end


return {
  "nvim-lualine/lualine.nvim",
  dependencies = {"nvim-tree/nvim-web-devicons"},
  config = function()
    local bg = vim.g.srcery_xgray1

    local colors = {
      red = vim.g.srcery_red,
      grey = vim.g.srcery_black,
      black = vim.g.srcery_hard_black,
      white = vim.g.srcery_bright_white,
      light_green = vim.g.srcery_bright_green,
      orange = vim.g.srcery_orange,
      green = vim.g.srcery_green,
    }

    local theme = {
      normal = {
        a = { fg = colors.white, bg = colors.black },
        b = { fg = colors.white, bg = colors.grey },
        c = { fg = colors.black, bg = bg },
        z = { fg = colors.white, bg = colors.black },
      },
      insert = { a = { fg = colors.black, bg = colors.light_green } },
      visual = { a = { fg = colors.black, bg = colors.orange } },
      replace = { a = { fg = colors.black, bg = colors.green } },
    }

    local extensions = {"man", "quickfix"}
    if pcall(require, "neo-tree") then
      table.insert(extensions, "neo-tree")
    end
    if pcall(require, "luapad.statusline") then
      table.insert(extensions, {
        sections = process_sections {
          lualine_a = {"mode"},
          lualine_b = {
            {
              "diagnostics",
              source = { "nvim" },
              sections = { "error" },
              diagnostics_color = { error = { bg = colors.red, fg = colors.white } },
            },
            {
              "diagnostics",
              source = { "nvim" },
              sections = { "warn" },
              diagnostics_color = { warn = { bg = colors.orange, fg = colors.white } },
            },
            { recording_macro, color = { bg = colors.orange } }
          },
          lualine_y = { search_result, "filetype" },
          lualine_z = { "%l:%c", "%p%%/%L" },
        },
        filetypes = {"lua.luapad"}
      })
    end

    require("lualine").setup {
      options = {
        icons_enabled = true,
        theme = theme,
        component_separators = "",
        section_separators = { left = "", right = "" },
        globalstatus = true
      },
      sections = process_sections {
        lualine_a = { "mode" },
        lualine_b = {
          "branch",
           "diff",
          {
            "diagnostics",
            source = { "nvim" },
            sections = { "error" },
            diagnostics_color = { error = { bg = colors.red, fg = colors.white } },
          },
          {
            "diagnostics",
            source = { "nvim" },
            sections = { "warn" },
            diagnostics_color = { warn = { bg = colors.orange, fg = colors.white } },
          },
          { "filename", file_status = false, path = 1 },
          { modified, color = { bg = colors.red } },
          { "%w", cond = function() return vim.wo.previewwindow end },
          { "%r", cond = function() return vim.bo.readonly end },
          { "%q", cond = function() return vim.bo.buftype == "quickfix" end },
          { recording_macro, color = { bg = colors.orange } }
        },
        lualine_c = {},
        lualine_x = {},
        lualine_y = { search_result, "filetype" },
        lualine_z = { "%l:%c", "%p%%/%L" },
      },
      inactive_sections = {
        lualine_a = {},
        lualine_b = {},
        lualine_c = {},
        lualine_x = {},
        lualine_y = {},
        lualine_z = {},
      },
      extensions = extensions
    }
  end
}