aboutsummaryrefslogtreecommitdiff
diff options
context:
space:
mode:
-rw-r--r--runtime/queries/rust/injections.scm27
1 files changed, 27 insertions, 0 deletions
diff --git a/runtime/queries/rust/injections.scm b/runtime/queries/rust/injections.scm
index 704d8494..d191a53a 100644
--- a/runtime/queries/rust/injections.scm
+++ b/runtime/queries/rust/injections.scm
@@ -51,3 +51,30 @@
[(string_literal) (raw_string_literal)] @injection.content
)
(#set! injection.language "sql"))
+
+; Highlight SQL in `sqlx::query_unchecked!()`
+(macro_invocation
+ macro: (scoped_identifier
+ path: (identifier) @_sqlx (#eq? @_sqlx "sqlx")
+ name: (identifier) @_query_as (#eq? @_query_as "query_unchecked"))
+ (token_tree
+ ; Only the first argument is SQL
+ .
+ [(string_literal) (raw_string_literal)] @injection.content
+ )
+ (#set! injection.language "sql"))
+
+; Highlight SQL in `sqlx::query_as_unchecked!()`
+(macro_invocation
+ macro: (scoped_identifier
+ path: (identifier) @_sqlx (#eq? @_sqlx "sqlx")
+ name: (identifier) @_query_as (#eq? @_query_as "query_as_unchecked"))
+ (token_tree
+ ; Only the second argument is SQL
+ .
+ ; Allow anything as the first argument in case the user has lower case type
+ ; names for some reason
+ (_)
+ [(string_literal) (raw_string_literal)] @injection.content
+ )
+ (#set! injection.language "sql"))