-- progress data
local clients = {}
local progress = { '⠋', '⠙', '⠸', '⢰', '⣠', '⣄', '⡆', '⠇' }

-- check for lsp progress data
local function is_lsp_loading(client) return client and clients[client] and clients[client].percentage < 100 end

-- update lsp progress
local function update_lsp_progress()
  local messages = vim.lsp.util.get_progress_messages()
  for _, message in ipairs(messages) do
    if not message.name then goto continue end

    local client_name = message.name

    if not clients[client_name] then clients[client_name] = { percentage = 0, progress_index = 0 } end

    if message.done then
      clients[client_name].percentage = 100
    else
      if message.percentage then clients[client_name].percentage = message.percentage end
    end

    if clients[client_name].percentage % 5 == 0 or clients[client_name].progress_index == 0 then
      vim.opt.statusline = vim.opt.statusline
      clients[client_name].progress_index = clients[client_name].progress_index + 1
    end

    if clients[client_name].progress_index > #progress then clients[client_name].progress_index = 1 end

    ::continue::
  end
end

-- get lsp client name for buffer
local function get_lsp_client_name()
  local active_clients = vim.lsp.get_clients({ bufnr = 0 })
  local client_name

  if #active_clients > 0 then client_name = active_clients[1].name end
  return client_name
end

-- configure feline
local function config(_, opts)
  local colorscheme = vim.g.colors_name
  local palette = require('nightfox.palette').load(colorscheme)
  local feline = require('feline')
  local vi_mode = require('feline.providers.vi_mode')
  local file = require('feline.providers.file')
  local separators = require('feline.defaults').statusline.separators.default_value
  local lsp = require('feline.providers.lsp')

  local theme = {
    fg = palette.fg1,
    bg = palette.bg1,
    black = palette.black.base,
    skyblue = palette.blue.bright,
    cyan = palette.cyan.base,
    green = palette.green.base,
    oceanblue = palette.blue.base,
    magenta = palette.magenta.base,
    orange = palette.orange.base,
    red = palette.red.base,
    violet = palette.magenta.bright,
    white = palette.white.base,
    yellow = palette.yellow.base,
  }

  local c = {

    -- local function git_diff(type)
    --  ---@diagnostic disable-next-line: undefined-field
    --  local gsd = vim.b.gitsigns_status_dict
    --  if gsd and gsd[type] and gsd[type] > 0 then return tostring(gsd[type]) end
    --  return nil
    -- end

    -- left
    vim_status = {
      provider = function()
        local s
        if require('lazy.status').has_updates() then
          s = require('lazy.status').updates()
        else
          s = ''
        end
        s = string.format(' %s ', s)
        return s
      end,
      hl = { fg = palette.bg0, bg = palette.blue.base },
      right_sep = {
        always_visible = true,
        str = separators.slant_right,
        hl = { fg = palette.blue.base, bg = palette.bg0 },
      },
    },

    file_name = {
      provider = {
        name = 'file_info',
        opts = { colored_icon = false },
      },
      hl = { fg = palette.bg0, bg = palette.white.base },
      left_sep = {
        always_visible = true,
        str = string.format('%s ', separators.slant_right),
        hl = { fg = palette.bg0, bg = palette.white.base },
      },
    },

    git_branch = {
      provider = function()
        local git = require('feline.providers.git')
        local branch, icon = git.git_branch()
        local s
        if #branch > 0 then
          s = string.format(' %s%s ', icon, branch)
        else
          s = string.format(' %s ', 'Untracked')
        end
        return s
      end,
      hl = { fg = palette.bg0, bg = palette.fg3 },
      left_sep = {
        always_visible = true,
        str = string.format('%s%s', separators.block, separators.slant_right),
        hl = { fg = palette.white.base, bg = palette.fg3 },
      },
      right_sep = {
        always_visible = true,
        str = separators.slant_right,
        hl = { fg = palette.fg3, bg = palette.bg0 },
      },
    },

    -- table.insert(components.active[left], {
    --  provider = function()
    --    local status = git_diff('added')
    --    local s
    --    if status then
    --      s = string.format(' %s %s ', '', status)
    --    else
    --      s = ''
    --    end
    --    return s
    --  end,
    --  hl = { fg = palette.bg0, bg = palette.green.base },
    --  left_sep = {
    --    always_visible = true,
    --    str = separators.slant_right,
    --    hl = { fg = palette.bg0, bg = palette.green.base },
    --  },
    -- })

    -- table.insert(components.active[left], {
    --  provider = function()
    --    local status = git_diff('changed')
    --    local s
    --    if status then
    --      s = string.format(' %s %s ', '', status)
    --    else
    --      s = ''
    --    end
    --    return s
    --  end,
    --  hl = { fg = palette.bg0, bg = palette.yellow.base },
    --  left_sep = {
    --    always_visible = true,
    --    str = separators.slant_right,
    --    hl = { fg = palette.green.base, bg = palette.yellow.base },
    --  },
    -- })

    -- table.insert(components.active[left], {
    --  provider = function()
    --    local status = git_diff('removed')
    --    local s
    --    if status then
    --      s = string.format(' %s %s ', '', status)
    --    else
    --      s = ''
    --    end
    --    return s
    --  end,
    --  hl = { fg = palette.bg0, bg = palette.red.base },
    --  left_sep = {
    --    always_visible = true,
    --    str = separators.slant_right,
    --    hl = { fg = palette.yellow.base, bg = palette.red.base },
    --  },
    --  right_sep = {
    --    always_visible = true,
    --    str = separators.slant_right,
    --    hl = { fg = palette.red.base, bg = palette.bg0 },
    --  },
    -- })

    lsp = {
      provider = function()
        if not lsp.is_lsp_attached() then return ' 󱏎 LSP ' end

        local client_name = get_lsp_client_name()
        if is_lsp_loading(client_name) then
          return string.format(' %s LSP ', progress[clients[client_name].progress_index])
        else
          return ' 󱁛 LSP '
        end
      end,
      hl = function()
        if not lsp.is_lsp_attached() then return { fg = palette.bg0, bg = palette.fg3 } end

        local client_name = get_lsp_client_name()
        if is_lsp_loading(client_name) then return { fg = palette.bg0, bg = palette.yellow.base } end

        return { fg = palette.bg0, bg = palette.green.base }
      end,
      left_sep = {
        always_visible = true,
        str = separators.slant_right,
        hl = function()
          if not lsp.is_lsp_attached() then return { fg = palette.bg0, bg = palette.fg3 } end

          local client_name = get_lsp_client_name()
          if is_lsp_loading(client_name) then return { fg = palette.bg0, bg = palette.yellow.base } end

          return { fg = palette.bg0, bg = palette.green.base }
        end,
      },
      right_sep = {
        always_visible = true,
        str = separators.slant_right,
        hl = function()
          if not lsp.is_lsp_attached() then return { fg = palette.fg3, bg = 'none' } end

          local client_name = get_lsp_client_name()
          if is_lsp_loading(client_name) then return { fg = palette.yellow.base, bg = 'none' } end

          return { fg = palette.green.base, bg = 'none' }
        end,
      },
    },

    -- table.insert(components.active[left], {
    --  provider = function()
    --    local s
    --    local count = vim.tbl_count(vim.diagnostic.get(0, { severity = vim.diagnostic.severity.ERROR }))
    --    if count > 0 then
    --      s = string.format(' %s %d ', '', count)
    --    else
    --      s = ''
    --    end
    --    return s
    --  end,
    --  hl = { fg = palette.bg0, bg = palette.red.base },
    --  left_sep = {
    --    always_visible = true,
    --    str = separators.slant_right,
    --    hl = { fg = palette.bg0, bg = palette.red.base },
    --  },
    -- })

    -- table.insert(components.active[left], {
    --  provider = function()
    --    local s
    --    local count = vim.tbl_count(vim.diagnostic.get(0, { severity = vim.diagnostic.severity.WARN }))
    --    if count > 0 then
    --      s = string.format(' %s %d ', '', count)
    --    else
    --      s = ''
    --    end
    --    return s
    --  end,
    --  hl = { fg = palette.bg0, bg = palette.magenta.base },
    --  left_sep = {
    --    always_visible = true,
    --    str = separators.slant_right,
    --    hl = { fg = palette.red.base, bg = palette.magenta.base },
    --  },
    -- })

    -- table.insert(components.active[left], {
    --  provider = function()
    --    local s
    --    local count = vim.tbl_count(vim.diagnostic.get(0, { severity = vim.diagnostic.severity.INFO }))
    --    if count > 0 then
    --      s = string.format(' %s %d ', '', count)
    --    else
    --      s = ''
    --    end
    --    return s
    --  end,
    --  hl = { fg = palette.bg0, bg = palette.blue.base },
    --  left_sep = {
    --    always_visible = true,
    --    str = separators.slant_right,
    --    hl = { fg = palette.magenta.base, bg = palette.blue.base },
    --  },
    -- })

    -- table.insert(components.active[left], {
    --  provider = function()
    --    local s
    --    local count = vim.tbl_count(vim.diagnostic.get(0, { severity = vim.diagnostic.severity.HINT }))
    --    if count > 0 then
    --      s = string.format(' %s %d ', '', count)
    --    else
    --      s = ''
    --    end
    --    return s
    --  end,
    --  hl = { fg = palette.bg0, bg = palette.orange.base },
    --  left_sep = {
    --    always_visible = true,
    --    str = separators.slant_right,
    --    hl = { fg = palette.blue.base, bg = palette.orange.base },
    --  },
    --  right_sep = {
    --    always_visible = true,
    --    str = separators.slant_right,
    --    hl = { fg = palette.orange.base, bg = 'none' },
    --  },
    -- })

    -- right
    vi_mode = {
      provider = function() return string.format(' %s ', vi_mode.get_vim_mode()) end,
      hl = function() return { fg = palette.bg0, bg = vi_mode.get_mode_color() } end,
      left_sep = {
        always_visible = true,
        str = separators.slant_left,
        hl = function() return { fg = vi_mode.get_mode_color(), bg = 'none' } end,
      },
      right_sep = {
        always_visible = true,
        str = separators.slant_left,
        hl = function() return { fg = palette.bg0, bg = vi_mode.get_mode_color() } end,
      },
    },

    macro = {
      provider = function()
        local s
        local recording_register = vim.fn.reg_recording()
        if #recording_register == 0 then
          s = ''
        else
          s = string.format(' Recording @%s ', recording_register)
        end
        return s
      end,
      hl = { fg = palette.bg0, bg = palette.fg3 },
      left_sep = {
        always_visible = true,
        str = separators.slant_left,
        hl = function() return { fg = palette.fg3, bg = palette.bg0 } end,
      },
    },

    search_count = {
      provider = function()
        if vim.v.hlsearch == 0 then return '' end

        local ok, result = pcall(vim.fn.searchcount, { maxcount = 999, timeout = 250 })
        if not ok then return '' end
        if next(result) == nil then return '' end

        local denominator = math.min(result.total, result.maxcount)
        return string.format(' [%d/%d] ', result.current, denominator)
      end,
      hl = { fg = palette.bg0, bg = palette.white.base },
      left_sep = {
        always_visible = true,
        str = separators.slant_left,
        hl = function() return { fg = palette.white.base, bg = palette.fg3 } end,
      },
      right_sep = {
        always_visible = true,
        str = separators.slant_left,
        hl = { fg = palette.bg0, bg = palette.white.base },
      },
    },

    cursor_position = {
      provider = {
        name = 'position',
        opts = { padding = true },
      },
      hl = { fg = palette.bg0, bg = palette.blue.base },
      left_sep = {
        always_visible = true,
        str = string.format('%s%s', separators.slant_left, separators.block),
        hl = function() return { fg = palette.blue.base, bg = palette.bg0 } end,
      },
      right_sep = {
        always_visible = true,
        str = ' ',
        hl = { fg = palette.bg0, bg = palette.blue.base },
      },
    },

    scroll_bar = {
      provider = {
        name = 'scroll_bar',
        opts = { reverse = true },
      },
      hl = { fg = palette.blue.dim, bg = palette.blue.base },
    },

    -- inactive statusline
    in_file_info = {
      provider = function()
        if vim.api.nvim_buf_get_name(0) ~= '' then
          return file.file_info({}, { colored_icon = false })
        else
          return file.file_type({}, { colored_icon = false, case = 'lowercase' })
        end
      end,
      hl = { fg = palette.bg0, bg = palette.blue.base },
      left_sep = {
        always_visible = true,
        str = string.format('%s%s', separators.slant_left, separators.block),
        hl = { fg = palette.blue.base, bg = 'none' },
      },
      right_sep = {
        always_visible = true,
        str = ' ',
        hl = { fg = palette.bg0, bg = palette.blue.base },
      },
    },
  }

  local active = {
    { -- left
      c.vim_status,
      c.file_name,
      c.git_branch,
      c.lsp,
    },
    { -- right
      c.vi_mode,
      c.macro,
      c.search_count,
      c.cursor_position,
      c.scroll_bar,
    },
  }

  local inactive = {
    { -- left
    },
    { -- right
      c.in_file_info,
    },
  }

  opts.components = { active = active, inactive = inactive }

  feline.setup(opts)
  feline.use_theme(theme)
end

return {
  'freddiehaddad/feline.nvim',
  config = config,
  dependencies = { 'EdenEast/nightfox.nvim', 'lewis6991/gitsigns.nvim', 'nvim-tree/nvim-web-devicons' },
  init = function()
    -- use a global statusline
    -- vim.opt.laststatus = 3

    -- update statusbar when there's a plugin update
    vim.api.nvim_create_autocmd('User', {
      pattern = 'LazyCheck',
      callback = function() vim.opt.statusline = vim.opt.statusline end,
    })

    -- update statusbar with LSP progress
    vim.api.nvim_create_autocmd('User', {
      pattern = 'LspProgressUpdate',
      callback = function() update_lsp_progress() end,
    })

    -- hide the mode
    vim.opt.showmode = false

    -- hide search count on command line
    vim.opt.shortmess:append({ S = true })
  end,
  opts = {
    force_inactive = { filetypes = { '^dapui_*', '^help$', '^neotest*', '^NvimTree$', '^qf$' } },
    disable = { filetypes = { '^alpha$' } },
  },
}