convert typst module

modules/plugins/languages/typst.nix

@@ -6,52 +6,87 @@
 }: let
   inherit (lib.options) mkEnableOption mkOption;
   inherit (lib.modules) mkIf mkMerge;
-  inherit (lib.lists) isList;
+  inherit (lib.types) nullOr enum attrsOf listOf package str;
-  inherit (lib.types) nullOr enum either attrsOf listOf package str;
   inherit (lib.attrsets) attrNames;
   inherit (lib.meta) getExe;
-  inherit (lib.nvim.lua) expToLua toLuaObject;
   inherit (lib.nvim.types) mkGrammarOption mkPluginSetupOption;
   inherit (lib.nvim.dag) entryAnywhere;
+  inherit (lib.nvim.lua) toLuaObject;
+  inherit (lib.nvim.attrsets) mapListToAttrs;
+  inherit (lib.generators) mkLuaInline;
 
   cfg = config.vim.languages.typst;
 
-  defaultServer = "tinymist";
+  defaultServers = ["tinymist"];
   servers = {
-    typst-lsp = {
+    typst_lsp = {
-      package = pkgs.typst-lsp;
+      enable = true;
-      lspConfig = ''
+      cmd = [(getExe pkgs.typst-lsp)];
-        lspconfig.typst_lsp.setup {
+      filetypes = ["typst"];
-          capabilities = capabilities,
+      root_markers = [".git"];
-          on_attach = function(client, bufnr)
+      on_attach = mkLuaInline ''
-            -- Disable semantic tokens as a workaround for a semantic token error when using non-english characters
+        function(client, bufnr)
-            client.server_capabilities.semanticTokensProvider = nil
+          -- Disable semantic tokens as a workaround for a semantic token error when using non-english characters
-          end,
+          client.server_capabilities.semanticTokensProvider = nil
-          cmd = ${
+        end
-          if isList cfg.lsp.package
-          then expToLua cfg.lsp.package
-          else ''{"${cfg.lsp.package}/bin/typst-lsp"}''
-        },
-        }
       '';
     };
 
     tinymist = {
-      package = pkgs.tinymist;
+      enable = true;
-      lspConfig = ''
+      cmd = [(getExe pkgs.tinymist)];
-        lspconfig.tinymist.setup {
+      filetypes = ["typst"];
-          capabilities = capabilities,
+      root_markers = [".git"];
-          single_file_support = true,
+      on_attach = mkLuaInline ''
-          on_attach = function(client, bufnr)
+        function(client, bufnr)
-            -- Disable semantic tokens as a workaround for a semantic token error when using non-english characters
+          local function create_tinymist_command(command_name, client, bufnr)
-            client.server_capabilities.semanticTokensProvider = nil
+            local export_type = command_name:match 'tinymist%.export(%w+)'
-          end,
+            local info_type = command_name:match 'tinymist%.(%w+)'
-          cmd = ${
+            if info_type and info_type:match '^get' then
-          if isList cfg.lsp.package
+              info_type = info_type:gsub('^get', 'Get')
-          then expToLua cfg.lsp.package
+            end
-          else ''{"${cfg.lsp.package}/bin/tinymist"}''
+            local cmd_display = export_type or info_type
-        },
+            local function run_tinymist_command()
-        }
+              local arguments = { vim.api.nvim_buf_get_name(bufnr) }
+              local title_str = export_type and ('Export ' .. cmd_display) or cmd_display
+              local function handler(err, res)
+                if err then
+                  return vim.notify(err.code .. ': ' .. err.message, vim.log.levels.ERROR)
+                end
+                vim.notify(export_type and res or vim.inspect(res), vim.log.levels.INFO)
+              end
+              if vim.fn.has 'nvim-0.11' == 1 then
+                return client:exec_cmd({
+                  title = title_str,
+                  command = command_name,
+                  arguments = arguments,
+                }, { bufnr = bufnr }, handler)
+              else
+                return vim.notify('Tinymist commands require Neovim 0.11+', vim.log.levels.WARN)
+              end
+            end
+            local cmd_name = export_type and ('LspTinymistExport' .. cmd_display) or ('LspTinymist' .. cmd_display)
+            local cmd_desc = export_type and ('Export to ' .. cmd_display) or ('Get ' .. cmd_display)
+            return run_tinymist_command, cmd_name, cmd_desc
+          end
+
+          for _, command in ipairs {
+            'tinymist.exportSvg',
+            'tinymist.exportPng',
+            'tinymist.exportPdf',
+            'tinymist.exportMarkdown',
+            'tinymist.exportText',
+            'tinymist.exportQuery',
+            'tinymist.exportAnsiHighlight',
+            'tinymist.getServerInfo',
+            'tinymist.getDocumentTrace',
+            'tinymist.getWorkspaceLabels',
+            'tinymist.getDocumentMetrics',
+          } do
+            local cmd_func, cmd_name, cmd_desc = create_tinymist_command(command, client, bufnr)
+            vim.api.nvim_buf_create_user_command(bufnr, cmd_name, cmd_func, { nargs = 0, desc = cmd_desc })
+          end
+        end
       '';
     };
   };
@@ -78,17 +113,10 @@ in {
     lsp = {
       enable = mkEnableOption "Typst LSP support (typst-lsp)" // {default = config.vim.lsp.enable;};
 
-      server = mkOption {
+      servers = mkOption {
         description = "Typst LSP server to use";
-        type = enum (attrNames servers);
+        type = listOf (enum (attrNames servers));
-        default = defaultServer;
+        default = defaultServers;
-      };
-
-      package = mkOption {
-        description = "typst-lsp package, or the command to run as a list of strings";
-        example = ''[lib.getExe pkgs.jdt-language-server "-data" "~/.cache/jdtls/workspace"]'';
-        type = either package (listOf str);
-        default = servers.${cfg.lsp.server}.package;
       };
     };
 
@@ -131,7 +159,7 @@ in {
           dependencies_bin = mkOption {
             type = attrsOf str;
             default = {
-              "tinymist" = getExe servers.tinymist.package;
+              "tinymist" = getExe pkgs.tinymist;
               "websocat" = getExe pkgs.websocat;
             };
 
@@ -169,8 +197,12 @@ in {
     })
 
     (mkIf cfg.lsp.enable {
-      vim.lsp.lspconfig.enable = true;
+      vim.lsp.servers =
-      vim.lsp.lspconfig.sources.typst-lsp = servers.${cfg.lsp.server}.lspConfig;
+        mapListToAttrs (n: {
+          name = n;
+          value = servers.${n};
+        })
+        cfg.lsp.servers;
     })
 
     # Extensions