dotfiles-ansible/roles/nvim/files/nvim/lua/modules/ui/components.lua

local M = {}
local colors = require('themes.' .. vim.g.colors_name .. '.colors')
local vi_mode_utils = require('feline.providers.vi_mode')

local function file_osinfo()
    local os = vim.bo.fileformat:upper()
    local icon = {}
    if os == 'UNIX' then
        icon = ' '
    elseif os == 'MAC' then
        icon = ' '
    else
        icon = ' '
    end
    return icon .. os
end

local function buffer_not_empty()
    if vim.fn.empty(vim.fn.expand('%:t')) ~= 1 then
        return true
    end
    return false
end

local function is_lsp_attached()
    return next(vim.lsp.buf_get_clients(0)) ~= nil
end

local function lsp_active_clients()
    local clients = {}

    -- Typical lsp clients
    for _, client in pairs(vim.lsp.buf_get_clients(0)) do
        if client.name ~= 'null-ls' then
            table.insert(clients, client.name)
        end
    end

    -- null-ls sources
    local null_ls_methods = require('null-ls.methods')
    local active_sources = require('null-ls.info').get_active_sources()

    local linter_method = null_ls_methods.internal['DIAGNOSTICS']
    local active_linters = active_sources[linter_method] or {}
    vim.list_extend(clients, active_linters)

    local formatter_method = null_ls_methods.internal['FORMATTING']
    local active_formatters = active_sources[formatter_method] or {}
    vim.list_extend(clients, active_formatters)

    return table.concat(clients, ', ')
end

local function lsp_diagnostic_count(bufnr)
    bufnr = bufnr or 0
    local diagnostics = vim.diagnostic.get(bufnr)
    local count = {0, 0, 0, 0}

    for _, diagnostic in ipairs(diagnostics) do
        count[diagnostic.severity] = count[diagnostic.severity] + 1
    end

    return count[vim.diagnostic.severity.ERROR],
        count[vim.diagnostic.severity.WARN],
        count[vim.diagnostic.severity.INFO],
        count[vim.diagnostic.severity.HINT]
end

-- Default colors for statusline --
M['normal_colors'] = {bg = colors.grey1, fg = colors.fg}
M['vi_mode_colors'] = {
    NORMAL        = colors.green,
    OP            = colors.green,
    INSERT        = colors.blue,
    VISUAL        = colors.yellow,
    LINES         = colors.yellow,
    BLOCK         = colors.yellow,
    REPLACE       = colors.red,
    ['V-REPLACE'] = colors.red,
    ENTER         = colors.cyan,
    MORE          = colors.cyan,
    SELECT        = colors.orange,
    COMMAND       = colors.purple,
    SHELL         = colors.green,
    TERM          = colors.green,
    NONE          = colors.white2
}

-- Components --
M['dummy'] = {
    provider = '▊',
    hl = {fg = colors.blue},
    right_sep = ' '
}

M['vi_mode'] = {
    provider = ' ',
    hl = function()
        return {
            name = vi_mode_utils.get_mode_highlight_name(),
            fg = vi_mode_utils.get_mode_color()
        }
    end,
    right_sep = ' '
}

M['filesize'] = {
    provider = 'file_size',
    enabled = buffer_not_empty,
    hl = {fg = colors.fg, style = 'bold'},
    right_sep = ' '
}

M['fileinfo'] = {
    provider = {
        name = 'file_info',
        opts = {
            file_modified_icon = ' ',
            file_readonly_icon = ' ',
            type = 'base-only'
        }
    },
    enabled = buffer_not_empty,
    hl = {fg = colors.blue, style = 'bold'},
    right_sep = ' '
}

M['lineinfo'] = {
    provider = 'position',
    hl = {fg = colors.fg},
    right_sep = ' '
}

M['percent'] = {
    provider = 'line_percentage',
    hl = {fg = colors.fg, style = 'bold'},
    right_sep = '  '
}

-- M['gps'] = {
--     provider = function()
--         return require('nvim-gps').get_location()
--     end,
--     enabled = function()
--         local gps_present, gps = pcall(require, 'nvim-gps')
--         if gps_present then
--             return gps.is_available()
--         else
--             return false
--         end
--     end,
--     hl = {fg = colors.cyan}
-- }

M['codeact'] = {
    provider = function()
        return require('nvim-lightbulb').get_status_text()
    end,
    enabled = function()
        local lightbulb_present, _ = pcall(require, 'nvim-lightbulb')
        return lightbulb_present
    end,
    hl = {fg = colors.green, style = 'bold'}
}

M['diagerr'] = {
    provider = function()
        local error, _, _, _ = lsp_diagnostic_count(0)
        return error ~= 0 and tostring(error) or ''
    end,
    icon = ' ',
    enabled = is_lsp_attached,
    hl = {fg = colors.red},
    left_sep = ' '
}

M['diagwarn'] = {
    provider = function()
        local _, warn, _, _ = lsp_diagnostic_count(0)
        return warn ~= 0 and tostring(warn) or ''
    end,
    icon = ' ',
    enabled = is_lsp_attached,
    hl = {fg = colors.yellow},
    left_sep = ' '
}

M['diaghint'] = {
    provider = function()
        local _, _, _, hint = lsp_diagnostic_count(0)
        return hint ~= 0 and tostring(hint) or ''
    end,
    icon = ' ',
    enabled = is_lsp_attached,
    hl = {fg = colors.cyan},
    left_sep = ' '
}

M['diaginfo'] = {
    provider = function()
        local _, _, info, _ = lsp_diagnostic_count(0)
        return info ~= 0 and tostring(info) or ''
    end,
    icon = ' ',
    enabled = is_lsp_attached,
    hl = {fg = colors.blue},
    left_sep = ' '
}

M['lspclient'] = {
    provider = lsp_active_clients,
    enabled = is_lsp_attached,
    icon = '  LSP:',
    hl = {fg = colors.purple, style = 'bold'},
    left_sep = '  '
}

M['format'] = {
    provider = file_osinfo,
    hl = {fg = colors.cyan},
    left_sep = '  '
}

M['encode'] = {
    provider = 'file_encoding',
    hl = {fg = colors.fg},
    left_sep = ' '
}

M['filetype'] = {
    provider = 'file_type',
    hl = {fg = colors.blue, style = 'bold'},
    left_sep = '  '
}

M['gitbranch'] = {
    provider = 'git_branch',
    icon = ' ',
    hl = {fg = colors.green, style = 'bold'},
    left_sep = '  '
}

M['diffadd'] = {
    provider = 'git_diff_added',
    icon = ' ',
    hl = {fg = colors.green},
    left_sep = ' '
}

M['diffchange'] = {
    provider = 'git_diff_changed',
    icon = '柳',
    hl = {fg = colors.yellow},
    left_sep = ' '
}

M['diffremove'] = {
    provider = 'git_diff_removed',
    icon = ' ',
    hl = {fg = colors.red},
    left_sep = ' '
}

return M