nvqw

1 files changed, 116 insertions, 0 deletions

diff --git a/.config/nvim/lsp/zk.lua b/.config/nvim/lsp/zk.lua
new file mode 100644
index 0000000..3faa3ba
--- /dev/null
+++ b/.config/nvim/lsp/zk.lua
@@ -0,0 +1,116 @@
+---@brief
+---
+--- https://github.com/zk-org/zk
+---
+--- A plain text note-taking assistant
+
+---List notes
+---@param client vim.lsp.Client
+---@param bufnr integer
+---@param opts table
+---@param action fun(path: string, title: string)
+local function zk_list(client, bufnr, opts, action)
+  opts = vim.tbl_extend('keep', { select = { 'path', 'title' } }, opts or {})
+  client:exec_cmd({
+    title = 'ZkList',
+    command = 'zk.list',
+    arguments = { vim.api.nvim_buf_get_name(bufnr), opts },
+  }, { bufnr = bufnr }, function(err, result)
+    if err ~= nil then
+      vim.api.nvim_echo({ { 'zk.list error\n' }, { vim.inspect(err) } }, true, {})
+      return
+    end
+    if result == nil then
+      return
+    end
+
+    vim.ui.select(result, {
+      format_item = function(item)
+        return item.title
+      end,
+    }, function(item)
+      if item ~= nil then
+        action(vim.fs.joinpath(client.root_dir, item.path), item.title)
+      end
+    end)
+  end)
+end
+
+---@type vim.lsp.Config
+return {
+  cmd = { 'zk', 'lsp' },
+  filetypes = { 'markdown' },
+  root_markers = { '.zk' },
+  workspace_required = true,
+  on_attach = function(client, bufnr)
+    vim.api.nvim_buf_create_user_command(bufnr, 'LspZkIndex', function()
+      client:exec_cmd({
+        title = 'ZkIndex',
+        command = 'zk.index',
+        arguments = { vim.api.nvim_buf_get_name(bufnr) },
+      }, { bufnr = bufnr }, function(err, result)
+        if err ~= nil then
+          vim.api.nvim_echo({ { 'zk.index error\n' }, { vim.inspect(err) } }, true, {})
+          return
+        end
+        if result ~= nil then
+          vim.api.nvim_echo({ { vim.inspect(result) } }, false, {})
+        end
+      end)
+    end, { desc = 'ZkIndex' })
+
+    vim.api.nvim_buf_create_user_command(bufnr, 'LspZkList', function()
+      zk_list(client, bufnr, {}, function(path)
+        vim.cmd('edit ' .. path)
+      end)
+    end, { desc = 'ZkList' })
+
+    vim.api.nvim_buf_create_user_command(bufnr, 'LspZkTagList', function()
+      client:exec_cmd({
+        title = 'ZkTagList',
+        command = 'zk.tag.list',
+        arguments = { vim.api.nvim_buf_get_name(bufnr) },
+      }, { bufnr = bufnr }, function(err, result)
+        if err ~= nil then
+          vim.api.nvim_echo({ { 'zk.tag.list error\n' }, { vim.inspect(err) } }, true, {})
+          return
+        end
+        if result == nil then
+          return
+        end
+
+        vim.ui.select(result, {
+          format_item = function(item)
+            return item.name
+          end,
+        }, function(item)
+          if item ~= nil then
+            zk_list(client, bufnr, { tags = { item.name } }, function(path)
+              vim.cmd('edit ' .. path)
+            end)
+          end
+        end)
+      end)
+    end, { desc = 'ZkTagList' })
+
+    vim.api.nvim_buf_create_user_command(bufnr, 'LspZkNew', function(args)
+      local title = #args.fargs >= 1 and args.fargs[1] or ''
+      local dir = #args.fargs >= 2 and args.fargs[2] or ''
+      client:exec_cmd({
+        title = 'ZkNew',
+        command = 'zk.new',
+        arguments = {
+          vim.api.nvim_buf_get_name(bufnr),
+          { title = title, dir = dir },
+        },
+      }, { bufnr = bufnr }, function(err, result)
+        if err ~= nil then
+          vim.api.nvim_echo({ { 'zk.new error\n' }, { vim.inspect(err) } }, true, {})
+          return
+        end
+
+        vim.cmd('edit ' .. result.path)
+      end)
+    end, { desc = 'ZkNew [title] [dir]', nargs = '*' })
+  end,
+}