require 'thor'
require 'aws-sdk'
require 'timeout'
require 'mikoshi/plan'
require 'mikoshi/plan/task_definition'
require 'mikoshi/plan/service'

module Mikoshi
  class Cli < Thor
    TASK_DEFINITION_PATH = 'task_definitions'.freeze
    SERVICE_PATH = 'services'.freeze
    PLAN_EXT = '.yml.erb'.freeze
    FETCH_INTERVAL = 10
    DEPLOY_TIMEOUT = 300

    class_option :region, type: :string, desc: 'aws region'

    desc 'update_task TASK_NAME', 'Update task definition'
    def update_task_definition(task_name)
      task = ::Mikoshi::Plan::TaskDefinition.new(
        yaml_path:  File.join(TASK_DEFINITION_PATH, task_name + PLAN_EXT),
        client:     aws_client,
      )
      puts "Update task definition: #{task_name}"
      task.register_task_definition
      puts "Done update task definition: #{task_name} revision: #{ENV['TASK_DEF_REVISION']}"
    end

    desc 'update_service SERVICE_NAME', 'Update service'
    def update_service(service_name)
      service = ::Mikoshi::Plan::Service.new(
        yaml_path:  File.join(SERVICE_PATH, service_name + PLAN_EXT),
        client:     aws_client,
      )
      puts "Update service : #{service_name}"
      service.deploy_service
      begin
        Timeout.timeout(DEPLOY_TIMEOUT) do
          loop do
            puts "Waiting for #{FETCH_INTERVAL} sec..."
            sleep FETCH_INTERVAL
            break if service.deployed?
          end
        end
      rescue Timeout::Error
        puts "Update failed by timeout(#{DEPLOY_TIMEOUT} sec)"
        exit(false)
      end

      puts "Done update service #{service_name}"
    end

    desc 'deploy', 'Deploy task definition and service'
    method_option :task_definition, type: :string, desc: 'task_definition name', aliases: '-t'
    method_option :service, type: :string, desc: 'service name', aliases: '-s'
    def deploy
      update_task_definition(options[:task_definition]) unless options[:task_definition].nil?
      update_service(options[:service]) unless options[:service].nil?
    end

    no_tasks do
      def aws_client
        opt =
          if options[:region].nil?
            {}
          else
            { region: options[:region] }
          end

        Aws::ECS::Client.new(opt)
      end
    end
  end
end