module Kontena::Cli::Stacks
  module StacksHelper

    def wait_for_deployment_to_start(deployment, timeout = 600)
      started = false
      Timeout::timeout(timeout) do
        until started
          deployment = client.get("stacks/#{deployment['stack_id']}/deploys/#{deployment['id']}")
          started = true if deployment['service_deploys'].size > 0
          sleep 1
        end
        if deployment['state'] == 'error'
          deployment['service_deploys'].each do |service_deploy|
            if service_deploy['state'] == 'error'
              puts "        #{service_deploy['reason']}"
            end
          end

          raise 'deploy failed'
        end
      end

      started
    rescue Timeout::Error
      raise 'deploy timed out'
    end

    # @param [Hash] deployment
    # @return [Boolean]
    def wait_for_deploy_to_finish(deployment, timeout = 600)
      deployed = false
      states = %w(success error)
      tracked_services = []
      Timeout::timeout(timeout) do
        until deployed
          deployment = client.get("stacks/#{deployment['stack_id']}/deploys/#{deployment['id']}")
          service_deploy = deployment['service_deploys'].find{ |s| s['state'] == 'ongoing' }
          if service_deploy
            tracked_services << service_deploy['id']
            wait_for_service_deploy(service_deploy, states)
          end
          if states.include?(deployment['state'])
            deployed = true
            deployment['service_deploys'].select{ |s| !tracked_services.include?(s['id']) }.each do |s|
              wait_for_service_deploy(s, states)
            end
          else
            sleep 1
          end
        end
        if deployment['state'] == 'error'
          deployment['service_deploys'].each do |service_deploy|
            if service_deploy['state'] == 'error'
              error service_deploy['reason']
            end
          end
          abort
        end
      end

      deployed
    rescue Timeout::Error
      raise 'deploy timed out'
    end

    def wait_for_service_deploy(service_deploy, states)
      service_deployed = false
      name = service_deploy['service_id'].split('/')[-1]
      spinner "Deploying service #{pastel.cyan(name)}" do
        until service_deployed
          r = client.get("services/#{service_deploy['service_id']}/deploys/#{service_deploy['id']}")
          if states.include?(r['state'])
            service_deployed = true
          else
            sleep 1
          end
        end
      end
    end
  end
end