class RubyLsp::Listeners::FoldingRanges

# File lib/ruby_lsp/listeners/folding_ranges.rb, line 17
def initialize(response_builder, comments, dispatcher)
  @response_builder = response_builder
  @requires = T.let([], T::Array[Prism::CallNode])
  @comments = comments

  dispatcher.register(
    self,
    :on_if_node_enter,
    :on_in_node_enter,
    :on_rescue_node_enter,
    :on_when_node_enter,
    :on_interpolated_string_node_enter,
    :on_array_node_enter,
    :on_block_node_enter,
    :on_case_node_enter,
    :on_case_match_node_enter,
    :on_class_node_enter,
    :on_module_node_enter,
    :on_for_node_enter,
    :on_hash_node_enter,
    :on_singleton_class_node_enter,
    :on_unless_node_enter,
    :on_until_node_enter,
    :on_while_node_enter,
    :on_else_node_enter,
    :on_ensure_node_enter,
    :on_begin_node_enter,
    :on_def_node_enter,
    :on_call_node_enter,
    :on_lambda_node_enter,
  )
end

# File lib/ruby_lsp/listeners/folding_ranges.rb, line 51
def finalize_response!
  push_comment_ranges
  emit_requires_range
end

# File lib/ruby_lsp/listeners/folding_ranges.rb, line 85
def on_array_node_enter(node)
  add_simple_range(node)
end

# File lib/ruby_lsp/listeners/folding_ranges.rb, line 155
def on_begin_node_enter(node)
  add_simple_range(node)
end

# File lib/ruby_lsp/listeners/folding_ranges.rb, line 90
def on_block_node_enter(node)
  add_simple_range(node)
end

# File lib/ruby_lsp/listeners/folding_ranges.rb, line 175
def on_call_node_enter(node)
  # If we find a require, don't visit the child nodes (prevent `super`), so that we can keep accumulating into
  # the `@requires` array and then push the range whenever we find a node that isn't a CallNode
  if require?(node)
    @requires << node
    return
  end

  location = node.location
  add_lines_range(location.start_line, location.end_line - 1)
end

# File lib/ruby_lsp/listeners/folding_ranges.rb, line 100
def on_case_match_node_enter(node)
  add_simple_range(node)
end

# File lib/ruby_lsp/listeners/folding_ranges.rb, line 95
def on_case_node_enter(node)
  add_simple_range(node)
end

# File lib/ruby_lsp/listeners/folding_ranges.rb, line 105
def on_class_node_enter(node)
  add_simple_range(node)
end

# File lib/ruby_lsp/listeners/folding_ranges.rb, line 160
def on_def_node_enter(node)
  params = node.parameters
  parameter_loc = params&.location
  location = node.location

  if params && parameter_loc.end_line > location.start_line
    # Multiline parameters
    add_lines_range(location.start_line, parameter_loc.end_line)
    add_lines_range(parameter_loc.end_line + 1, location.end_line - 1)
  else
    add_lines_range(location.start_line, location.end_line - 1)
  end
end

# File lib/ruby_lsp/listeners/folding_ranges.rb, line 145
def on_else_node_enter(node)
  add_simple_range(node)
end

# File lib/ruby_lsp/listeners/folding_ranges.rb, line 150
def on_ensure_node_enter(node)
  add_simple_range(node)
end

# File lib/ruby_lsp/listeners/folding_ranges.rb, line 115
def on_for_node_enter(node)
  add_simple_range(node)
end

# File lib/ruby_lsp/listeners/folding_ranges.rb, line 120
def on_hash_node_enter(node)
  add_simple_range(node)
end

# File lib/ruby_lsp/listeners/folding_ranges.rb, line 57
def on_if_node_enter(node)
  add_statements_range(node)
end

# File lib/ruby_lsp/listeners/folding_ranges.rb, line 62
def on_in_node_enter(node)
  add_statements_range(node)
end

# File lib/ruby_lsp/listeners/folding_ranges.rb, line 77
def on_interpolated_string_node_enter(node)
  opening_loc = node.opening_loc || node.location
  closing_loc = node.closing_loc || node.parts.last&.location || node.location

  add_lines_range(opening_loc.start_line, closing_loc.start_line - 1)
end

# File lib/ruby_lsp/listeners/folding_ranges.rb, line 188
def on_lambda_node_enter(node)
  add_simple_range(node)
end

# File lib/ruby_lsp/listeners/folding_ranges.rb, line 110
def on_module_node_enter(node)
  add_simple_range(node)
end

# File lib/ruby_lsp/listeners/folding_ranges.rb, line 67
def on_rescue_node_enter(node)
  add_statements_range(node)
end

# File lib/ruby_lsp/listeners/folding_ranges.rb, line 125
def on_singleton_class_node_enter(node)
  add_simple_range(node)
end

# File lib/ruby_lsp/listeners/folding_ranges.rb, line 130
def on_unless_node_enter(node)
  add_simple_range(node)
end

# File lib/ruby_lsp/listeners/folding_ranges.rb, line 135
def on_until_node_enter(node)
  add_simple_range(node)
end

# File lib/ruby_lsp/listeners/folding_ranges.rb, line 72
def on_when_node_enter(node)
  add_statements_range(node)
end

# File lib/ruby_lsp/listeners/folding_ranges.rb, line 140
def on_while_node_enter(node)
  add_simple_range(node)
end