golang highlight sql strings in query methods

gopad-dev · Jun 23, 2024 · 2d15c85 · 2d15c85
1 parent d7b50e3
commit 2d15c85
Showing 1 changed file with 12 additions and 0 deletions.
diff --git a/config/queries/go/injections.scm b/config/queries/go/injections.scm
@@ -30,3 +30,15 @@
                 (interpreted_string_literal) @injection.content))
   (#any-of? @_method "Fprintf" "Fscanf" "Appendf" "Sscanf")
   (#set! injection.language "printf"))
+
+(call_expression
+  (selector_expression) @_function
+  (#match? @_function "^[a-zA-Z_][a-zA-Z0-9_]*.(Query|Exec|Prepare|Get|Select)(Context)?$")
+  (argument_list
+    .
+    [
+      (raw_string_literal)
+      (interpreted_string_literal)
+    ] @injection.content
+	(#offset! @injection.content 0 1 0 -1)
+    (#set! injection.language "sql")))