
require 'trollop'

module SimpleDeploy
  module CLI

    class Protect
      include Shared

      def protect
        @opts = Trollop::options do
          version SimpleDeploy::VERSION
          banner <<-EOS

Protect/Unprotect one or more stacks.

simple_deploy protect -n STACK_NAME1 -n STACK_NAME2 -e ENVIRONMENT -p on_off

EOS
          opt :help, "Display Help"
          opt :environment, "Set the target environment", :type => :string
          opt :protection, "Enable/Disable protection using on/off", :type  => :string
          opt :log_level, "Log level:  debug, info, warn, error", :type    => :string,
                                                                  :default => 'info'
          opt :name, "Stack name(s) of stacks to protect", :type => :string,
                                                           :multi => true
          opt :read_from_env, "Read credentials and region from environment variables"
        end

        valid_options? :provided => @opts,
                       :required => [:environment, :name, :read_from_env]

        config_arg = @opts[:read_from_env] ? :read_from_env : @opts[:environment]
        SimpleDeploy.create_config config_arg
        SimpleDeploy.logger @opts[:log_level]

        @opts[:name].each do |name|
          stack = Stack.new :name        => name,
                            :environment => @opts[:environment]
          rescue_exceptions_and_exit do
            stack.update :attributes => [{ 'protection' => @opts[:protection] }]
          end
        end
      end

      def command_summary
        'Protect/Unprotect one or more stacks'
      end

    end

  end
end
