# -*- coding: UTF-8 -*-

import os
import re
import sys
import subprocess
from abc import ABCMeta, abstractmethod

DEFAULT_UT_FILE = 'test/test_npu/test_add.py'

class AccurateTest(metaclass=ABCMeta):
    @abstractmethod
    def identify(self, modify_files):
        """
        该接口提供代码对应的UT的路径信息
        """
        raise Exception("abstract method.")
    
    def find_ut_by_regex(self, regex):
        ut_files = []
        cmd = "find {} -name {}".format('test/test_npu', regex)
        status, output = subprocess.getstatusoutput(cmd)
        if status:
            pass # 对于找不到的暂时不作处理
        else:
            if output:
                files = output.split('\n')
                for ut_file in files:
                    if ut_file.endswith(".py"):
                        ut_files.append(ut_file)
        return ut_files


class OpStrategy(AccurateTest):
    """
    通过对适配层的代码的识别
    """
    def identify(self, modify_file):
        """
        通过对于算子实现文件的文件名解析获取其单元测试的名字，比如：
        BinaryCrossEntropyWithLogitsBackwardKernelNpu.cpp
        针对这个文件，先识别关键字BinaryCrossEntropyWithLogitsBackward
        然后，获取其正则表达式*binary*cross*entropy*with*logits*backward*识别到符合要求的测试用例
        具体方法：通过大写字母切分关键字，再识别包含所有这些关键字的测试文件名。
        """
        filename = os.path.basename(modify_file)
        if filename.find('KernelNpu') >= 0: 
            feature_line = filename.split('KernelNpu')[0]
            features = re.findall('[A-Z][^A-Z]*', feature_line)
            regex = '*' + '*'.join(["%s" % feature.lower() for feature in features]) + '*'
            return self.find_ut_by_regex(regex)
        return []


class DirectoryStrategy(AccurateTest):
    """
    通过识别测试文件的目录确认需要进行UT的文件
    """
    def identify(self, modify_file):
        second_dir = modify_file.split("/")[0]
        if second_dir == 'test':
            return [modify_file]
        return []


class CopyOptStrategy(AccurateTest):
    """
    通过识别非连续转连续的测试用例
    """
    def identify(self, modify_file):
        if modify_file.find('contiguous') > 0:
            regex = '*contiguous*'
            return self.find_ut_by_regex(regex)
        return []


class TestMgr():
    def __init__(self):
        self.modify_files = []
        self.ut_files = []

    def load(self, modify_files):
        with open(modify_files) as f:
            for line in f:
                line = line.strip()
                self.modify_files.append(line)

    def analyze(self):
        for modify_file in self.modify_files:
            self.ut_files += DirectoryStrategy().identify(modify_file)
            self.ut_files += OpStrategy().identify(modify_file)
            self.ut_files += CopyOptStrategy().identify(modify_file)
        unique_files = set(self.ut_files)

        exist_ut_file = []
        for changed_file in unique_files:
            if os.path.exists(changed_file):
                exist_ut_file.append(changed_file)
        self.ut_files = exist_ut_file

        if len(self.ut_files) == 0:
            self.ut_files.append(DEFAULT_UT_FILE)

    def get_ut_files(self):
        return self.ut_files

    def print_modify_files(self):
        print("modify files:")
        for modify_file in self.modify_files:
            print(modify_file)

    def print_ut_files(self):
        print("ut files:")
        for ut_file in ut_files:
            print(ut_file)


def exec_ut(ut_files):
    """
    执行单元测试文件，其中存在失败，则标识异常并打印相关信息
    """
    def change_dir_and_exec(ut_path):
        ut_dir = os.path.dirname(ut_path)
        ut_file = os.path.basename(ut_path)
        os.chdir(ut_dir)
        cmd = "python3 {}".format(ut_file)
        p = subprocess.Popen(cmd, stderr=subprocess.STDOUT, stdout=subprocess.PIPE, shell=True)
        try:
            msg = p.communicate(timeout=300)
            ret = p.poll()
            if ret:
                print(msg)

        except subprocess.TimeoutExpired:
            p.kill()
            p.terminate()
            ret = 1
            print("Timeout: Command '" + cmd + "' timed out after 300 seconds")
        except Exception as err:
            ret = 1
            print(err)
        return ret



    ret_status = 0
    cur_dir = os.path.abspath(os.path.dirname(__file__))
    exec_infos = []
    for ut_file in ut_files:
        os.chdir(cur_dir)
        ret = change_dir_and_exec(ut_file)
        if ret:
            ret_status = ret
            exec_infos.append("exec ut {} failed.".format(ut_file))
        else:
            exec_infos.append("exec ut {} success.".format(ut_file))
    
    print("***** Total result:")
    for exec_info in exec_infos:
        print(exec_info)
    return ret_status


if __name__ == "__main__":
    cur_dir = os.path.abspath(os.path.dirname(__file__))
    modify_files = os.path.join(cur_dir, 'modify_files.txt')
    test_mgr = TestMgr()
    test_mgr.load(modify_files)
    test_mgr.analyze()
    ut_files = test_mgr.get_ut_files()

    test_mgr.print_modify_files()
    test_mgr.print_ut_files()

    ret = exec_ut(ut_files)
    sys.exit(ret)