module Aptible
  module CLI
    module Subcommands
      module Apps
        def self.included(thor)
          thor.class_eval do
            include Helpers::App
            include Helpers::Environment
            include Helpers::Token

            desc 'apps', 'List all applications'
            option :environment
            def apps
              Formatter.render(Renderer.current) do |root|
                root.grouped_keyed_list(
                  { 'environment' => 'handle' },
                  'handle'
                ) do |node|
                  scoped_environments(options).each do |account|
                    account.each_app do |app|
                      node.object do |n|
                        ResourceFormatter.inject_app(n, app, account)
                      end
                    end
                  end
                end
              end
            end

            desc 'apps:create HANDLE', 'Create a new application'
            option :environment
            define_method 'apps:create' do |handle|
              environment = ensure_environment(options)
              app = environment.create_app(handle: handle)

              if app.errors.any?
                raise Thor::Error, app.errors.full_messages.first
              else
                CLI.logger.info "App #{handle} created!"

                Formatter.render(Renderer.current) do |root|
                  root.object do |o|
                    o.value('git_remote', app.git_repo)
                  end
                end
              end
            end

            desc 'apps:scale SERVICE ' \
                 '[--container-count COUNT] [--container-size SIZE_MB]',
                 'Scale a service'
            app_options
            option :container_count, type: :numeric
            option :container_size, type: :numeric
            option :size, type: :numeric,
                          desc: 'DEPRECATED, use --container-size'
            define_method 'apps:scale' do |type, *more|
              service = ensure_service(options, type)

              container_count = options[:container_count]
              container_size = options[:container_size]

              # There are two legacy options we have to process here:
              # - We used to accept apps:scale SERVICE COUNT
              # - We used to accept --size
              case more.size
              when 0
                # Noop
              when 1
                if container_count.nil?
                  m = 'Passing container count as a positional ' \
                      'argument is deprecated, use --container-count'
                  CLI.logger.warn(m)
                  container_count = Integer(more.first)
                else
                  raise Thor::Error, 'Container count was passed via both ' \
                                     'the --container-count keyword argument ' \
                                     'and a positional argument. ' \
                                     'Use only --container-count to proceed.'
                end
              else
                # Unfortunately, Thor does not want to let us easily hook into
                # its usage formatting, so we have to work around it here.
                command = thor.commands.fetch('apps:scale')
                error = ArgumentError.new
                args = [type] + more
                thor.handle_argument_error(command, error, args, 1)
              end

              if options[:size]
                if container_size.nil?
                  m = 'Passing container size via the --size keyword ' \
                      'argument is deprecated, use --container-size'
                  CLI.logger.warn(m)
                  container_size = options[:size]
                else
                  raise Thor::Error, 'Container size was passed via both ' \
                                     '--container-size and --size. ' \
                                     'Use only --container-size to proceed.'
                end
              end

              if container_count.nil? && container_size.nil?
                raise Thor::Error,
                      'Provide at least --container-count or --container-size'
              end

              # We don't validate any parameters here: API will do that for us.
              opts = { type: 'scale' }
              opts[:container_count] = container_count if container_count
              opts[:container_size] = container_size if container_size

              op = service.create_operation!(opts)
              attach_to_operation_logs(op)
            end

            desc 'apps:deprovision', 'Deprovision an app'
            app_options
            define_method 'apps:deprovision' do
              app = ensure_app(options)
              CLI.logger.info "Deprovisioning #{app.handle}..."
              app.create_operation!(type: 'deprovision')
            end
          end
        end
      end
    end
  end
end