#!/usr/bin/env python
#
# jose.lausuch@ericsson.com
# 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
#

""" global variables """

import os
import pkg_resources

import click

import functest.ci.tier_builder as tb
from functest.utils.constants import CONST
import functest.utils.functest_utils as ft_utils


class Tier(object):

    def __init__(self):
        self.tiers = tb.TierBuilder(
            CONST.__getattribute__('INSTALLER_TYPE'),
            CONST.__getattribute__('DEPLOY_SCENARIO'),
            pkg_resources.resource_filename('functest', 'ci/testcases.yaml'))

    def list(self):
        summary = ""
        for tier in self.tiers.get_tiers():
            summary += ("    - %s. %s:\n\t   %s\n"
                        % (tier.get_order(),
                           tier.get_name(),
                           tier.get_test_names()))
        return summary

    def show(self, tiername):
        tier = self.tiers.get_tier(tiername)
        if tier is None:
            return None
        else:
            tier_info = self.tiers.get_tier(tiername)
            return tier_info

    def gettests(self, tiername):
        tier = self.tiers.get_tier(tiername)
        if tier is None:
            return None
        else:
            tests = tier.get_test_names()
            return tests

    @staticmethod
    def run(tiername, noclean=False, report=False):

        flags = ""
        if noclean:
            flags += "-n "
        if report:
            flags += "-r "

        if not os.path.isfile(CONST.__getattribute__('env_active')):
            click.echo("Functest environment is not ready. "
                       "Run first 'functest env prepare'")
        else:
            cmd = "run_tests {}-t {}".format(flags, tiername)
            ft_utils.execute_command(cmd)


class CliTier(Tier):

    def __init__(self):
        super(CliTier, self).__init__()

    def list(self):
        click.echo(super(CliTier, self).list())

    def show(self, tiername):
        tier_info = super(CliTier, self).show(tiername)
        if tier_info:
            click.echo(tier_info)
        else:
            tier_names = self.tiers.get_tier_names()
            click.echo("The tier with name '%s' does not exist. "
                       "Available tiers are:\n  %s\n" % (tiername, tier_names))

    def gettests(self, tiername):
        tests = super(CliTier, self).gettests(tiername)
        if tests:
            click.echo("Test cases in tier '%s':\n %s\n" % (tiername, tests))
        else:
            tier_names = self.tiers.get_tier_names()
            click.echo("The tier with name '%s' does not exist. "
                       "Available tiers are:\n  %s\n" % (tiername, tier_names))