From c353077c0e2f34235fea12a9df66d773943ac863 Mon Sep 17 00:00:00 2001
From: Andreas Louv <andreas@louv.dk>
Date: Wed, 12 Apr 2023 10:24:12 +0200
Subject: [PATCH] Add support for a 'codeLens' language server

---
 autoload/lsp/buffer.vim | 1 +
 autoload/lsp/lsp.vim    | 2 +-
 doc/lsp.txt             | 3 ++-
 3 files changed, 4 insertions(+), 2 deletions(-)

diff --git a/autoload/lsp/buffer.vim b/autoload/lsp/buffer.vim
index 914fbb8..e341a65 100644
--- a/autoload/lsp/buffer.vim
+++ b/autoload/lsp/buffer.vim
@@ -56,6 +56,7 @@ export def BufLspServerGet(bnr: number, domain: string = null_string): dict<any>
     'documentFormatting': (lspserver) => lspserver.isDocumentFormattingProvider,
     'rename': (lspserver) => lspserver.isRenameProvider,
     'codeAction': (lspserver) => lspserver.isCodeActionProvider,
+    'codeLens': (lspserver) => lspserver.isCodeLensProvider,
   }
 
   if !SupportedCheckFns->has_key(domain)
diff --git a/autoload/lsp/lsp.vim b/autoload/lsp/lsp.vim
index 9d29193..746a3f3 100644
--- a/autoload/lsp/lsp.vim
+++ b/autoload/lsp/lsp.vim
@@ -918,7 +918,7 @@ enddef
 # Code lens
 # Uses LSP "textDocument/codeLens" request
 export def CodeLens()
-  var lspserver: dict<any> = buf.CurbufGetServerChecked()
+  var lspserver: dict<any> = buf.CurbufGetServerChecked('codeLens')
   if lspserver->empty()
     return
   endif
diff --git a/doc/lsp.txt b/doc/lsp.txt
index 7d0bb8c..e1ca37f 100644
--- a/doc/lsp.txt
+++ b/doc/lsp.txt
@@ -1373,7 +1373,8 @@ servers should be used for a given method.  The following flags are supported
 	references: true,
 	documentFormatting: true,
 	rename: true,
-	codeAction: true
+	codeAction: true,
+	codeLens: true
 }
 <
 
-- 
2.50.0