#!/usr/bin/env python
# -*- coding: utf-8 -*-

###############################################################################
#  Copyright 2016 Kitware Inc.
#
#  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.
###############################################################################

import time
import json

import argparse
from girder_client import GirderClient, HttpError

def wait_for_status(client, taskflow_id, status):
    current_status = None
    status_url = 'taskflows/%s/status' % (taskflow_id)
    tasks_url = 'taskflows/%s/tasks' % (taskflow_id)
    while current_status != status:
        r = client.get(status_url)
        current_status = r['status']
        print('Taskflow status: %s' % current_status)
        r = client.get(tasks_url)
        print('Tasks in flow: %d' % len(r))
        time.sleep(1)


def wait_for_complete(client, taskflow_id):
    wait_for_status(client, taskflow_id, 'complete')

def wait_for_terminated(client, taskflow_id):
    wait_for_status(client, taskflow_id, 'terminated')

def wait_for_deletion(client, taskflow_id):
    try:
        wait_for_status(client, taskflow_id, 'deleted')
    except HttpError as ex:
        pass

def create_taskflow(client, cls_name):
    url = 'taskflows'
    body = {
        'taskFlowClass': cls_name
    }
    r = client.post(url, data=json.dumps(body))

    return r['_id']

def main(config):
    client = GirderClient(apiUrl=config.girder_api_url)
    client.authenticate(config.girder_user,
                        config.girder_password)

    # Load any parameters
    params = {}
    if config.taskflow_start_params is not None:
        with open(config.taskflow_start_params) as fp:
            params = json.load(fp)

    print params

    try:
        print ('Running %s taskflow ...' % config.taskflow_start_params)
        taskflow_id = create_taskflow(
            client, config.taskflow_class)

        # Start the task flow
        url = 'taskflows/%s/start' % (taskflow_id)
        client.put(url, data=json.dumps(params))

        # Wait for it to complete
        wait_for_complete(client, taskflow_id)

    except HttpError as ex:
        print( ex.responseText)

if __name__ == '__main__':
    parser = argparse.ArgumentParser(description='Run test')
    parser.add_argument('-a', '--girder_api_url', help='', required=True)
    parser.add_argument('-u', '--girder_user', help='', required=True)
    parser.add_argument('-p', '--girder_password', help='', required=True)
    parser.add_argument('-c', '--taskflow_class', help='', required=True)
    parser.add_argument('-v', '--taskflow_start_params', help='', required=False, default=None)
    args = parser.parse_args()

    main(args)