class RubyLsp::Listeners::CodeLens

# File lib/ruby_lsp/listeners/code_lens.rb, line 35
def initialize(response_builder, global_state, uri, dispatcher)
  @response_builder = response_builder
  @global_state = global_state
  @uri = T.let(uri, URI::Generic)
  @path = T.let(uri.to_standardized_path, T.nilable(String))
  # visibility_stack is a stack of [current_visibility, previous_visibility]
  @visibility_stack = T.let([[:public, :public]], T::Array[T::Array[T.nilable(Symbol)]])
  @group_stack = T.let([], T::Array[String])
  @group_id = T.let(1, Integer)
  @group_id_stack = T.let([], T::Array[Integer])

  dispatcher.register(
    self,
    :on_class_node_enter,
    :on_class_node_leave,
    :on_module_node_enter,
    :on_module_node_leave,
    :on_def_node_enter,
    :on_call_node_enter,
    :on_call_node_leave,
  )
end

# File lib/ruby_lsp/listeners/code_lens.rb, line 123
def on_call_node_enter(node)
  name = node.name
  arguments = node.arguments

  # If we found `private` by itself or `private def foo`
  if ACCESS_MODIFIERS.include?(name)
    if arguments.nil?
      @visibility_stack.pop
      @visibility_stack.push([name, name])
    elsif arguments.arguments.first.is_a?(Prism::DefNode)
      visibility, _ = @visibility_stack.pop
      @visibility_stack.push([name, visibility])
    end

    return
  end

  if [DESCRIBE_KEYWORD, IT_KEYWORD].include?(name)
    case name
    when DESCRIBE_KEYWORD
      add_spec_code_lens(node, kind: :group)
      @group_id_stack.push(@group_id)
      @group_id += 1
    when IT_KEYWORD
      add_spec_code_lens(node, kind: :example)
    end
  end
end

# File lib/ruby_lsp/listeners/code_lens.rb, line 153
def on_call_node_leave(node)
  _, prev_visibility = @visibility_stack.pop
  @visibility_stack.push([prev_visibility, prev_visibility])
  if node.name == DESCRIBE_KEYWORD
    @group_id_stack.pop
  end
end

# File lib/ruby_lsp/listeners/code_lens.rb, line 59
def on_class_node_enter(node)
  @visibility_stack.push([:public, :public])
  class_name = node.constant_path.slice
  @group_stack.push(class_name)

  if @path && class_name.end_with?("Test")
    add_test_code_lens(
      node,
      name: class_name,
      command: generate_test_command(group_stack: @group_stack),
      kind: :group,
    )

    @group_id_stack.push(@group_id)
    @group_id += 1
  end
end

# File lib/ruby_lsp/listeners/code_lens.rb, line 78
def on_class_node_leave(node)
  @visibility_stack.pop
  @group_stack.pop

  class_name = node.constant_path.slice

  if @path && class_name.end_with?("Test")
    @group_id_stack.pop
  end
end

# File lib/ruby_lsp/listeners/code_lens.rb, line 90
def on_def_node_enter(node)
  class_name = @group_stack.last
  return unless class_name&.end_with?("Test")

  visibility, _ = @visibility_stack.last
  if visibility == :public
    method_name = node.name.to_s
    if @path && method_name.start_with?("test_")
      add_test_code_lens(
        node,
        name: method_name,
        command: generate_test_command(method_name: method_name, group_stack: @group_stack),
        kind: :example,
      )
    end
  end
end

# File lib/ruby_lsp/listeners/code_lens.rb, line 109
def on_module_node_enter(node)
  if (path = namespace_constant_name(node))
    @group_stack.push(path)
  else
    @group_stack.push(DYNAMIC_REFERENCE_MARKER)
  end
end

# File lib/ruby_lsp/listeners/code_lens.rb, line 118
def on_module_node_leave(node)
  @group_stack.pop
end