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 assign_properties(*)
  super
  str = with_interpolations_blanked
  begin
    @parsed_tree = Regexp::Parser.parse(str, options: options)
  rescue StandardError
    @parsed_tree = nil
  else
    origin = loc.begin.end
    source = @parsed_tree.to_s
    @parsed_tree.each_expression(true) do |e|
      e.origin = origin
      e.source = source
    end
  end
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 event == :enter &&
                  named == exp.respond_to?(:name) &&
                  exp.respond_to?(:capturing?) &&
                  exp.capturing?
  end
  self
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