# -*- coding: utf-8 -*-
# this script is use to collect aegis client and update logs
import os
import sys
import shutil
import logging

from aegis_checker.common.common_path import get_aegis_globalcfg_dir, get_aegis_update_dir, get_aegis_log_dir_in_checker, get_aegis_root_folder
from aegis_checker.common.print_log import log_warning


def copy_aegis_log(src, dest):
    if os.path.exists(src) and os.path.isdir(src):
        # clear dest folder first
        if os.path.exists(dest):
            shutil.rmtree(dest)

        shutil.copytree(src, dest)
        logging.debug("copy log success, src is %s, dest is %s", src, dest)
    else:
        log_warning("copy log fail, %s is not exists or is not dir" % src)


def collect_aegis_log():
    aegis_root_dir = get_aegis_root_folder()
    aegis_update_log_dir = os.path.join(get_aegis_update_dir(), "data")
    globalcf_dir = get_aegis_globalcfg_dir()
    aegis_log_dir = get_aegis_log_dir_in_checker()

    copy_logs = [
        (aegis_update_log_dir, os.path.join(aegis_log_dir, "aegis_update")),
        (os.path.join(aegis_root_dir, "PythonLoader", "data"), os.path.join(aegis_log_dir, "PythonLoader")),
        (os.path.join(globalcf_dir, "AliNet", "log"), os.path.join(aegis_log_dir, "AliNet")),
        # (os.path.join(globalcf_dir, "AliHips", "log"), os.path.join(aegis_log_dir, "AliHips")),
        (os.path.join(globalcf_dir, "AliSecGuard", "log"), os.path.join(aegis_log_dir, "AliSecGuard")),
    ]

    aegis_client_dir = os.path.join(aegis_root_dir, "aegis_client")
    if os.path.exists(aegis_client_dir):
        aegis_client_ver_dirs = os.listdir(aegis_client_dir)
        if not aegis_client_ver_dirs:
            log_warning("there is no aegis client")

        for aegis_client_ver_dir_name in aegis_client_ver_dirs:
            aegis_client_ver_dir = os.path.join(aegis_client_dir, aegis_client_ver_dir_name)
            if os.path.isdir(aegis_client_ver_dir) and aegis_client_ver_dir_name.startswith("aegis_"):
                copy_logs.append((os.path.join(aegis_client_ver_dir, "data"), os.path.join(aegis_log_dir, aegis_client_ver_dir_name)))
    else:
        log_warning(aegis_client_dir + " is not exists")

    for src, dest in copy_logs:
        copy_aegis_log(src, dest)

    # copy file
    copy_files = [
        os.path.join(globalcf_dir, "ui_ud"),
        os.path.join(globalcf_dir, "aegis.crt"),
        os.path.join(globalcf_dir, "domaincfg.ini")
    ]

    for copy_file in copy_files:
        if os.path.exists(copy_file):
            shutil.copy(copy_file, aegis_log_dir)
        else:
            logging.debug("%s is not exists", copy_file)


if __name__ == '__main__':
    collect_aegis_log()