1 files changed, 315 insertions, 0 deletions
diff --git a/runtime/queries/solidity/highlights.scm b/runtime/queries/solidity/highlights.scm
new file mode 100644
index 000000000..ca6988edb
--- /dev/null
+++ b/runtime/queries/solidity/highlights.scm
@@ -0,0 +1,315 @@
+; Pragma
+[
+  "pragma"
+  "solidity"
+] @keyword.directive
+
+(solidity_pragma_token
+  "||" @string.special.symbol)
+
+(solidity_pragma_token
+  "-" @string.special.symbol)
+
+(solidity_version_comparison_operator) @operator
+
+(solidity_version) @string.special
+
+; Literals
+[
+  (string)
+  (yul_string_literal)
+] @string
+
+(hex_string_literal
+  "hex" @string.special.symbol
+  (_) @string)
+
+(unicode_string_literal
+  "unicode" @string.special.symbol
+  (_) @string)
+
+[
+  (number_literal)
+  (yul_decimal_number)
+  (yul_hex_number)
+] @number
+
+(yul_boolean) @boolean
+
+; Variables
+[
+  (identifier)
+  (yul_identifier)
+] @variable
+
+; Types
+(type_name
+  (identifier) @type)
+
+(type_name
+  (user_defined_type
+    (identifier) @type))
+
+(type_name
+  "mapping" @function.builtin)
+
+[
+  (primitive_type)
+  (number_unit)
+] @type.builtin
+
+(contract_declaration
+  name: (identifier) @type)
+
+(struct_declaration
+  name: (identifier) @type)
+
+(struct_member
+  name: (identifier) @variable.member)
+
+(enum_declaration
+  name: (identifier) @type)
+
+(emit_statement
+  .
+  (expression
+    (identifier)) @type)
+
+; Handles ContractA, ContractB in function foo() override(ContractA, contractB) {}
+(override_specifier
+  (user_defined_type) @type)
+
+; Functions and parameters
+(function_definition
+  name: (identifier) @function)
+
+(modifier_definition
+  name: (identifier) @function)
+
+(yul_evm_builtin) @function.builtin
+
+; Use constructor coloring for special functions
+(constructor_definition
+  "constructor" @constructor)
+
+(modifier_invocation
+  (identifier) @function)
+
+; Handles expressions like structVariable.g();
+(call_expression
+  .
+  (expression
+    (member_expression
+      (identifier) @function.method.call)))
+
+; Handles expressions like g();
+(call_expression
+  .
+  (expression
+    (identifier) @function.call))
+
+; Function parameters
+(event_parameter
+  name: (_) @variable.parameter)
+
+(parameter
+  name: (_) @variable.parameter)
+
+; Yul functions
+(yul_function_call
+  function: (yul_identifier) @function.call)
+
+; Yul function parameters
+(yul_function_definition
+  .
+  (yul_identifier) @function
+  (yul_identifier) @variable.parameter)
+
+(meta_type_expression
+  "type" @keyword)
+
+(member_expression
+  property: (_) @variable.member)
+
+(call_struct_argument
+  name: (_) @variable.member)
+
+(struct_field_assignment
+  name: (identifier) @variable.member)
+
+(enum_value) @constant
+
+; Keywords
+[
+  "abstract"
+  "library"
+  "is"
+  "event"
+  "assembly"
+  "emit"
+  "override"
+  "modifier"
+  "var"
+  "let"
+  "emit"
+  "error"
+  "fallback"
+  "receive"
+  (virtual)
+] @keyword
+
+[
+  "enum"
+  "struct"
+  "contract"
+  "interface"
+] @keyword.type
+
+; FIXME: update grammar
+; (block_statement "unchecked" @keyword)
+(event_parameter
+  "indexed" @keyword)
+
+[
+  "public"
+  "internal"
+  "private"
+  "external"
+  "pure"
+  "view"
+  "payable"
+  (immutable)
+] @keyword.modifier
+
+[
+  "memory"
+  "storage"
+  "calldata"
+  "constant"
+] @keyword.modifier
+
+[
+  "for"
+  "while"
+  "do"
+  "break"
+  "continue"
+] @keyword.repeat
+
+[
+  "if"
+  "else"
+  "switch"
+  "case"
+  "default"
+] @keyword.conditional
+
+(ternary_expression
+  "?" @keyword.conditional.ternary
+  ":" @keyword.conditional.ternary)
+
+[
+  "try"
+  "catch"
+  "revert"
+] @keyword.exception
+
+[
+  "return"
+  "returns"
+  (yul_leave)
+] @keyword.return
+
+"function" @keyword.function
+
+[
+  "import"
+  "using"
+] @keyword.import
+
+(import_directive
+  "as" @keyword.import)
+
+(import_directive
+  "from" @keyword.import)
+
+((import_directive
+  source: (string) @string.special.path)
+  (#offset! @string.special.path 0 1 0 -1))
+
+; Punctuation
+[
+  "{"
+  "}"
+] @punctuation.bracket
+
+[
+  "["
+  "]"
+] @punctuation.bracket
+
+[
+  "("
+  ")"
+] @punctuation.bracket
+
+[
+  "."
+  ","
+  ":"
+  ; FIXME: update grammar
+  ; (semicolon)
+  "->"
+  "=>"
+] @punctuation.delimiter
+
+; Operators
+[
+  "&&"
+  "||"
+  ">>"
+  "<<"
+  "&"
+  "^"
+  "|"
+  "+"
+  "-"
+  "*"
+  "/"
+  "%"
+  "**"
+  "="
+  "<"
+  "<="
+  "=="
+  "!="
+  ">="
+  ">"
+  "!"
+  "~"
+  "-"
+  "+"
+  "++"
+  "--"
+  ":="
+] @operator
+
+[
+  "delete"
+  "new"
+] @keyword.operator
+
+(import_directive
+  "*" @character.special)
+
+; Comments
+(comment) @comment @spell
+
+((comment) @comment.documentation
+  (#lua-match? @comment.documentation "^///[^/]"))
+
+((comment) @comment.documentation
+  (#lua-match? @comment.documentation "^///$"))
+
+((comment) @comment.documentation
+  (#lua-match? @comment.documentation "^/[*][*][^*].*[*]/$"))