From 3d230e701d4771377a6b3f3b8c68527af29ee066 Mon Sep 17 00:00:00 2001 From: Maxicarlos08 Date: Fri, 10 Mar 2023 22:57:52 +0000 Subject: [PATCH] Also highlight unchecked variants of the sqlx query methods (#6256) --- runtime/queries/rust/injections.scm | 27 +++++++++++++++++++++++++++ 1 file changed, 27 insertions(+) diff --git a/runtime/queries/rust/injections.scm b/runtime/queries/rust/injections.scm index 704d84947..d191a53ad 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"))