# Copyright 2016 Google Inc. All Rights Reserved.
#
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at
#
#    http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.

"""The `app operations wait` command."""

from __future__ import absolute_import
from __future__ import unicode_literals
from googlecloudsdk.api_lib.app import appengine_api_client
from googlecloudsdk.api_lib.app import operations_util
from googlecloudsdk.calliope import base
from googlecloudsdk.core import log
from googlecloudsdk.core.console import progress_tracker


class Wait(base.Command):
  """Polls an operation until completion."""

  detailed_help = {
      'EXAMPLES': """\
          To wait for an App Engine operation called o1 to complete, run:

              $ {command} o1
          """,
  }

  @staticmethod
  def Args(parser):
    parser.add_argument('operation', help='ID of operation.')

  def Run(self, args):
    api_client = appengine_api_client.GetApiClientForTrack(self.ReleaseTrack())
    operation = api_client.GetOperation(args.operation)
    if operation.done:
      log.status.Print(
          'Operation [{0}] is already done.'.format(args.operation))
      return operation
    else:
      with progress_tracker.ProgressTracker(
          'Waiting for operation [{0}] to complete.'.format(args.operation)):
        return operations_util.WaitForOperation(
            api_client.client.apps_operations, operation)
