module EYAML
  class InvalidFormatError < StandardError; end

  class CLI < Thor
    class_option :keydir, aliases: "-k", type: :string, desc: "Directory containing EYAML keys"

    desc "encrypt", "(Re-)encrypt one or more EYAML files"
    def encrypt(*files)
      files.each do |file|
        file_path = Pathname.new(file)
        next unless file_path.exist?

        bytes_written = EYAML.encrypt_file_in_place(
          file_path,
          keydir: options.fetch(:keydir, nil)
        )

        puts "Wrote #{bytes_written} bytes to #{file_path}."
      end
    end

    method_option :output, type: :string, desc: "print output to the provided file, rather than stdout", aliases: "-o"
    method_option :"key-from-stdin", type: :boolean, desc: "read the private key from STDIN", default: false
    desc "decrypt", "Decrypt an EYAML file"
    def decrypt(file)
      file_path = Pathname.new(file)
      unless file_path.exist?
        puts "#{file} doesn't exist"
        return
      end

      key_options = if options.fetch(:"key-from-stdin")
        # Read key from STDIN
        {private_key: $stdin.gets}
      else
        {keydir: options.fetch(:keydir, nil)}
      end

      eyaml = EYAML.decrypt_file(file, **key_options)

      if options.has_key?("output")
        output_file = Pathname.new(options.fetch(:output))
        File.write(output_file, eyaml)
        return
      end

      puts eyaml
    end

    method_option :write, type: :boolean, aliases: "-w", desc: "rather than printing both keys, print the public and write the private into the keydir", default: false
    desc "keygen", "Generate a new EYAML keypair"
    def keygen
      public_key, private_key = EYAML.generate_keypair(
        save: options.fetch(:write),
        keydir: options.fetch(:keydir, nil)
      )

      puts "Public Key: #{public_key}"
      puts "Private Key: #{private_key}" unless options.fetch(:write)
    end

    map e: :encrypt
    map d: :decrypt
    map g: :keygen

    def self.exit_on_failure?
      true
    end
  end
end