ipynb support

Author: rchhong

conda/.condarc

@@ -1,4 +1,4 @@
-auto_activate_base: false
+auto_activate_base: true
 changeps1: false
 channel_priority: strict
 channels:

mamba/.mambarc

@@ -1,4 +1,4 @@
-auto_activate_base: false
+auto_activate_base: true
 changeps1: false
 channel_priority: strict
 channels:

nvim/.config/nvim/lua/plugins/code/init.lua

@@ -1,3 +1,4 @@
 return {
-    {import = 'plugins.code.markdown'}
-}
+    {import = 'plugins.code.markdown'},
+    {import = 'plugins.code.jupyter'}
+}

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

@@ -0,0 +1,26 @@
+return {
+    "kiyoon/jupynium.nvim",
+    build = "micromamba install -n base --yes nb_conda_kernels 2> /dev/null && micromamba run -a \"\" -n base pip install ~/.local/share/nvim/lazy/jupytext.nvim/.",
+    enabled = vim.fn.isdirectory(vim.fn.expand "~/micromamba/envs/base"),
+    dependencies = {
+      "rcarriga/nvim-notify",   -- optional
+      "stevearc/dressing.nvim", -- optional, UI for :JupyniumKernelSelect
+    },
+    opts = {
+        python_host = { "micromamba", "run", "-a", "\"\"", "-n", "base", "python" },
+        jupyter_command = { "micromamba", "run", "-a", "\"\"", "-n", "base", "jupyter" },
+        auto_start_server = {
+            enable = true,
+            file_pattern = { "*.ju.*", "*.ipynb" },
+        },
+        auto_attach_to_server = {
+            enable = true,
+            file_pattern = { "*.ju.*", "*.ipynb", "*.md" },
+        },
+        auto_sync_server = {
+            enable = true,
+            file_pattern = { "*.ju.*", "*.ipynb", "*.md"},
+        },
+        jupynium_file_pattern = { "*.ju.*", "*.ipynb" },
+    }
+}

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

@@ -0,0 +1,7 @@
+return {
+  "GCBallesteros/jupytext.nvim",
+  config = true,
+  build = "pip3 install --user jupytext"
+  -- Depending on your nvim distro or config you may need to make the loading not lazy
+  -- lazy=false,
+}

nvim/.config/nvim/lua/plugins/code/markdown/markdown-preview.lua

@@ -7,4 +7,4 @@ return {
     },
     ft = { "markdown" },
     build = function() vim.fn["mkdp#util#install"]() end,
-}
+}

nvim/.config/nvim/lua/plugins/editor/cmp.lua

@@ -26,6 +26,7 @@ return {
 
         cmp.setup({
           sources = {
+            {name = "jupynium", priority = 1000 },
             {name = 'nvim_lsp'},
             {name = 'luasnip'},
             {name = 'path'},
@@ -98,6 +99,14 @@ return {
               require('luasnip').lsp_expand(args.body)
             end,
           },
+          sorting = {
+            priority_weight = 1.0,
+            comparators = {
+              cmp.config.compare.score,            -- Jupyter kernel completion shows prior to LSP
+              cmp.config.compare.recently_used,
+              cmp.config.compare.locality,
+            },
+          }
         })
     end
 }