require 'capistrano'
require 'capistrano/cli'

require 'simple_deploy/stack/deployment/status'

module SimpleDeploy
  class Stack
    class Deployment
      def initialize(args)
        @config = args[:config]
        @instances = args[:instances]
        @environment = args[:environment]
        @ssh_user = args[:ssh_user]
        @ssh_key = args[:ssh_key]
        @stack = args[:stack]
        @name = args[:name]
        @attributes = @stack.attributes
        @logger = @config.logger
        @region = @config.region @environment
      end

      def create_deployment 
        if @instances.nil? || @instances.empty?
          raise "There are no running instances to deploy to"
        end

        @deployment = Capistrano::Configuration.new :output => @logger
        @deployment.logger.level = 3
        @logger.info "Creating deployment to #{@name}."

        set_ssh_gateway
        set_ssh_user
        set_ssh_options
        set_instances
        set_deploy_command
      end

      def execute(force=false)
        if clear_for_deployment?
          status.set_deployment_in_progress
          @logger.info 'Starting deployment.'
          @deployment.simpledeploy
          @logger.info 'Deployment complete.'
          status.unset_deployment_in_progress
          true
        else
          @logger.error "Not clear to deploy."
          false
        end
      end

      def clear_for_deployment?
        status.clear_for_deployment?
      end

      def clear_deployment_lock(force=false)
        status.clear_deployment_lock(force)
      end

      def ssh
        @stack.instances.map do |instance|
          info = instance['instancesSet'].first
          if info['vpcId']
            gw = @attributes['ssh_gateway']
            "\nssh -i #{@ssh_key} -l #{@ssh_user} -L 9998:#{info['privateIpAddress']}:22 -N #{gw} &\nssh -p 9998 localhost"
          else
            "\nssh -i #{@ssh_key} -l #{@ssh_user} #{info['ipAddress']}"
          end
        end
      end

      private

      def set_deploy_command
        cmd = 'env '
        get_artifact_endpoints.each_pair do |key,value|
          cmd += "#{key}=#{value} "
        end
        cmd += "PRIMARY_HOST=#{primary_instance} #{deploy_script}"

        @logger.info "Deploy command: '#{cmd}'."
        @deployment.load :string => "task :simpledeploy do
        sudo '#{cmd}'
        end"
      end

      def get_artifact_endpoints
        h = {}
        @config.artifacts.each do |artifact|
          variable = @config.artifact_deploy_variable artifact
          bucket_prefix = @attributes["#{artifact}_bucket_prefix"]
          domain = @attributes["#{artifact}_domain"]

          artifact = Artifact.new :name          => artifact,
                                  :id            => @attributes[artifact],
                                  :region        => @region,
                                  :domain        => domain,
                                  :bucket_prefix => bucket_prefix

          h[variable] = artifact.endpoints['s3']
        end
        h
      end

      def set_instances
        @instances.each do |instance| 
          @logger.debug "Deploying to instance #{instance}."
          @deployment.server instance, :instances
        end
      end

      def set_ssh_options
        @logger.debug "Setting key to #{@ssh_key}."
        @deployment.variables[:ssh_options] = { :keys     => @ssh_key, 
                                                :paranoid => false }
      end

      def set_ssh_gateway
        ssh_gateway = @attributes['ssh_gateway']
        if ssh_gateway && !ssh_gateway.empty?
          @deployment.set :gateway, ssh_gateway
          @logger.info "Proxying via gateway #{ssh_gateway}."
        else
          @logger.debug "Not using an ssh gateway."
        end
      end

      def set_ssh_user
        @logger.debug "Setting user to #{@ssh_user}."
        @deployment.set :user, @ssh_user
      end

      def primary_instance 
        if @stack.instances.any?
          @stack.instances.first['instancesSet'].first['privateIpAddress']
        end
      end

      def deploy_script
        @config.deploy_script
      end

      def status
        options = { :name        => @name,
                    :environment => @environment,
                    :ssh_user    => @ssh_user,
                    :config      => @config,
                    :stack       => @stack }
        @status ||= SimpleDeploy::Stack::Deployment::Status.new options
      end

    end
  end
end