module RuboCop::Ext::RegexpNode

def assign_properties(*)
  super
  str = with_interpolations_blanked
  @parsed_tree = begin
    Regexp::Parser.parse(str, options: options)
  rescue StandardError
    nil
  end
  origin = loc.begin.end
  @parsed_tree&.each_expression(true) { |e| e.origin = origin }
end

def each_capture(named: ANY)
  return enum_for(__method__, named: named) unless block_given?
  parsed_tree&.traverse do |event, exp, _index|
    yield(exp) if named_capturing?(exp, event, named)
  end
  self
end

def named_capturing?(exp, event, named)
  event == :enter &&
    named == exp.respond_to?(:name) &&
    !exp.text.start_with?('(?<=') &&
    exp.respond_to?(:capturing?) &&
    exp.capturing?
end

def with_interpolations_blanked
  # Ignore the trailing regopt node
  children[0...-1].map do |child|
    source = child.source
    # We don't want to consider the contents of interpolations as part of the pattern source,
    # but need to preserve their width, to allow offsets to correctly line up with the
    # original source: spaces have no effect, and preserve width.
    if child.begin_type?
      ' ' * source.length
    else
      source
    end
  end.join
end