class Pry::Command::ReloadCode

def check_for_reloadability(code_object, identifier)
  if !code_object || !code_object.source_file
    raise CommandError, "Cannot locate #{identifier}!"
  end
  return if File.exist?(code_object.source_file)
  raise CommandError,
        "Cannot reload #{identifier} as it has no associated file on disk. " \
        "File found was: #{code_object.source_file}"
end

def current_file
  file =
    if target.respond_to?(:source_location)
      target.source_location.first
    else
      target.eval("__FILE__")
    end
  File.expand_path file
end

def process
  if !args.empty?
    reload_object(args.join(" "))
  elsif internal_binding?(target)
    reload_object("self")
  else
    reload_current_file
  end
end

def reload_current_file
  unless File.exist?(current_file)
    raise CommandError, "Current file: #{current_file} cannot be found on disk!"
  end
  load current_file
  output.puts "The current file: #{current_file} was reloaded!"
end

def reload_object(identifier)
  code_object = Pry::CodeObject.lookup(identifier, pry_instance)
  check_for_reloadability(code_object, identifier)
  load code_object.source_file
  output.puts "#{identifier} was reloaded!"
end