init.lua

local _config = require('one.config').config
local kindSymbolMap = _config.kindSymbolMap
local api = vim.api

local function configFormating(conf)
	return {
		-- https://github.com/hrsh7th/nvim-cmp/blob/main/lua/cmp/config/default.lua#L42-L47
		fields = conf.menu.fields, -- fields see ":h complete-items"
		format = function(entry, item)
			-- vim_item.kind = string.format("%s %s", kindSymbolMap[vim_item.kind], vim_item.kind)
			item.kind = kindSymbolMap[item.kind]

			local srcName = entry.source.name
			item.menu = conf.menu.kindLabels[srcName] or srcName:upper()

			local maxAbbrWidth = conf.menu.maxAbbrWidth
			local abbr = item.abbr

			if #abbr > maxAbbrWidth then item.abbr = vim.fn.strcharpart(abbr, 0, maxAbbrWidth) .. '…' end

			return item
		end,
	}
end

local function has_words_before()
	local line, col = table.unpack(api.nvim_win_get_cursor(0))
	return col ~= 0 and api.nvim_buf_get_lines(0, line - 1, line, true)[1]:sub(col, col):match('%s') == nil
end

local function configMapping(cmp, config)
	local snippy = require('snippy')
	local useFallback = function(fallback)
		fallback()
	end

	-- SelectBehavior : { Insert = "insert", Select = "select" }
	-- :lua print(vim.inspect(require('cmp').SelectBehavior))
	local behavior = cmp.SelectBehavior.Insert

	local selectNext = {
		i = function(fallback)
			if cmp.visible() then
				cmp.select_next_item({ behavior = behavior })
			elseif snippy.can_jump(1) then
				snippy.next()
			elseif snippy.can_expand(1) then
				snippy.expand()
			elseif has_words_before() then
				cmp.complete()
			else
				fallback()
			end
		end,

		s = function(fallback)
			if cmp.visible() then
				cmp.select_next_item({ behavior = behavior })
			elseif snippy.can_jump(1) then
				snippy.next()
			elseif snippy.can_expand(1) then
				snippy.expand()
			elseif has_words_before() then
				cmp.complete()
			else
				fallback()
			end
		end,

		c = function(fallback)
			if cmp.visible() then
				cmp.select_next_item({ behavior = behavior })
			elseif has_words_before() then
				cmp.complete()
			else
				fallback()
			end
		end,
	}

	local selectPrev = {
		i = function(fallback)
			if cmp.visible() then
				cmp.select_prev_item({ behavior = behavior })
			elseif snippy.can_jump(-1) then
				snippy.previous()
			else
				fallback()
			end
		end,

		s = function(fallback)
			if cmp.visible() then
				cmp.select_prev_item({ behavior = behavior })
			elseif snippy.can_jump(-1) then
				snippy.previous()
			else
				fallback()
			end
		end,

		c = function(fallback)
			if cmp.visible() then
				cmp.select_prev_item({ behavior = behavior })
			elseif has_words_before() then
				cmp.complete()
			else
				fallback()
			end
		end,
	}

	local selectPrevOrHistoryPrev = { c = useFallback }
	selectPrevOrHistoryPrev.i = selectPrev.i
	selectPrevOrHistoryPrev.s = selectPrev.s

	local selectNextOrHistoryNext = { c = useFallback }
	selectNextOrHistoryNext.i = selectNext.i
	selectNextOrHistoryNext.s = selectNext.s

	local mapping = cmp.mapping

	local selectPageUp = mapping(function()
		if cmp.visible() then
			cmp.select_prev_item({ behavior = behavior, count = config.completion.pageScrollLines })
		else
			vim.api.nvim_feedkeys(vim.api.nvim_replace_termcodes('<Left>', true, false, true), 't', false)
		end
	end, { 'i', 'c', 's' })

	local selectPageDown = mapping(function()
		if cmp.visible() then
			cmp.select_next_item({ behavior = behavior, count = config.completion.pageScrollLines })
		else
			vim.api.nvim_feedkeys(vim.api.nvim_replace_termcodes('<Right>', true, false, true), 't', false)
		end
	end, { 'i', 'c', 's' })

	local confirm = mapping(function(fallback)
		if cmp.visible() then
			local entry = cmp.get_selected_entry()
			if entry then
				-- :lua print(vim.inspect(require('cmp').ConfirmBehavior))
				-- { Insert = "insert", Replace = "replace" }
				cmp.confirm {
					behavior = cmp.ConfirmBehavior.Replace,
					select = false, -- If select = true, automatically select the first item when completion.
				}
			else
				fallback()
			end
		else
			fallback()
		end
	end, { 'i', 'c', 's' })

	return mapping.preset.insert {
		['<CR>'] = confirm,
		['<C-o>'] = confirm,
		['<Down>'] = selectNext,
		['<Up>'] = selectPrev,
		['<C-n>'] = selectNextOrHistoryNext,
		['<C-p>'] = selectPrevOrHistoryPrev,

		['<Tab>'] = selectNext,
		['<S-Tab>'] = selectPrev,
		['<C-j>'] = selectNext,
		['<C-k>'] = selectPrev,
		['<C-f>'] = selectPageDown,
		['<C-b>'] = selectPageUp,
		['<C-u>'] = mapping(mapping.scroll_docs(-4), { 'i', 's' }), -- scroll preview up
		['<C-d>'] = mapping(mapping.scroll_docs(4), { 'i', 's' }), -- scroll preview down
		['<M-c>'] = mapping(mapping.abort(), { 'i', 'c', 's' }), -- abort completion
		['<C-e>'] = { i = false },
	}
end

local M = {
	'completion',

	requires = { 'hrsh7th/nvim-cmp' },

	deps = {
		'hrsh7th/cmp-nvim-lsp', -- LSP source for nvim-cmp
		'hrsh7th/cmp-buffer', -- buffer source for nvim-cmp
		'hrsh7th/cmp-path', -- path source for nvim-cmp
		'hrsh7th/cmp-nvim-lua',
		'f3fora/cmp-spell',
		'hrsh7th/cmp-cmdline',
		{ 'dmitmel/cmp-cmdline-history', disable = true },
		'David-Kunz/cmp-npm',
		'petertriho/cmp-git',
		'ray-x/cmp-treesitter',

		require('one.plugins.completion.dynamic'),
		require('one.plugins.completion.copilot'),
		require('one.plugins.completion.tabnine'),
		require('one.plugins.completion.hover'),
		require('one.plugins.completion.snippet'),
		require('one.plugins.completion.codeium'),

	},

	highlights = require('one.plugins.completion.highlights'),
}

function M.config(config)
	local cmp = require('cmp')
	local conf = config.completion
	local normalSources = conf.sources.normal

	local function addNormalSrc(src, group_index)
		normalSources[#normalSources + 1] = { name = src, group_index = group_index or 1 }
	end

	local snippet = {}

	if pcall(require, 'cmp_tabnine') then addNormalSrc('tabnine') end
	if pcall(require, 'cmp_copilot') then addNormalSrc('copilot') end
	if pcall(require, 'cmp_treesitter') then addNormalSrc('treesitter', 2) end
	if pcall(require, 'codeium') then addNormalSrc('codeium') end
	if pcall(require, 'snippy') then
		snippet.expand = function(args)
			require('snippy').expand_snippet(args.body)
		end
	end

	cmp.setup {
		-- https://github.com/hrsh7th/nvim-cmp/wiki/Advanced-techniques#disabling-completion-in-certain-contexts-such-as-comments
		enabled = function()
			local filetype = vim.api.nvim_buf_get_option(0, 'filetype')
			return vim.fn.index({ 'TelescopePrompt', 'neo-tree', 'neo-tree-popup' }, filetype) == -1
		end,
		mapping = conf.mapping,
		formatting = configFormating(conf),
		sources = normalSources,
		snippet = snippet,
		experimental = conf.experimental,
		window = conf.window,
	}

	for _, cmd_type in pairs({ '/', '?' }) do
		cmp.setup.cmdline(cmd_type, {
			-- mapping = cmp.mapping.preset.cmdline(),
			sources = conf.sources.search,
		})
	end

	cmp.setup.cmdline(':', {
		-- mapping = cmp.mapping.preset.cmdline(),
		sources = conf.sources.cmdline,
	})

	for ft, props in pairs(conf.filetypes) do
		-- Set configuration for specific filetype.
		cmp.setup.filetype(ft, props)
	end

	vim.o.pumheight = conf.pumheight
end

M.defaultConfig = function(config)
	local cmp = require('cmp')

	return {
		'completion',
		{
			pumheight = 20, -- The window height of cmdline completion
			pageScrollLines = 8, -- Page down/up scroll lines

			experimental = {
				ghost_text = false, -- this feature conflict with copilot.vim's preview.
			},

			mapping = configMapping(cmp, config),

			-- You can specify multiple source arrays. The sources are grouped in the group_index order you specify,
			-- and the groups are displayed as a fallback, like chain completion.
			-- See ":h cmp-config.sources" for full source properties.
			sources = {
				normal = {
					{ name = 'dynamic', group_index = 1 },
					{ name = 'path', group_index = 1 },
					{ name = 'nvim_lua', group_index = 1 },
					{ name = 'snippy', group_index = 1 },
					{ name = 'nvim_lsp', group_index = 1 },
					{ name = 'buffer', group_index = 1 },
					{ name = 'spell', group_index = 2 },
				},

				search = { -- Use buffer source for `/` and '?' (if you enabled `native_menu`, this won't work anymore).
					{ name = 'buffer', group_index = 1 },
					{ name = 'path', group_index = 1 },
					{ name = 'cmdline_history', group_index = 1 },
				},

				cmdline = { -- Use cmdline & path source for ':' (if you enabled `native_menu`, this won't work anymore).
					{ name = 'path', group_index = 1 },
					{ name = 'cmdline', group_index = 1 },
					{ name = 'cmdline_history', group_index = 1 },
					{ name = 'nvim_lua', group_index = 1 },
				},
			},

			filetypes = {
				gitcommit = {
					sources = {
						{ name = 'git', group_index = 1 },
						{ name = 'buffer', group_index = 1 },
						{ name = 'path', group_index = 1 },
					},
				},
			},

			window = {
				-- https://github.com/hrsh7th/nvim-cmp/blob/main/lua/cmp/config/window.lua
				completion = cmp.config.window.bordered({
					winhighlight = 'Normal:CmpNormal,FloatBorder:CmpFloatBorder,CursorLine:CmpCursorLine,Search:None',
				}),

				documentation = cmp.config.window.bordered({
					winhighlight = 'Normal:CmpNormal,FloatBorder:CmpFloatBorder,CursorLine:CmpCursorLine,Search:None',
				}),
			},

			menu = {
				fields = { 'kind', 'abbr', 'menu' },

				kindLabels = {
					buffer = 'BUF',
					nvim_lsp = 'LSP',
					nvim_lua = 'LUA',
					neorg = 'ORG',
					tabnine = 'TABN',
					copilot = 'COPI',
					snippy = 'SNIP',
					treesitter = 'TREE',
					cmdline = 'CMD',
					cmdline_history = 'HIST',
					dynamic = 'DYNA',
				},

				maxAbbrWidth = 30,
			},
		},
	}
end

return M