class RubyLsp::Listeners::SemanticHighlighting

# File lib/ruby_lsp/listeners/semantic_highlighting.rb, line 28
def initialize(dispatcher, response_builder, range: nil)
  @response_builder = response_builder
  @range = range
  @special_methods = T.let(nil, T.nilable(T::Array[String]))
  @current_scope = T.let(ParameterScope.new, ParameterScope)
  @inside_regex_capture = T.let(false, T::Boolean)
  @inside_implicit_node = T.let(false, T::Boolean)

  dispatcher.register(
    self,
    :on_call_node_enter,
    :on_class_node_enter,
    :on_def_node_enter,
    :on_def_node_leave,
    :on_block_node_enter,
    :on_block_node_leave,
    :on_self_node_enter,
    :on_module_node_enter,
    :on_local_variable_write_node_enter,
    :on_local_variable_read_node_enter,
    :on_block_parameter_node_enter,
    :on_required_keyword_parameter_node_enter,
    :on_optional_keyword_parameter_node_enter,
    :on_keyword_rest_parameter_node_enter,
    :on_optional_parameter_node_enter,
    :on_required_parameter_node_enter,
    :on_rest_parameter_node_enter,
    :on_constant_read_node_enter,
    :on_constant_write_node_enter,
    :on_constant_and_write_node_enter,
    :on_constant_operator_write_node_enter,
    :on_constant_or_write_node_enter,
    :on_constant_target_node_enter,
    :on_local_variable_and_write_node_enter,
    :on_local_variable_operator_write_node_enter,
    :on_local_variable_or_write_node_enter,
    :on_local_variable_target_node_enter,
    :on_block_local_variable_node_enter,
    :on_match_write_node_enter,
    :on_match_write_node_leave,
    :on_implicit_node_enter,
    :on_implicit_node_leave,
  )
end

# File lib/ruby_lsp/listeners/semantic_highlighting.rb, line 173
def on_block_local_variable_node_enter(node)
  @response_builder.add_token(node.location, :variable)
end

# File lib/ruby_lsp/listeners/semantic_highlighting.rb, line 163
def on_block_node_enter(node)
  @current_scope = ParameterScope.new(@current_scope)
end

# File lib/ruby_lsp/listeners/semantic_highlighting.rb, line 168
def on_block_node_leave(node)
  @current_scope = T.must(@current_scope.parent)
end

# File lib/ruby_lsp/listeners/semantic_highlighting.rb, line 178
def on_block_parameter_node_enter(node)
  name = node.name
  @current_scope << name.to_sym if name
end

# File lib/ruby_lsp/listeners/semantic_highlighting.rb, line 74
def on_call_node_enter(node)
  return if @inside_implicit_node
  return unless visible?(node, @range)

  message = node.message
  return unless message

  # We can't push a semantic token for [] and []= because the argument inside the brackets is a part of
  # the message_loc
  return if message.start_with?("[") && (message.end_with?("]") || message.end_with?("]="))
  return if message == "=~"
  return if special_method?(message)

  type = Requests::Support::Sorbet.annotation?(node) ? :type : :method
  @response_builder.add_token(T.must(node.message_loc), type)
end

# File lib/ruby_lsp/listeners/semantic_highlighting.rb, line 302
def on_class_node_enter(node)
  return unless visible?(node, @range)

  @response_builder.add_token(node.constant_path.location, :class, [:declaration])

  superclass = node.superclass
  @response_builder.add_token(superclass.location, :class) if superclass
end

# File lib/ruby_lsp/listeners/semantic_highlighting.rb, line 122
def on_constant_and_write_node_enter(node)
  return unless visible?(node, @range)

  @response_builder.add_token(node.name_loc, :namespace)
end

# File lib/ruby_lsp/listeners/semantic_highlighting.rb, line 129
def on_constant_operator_write_node_enter(node)
  return unless visible?(node, @range)

  @response_builder.add_token(node.name_loc, :namespace)
end

# File lib/ruby_lsp/listeners/semantic_highlighting.rb, line 136
def on_constant_or_write_node_enter(node)
  return unless visible?(node, @range)

  @response_builder.add_token(node.name_loc, :namespace)
end

# File lib/ruby_lsp/listeners/semantic_highlighting.rb, line 107
def on_constant_read_node_enter(node)
  return if @inside_implicit_node
  return unless visible?(node, @range)

  @response_builder.add_token(node.location, :namespace)
end

# File lib/ruby_lsp/listeners/semantic_highlighting.rb, line 143
def on_constant_target_node_enter(node)
  return unless visible?(node, @range)

  @response_builder.add_token(node.location, :namespace)
end

# File lib/ruby_lsp/listeners/semantic_highlighting.rb, line 115
def on_constant_write_node_enter(node)
  return unless visible?(node, @range)

  @response_builder.add_token(node.name_loc, :namespace)
end

# File lib/ruby_lsp/listeners/semantic_highlighting.rb, line 150
def on_def_node_enter(node)
  @current_scope = ParameterScope.new(@current_scope)
  return unless visible?(node, @range)

  @response_builder.add_token(node.name_loc, :method, [:declaration])
end

# File lib/ruby_lsp/listeners/semantic_highlighting.rb, line 158
def on_def_node_leave(node)
  @current_scope = T.must(@current_scope.parent)
end

# File lib/ruby_lsp/listeners/semantic_highlighting.rb, line 319
def on_implicit_node_enter(node)
  return unless visible?(node, @range)

  @inside_implicit_node = true
end

# File lib/ruby_lsp/listeners/semantic_highlighting.rb, line 326
def on_implicit_node_leave(node)
  @inside_implicit_node = false
end

# File lib/ruby_lsp/listeners/semantic_highlighting.rb, line 202
def on_keyword_rest_parameter_node_enter(node)
  name = node.name

  if name
    @current_scope << name.to_sym

    @response_builder.add_token(T.must(node.name_loc), :parameter) if visible?(node, @range)
  end
end

# File lib/ruby_lsp/listeners/semantic_highlighting.rb, line 268
def on_local_variable_and_write_node_enter(node)
  return unless visible?(node, @range)

  @response_builder.add_token(node.name_loc, @current_scope.type_for(node.name))
end

# File lib/ruby_lsp/listeners/semantic_highlighting.rb, line 275
def on_local_variable_operator_write_node_enter(node)
  return unless visible?(node, @range)

  @response_builder.add_token(node.name_loc, @current_scope.type_for(node.name))
end

# File lib/ruby_lsp/listeners/semantic_highlighting.rb, line 282
def on_local_variable_or_write_node_enter(node)
  return unless visible?(node, @range)

  @response_builder.add_token(node.name_loc, @current_scope.type_for(node.name))
end

# File lib/ruby_lsp/listeners/semantic_highlighting.rb, line 254
def on_local_variable_read_node_enter(node)
  return if @inside_implicit_node
  return unless visible?(node, @range)

  # Numbered parameters
  if /_\d+/.match?(node.name)
    @response_builder.add_token(node.location, :parameter)
    return
  end

  @response_builder.add_token(node.location, @current_scope.type_for(node.name))
end

# File lib/ruby_lsp/listeners/semantic_highlighting.rb, line 289
def on_local_variable_target_node_enter(node)
  # If we're inside a regex capture, Prism will add LocalVariableTarget nodes for each captured variable.
  # Unfortunately, if the regex contains a backslash, the location will be incorrect and we'll end up highlighting
  # the entire regex as a local variable. We process these captures in process_regexp_locals instead and then
  # prevent pushing local variable target tokens. See https://github.com/ruby/prism/issues/1912
  return if @inside_regex_capture

  return unless visible?(node, @range)

  @response_builder.add_token(node.location, @current_scope.type_for(node.name))
end

# File lib/ruby_lsp/listeners/semantic_highlighting.rb, line 247
def on_local_variable_write_node_enter(node)
  return unless visible?(node, @range)

  @response_builder.add_token(node.name_loc, @current_scope.type_for(node.name))
end

# File lib/ruby_lsp/listeners/semantic_highlighting.rb, line 92
def on_match_write_node_enter(node)
  call = node.call

  if call.message == "=~"
    @inside_regex_capture = true
    process_regexp_locals(call)
  end
end

# File lib/ruby_lsp/listeners/semantic_highlighting.rb, line 102
def on_match_write_node_leave(node)
  @inside_regex_capture = true if node.call.message == "=~"
end

# File lib/ruby_lsp/listeners/semantic_highlighting.rb, line 312
def on_module_node_enter(node)
  return unless visible?(node, @range)

  @response_builder.add_token(node.constant_path.location, :namespace, [:declaration])
end

# File lib/ruby_lsp/listeners/semantic_highlighting.rb, line 193
def on_optional_keyword_parameter_node_enter(node)
  @current_scope << node.name
  return unless visible?(node, @range)

  location = node.name_loc
  @response_builder.add_token(location.copy(length: location.length - 1), :parameter)
end

# File lib/ruby_lsp/listeners/semantic_highlighting.rb, line 213
def on_optional_parameter_node_enter(node)
  @current_scope << node.name
  return unless visible?(node, @range)

  @response_builder.add_token(node.name_loc, :parameter)
end

# File lib/ruby_lsp/listeners/semantic_highlighting.rb, line 184
def on_required_keyword_parameter_node_enter(node)
  @current_scope << node.name
  return unless visible?(node, @range)

  location = node.name_loc
  @response_builder.add_token(location.copy(length: location.length - 1), :parameter)
end

# File lib/ruby_lsp/listeners/semantic_highlighting.rb, line 221
def on_required_parameter_node_enter(node)
  @current_scope << node.name
  return unless visible?(node, @range)

  @response_builder.add_token(node.location, :parameter)
end

# File lib/ruby_lsp/listeners/semantic_highlighting.rb, line 229
def on_rest_parameter_node_enter(node)
  name = node.name

  if name
    @current_scope << name.to_sym

    @response_builder.add_token(T.must(node.name_loc), :parameter) if visible?(node, @range)
  end
end

# File lib/ruby_lsp/listeners/semantic_highlighting.rb, line 240
def on_self_node_enter(node)
  return unless visible?(node, @range)

  @response_builder.add_token(node.location, :variable, [:default_library])
end