1 files changed, 66 insertions, 12 deletions

diff --git a/lua/lspconfig/denols.lua b/lua/lspconfig/denols.lua
index 9f07ed83..df8214b6 100644
--- a/lua/lspconfig/denols.lua
+++ b/lua/lspconfig/denols.lua
@@ -4,6 +4,24 @@ local lsp = vim.lsp
 
 local server_name = "denols"
 
+local function deno_uri_to_uri(uri)
+  -- denols returns deno:/https/deno.land/std%400.85.0/http/server.ts
+  -- nvim-lsp only handles deno://
+  if string.sub(uri, 1, 6) == "deno:/" and string.sub(uri, 7, 7) ~= "/" then
+    return string.gsub(uri, "^deno:/", "deno://", 1)
+  end
+  return uri
+end
+
+local function uri_to_deno_uri(uri)
+  -- denols use deno:/ and nvim-lsp use deno:// as buffer_uri.
+  -- When buffer_uri is deno://, change uri to deno:/.
+  if string.sub(uri, 1, 7) == "deno://" and string.sub(uri, 8, 8) ~= "/" then
+    return string.gsub(uri, "^deno://", "deno:/", 1)
+  end
+  return uri
+end
+
 local function buf_cache(bufnr)
   local params = {}
   params["referrer"] = { uri = vim.uri_from_bufnr(bufnr) }
@@ -18,7 +36,7 @@ local function virtual_text_document_handler(uri, result)
 
   for client_id, res in pairs(result) do
     local lines = vim.split(res.result, "\n")
-    local bufnr = vim.uri_to_bufnr(uri)
+    local bufnr = vim.uri_to_bufnr(deno_uri_to_uri(uri))
 
     local current_buf = vim.api.nvim_buf_get_lines(bufnr, 0, -1, false)
     if #current_buf ~= 0 then return nil end
@@ -26,6 +44,7 @@ local function virtual_text_document_handler(uri, result)
     vim.api.nvim_buf_set_lines(bufnr, 0, -1, nil, lines)
     vim.api.nvim_buf_set_option(bufnr, "readonly", true)
     vim.api.nvim_buf_set_option(bufnr, "modified", false)
+    vim.api.nvim_buf_set_option(bufnr, "modifiable", false)
     lsp.buf_attach_client(bufnr, client_id)
   end
 end
@@ -40,6 +59,38 @@ local function virtual_text_document(uri)
   virtual_text_document_handler(uri, result)
 end
 
+local function denols_handler(err, method, result)
+  if not result or vim.tbl_isempty(result) then return nil end
+
+  for _, res in pairs(result) do
+    local uri = res.uri or res.targetUri
+    if string.sub(uri, 1, 6) == "deno:/" then
+      virtual_text_document(uri)
+      res['uri'] = deno_uri_to_uri(uri)
+      res['targetUri'] = deno_uri_to_uri(uri)
+    end
+  end
+
+  lsp.handlers[method](err, method, result)
+end
+
+local function denols_definition()
+  local params = lsp.util.make_position_params()
+  params.textDocument.uri = uri_to_deno_uri(params.textDocument.uri)
+  lsp.buf_request(0, "textDocument/definition", params)
+end
+
+local function denols_references(context)
+  vim.validate { context = { context, 't', true } }
+  local params = lsp.util.make_position_params()
+  params.context = context or {
+    includeDeclaration = true;
+  }
+  params[vim.type_idx] = vim.types.dictionary
+  params.textDocument.uri = uri_to_deno_uri(params.textDocument.uri)
+  lsp.buf_request(0, 'textDocument/references', params)
+end
+
 configs[server_name] = {
   default_config = {
     cmd = {"deno", "lsp"};
@@ -51,20 +102,21 @@ configs[server_name] = {
       unstable = false;
     };
     handlers = {
-      ["textDocument/definition"] = function(err, method, result)
-        if not result or vim.tbl_isempty(result) then return nil end
-
-        for _, res in pairs(result) do
-          if string.sub(res.targetUri, 1, 7) == "deno://" then
-            virtual_text_document(res.targetUri)
-          end
-        end
-
-        lsp.handlers["textDocument/definition"](err, method, result)
-      end;
+      ["textDocument/definition"] = denols_handler;
+      ["textDocument/references"] = denols_handler;
     };
   };
   commands = {
+    DenolsDefinition = {
+      denols_definition;
+      description = "Jump to definition. This handle deno:/ schema in deno:// buffer."
+    };
+    DenolsReferences = {
+      function()
+       denols_references({ includeDeclaration = true })
+      end;
+      description = "List references. This handle deno:/ schema in deno:// buffer."
+    };
     DenolsCache = {
       function()
         buf_cache(0)
@@ -84,6 +136,8 @@ Deno's built-in language server
   };
 }
 
+configs.denols.definition = denols_definition
+configs.denols.references = denols_references
 configs.denols.buf_cache = buf_cache
 configs.denols.virtual_text_document = virtual_text_document
 -- vim:et ts=2 sw=2