class RuboCop::Cop::RSpec::ReturnFromStub::AndReturnCallCorrector

def call(corrector)
  # Heredoc autocorrection is not yet implemented.
  return if heredoc?
  corrector.replace(range, " { #{replacement} }")
end

def hash_without_braces?
  arg.hash_type? && !arg.braces?
end

def heredoc?
  arg.loc.is_a?(Parser::Source::Map::Heredoc)
end

def initialize(node)
  @node = node
  @receiver = node.receiver
  @arg = node.first_argument
end

def range
  Parser::Source::Range.new(
    node.source_range.source_buffer,
    receiver.source_range.end_pos,
    node.source_range.end_pos
  )
end

def replacement
  if hash_without_braces?
    "{ #{arg.source} }"
  else
    arg.source
  end
end