class Rails::Command::CredentialsCommand

def change_credentials_in_system_editor
  using_system_editor do
    say "Editing #{content_path}..."
    credentials.change { |tmp_path| system_editor(tmp_path) }
    say "File encrypted and saved."
    warn_if_credentials_are_invalid
  end
rescue ActiveSupport::EncryptedFile::MissingKeyError => error
  say error.message
rescue ActiveSupport::MessageEncryptor::InvalidMessage
  say "Couldn't decrypt #{content_path}. Perhaps you passed the wrong key?"
end

def config
  Rails.application.config.credentials
end

def content_path
  @content_path ||= relative_path(config.content_path)
end

def credentials
  @credentials ||= Rails.application.encrypted(content_path, key_path: key_path)
end

def diff(content_path = nil)
  if @content_path = content_path
    self.environment = extract_environment_from_path(content_path)
    load_environment_config!
    say credentials.read.presence || credentials.content_path.read
  else
    disenroll_project_from_credentials_diffing if options[:disenroll]
    enroll_project_in_credentials_diffing if options[:enroll]
  end
rescue ActiveSupport::MessageEncryptor::InvalidMessage
  say credentials.content_path.read
end

def edit
  load_environment_config!
  load_generators
  if environment_specified?
    @content_path = "config/credentials/#{environment}.yml.enc" unless config.key?(:content_path)
    @key_path = "config/credentials/#{environment}.key" unless config.key?(:key_path)
  end
  ensure_encryption_key_has_been_added
  ensure_credentials_have_been_added
  ensure_diffing_driver_is_configured
  change_credentials_in_system_editor
end

def ensure_credentials_have_been_added
  require "rails/generators/rails/credentials/credentials_generator"
  Rails::Generators::CredentialsGenerator.new(
    [content_path, key_path],
    skip_secret_key_base: environment_specified? && %w[development test].include?(environment),
    quiet: true
  ).invoke_all
end

def ensure_encryption_key_has_been_added
  return if credentials.key?
  require "rails/generators/rails/encryption_key_file/encryption_key_file_generator"
  encryption_key_file_generator = Rails::Generators::EncryptionKeyFileGenerator.new
  encryption_key_file_generator.add_key_file(key_path)
  encryption_key_file_generator.ignore_key_file(key_path)
end

def extract_environment_from_path(path)
  available_environments.find { |env| path.end_with?("#{env}.yml.enc") }
end

def key_path
  @key_path ||= relative_path(config.key_path)
end

def missing_credentials_message
  if !credentials.key?
    "Missing '#{key_path}' to decrypt credentials. See `#{executable(:help)}`."
  else
    "File '#{content_path}' does not exist. Use `#{executable(:edit)}` to change that."
  end
end

def relative_path(path)
  Rails.root.join(path).relative_path_from(Rails.root).to_s
end

def show
  load_environment_config!
  say credentials.read.presence || missing_credentials_message
end

def warn_if_credentials_are_invalid
  credentials.validate!
rescue ActiveSupport::EncryptedConfiguration::InvalidContentError => error
  say "WARNING: #{error.message}", :red
  say ""
  say "Your application will not be able to load '#{content_path}' until the error has been fixed.", :red
end