require 'aws-sdk'
require 'thor'

require_relative '../control/data'
require_relative '../interface/packer'
require_relative '../patch/thor-actions'

module Builderator
  module Tasks
    ##
    # Wrap Packer commands
    ##
    class Packer < Thor
      include Thor::Actions

      def self.exit_on_failure?
        true
      end

      class_option :debug, :type => :boolean

      desc 'configure [PROFILE=default]', 'Generate a packer configuration'
      def configure(profile = :default)
        Config.profile.use(profile)

        invoke Tasks::Version, :current, [], options
        puts Interface.packer.render if options['debug']
      end

      desc 'build [PROFILE=default *ARGS]', 'Run a build with the installed version of packer'
      def build(profile = :default, *args)
        invoke :configure, [profile], options
        run_with_input "#{Interface.packer.command} build - #{ args.join('') }", Interface.packer.render
      end

      desc 'copy PROFILE', 'Copy AMIs generated by packer to other regions'
      def copy(profile)
        invoke :configure, [profile], options

        images.each do |image_name, (image, build)|
          parameters = {
            :source_region => Config.aws.region,
            :source_image_id => image.image_id,
            :name => image_name,
            :description => image.description
          }

          build.ami_regions.each do |region|
            say_status :copy, "image #{image_name} (#{image.image_id}) from #{Config.aws.region} to #{region}"
            Util.ec2(region).copy_image(parameters)
          end
        end

        invoke :wait, [profile], options
        invoke :tag, [profile], options
      end

      desc 'tag PROFILE', 'Tag AMIs in other regions'
      def tag(profile)
        invoke :configure, [profile], options

        images.each do |image_name, (image, build)|
          filters = [{
            :name => 'name',
            :values => [image_name]
          }]

          ## Add some additional tags about the regional source
          image.tags << {
            :key => 'source_region',
            :value => Config.aws.region
          }
          image.tags << {
            :key => 'source_ami',
            :value => image.image_id
          }

          build.ami_regions.each do |region|
            regional_image = Util.ec2(region).describe_images(:filters => filters).images.first

            say_status :tag, "AMI #{image_name} (#{regional_image.image_id}) in #{region}"
            Util.ec2(region).create_tags(:resources => [regional_image.image_id], :tags => image.tags)
          end
        end
      end

      desc 'wait PROFILE', 'Wait for copied AMIs to become available in another region'
      def wait(profile)
        invoke :configure, [profile], options

        waiting = true

        images.each do |image_name, (image, build)|
          say_status :wait, "for #{image.image_id} (#{image_name}) to be available in #{build.ami_regions.join(', ')}", :yellow
        end

        while waiting
          waiting = false

          images.each do |image_name, (image, build)|
            filters = [{
              :name => 'name',
              :values => [image_name]
            }]

            build.ami_regions.each do |region|
              regional_image = Util.ec2(region).describe_images(:filters => filters).images.first

              ## It takes a few seconds for the new AMI to show up in the `describe_images` response-set
              state = regional_image.nil? ? 'unknown' : regional_image.state
              image_id = regional_image.nil? ? 'unknown' : regional_image.image_id

              waiting = (state != 'available') || waiting
              status_color = case state
                             when 'pending', 'unknown' then :yellow
                             when 'available' then :green
                             else :red
                             end

              say_status :image, "#{image_id} (#{image.name}) is #{state} in #{region}", status_color
            end
          end

          ## If waiting == false, loop immediately to break
          sleep(10) if waiting
        end

        say_status :complete, 'All copied images are available'
      end

      private

      ## Find details for generated images in current region
      def images
        @images ||= Config.profile.current.packer.build.each_with_object({}) do |(_, build), memo|
          memo[build.ami_name] = [Control::Data.lookup(:image, :name => build.ami_name).first, build]
        end
      end
    end
  end
end