Also highlight unchecked variants of the sqlx query methods (#6256)
This commit is contained in:
parent
3907358103
commit
3d230e701d
1 changed files with 27 additions and 0 deletions
|
@ -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"))
|
||||
|
|
Loading…
Reference in a new issue