nvim: fix lsp features in ipynb

Author: rchhong

nvim/.config/nvim/ftplugin/markdown.lua

@@ -1,2 +1,38 @@
--- Needed for notebooks
-require("quarto").activate()
+vim.b.slime_cell_delimiter = "```"
+
+local config = require("quarto.config").config
+local quarto = require("quarto")
+
+local function set_keymaps()
+  local b = vim.api.nvim_get_current_buf()
+  local function set(lhs, rhs)
+    vim.api.nvim_buf_set_keymap(b, "n", lhs, rhs, { silent = true, noremap = true })
+  end
+  set(config.keymap.definition, ":lua require'otter'.ask_definition()<cr>")
+  set(config.keymap.type_definition, ":lua require'otter'.ask_type_definition()<cr>")
+  set(config.keymap.hover, ":lua require'otter'.ask_hover()<cr>")
+  set(config.keymap.rename, ":lua require'otter'.ask_rename()<cr>")
+  set(config.keymap.references, ":lua require'otter'.ask_references()<cr>")
+  set(config.keymap.document_symbols, ":lua require'otter'.ask_document_symbols()<cr>")
+  set(config.keymap.format, ":lua require'otter'.ask_format()<cr>")
+end
+
+if config.lspFeatures.enabled then
+  quarto.activate()
+  set_keymaps()
+  -- set the keymap again if a language server attaches
+  -- directly to this buffer
+  -- because it probably overwrites these in `LspAttach`
+  -- TODO: make this more robust
+  -- This currently only works if 'LspAttach' is used
+  -- directly, e.g. in LazyVim
+  -- It does no work if the `on_attach` callback
+  -- is used in the lspconfig setup
+  -- because this gets executed after the `LspAttach` autocommand
+  -- <https://github.com/neovim/neovim/blob/d0d132fbd055834cbecb3d4e3a123a6ea8f099ec/runtime/lua/vim/lsp.lua#L1702-L1711>
+  vim.api.nvim_create_autocmd("LspAttach", {
+    buffer = vim.api.nvim_get_current_buf(),
+    group = vim.api.nvim_create_augroup("QuartoKeymapSetup", {}),
+    callback = set_keymaps,
+  })
+end

nvim/.config/nvim/lua/plugins/code/jupyter/otter.lua

@@ -6,5 +6,5 @@ return {
           set_filetype = true,
           write_to_disk = true,
         },
-    },
+    }
 }

nvim/.config/nvim/lua/plugins/code/jupyter/quarto.lua

@@ -1,10 +1,10 @@
 return {
     'quarto-dev/quarto-nvim',
     dependencies = {
-        'jmbuhr/otter.nvim',
         'hrsh7th/nvim-cmp',
         'neovim/nvim-lspconfig',
         'nvim-treesitter/nvim-treesitter',
+        'jmbuhr/otter.nvim',
     },
     keys = {
         {"<leader>qrc", function() require("quarto.runner").run_cell() end, desc="Quarto: run cell", silent=true},
@@ -37,15 +37,21 @@ return {
         },
         keymap = {
             -- NOTE: setup your own keymaps:
-            hover = "H",
+            hover = "K",
             definition = "gd",
+            type_definition = "go",
             rename = "<F2>",
             references = "gr",
             format = "<F3>",
+            document_symbols = "gS",
         },
         codeRunner = {
             enabled = true,
             default_method = "molten",
         },
-    }
+    },
+    config = function(_, opts)
+        require('quarto').setup(opts)
+        require('quarto').activate()
+    end
 }

nvim/.config/nvim/lua/plugins/lsp/lsp.lua

@@ -25,6 +25,7 @@ return {
     {"go", function() vim.lsp.buf.type_definition() end, mode = {"n"}, desc = "LSP: go to type definition"},
     {"gr", function() vim.lsp.buf.references() end, mode = {"n"}, desc = "LSP: list all references"},
     {"gs", function() vim.lsp.buf.signature_help() end, mode = {"n"}, desc = "LSP: signature help"},
+    {"gS", function() vim.lsp.buf.document_symbol() end, mode = {"n"}, desc = "LSP: show document symbols"},
     {"<F2>", function() vim.lsp.buf.rename() end, mode = {"n"}, desc = "LSP: rename variable"},
     {"<F4>", function() vim.lsp.buf.code_action() end, mode = {"n"}, desc = "LSP: code actions"},
     {"gl", function() vim.diagnostic.open_float() end, mode = {"n"}, desc = "LSP: open diagnostics float window"},