chore(injections): add support injections for mkBefore/mkAfter in spesific namespace

Author: fmway

pkgs/injections-scm/default.nix

@@ -14,7 +14,7 @@
 
   res =
     # add mkRawFn, mkLuaFn highlight
-    lib.replaceString "mkRaw" "(mkRaw|mkRawFn|mkLuaFn)" ctx
+    lib.replaceStrings ["mkRaw" "__raw"] ["(mkRaw|mkRawFn|mkLuaFn)" "__raw|capabilities"] ctx
   + lib.concatStrings (map builtins.readFile list-scm)
   ;
 

pkgs/injections-scm/extras.scm

@@ -0,0 +1,82 @@
+(binding
+  attrpath: (attrpath
+    (identifier) @_path)
+  expression: [
+    (apply_expression
+      function: (_) @_func
+      argument: [
+        (string_expression
+          ((string_fragment) @injection.content
+            (#set! injection.language "lua")))
+        (indented_string_expression
+          ((string_fragment) @injection.content
+            (#set! injection.language "lua")))
+      ]
+      (#match? @_func "(^|\\.)(mkBefore|mkAfter)$"))
+  ]
+  (#match? @_path "(^(extraConfigLua(Pre|Post)?|capabilities))$"))
+
+
+(binding
+  attrpath: (attrpath
+    (identifier) @_path)
+  expression: [
+    (apply_expression
+      function: (_) @_func
+      argument: [
+        (string_expression
+          ((string_fragment) @injection.content
+            (#set! injection.language "vim")))
+        (indented_string_expression
+          ((string_fragment) @injection.content
+            (#set! injection.language "vim")))
+      ]
+      (#match? @_func "(^|\\.)(mkBefore|mkAfter)$"))
+  ]
+  (#match? @_path "(^extraConfigVim(Pre|Post)?)$"))
+
+(binding
+  attrpath: (attrpath
+    (identifier) @namespace
+    (identifier) @name)
+  expression: [
+    (apply_expression
+      function: (_) @_func
+      argument: [
+        (string_expression
+          ((string_fragment) @injection.content
+            (#set! injection.language "lua")))
+        (indented_string_expression
+          ((string_fragment) @injection.content
+            (#set! injection.language "lua")))
+      ]
+      (#match? @_func "(^|\\.)(mkBefore|mkAfter)$"))
+  ]
+  (#match? @namespace "^luaConfig$")
+  (#match? @name "^(pre|post|content)$"))
+
+(binding
+  attrpath: (attrpath
+    (identifier) @_path)
+  expression: [
+    (attrset_expression
+      (binding_set
+        (binding
+          attrpath: (attrpath
+            (identifier) @_nested_path)
+          expression: [
+            (apply_expression
+              function: (_) @_func
+              argument: [
+                (string_expression
+                  ((string_fragment) @injection.content
+                    (#set! injection.language "lua")))
+                (indented_string_expression
+                  ((string_fragment) @injection.content
+                    (#set! injection.language "lua")))
+              ]
+              (#match? @_func "(^|\\.)(mkBefore|mkAfter)$"))
+          ]
+          (#match? @_nested_path "^(pre|post|content)$"))))
+  ]
+  (#match? @_path "^luaConfig$"))

pkgs/injections-scm/onAttach-function.scm

@@ -10,6 +10,17 @@
     (indented_string_expression
       ((string_fragment) @injection.content
         (#set! injection.language "lua")))
+    (apply_expression
+      function: (_) @_func
+      argument: [
+        (string_expression
+          ((string_fragment) @injection.content
+            (#set! injection.language "lua")))
+        (indented_string_expression
+          ((string_fragment) @injection.content
+            (#set! injection.language "lua")))
+      ]
+      (#match? @_func "(^|\\.)(mkBefore|mkAfter)$"))
   ]
   (#match? @namespace "^onAttach$")
   (#match? @name "^function$"))
@@ -30,6 +41,17 @@
             (indented_string_expression
               ((string_fragment) @injection.content
                 (#set! injection.language "lua")))
+            (apply_expression
+              function: (_) @_func
+              argument: [
+                (string_expression
+                  ((string_fragment) @injection.content
+                    (#set! injection.language "lua")))
+                (indented_string_expression
+                  ((string_fragment) @injection.content
+                    (#set! injection.language "lua")))
+              ]
+              (#match? @_func "(^|\\.)(mkBefore|mkAfter)$"))
           ]
           (#match? @_nested_path "^function$"))))
   ]