##############################################################################
# Copyright (c) 2016 Huawei Technologies Co.,Ltd and others.
#
# All rights reserved. This program and the accompanying materials
# are made available under the terms of the Apache License, Version 2.0
# which accompanies this distribution, and is available at
# http://www.apache.org/licenses/LICENSE-2.0
##############################################################################
import os
import sys
import yaml
import click
import six

import dovetail.utils.dovetail_utils as dt_utils
from dovetail.utils.dovetail_config import DovetailConfig as dt_cfg
from dovetail.testcase import Testsuite


class CliTestcase(object):

    @classmethod
    def testsuite_load(cls):
        dt_cfg.load_config_files()
        Testsuite.load()

    @classmethod
    def get_path(cls, path):
        if isinstance(path, six.string_types):
            dt_cfg.load_config_files()
            dovetail_dir = os.path.abspath(
                os.path.join(os.path.dirname(__file__),
                             os.pardir, os.pardir))
            relative_path = dt_cfg.dovetail_config[path]
            abs_path = os.path.join(dovetail_dir, relative_path)
        else:
            click.echo("input %s is not a string" % path)
            sys.exit(1)
        return abs_path

    def list_testcase(self, testsuite):
        self.testsuite_load()
        if testsuite:
            testsuite_stream = Testsuite.get(testsuite)
            if testsuite_stream:
                testcase_list = []
                for value in testsuite_stream['testcases_list']:
                    if value is not None:
                        testcase_list.append(value)
                testarea_list = []
                for testcase in testcase_list:
                    testarea = testcase.split('.')[1]
                    if testarea not in testarea_list:
                        testarea_list.append(testarea)
                for testarea in testarea_list:
                    click.echo("- %s" % testarea)
                    for testcase in testcase_list:
                        if testarea in testcase:
                            click.echo("    %s" % testcase)
            else:
                click.echo("testsuite %s does not exist or not supported"
                           % testsuite)
        else:
            testsuite_json = Testsuite.get_all()
            if testsuite_json:
                for key, value in testsuite_json.items():
                    click.echo("- %s" % key)
                    testsuite_stream = Testsuite.get(key)
                    if testsuite_stream['testcases_list']:
                        for testcase in testsuite_stream['testcases_list']:
                            click.echo("    %s" % testcase)
                    else:
                        click.echo("No testcase in testsuite %s" % key)
            else:
                click.echo("No testsuite defined yet in dovetail!!!")

    def show_testcase(self, testcase):
        abs_testcase_path = self.get_path('TESTCASE_PATH')
        if testcase.startswith("dovetail."):
            testcase_yml = testcase[9:] + '.yml'
        else:
            testcase_yml = testcase + '.yml'
        for root, dirs, files in os.walk(abs_testcase_path):
            if testcase_yml in files:
                testcase_path = abs_testcase_path + testcase_yml
                with open(testcase_path, 'r') as stream:
                    try:
                        click.echo(stream.read())
                    except yaml.YAMLError as exc:
                        click.echo(exc)
            else:
                click.echo("testcase %s does not exist or not supported"
                           % testcase)

    def run(self, args_str):
        options = ''
        if args_str:
            options = options + args_str

        repo_dir = os.path.abspath(
            os.path.join(os.path.dirname(__file__),
                         os.pardir, os.pardir))

        cmd = ("python %s/run.py"
               " %s" % (repo_dir, options))
        dt_utils.exec_cmd(cmd, exit_on_error=False,
                          exec_msg_on=False, info=True)