module Pry::Command::Edit::FileAndLineLocator

def from_binding(target)
  if target.respond_to?(:source_location)
    target.source_location
  else
    target.eval("[__FILE__, __LINE__]")
  end
end

def from_code_object(code_object, filename_argument)
  unless File.exist?(code_object.source_file.to_s)
    raise CommandError, "Cannot find a file for #{filename_argument}!"
  end
  [code_object.source_file, code_object.source_line]
end

def from_exception(exception, backtrace_level)
  raise CommandError, "No exception found." if exception.nil?
  file_name, line = exception.bt_source_location_for(backtrace_level)
  raise CommandError, "Exception has no associated file." if file_name.nil?
  if Pry.eval_path == file_name
    raise CommandError, "Cannot edit exceptions raised in REPL."
  end
  [file_name, line]
end

def from_filename_argument(filename_argument)
  f = File.expand_path(filename_argument)
  l = f.sub!(/:(\d+)$/, "") ? Regexp.last_match(1).to_i : 1
  [f, l]
end