"======================================================================
"
" lsp.vim - 
"
" Created by skywind on 2023/08/29
" Last Modified: 2023/08/29 17:10:58
"
"======================================================================


"----------------------------------------------------------------------
" internal
"----------------------------------------------------------------------
let s:lsp_servers = {}


"----------------------------------------------------------------------
" register lsp server
"----------------------------------------------------------------------
function! module#lsp#register(name, opts) abort
	let ni = {}
	let ni.path = get(a:opts, 'path', '')
	let ni.args = get(a:opts, 'args', [])
	let ni.root = get(a:opts, 'root', [])
	let ni.filetype = get(a:opts, 'filetype', [])
	let ni.options = get(a:opts, 'options', {})
	let ni.workspace = get(a:opts, 'workspace', {})
	let ni.config = get(a:opts, 'config', {})
	let ni.environ = get(a:opts, 'environ', {})
	let s:lsp_servers[a:name] = ni
endfunc


"----------------------------------------------------------------------
" list lsp servers
"----------------------------------------------------------------------
function! module#lsp#list(format) abort
	let output = []
	for [name, info] in items(s:lsp_servers)
		let item = {'name': name}
		if a:format == ''
			for key in keys(info)
				let item[key] = info[key]
			endfor
		elseif a:format == 'lsp'
			let item.cmd = [info.path] + info.args
		elseif a:format == 'yageppan'
			let item.path = info.path
			let item.args = info.args
			let item.filetype = info.filetype
			let item.initializationOptions = info.options
			let item.workspaceConfig = info.workspace
			let root = get(info, 'root', [])
			if len(root) > 0
				let markers = []
				for marker in root
					call add(markers, marker)
					call add(markers, marker . '/')
				endfor
				let item.rootSearch = markers
			endif
		endif
		call add(output, item)
	endfor
	return output
endfunc


"----------------------------------------------------------------------
" check type
"----------------------------------------------------------------------
function! module#lsp#type()
	if exists(':CocInstall')
		return 'coc'
	elseif exists(':LspDefinition') && exists(':LspNextDiagnostic')
		return 'lsp'
	elseif exists(':LspGotoDefinition') && exists(':LspShowSignature')
		return 'yegappan'
	elseif exists(':YcmCompleter')
		return 'ycm'
	elseif exists(':CmpStatus')
		return 'cmp'
	endif
	return ''
endfunc


"----------------------------------------------------------------------
" hover
"----------------------------------------------------------------------
function! module#lsp#hover() abort
	let tt = module#lsp#type()
	if tt == 'coc'
		if CocAction('hasProvider', 'hover')
			call CocActionAsync('doHover')
		elseif &ft == 'vim'
			call feedkeys('K', 'ni')
		endif
	elseif tt == 'yegappan' || tt == 'lsp'
		exec 'LspHover'
	elseif tt == 'ycm'
		exec "normal \<Plug>(YCMHover)"
	elseif tt == 'cmp'
		lua vim.lsp.buf.hover()
	endif
endfunc


"----------------------------------------------------------------------
" signature help
"----------------------------------------------------------------------
function! module#lsp#signature_help() abort
	let tt = module#lsp#type()
	if tt == 'coc'
	elseif tt == 'cmp'
		lua vim.lsp.buf.signature_help()
	endif
endfunc


"----------------------------------------------------------------------
" get doc
"----------------------------------------------------------------------
function! module#lsp#get_document() abort
	let tt = module#lsp#type()
	if tt == 'ycm'
		exec 'YcmCompleter GetDoc'
	elseif tt == 'coc'
	endif
endfunc


"----------------------------------------------------------------------
" 
"----------------------------------------------------------------------
function! module#lsp#get_type() abort
	let tt = module#lsp#type()
	if tt == 'ycm'
	elseif tt == 'coc'
	endif
endfunc


"----------------------------------------------------------------------
" goto definition
"----------------------------------------------------------------------
function! module#lsp#goto_definition() abort
	let tt = module#lsp#type()
	if tt == 'ycm'
		exec 'YcmCompleter GoToDefinitionElseDeclaration'
	elseif tt == 'coc'
		call CocActionAsync('jumpDefinition')
	elseif tt == 'lsp'
		exec 'LspDefinition'
	elseif tt == 'yegappan'
		exec 'LspGotoDefinition'
	endif
endfunc


"----------------------------------------------------------------------
" goto declaration
"----------------------------------------------------------------------
function! module#lsp#goto_declaration() abort
	let tt = module#lsp#type()
	if tt == 'ycm'
		exec 'YcmCompleter GoToDeclaration'
	elseif tt == 'coc'
		call CocActionAsync('jumpDeclaration')
	elseif tt == 'lsp'
		exec 'LspDeclaration'
	elseif tt == 'yegappan'
		exec 'LspGotoDeclaration'
	endif
endfunc


"----------------------------------------------------------------------
" goto references
"----------------------------------------------------------------------
function! module#lsp#goto_references() abort
	let tt = module#lsp#type()
	if tt == 'ycm'
		exec 'YcmCompleter GoToReferences'
	elseif tt == 'coc'
		call CocActionAsync('jumpReferences')
	elseif tt == 'lsp'
		exec 'LspReferences'
	elseif tt == 'yegappan'
		exec 'LspShowReferences'
	endif
endfunc


"----------------------------------------------------------------------
" goto implementation
"----------------------------------------------------------------------
function! module#lsp#goto_implementation() abort
	let tt = module#lsp#type()
	if tt == 'ycm'
		exec 'YcmCompleter GoToImplementation'
	elseif tt == 'coc'
		call CocActionAsync('jumpImplementation')
	elseif tt == 'lsp'
		exec 'LspImplementation'
	elseif tt == 'yegappan'
		exec 'LspGotoImpl'
	endif
endfunc


"----------------------------------------------------------------------
" goto type definition
"----------------------------------------------------------------------
function! module#lsp#goto_type_definition() abort
	let tt = module#lsp#type()
	if tt == 'ycm'
		exec 'YcmCompleter GoToTypeDefinition'
	elseif tt == 'coc'
		call CocActionAsync('jumpTypeDefinition')
	endif
endfunc



"----------------------------------------------------------------------
" diag
"----------------------------------------------------------------------
function! module#lsp#diag_current() abort
	let tt = module#lsp#type()
	if tt == 'ycm'
	elseif tt == 'yegappan'
		exec 'LspDiag current'
	endif
endfunc