DeepCodeGeniusWeb-tools/tree-sitter-queries/rust/textobjects.scm

; functions
(function_signature_item) @function.outer

(function_item) @function.outer

(function_item
  body: (block
    .
    "{"
    .
    (_) @_start @_end
    (_)? @_end
    .
    "}"
    (#make-range! "function.inner" @_start @_end)))

; quantifies as class(es)
(struct_item) @class.outer

(struct_item
  body: (field_declaration_list
    .
    "{"
    .
    (_) @_start
    [
      (_)
      ","
    ]? @_end
    .
    "}"
    (#make-range! "class.inner" @_start @_end)))

(enum_item) @class.outer

(enum_item
  body: (enum_variant_list
    .
    "{"
    .
    (_) @_start
    [
      (_)
      ","
    ]? @_end
    .
    "}"
    (#make-range! "class.inner" @_start @_end)))

(union_item) @class.outer

(union_item
  body: (field_declaration_list
    .
    "{"
    .
    (_) @_start
    [
      (_)
      ","
    ]? @_end
    .
    "}"
    (#make-range! "class.inner" @_start @_end)))

(trait_item) @class.outer

(trait_item
  body: (declaration_list
    .
    "{"
    .
    (_) @_start @_end
    (_)? @_end
    .
    "}"
    (#make-range! "class.inner" @_start @_end)))

(impl_item) @class.outer

(impl_item
  body: (declaration_list
    .
    "{"
    .
    (_) @_start @_end
    (_)? @_end
    .
    "}"
    (#make-range! "class.inner" @_start @_end)))

(mod_item) @class.outer

(mod_item
  body: (declaration_list
    .
    "{"
    .
    (_) @_start @_end
    (_)? @_end
    .
    "}"
    (#make-range! "class.inner" @_start @_end)))

; conditionals
(if_expression
  alternative: (_
    (_) @conditional.inner)?) @conditional.outer

(if_expression
  alternative: (else_clause
    (block) @conditional.inner))

(if_expression
  condition: (_) @conditional.inner)

(if_expression
  consequence: (block) @conditional.inner)

(match_arm
  (_)) @conditional.inner

(match_expression) @conditional.outer

; loops
(loop_expression
  body: (block
    .
    "{"
    .
    (_) @_start
    [
      (_)
      ","
    ]? @_end
    .
    "}"
    (#make-range! "loop.inner" @_start @_end))) @loop.outer

(while_expression
  body: (block
    .
    "{"
    .
    (_) @_start
    [
      (_)
      ","
    ]? @_end
    .
    "}"
    (#make-range! "loop.inner" @_start @_end))) @loop.outer

(for_expression
  body: (block
    .
    "{"
    .
    (_) @_start
    [
      (_)
      ","
    ]? @_end
    .
    "}"
    (#make-range! "loop.inner" @_start @_end))) @loop.outer

; blocks
(_
  (block) @block.inner) @block.outer

(unsafe_block
  (_)? @block.inner) @block.outer

; calls
(macro_invocation) @call.outer

(macro_invocation
  (token_tree
    .
    "("
    .
    (_) @_start
    (_)? @_end
    .
    ")"
    (#make-range! "call.inner" @_start @_end)))

(call_expression) @call.outer

(call_expression
  arguments: (arguments
    .
    "("
    .
    (_) @_start
    (_)? @_end
    .
    ")"
    (#make-range! "call.inner" @_start @_end)))

; returns
(return_expression
  (_)? @return.inner) @return.outer

; statements
(block
  (_) @statement.outer)

; comments
(line_comment) @comment.outer

(block_comment) @comment.outer

; parameter
((parameters
  "," @_start
  .
  (parameter) @parameter.inner)
  (#make-range! "parameter.outer" @_start @parameter.inner))

((parameters
  .
  (parameter) @parameter.inner
  .
  ","? @_end)
  (#make-range! "parameter.outer" @parameter.inner @_end))

((parameters
  "," @_start
  .
  (type_identifier) @parameter.inner)
  (#make-range! "parameter.outer" @_start @parameter.inner))

((parameters
  .
  (type_identifier) @parameter.inner
  .
  ","? @_end)
  (#make-range! "parameter.outer" @parameter.inner @_end))

((type_parameters
  "," @_start
  .
  (_) @parameter.inner)
  (#make-range! "parameter.outer" @_start @parameter.inner))

((type_parameters
  .
  (_) @parameter.inner
  .
  ","? @_end)
  (#make-range! "parameter.outer" @parameter.inner @_end))

((tuple_pattern
  "," @_start
  .
  (identifier) @parameter.inner)
  (#make-range! "parameter.outer" @_start @parameter.inner))

((tuple_pattern
  .
  (identifier) @parameter.inner
  .
  ","? @_end)
  (#make-range! "parameter.outer" @parameter.inner @_end))

((tuple_struct_pattern
  "," @_start
  .
  (identifier) @parameter.inner)
  (#make-range! "parameter.outer" @_start @parameter.inner))

((tuple_struct_pattern
  .
  (identifier) @parameter.inner
  .
  ","? @_end)
  (#make-range! "parameter.outer" @parameter.inner @_end))

(tuple_expression
  "," @_start
  .
  (_) @parameter.inner
  (#make-range! "parameter.outer" @_start @parameter.inner))

(tuple_expression
  .
  (_) @parameter.inner
  .
  ","? @_end
  (#make-range! "parameter.outer" @parameter.inner @_end))

((tuple_type
  "," @_start
  .
  (_) @parameter.inner)
  (#make-range! "parameter.outer" @_start @parameter.inner))

((tuple_type
  .
  (_) @parameter.inner
  .
  ","? @_end)
  (#make-range! "parameter.outer" @parameter.inner @_end))

(struct_item
  body: (field_declaration_list
    "," @_start
    .
    (_) @parameter.inner
    (#make-range! "parameter.outer" @_start @parameter.inner)))

(struct_item
  body: (field_declaration_list
    .
    (_) @parameter.inner
    .
    ","? @_end
    (#make-range! "parameter.outer" @parameter.inner @_end)))

(struct_expression
  body: (field_initializer_list
    "," @_start
    .
    (_) @parameter.inner
    (#make-range! "parameter.outer" @_start @parameter.inner)))

(struct_expression
  body: (field_initializer_list
    .
    (_) @parameter.inner
    .
    ","? @_end
    (#make-range! "parameter.outer" @parameter.inner @_end)))

((closure_parameters
  "," @_start
  .
  (_) @parameter.inner)
  (#make-range! "parameter.outer" @_start @parameter.inner))

((closure_parameters
  .
  (_) @parameter.inner
  .
  ","? @_end)
  (#make-range! "parameter.outer" @parameter.inner @_end))

((arguments
  "," @_start
  .
  (_) @parameter.inner)
  (#make-range! "parameter.outer" @_start @parameter.inner))

((arguments
  .
  (_) @parameter.inner
  .
  ","? @_end)
  (#make-range! "parameter.outer" @parameter.inner @_end))

((type_arguments
  "," @_start
  .
  (_) @parameter.inner)
  (#make-range! "parameter.outer" @_start @parameter.inner))

((type_arguments
  .
  (_) @parameter.inner
  .
  ","? @_end)
  (#make-range! "parameter.outer" @parameter.inner @_end))

((token_tree
  "," @_start
  .
  (_) @parameter.inner)
  (#make-range! "parameter.outer" @_start @parameter.inner))

((token_tree
  .
  (_) @parameter.inner
  .
  ","? @_end)
  (#make-range! "parameter.outer" @parameter.inner @_end))

(scoped_use_list
  list: (use_list
    "," @_start
    .
    (_) @parameter.inner
    (#make-range! "parameter.outer" @_start @parameter.inner)))

(scoped_use_list
  list: (use_list
    .
    (_) @parameter.inner
    .
    ","? @_end
    (#make-range! "parameter.outer" @parameter.inner @_end)))

[
  (integer_literal)
  (float_literal)
] @number.inner

(let_declaration
  pattern: (_) @assignment.lhs
  value: (_) @assignment.inner @assignment.rhs) @assignment.outer

(let_declaration
  pattern: (_) @assignment.inner)

(assignment_expression
  left: (_) @assignment.lhs
  right: (_) @assignment.inner @assignment.rhs) @assignment.outer

(assignment_expression
  left: (_) @assignment.inner)