source.lua

local compe = require'compe'
local util = require'vim.lsp.util'

local Source = {}

function Source.new(client, filetype)
  local self = setmetatable({}, { __index = Source })
  self.client = client
  self.request_ids = {}
  self.filetype = filetype
  return self
end

function Source.get_metadata(self)
  return {
    priority = 1000;
    dup = 1;
    menu = '[LSP]';
    filetypes = { self.filetype };
  }
end

--- determine
function Source.determine(self, context)
  return compe.helper.determine(context, {
    trigger_characters = self:_get_paths(self.client.server_capabilities, { 'completionProvider', 'triggerCharacters' }) or {};
  })
end

--- complete
function Source.complete(self, args)
  if vim.lsp.client_is_stopped(self.client.id) then
    return args.abort()
  end
  if not self:_get_paths(self.client.server_capabilities, { 'completionProvider' }) then
    return args.abort()
  end

  local params = vim.lsp.util.make_position_params()
  params.context = {}
  params.context.triggerKind = (args.trigger_character_offset > 0 and 2 or (args.incomplete and 3 or 1))
  if args.trigger_character_offset > 0 then
    params.context.triggerCharacter = args.context.before_char
  end

  self:_request('textDocument/completion', params, function(err, response)
    if err or response == nil then
      return args.abort()
    end
    args.callback(compe.helper.convert_lsp({
      keyword_pattern_offset = args.keyword_pattern_offset,
      context = args.context,
      request = params,
      response = response,
    }))
  end)
end

--- resolve
function Source.resolve(self, args)
  local completion_item = self:_get_paths(args, { 'completed_item', 'user_data', 'compe', 'completion_item' })
  local has_resolve = self:_get_paths(self.client.server_capabilities, { 'completionProvider', 'resolveProvider' })
  if has_resolve and completion_item then
    self:_request('completionItem/resolve', completion_item, function(err, result)
      if not err and result then
        args.completed_item.user_data.compe.completion_item = result
      end
      args.callback(args.completed_item)
    end)
  else
    args.callback(args.completed_item)
  end
end

--- confirm
function Source.confirm(self, args)
  local completed_item = args.completed_item
  local completion_item = self:_get_paths(completed_item, { 'user_data', 'compe', 'completion_item' })
  local request_position = self:_get_paths(completed_item, { 'user_data', 'compe', 'request_position' })
  if completion_item then
    vim.call('compe#confirmation#lsp', {
      completed_item = completed_item,
      completion_item = completion_item,
      request_position = request_position,
    })
  end
end

--- documentation
function Source.documentation(self, args)
  local completion_item = self:_get_paths(args, { 'completed_item', 'user_data', 'compe', 'completion_item' })
  if completion_item then
    local document = self:_create_document(args.context.filetype, completion_item)
    if #document > 0 then
      args.callback(document)
    else
      args.abort()
    end
  end
end

--- _create_document
function Source._create_document(_, filetype, completion_item)
  local detail = (function()
    if completion_item.detail and completion_item.detail ~= '' then
      return string.format("```%s\n%s\n```", filetype, completion_item.detail)
    end
  end)()
  local doc = (function()
    local doc = completion_item.documentation or {}
    if type(doc) == "string" then
      if doc == "" then
        doc = nil
      else
        doc = string.format("```%s\n%s\n```", filetype, doc)
      end
    else
      doc = vim.tbl_deep_extend('force', {}, doc)
    end
    return doc
  end)()
  local items = {}
  if detail then
    table.insert(items, detail)
  end
  if doc then
    table.insert(items, doc)
  end
  return util.convert_input_to_markdown_lines(items) or {}
end

function Source._request(self, method, params, callback)
  if self.request_ids[method] ~= nil then
    self.client.cancel_request(self.request_ids[method])
    self.request_ids[method] = nil
  end

  local _, request_id
  _, request_id = self.client.request(method, params, function(arg1, arg2, arg3, arg4)
    if self.request_ids[method] ~= request_id then
      return
    end
    if type(arg4) == 'number' then
      callback(arg1, arg3) -- old signature
    else
      callback(arg1, arg2) -- new signature
    end
  end)
  self.request_ids[method] = request_id
end

--- _get_paths
function Source._get_paths(self, root, paths)
  local c = root
  for _, path in ipairs(paths) do
    c = c[path]
    if not c then
      return nil
    end
  end
  return c
end

return Source