local blink = require('blink.cmp')
local icons = require('lib.icons')

blink.setup({
    -- 'default', 'super-tab', 'enter'
    keymap = {
        preset = 'enter',
        ['<Tab>'] = { 'select_next', 'snippet_forward', 'fallback' },
        ['<S-Tab>'] = { 'select_prev', 'snippet_backward', 'fallback' },
        ['<C-space>'] = { 'show', 'hide' },
        ['<C-y>'] = { 'show_documentation', 'hide_documentation' },
    },
    snippets = { preset = 'luasnip' },
    completion = {
        list = {
            selection = {
                auto_insert = true,
                preselect = true,
            },
        },
        menu = { border = 'rounded' },
        documentation = { window = { border = 'rounded' } },
    },
    signature = { window = { border = 'rounded' } },
    appearance = {
        -- Sets the fallback highlight groups to nvim-cmp's highlight groups
        use_nvim_cmp_as_default = true,
        -- Set to 'mono' for 'Nerd Font Mono' or 'normal' for 'Nerd Font'
        nerd_font_variant = 'normal',
        kind_icons = icons.kind,
    },
    sources = {
        default = { 'avante', 'lazydev', 'lsp', 'path', 'snippets', 'buffer', 'copilot', 'spell' },
        providers = {
            avante = {
                module = 'blink-cmp-avante',
                name = 'Avante',
                opts = {},
            },
            copilot = {
                name = 'copilot',
                module = 'blink-cmp-copilot',
                score_offset = 90,
                async = true,
                transform_items = function(_, items)
                    local CompletionItemKind = require('blink.cmp.types').CompletionItemKind
                    local kind_idx = #CompletionItemKind + 1
                    CompletionItemKind[kind_idx] = 'Copilot'
                    for _, item in ipairs(items) do
                        item.kind = kind_idx
                    end
                    return items
                end,
            },
            lazydev = {
                name = 'LazyDev',
                module = 'lazydev.integrations.blink',
                -- make lazydev completions top priority (see `:h blink.cmp`)
                score_offset = 100,
            },
            spell = {
                name = 'Spell',
                module = 'blink-cmp-spell',
                opts = {
                    -- EXAMPLE: Only enable source in `@spell` captures, and disable it
                    -- in `@nospell` captures.
                    enable_in_context = function()
                        local curpos = vim.api.nvim_win_get_cursor(0)
                        local captures = vim.treesitter.get_captures_at_pos(0, curpos[1] - 1, curpos[2] - 1)
                        local in_spell_capture = false
                        for _, cap in ipairs(captures) do
                            if cap.capture == 'spell' then
                                in_spell_capture = true
                            elseif cap.capture == 'nospell' then
                                return false
                            end
                        end
                        return in_spell_capture
                    end,
                },
            },
        },
    },
})