intuit/heirloom

require 'tempfile'

module Heirloom
  module CLI
    class Upload

      include Heirloom::CLI::Shared

      def self.command_summary
        'Upload a directory to Heirloom'
      end

      def initialize
        @opts = read_options
        @logger = HeirloomLogger.new :log_level => @opts[:level]
        @config = load_config :logger => @logger,
                              :opts   => @opts
        @catalog = Heirloom::Catalog.new :name    => @opts[:name],
                                         :config  => @config
        ensure_valid_options :provided => @opts,
                             :required => [:name, :id, :directory],
                             :config   => @config
        ensure_catalog_domain_exists :config  => @config,
                                     :catalog => @catalog
        ensure_entry_exists_in_catalog :config  => @config,
                                       :catalog => @catalog,
                                       :entry   => @opts[:name]
        @archive = Archive.new :name   => @opts[:name],
                               :id     => @opts[:id],
                               :config => @config
        @regions = @catalog.regions
        @bucket_prefix = @catalog.bucket_prefix
      end

      def upload
        ensure_valid_metadata_region @config
        ensure_domain_exists :name   => @opts[:name],
                             :config => @config
        ensure_buckets_exist :bucket_prefix => @bucket_prefix,
                             :name          => @opts[:name],
                             :regions       => @regions,
                             :config        => @config
        ensure_path_is_directory :path   => @opts[:directory],
                                 :config => @config

        secret = read_secret :opts   => @opts,
                             :config => @config
        ensure_valid_secret :secret => secret,
                            :config => @config

        @archive.destroy if @archive.exists?

        @file = Tempfile.new('archive.tar.gz')

        unless @archive.build :bucket_prefix => @bucket_prefix,
                              :directory     => @opts[:directory],
                              :exclude       => @opts[:exclude],
                              :file          => @file.path,
                              :secret        => secret

          @logger.error "Build failed."
          exit 1
        end

        @archive.upload :bucket_prefix   => @bucket_prefix,
                        :regions         => @regions,
                        :public_readable => @opts[:public],
                        :file            => @file.path,
                        :secret          => secret

        @file.close!
      end

      private

      def read_options
        Trollop::options do
          version Heirloom::VERSION
          banner <<-EOS

#{Upload.command_summary}.

Usage:

heirloom upload -n NAME -i ID -d DIRECTORY_TO_UPLOAD

EOS
          opt :directory, "Source directory to upload.", :type  => :string
          opt :exclude, "File(s) or directorie(s) to exclude. \
Can be specified multiple times.", :type  => :string, :multi => true, :short => "-x"
          opt :help, "Display Help"
          opt :id, "ID for Heirloom.", :type => :string
          opt :level, "Log level [debug|info|warn|error].", :type    => :string,
                                                            :default => 'info'
          opt :metadata_region, "AWS region to store Heirloom metadata.", :type    => :string
          opt :name, "Name of Heirloom.", :type => :string
          opt :public, "Set this Heirloom as public readable?"
          opt :secret, "Encrypt the Heirloom with given secret.", :type => :string
          opt :secret_file, "Read secret from file.", :type  => :string,
                                                      :short => :none
          opt :aws_access_key, "AWS Access Key ID", :type  => :string,
                                                    :short => :none
          opt :aws_secret_key, "AWS Secret Access Key", :type  => :string,
                                                        :short => :none
          opt :use_iam_profile, "Use IAM EC2 Profile", :short => :none
          opt :environment, "Environment (defined in heirloom config file)", :type => :string
        end
      end

    end
  end
end