diff options
| author | Maxicarlos08 | 2023-03-10 22:57:52 +0000 |
|---|---|---|
| committer | GitHub | 2023-03-10 22:57:52 +0000 |
| commit | 3d230e701d4771377a6b3f3b8c68527af29ee066 (patch) | |
| tree | a23372a842308574b9da3b61008b2b9f6c437b17 | |
| parent | 39073581032a015e5091497530a1b357854b4e99 (diff) | |
Also highlight unchecked variants of the sqlx query methods (#6256)
| -rw-r--r-- | runtime/queries/rust/injections.scm | 27 |
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")) |