#!/usr/bin/env python3
# -----------------------------------------------------------------------------------------------------------
# Copyright (c) 2025 Huawei Technologies Co., Ltd.
# This program is free software, you can redistribute it and/or modify it under the terms and conditions of
# CANN Open Software License Agreement Version 2.0 (the "License").
# Please refer to the License for details. You may not use this file except in compliance with the License.
# THIS SOFTWARE IS PROVIDED ON AN "AS IS" BASIS, WITHOUT WARRANTIES OF ANY KIND, EITHER EXPRESS OR IMPLIED,
# INCLUDING BUT NOT LIMITED TO NON-INFRINGEMENT, MERCHANTABILITY, OR FITNESS FOR A PARTICULAR PURPOSE.
# See LICENSE in the root of the software repository for the full text of the License.
# -----------------------------------------------------------------------------------------------------------
"""
版本兼容性检查

检查当前代码仓与基础 CANN 包间的兼容性.
"""

import argparse
import logging
from pathlib import Path
from typing import NoReturn


class VersionChecker:
    @classmethod
    def main(cls) -> NoReturn:
        parser = argparse.ArgumentParser(description="Check Version Compatible", epilog="Best Regards!")
        sub_parser = parser.add_subparsers(help="Sub-Command")
        # 参数注册
        parser.add_argument("--cann_path", required=True, nargs=1, type=str, help="CANN install path")
        parser.add_argument("--cann_package_name", required=True, nargs=1, type=str, help="CANN package name")
        # 子命令行(Check)
        p_chk = sub_parser.add_parser("check_code_compatible", help="Check Version Compatible.")
        p_chk.add_argument(
            "--code_version_info_file", required=True, nargs=1, type=str, help="Code version info file path"
        )
        p_chk.set_defaults(func=VersionChecker._check_compatible)
        # 子命令行(Get)
        p_get = sub_parser.add_parser("get_package_version", help="Get Package Version")
        p_get.set_defaults(func=VersionChecker._get_package_version)
        # 参数处理
        args = parser.parse_args()
        # 基本合法性检查, 版本号获取
        cann_version_info_file = Path(args.cann_path[0], args.cann_package_name[0], "version.info").absolute()
        if not cann_version_info_file.exists():
            raise ValueError(f"CANN version info file({cann_version_info_file}) not exist.")
        ret, cann_version = cls._get_version_str(file=cann_version_info_file)
        if not ret:
            raise ValueError(f"Can't get version from CANN version info file({cann_version_info_file}).")
        rst = args.func(cann_version, args)
        return rst

    @classmethod
    def _check_compatible(cls, cann_version: str, args) -> str:
        code_version_info_file = Path(args.code_version_info_file[0]).absolute()
        if not code_version_info_file.exists():
            raise ValueError(f"Code version info file({code_version_info_file}) not exist.")
        ret, code_version = cls._get_version_str(file=code_version_info_file)
        if not ret:
            raise ValueError(f"Can't get version from Code version info file({code_version_info_file}).")
        # 兼容性检查
        cann_sub_version = cann_version.rsplit(".", 1)[0]
        code_sub_version = code_version.rsplit(".", 1)[0]
        if cann_sub_version != code_sub_version:
            raise ValueError(
                f"The version number of the current code is {code_sub_version}, "
                f"and the version number of the cann package used is {cann_sub_version}. "
                f"Please install version {code_sub_version} of the cann package."
            )
        return cann_sub_version

    @classmethod
    def _get_package_version(cls, cann_version: str, args) -> str:
        cann_sub_version = cann_version.rsplit(".", 1)[0]
        return cann_sub_version

    @classmethod
    def _get_version_str(cls, file: Path):
        with open(file) as fh:
            lines = fh.readlines()
            for line in lines:
                if not line.startswith("Version="):
                    continue
                version = line[8:].replace("\r", "").replace("\n", "")
                return True, version
        return False, ""


if __name__ == "__main__":
    logging.basicConfig(format="%(filename)s:%(lineno)d [%(levelname)s] %(message)s", level=logging.INFO)
    try:
        print(VersionChecker.main())
    except Exception as e:
        logging.error(e)
        exit(1)