go.nvim/lua/go/codelens.lua

local log = require"go.utils".log
local codelens = require "vim.lsp.codelens"
local api = vim.api

-- ONLY SUPPORT GOPLS

local function select(title, options, displayName)
  if #options == 0 then
    print('No code lenses available')
    return

  elseif #options == 1 then
    return options[1]
  end

  local options_strings = {title}
  for i, option in ipairs(options) do
    table.insert(options_strings, string.format('%d. %s', i, displayName(option)))
  end
  local choice = vim.fn.inputlist(options_strings)
  if choice < 1 or choice > #options then
    return
  end
  return options[choice]
end

local M = {}

function M.setup()

  vim.cmd('highlight! link LspCodeLens WarningMsg')
  vim.cmd('highlight! link LspCodeLensText WarningMsg')
  vim.cmd('highlight! link LspCodeLensTextSign LspCodeLensText')
  vim.cmd('highlight! link LspCodeLensTextSeparator Boolean')

  vim.cmd('augroup go.codelenses')
  vim.cmd('  autocmd!')
  vim.cmd('autocmd BufEnter,CursorHold,InsertLeave <buffer> lua vim.lsp.codelens.refresh()')
  vim.cmd('augroup end')
end

function M.run_action()
  log("run code len action")
  local line = api.nvim_win_get_cursor(0)[1]
  local bufnr = api.nvim_get_current_buf()
  local options = {}

  local lenses = codelens.get(bufnr)
  for _, lens in pairs(lenses) do
    if lens.range.start.line == (line - 1) then
      table.insert(options, {lens = lens})
    end
  end

  log(options)
  local option = select('Code lenses:', options, function(option)
    return option.lens.command.title
  end)
  if option then
    log(option)
    vim.lsp.buf.execute_command(option.lens.command)
  end
end

return M