module DPL
  class Provider
    class Deis < Provider

      def install_deploy_dependencies
        install_url = determine_install_url
        context.shell "curl -sSL #{install_url} | bash -x -s #{option(:cli_version)}"
      end

      #Default to installing the default v1 client. Otherwise determine if this is a v2 client
      def determine_install_url
        return "https://raw.githubusercontent.com/teamhephy/workflow-cli/master/install-v2.sh"
      end

      def needs_key?
        true
      end

      def check_auth
        unless context.shell "./deis login #{option(:controller)}" \
                      " --username=#{option(:username)}" \
                      " --password=#{option(:password)}"
          error 'Login failed.'
        end
      end

      def check_app
        unless context.shell "./deis apps:info --app=#{option(:app)}"
          error 'Application could not be verified.'
        end
      end

      def setup_key(file)
        unless context.shell "./deis keys:add #{file}"
          error 'Adding keys failed.'
        end
      end

      def setup_git_ssh(path, key_path)
        key_path = File.expand_path(key_path)
        path     = File.expand_path(path)

        File.open(path, 'w') do |file|
          file.write "#!/bin/sh\n"
          file.write "exec ssh #{verbose_flag} -o StrictHostKeychecking=no -o CheckHostIP=no -o UserKnownHostsFile=/dev/null -i #{key_path} \"$@\"\n"
        end

        chmod(0740, path)
        context.env['GIT_SSH'] = path

        wait_for_git_access
      end

      def wait_for_git_access()
        retry_count=0
        max_retries=30

        #Get the deis git remote host and port
        remote_uri=repository_url.split("ssh://")[1].split("/")[0]
        remote_host, remote_port = remote_uri.split(":")
        puts "Git remote is #{remote_host} at port #{remote_port}"

        #Try and connect to the github remote via ssh.
        while retry_count < max_retries
          puts "Waiting for ssh key to propagate..."
          if context.shell "#{context.env['GIT_SSH']} #{remote_host} -p #{remote_port}  2>&1 | grep -c 'PTY allocation request failed' > /dev/null"
            puts "SSH connection established."
            break
          end
          retry_count += 1
          sleep(1)
        end
      end

      def remove_key
        unless context.shell "./deis keys:remove #{option(:key_name)}"
          error 'Removing keys failed.'
        end
      end

      def repository_url
        "ssh://git@#{builder_hostname}:2222/#{option(:app)}.git"
      end

      def builder_hostname
        host_tokens = controller_host.split(".")
        host_tokens[0] = [host_tokens[0], "builder"].join("-")
        host_tokens.join(".")
      end

      def controller_host
        option(:controller).gsub(/https?:\/\//, "").split(":")[0]
      end

      def push_app
        unless context.shell "bash -c 'git push #{verbose_flag} #{repository_url} HEAD:refs/heads/master -f 2>&1 | tr -dc \"[:alnum:][:space:][:punct:]\" | sed -E \"s/remote: (\\[1G)+//\" | sed \"s/\\[K$//\"; exit ${PIPESTATUS[0]}'"
          error 'Deploying application failed.'
        end
      end

      def run(command)
        unless context.shell "./deis run -a #{option(:app)} -- #{command}"
          error 'Running command failed.'
        end
      end

      def cleanup
        return if options[:skip_cleanup]
        context.shell "mv deis ~/deis"
        super
        context.shell "mv ~/deis deis"
      end

      def verbose_flag
        '-v' if options[:verbose]
      end
    end
  end
end