tree-sitter-surrealdb

config/treesitter/treesitter.nix

@@ -6,24 +6,38 @@
       nixvimInjections = true;
 
       languageRegister.nu = "nu";
+      languageRegister.surrealdb = "surrealdb";
+
       grammarPackages = with pkgs;
         vimPlugins.nvim-treesitter.passthru.allGrammars
-        ++ [tree-sitter-grammars.tree-sitter-nu];
+        ++ (with tree-sitter-grammars; [tree-sitter-nu tree-sitter-surrealdb]);
     };
   };
 
   extraFiles = with pkgs.tree-sitter-grammars; {
     "/queries/nu/highlights.scm" = builtins.readFile "${tree-sitter-nu}/queries/nu/highlights.scm";
     "/queries/nu/injections.scm" = builtins.readFile "${tree-sitter-nu}/queries/nu/injections.scm";
+
+    "/queries/surrealdb/highlights.scm" = builtins.readFile "${tree-sitter-surrealdb}/queries/highlights.scm";
   };
 
-  filetype.extension.nu = "nu";
+  filetype = {
+    extension = {
+      nu = "nu";
+      surql = "surql";
+      surrealql = "surql";
+    };
+  };
 
   extraConfigLua = ''
     local parser_config = require("nvim-treesitter.parsers").get_parser_configs()
 
     parser_config.nu = {
       filetype = "nu",
     }
+
+    parser_config.surrealdb = {
+      filetype = "surql",
+    }
   '';
 }

flake.nix

@@ -12,6 +12,10 @@
       url = "github:nushell/tree-sitter-nu";
       flake = false;
     };
+    tree-sitter-surrealdb = {
+      url = "github:dariuscorvus/tree-sitter-surrealdb";
+      flake = false;
+    };
     pre-commit-hooks = {
       url = "github:cachix/pre-commit-hooks.nix";
       inputs.nixpkgs.follows = "nixpkgs";
@@ -23,6 +27,7 @@
     nixvim,
     flake-parts,
     tree-sitter-nu,
+    tree-sitter-surrealdb,
     pre-commit-hooks,
     ...
   } @ inputs:
@@ -55,7 +60,7 @@
           inherit system;
           overlays = builtins.attrValues {
             default = import ./overlay {
-              inherit nixvim tree-sitter-nu lib system;
+              inherit nixvim tree-sitter-nu tree-sitter-surrealdb lib system;
             };
           };
         };

overlay/default.nix

@@ -2,6 +2,7 @@
   lib,
   nixvim,
   tree-sitter-nu,
+  tree-sitter-surrealdb,
   ...
 }: let
   additions = final: _prev:
@@ -11,9 +12,15 @@
     };
 
   modifications = final: prev: {
-    tree-sitter-grammars.tree-sitter-nu = final.callPackage ../pkgs/tree-sitter-grammars/nushell.nix {
-      inherit (final.tree-sitter) buildGrammar;
-      inherit tree-sitter-nu;
+    tree-sitter-grammars = {
+      tree-sitter-nu = final.callPackage ../pkgs/tree-sitter-grammars/nushell.nix {
+        inherit (final.tree-sitter) buildGrammar;
+        inherit tree-sitter-nu;
+      };
+      tree-sitter-surrealdb = final.callPackage ../pkgs/tree-sitter-grammars/surrealdb.nix {
+        inherit (final.tree-sitter) buildGrammar;
+        inherit tree-sitter-surrealdb;
+      };
     };
   };
 in

pkgs/tree-sitter-grammars/surrealdb.nix

@@ -0,0 +1,12 @@
+{
+  tree-sitter-surrealdb,
+  buildGrammar,
+}: let
+  inherit ((builtins.fromJSON (builtins.readFile ../../flake.lock)).nodes.tree-sitter-surrealdb.locked) rev;
+  version = "0.0.0+${rev}";
+in
+  buildGrammar {
+    inherit version;
+    language = "surrealdb";
+    src = tree-sitter-surrealdb;
+  }