FollieHiyuki-dotfiles/home/.config/nvim/lua/modules/completion.lua

local M = {}

function M.cmp_conf()
    local cmp = require('cmp')
    local luasnip = require('luasnip')

    local has_words_before = function()
        local cursor = vim.api.nvim_win_get_cursor(0)
        return not vim.api.nvim_get_current_line():sub(1, cursor[2]):match('^%s$')
    end

    cmp.setup {
        formatting = {
            format = function(entry, vim_item)
                local lspkind_icons = {
                    Text = '',
                    Method = '',
                    Function = '',
                    Constructor = '',
                    Field = 'ﰠ',
                    Variable = '',
                    Class = 'ﴯ',
                    Interface = '',
                    Module = '',
                    Property = 'ﰠ',
                    Unit = '塞',
                    Value = '',
                    Enum = '',
                    Keyword = '',
                    Snippet = '',
                    Color = '',
                    File = '',
                    Reference = '',
                    Folder = '',
                    EnumMember = '',
                    Constant = '',
                    Struct = 'פּ',
                    Event = '',
                    Operator = '',
                    TypeParameter = ''
                }
                -- load lspkind icons
                vim_item.kind = string.format('%s %s', lspkind_icons[vim_item.kind], vim_item.kind)

                vim_item.menu = ({
                    luasnip = '[SNIP]',
                    path = '[PATH]',
                    buffer = '[BUF]',
                    -- calc = '[CALC]',
                    -- nuspell = '[SPELL]',
                    spell = '[SPELL]',
                    emoji = '[EMOJI]',
                    nvim_lsp = '[LSP]',
                    latex_symbols = '[TEX]',
                    tmux = '[TMUX]',
                    orgmode = '[ORG]'
                })[entry.source.name]

                return vim_item
            end
        },
        mapping = {
            ['<C-Space>'] = cmp.mapping.complete(),
            ['<C-p>'] = cmp.mapping.select_prev_item(),
            ['<C-n>'] = cmp.mapping.select_next_item(),
            ['<C-d>'] = cmp.mapping.scroll_docs(-4),
            ['<C-f>'] = cmp.mapping.scroll_docs(4),
            ['<C-e>'] = cmp.mapping.close(),
            -- Change choice nodes for luasnip
            ['<C-h>'] = cmp.mapping(function(fallback)
                if luasnip.choice_active() then
                    vim.api.nvim_feedkeys(vim.api.nvim_replace_termcodes('<Plug>luasnip-prev-choice', true, true, true), '')
                else
                    fallback()
                end
            end, {'i', 's'}),
            ['<C-l>'] = cmp.mapping(function(fallback)
                if luasnip.choice_active() then
                    vim.api.nvim_feedkeys(vim.api.nvim_replace_termcodes('<Plug>luasnip-next-choice', true, true, true), '')
                else
                    fallback()
                end
            end, {'i', 's'}),
            -- supertab-like mapping
            ['<Tab>'] = cmp.mapping(function(fallback)
                if vim.fn.pumvisible() == 1 then
                    vim.api.nvim_feedkeys(vim.api.nvim_replace_termcodes('<C-n>', true, true, true), 'n', true)
                elseif has_words_before() and luasnip.expand_or_jumpable() then
                    vim.api.nvim_feedkeys(vim.api.nvim_replace_termcodes('<Plug>luasnip-expand-or-jump', true, true, true), '')
                else
                    fallback()
                end
            end, {'i', 's'}),
            ['<S-Tab>'] = cmp.mapping(function(fallback)
                if vim.fn.pumvisible() == 1 then
                    vim.api.nvim_feedkeys(vim.api.nvim_replace_termcodes('<C-p>', true, true, true), 'n', true)
                elseif luasnip.jumpable(-1) then
                    vim.api.nvim_feedkeys(vim.api.nvim_replace_termcodes('<Plug>luasnip-jump-prev', true, true, true), '', true)
                else
                    fallback()
                end
            end, {'i', 's'})
        },
        snippet = {
            expand = function(args)
                luasnip.lsp_expand(args.body)
            end
        },
        sources = {
            {name = 'luasnip'},
            {name = 'path'},
            {name = 'buffer'},
            -- {name = 'calc'},
            -- {name = 'nuspell'},
            {name = 'spell'},
            {name = 'emoji'},
            {name = 'nvim_lsp'},
            {name = 'latex_symbols'},
            {name = 'tmux'},
            {name = 'orgmode'}
        }
    }
end

function M.autopairs_conf()
    require('nvim-autopairs').setup {fast_wrap = {}}
    require('nvim-autopairs.completion.cmp').setup({
        map_cr = true, --  map <CR> on insert mode
        map_complete = true, -- it will auto insert `(` after select function or method item
        auto_select = true -- automatically select the first item
    })
end

function M.snippets_conf()
    local types = require('luasnip.util.types')

    require('luasnip').config.set_config({
        updateevents = 'TextChanged, TextChangedI',
        history = true,
        ext_opts = {
            [types.choiceNode] = {
                active = {
                    virt_text = {{'●', 'Orange'}}
                }
            },
            [types.insertNode] = {
                active = {
                    virt_text = {{'●', 'Blue'}}
                }
            }
        }
    })

    -- Loading vscode-like snippets from 'friendly-snippets'
    require('luasnip/loaders/from_vscode').load()
end

function M.coq_conf()
    -- To add snippets from lsp servers, change lsp.lua:
    -----
    -- local coq = require('coq')
    -- lspconf.<server>.setup(...) --> lspconf.<server>.setup(coq.lsp_ensure_capabilities(...))
    vim.g.coq_settings = {
        auto_start = true,
        display = {
            icons = {
                mode = 'none'
            }
        }
    }
end

function M.autotag_conf()
    require('nvim-treesitter.configs').setup {
        autotag = {enable = true}
    }
end

return M