#!/usr/bin/env python

# Copyright (c) 2017 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

"""
Resources to handle tier related requests
"""

import pkg_resources
import re

from flask import jsonify
from flasgger.utils import swag_from

from functest.api.base import ApiResource
from functest.api.common import api_utils
from functest.cli.commands.cli_tier import Tier


class V1Tiers(ApiResource):
    """ V1Tiers Resource class """

    @swag_from(pkg_resources.resource_filename(
        'functest', 'api/swagger/tiers.yaml'))
    def get(self):
        # pylint: disable=no-self-use
        """ GET all tiers """
        tiers_list = Tier().list()
        data = re.split("[\n\t]", tiers_list)
        data = [i.strip() for i in data if i != '']
        data_dict = dict()
        for i in range(len(data) / 2):
            one_data = {data[i * 2].lstrip('- ').rstrip(':'): data[i * 2 + 1]}
            if i == 0:
                data_dict = one_data
            else:
                data_dict.update(one_data)
        result = {'tiers': data_dict}
        return jsonify(result)


class V1Tier(ApiResource):
    """ V1Tier Resource class """

    @swag_from(pkg_resources.resource_filename(
        'functest', 'api/swagger/tier.yaml'))
    def get(self, tier_name):  # pylint: disable=no-self-use
        """ GET the info of one tier """
        tier_info = Tier().show(tier_name)
        if not tier_info:
            return api_utils.result_handler(
                status=1,
                data="The tier with name '%s' does not exist." % tier_name)
        tier_info.__dict__.pop('name')
        tier_info.__dict__.pop('tests_array')
        tier_info.__dict__.pop('skipped_tests_array')
        testcases = Tier().gettests(tier_name)
        result = {'tier': tier_name, 'testcases': testcases}
        result.update(tier_info.__dict__)
        return jsonify(result)


class V1TestcasesinTier(ApiResource):
    """ V1TestcasesinTier Resource class """

    @swag_from(pkg_resources.resource_filename(
        'functest', 'api/swagger/testcases_in_tier.yaml'))
    def get(self, tier_name):  # pylint: disable=no-self-use
        """ GET all testcases within given tier """
        tier_info = Tier().show(tier_name)
        if not tier_info:
            return api_utils.result_handler(
                status=1,
                data="The tier with name '%s' does not exist." % tier_name)
        testcases = Tier().gettests(tier_name)
        result = {'tier': tier_name, 'testcases': testcases}
        return jsonify(result)