aboutsummaryrefslogtreecommitdiff
path: root/runtime/queries
diff options
context:
space:
mode:
authorMaxicarlos082023-03-10 22:57:52 +0000
committerGitHub2023-03-10 22:57:52 +0000
commit3d230e701d4771377a6b3f3b8c68527af29ee066 (patch)
treea23372a842308574b9da3b61008b2b9f6c437b17 /runtime/queries
parent39073581032a015e5091497530a1b357854b4e99 (diff)
Also highlight unchecked variants of the sqlx query methods (#6256)
Diffstat (limited to 'runtime/queries')
-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"))