#!/usr/bin/env python3

# -*- coding: utf-8 -*-

# Copyright (C) 2025 Huawei Device Co., Ltd.

# Licensed under the Apache License, Version 2.0 (the "License");

# you may not use this file except in compliance with the License.

# You may obtain a copy of the License at

#

#     http://www.apache.org/licenses/LICENSE-2.0

#

# Unless required by applicable law or agreed to in writing, software

# distributed under the License is distributed on an "AS IS" BASIS,

# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.

# See the License for the specific language governing permissions and

# limitations under the License.



import subprocess

import zipfile

import hashlib

import argparse

import os





def install_dependencies(requirements_file):

    try:

        subprocess.check_call(["pip", "install", "-r", requirements_file])

        print(f"install requirements.txt success")

    except subprocess.CalledProcessError as e:

        print(f"install dependence fail: {str(e)}")

 



def check_exist_file():

    file_list = []

    file_list.append("AACommand07.hap")

    file_list.append("libA_v10001.hsp")

    file_list.append("libB_v10001.hsp")

    url = "https://gitee.com/TaowerfulMAX/h2dij432sfa423o_debugfortest/releases/download/0.0.1-debug/package_0.0.1.zip"

    expected_md5 = "0147bba3851cbada9379f106cf0f7b54"

    from testModule.utils import GP

    for file in file_list:

        if not os.path.exists(os.path.join(GP.local_path, file)):

            if download_and_extract_zip(url, os.path.join(GP.local_path), expected_md5):

                print(f"{file} File Download Success")

                continue

            else:

                print(f"{file} File Download Failed")

            print(f"No {file} File!")

            print(f"请自行访问以下链接，下载package.zip中的安装包文件解压到当前脚本resource目录中，"

                  "操作完成该步骤后重新执行prepare.py脚本。")

            print("Please download from the url below and unzip package.zip to resource directory,"

                  "please rerun after operation.")

            print(f"url: {url}")

            exit(1)





def download_file(url, timeout=(10, 30)):

    try:

        import requests

    except ModuleNotFoundError:

        print("Please install requests module, command: [pip install requests]")

        exit(1)

    try:

        response = requests.get(url, timeout=timeout)

        response.raise_for_status()  # 将触发HTTPError，如果状态不是200

        return True, response.content

    except requests.exceptions.Timeout:

        return False, "请求超时"

    except requests.exceptions.HTTPError as err:

        return False, f"HTTP错误：{err}"

    except requests.exceptions.RequestException as e:

        return False, f"请求异常：{e}"





def save_to_file(content, filename):

    with open(filename, 'wb') as f:

        f.write(content)





def extract_zip(filename, extract_to='.'):

    with zipfile.ZipFile(filename, 'r') as zip_ref:

        zip_ref.extractall(extract_to)





def calculate_md5(filename):

    hash_md5 = hashlib.md5()

    try:

        with open(filename, 'rb') as f:

            for chunk in iter(lambda: f.read(4096), b""):

                hash_md5.update(chunk)

        return hash_md5.hexdigest()

    except PermissionError:

        return "PermissionError"

    except FileNotFoundError:

        return "FileNotFoundError"





def download_and_extract_zip(url, extract_to='.', expected_md5=None):

    # 下载ZIP文件

    is_success, content = download_file(url)

    if not is_success:

        print(f"download_file failed: {content}")

        return False

    # 获取ZIP文件名

    zip_filename = url.split('/')[-1]

    

    # 写入本地文件

    save_to_file(content, zip_filename)

    

    # 如果提供了预期的MD5值，则进行校验

    if expected_md5:

        file_md5 = calculate_md5(zip_filename)

        if file_md5 != expected_md5:

            raise Exception(f"MD5校验失败：预期的MD5为{expected_md5}，实际为{file_md5}")

        else:

            print("MD5校验成功")

    

    # 解压ZIP文件

    extract_zip(zip_filename, extract_to)

    

    # 可选：删除ZIP文件

    os.remove(zip_filename)

    print(f"文件已解压到：{extract_to}")

    return True





def prepare(args=None):

    if vars(args) == vars(parser.parse_args([])) or args.requirements:

        install_dependencies("requirements.txt")

    from testModule.utils import GP, gen_package_dir, update_source, prepare_source, rmdir

    if args is not None:

        if args.source:

            rmdir(os.path.join(GP.local_path, "version"))

        if args.config:

            rmdir(os.path.join(".hdctester.conf"))

    test_path = os.path.join(os.getcwd(), "testModule")

    if not os.path.exists(test_path):

        print("testModule not exist")

        return

    GP.init()

    prepare_source()

    update_source()

    check_exist_file()

    gen_package_dir()





if __name__ == "__main__":

    parser = argparse.ArgumentParser()

    parser.add_argument('--source', '-s', action='store_true', help='source update')

    parser.add_argument('--config', '-c', action='store_true', help='config update')

    parser.add_argument('--requirements', '-r', action='store_true', help='requirements install')

    args = parser.parse_args()

    prepare(args)