#!/usr/bin/env python3
# coding=UTF-8
# Copyright (c) 2025 Huawei Technologies Co., Ltd

import argparse
import os

import tasks
import utils

log = utils.stream_logger()
# 计算根目录，期望的目录结构为 yuanrong-functionsystem/scripts/executor/make_functionsystem.py
ROOT_DIR = os.path.realpath(os.path.join(os.path.dirname(__file__), "..", ".."))


def make():
    parser, args = parser_args()
    # 执行对应的函数
    if hasattr(args, "func"):
        args.func(args)
    else:
        parser.print_help()


def parser_args():
    default_jobs = os.cpu_count() or 1
    parser = argparse.ArgumentParser(description="Build script for function system")
    subparsers = parser.add_subparsers(dest="subcommand", required=False)
    # 程序编译执行参数
    build_parser = subparsers.add_parser("build", help="Build all components of function system")
    build_parser.add_argument(
        "-j", "--job_num", type=int, default=default_jobs, help="Set the number of jobs for compile"
    )
    build_parser.add_argument(
        "-v", "--version", type=str, default="0.0.0", help="Set the version for function system build"
    )
    build_parser.add_argument(
        "--build_type",
        type=str,
        choices=["release", "debug"],
        default="release",
        help="Set program compilation mode(release/debug). Default: release",
    )
    build_parser.add_argument(
        "--cmake_args",
        type=utils.parse_kv_args,
        nargs='+',
        default = {},
        help="Add CMAKE compilation parameters. Format: <key>=<val>",
    )
    build_parser.add_argument(
        "--component",
        type=str,
        choices=[
            "all",
            "cli",
            "iam_server",
            "meta_service",
            "function_master",
            "domain_scheduler",
            "runtime_manager",
            "function_proxy",
            "function_agent",
        ],
        default="all",
        help="Build a specific component. Support cpp binaries and go modules(cli/meta_service). Default: all",
    )
    build_parser.add_argument(
        "--linker",
        type=str,
        choices=["auto", "gold", "lld", "mold"],
        default="auto",
        help="Select cpp linker backend. Default: auto (legacy behavior, no mold unless explicitly selected)",
    )
    build_parser.set_defaults(func=lambda func_args: tasks.run_build(ROOT_DIR, func_args))

    # 清理缓存执行参数
    clean_parser = subparsers.add_parser("clean", help="Clean all build artifacts and caches")
    clean_parser.add_argument(
        "--skip_vendor", type=bool, default=True, help="Skip clean vendor build cache and installed files"
    )
    clean_parser.add_argument(
        "--skip_change", type=bool, default=True, help="Skip clean all code change and the files that ignored by git"
    )
    clean_parser.set_defaults(func=lambda func_args: tasks.run_clean(ROOT_DIR, func_args))

    # 测试用例执行参数
    test_parser = subparsers.add_parser("test", help="Run tests for function system")
    test_parser.add_argument(
        "-a",
        "--action",
        type=str,
        choices=["make", "exec", "gcov"],
        default="all",
        help="Choose a specific test action to run: all/make/exec/gcov. Default: all",
    )
    test_parser.add_argument(
        "-j", "--job_num", type=int, default=default_jobs, help="Set the number of jobs for compile"
    )
    test_parser.add_argument(
        "-s", "--test_suite", type=str, default="*", help="Run the specified test suite name. Default: *"
    )
    test_parser.add_argument(
        "-c", "--test_case", type=str, default="*", help="Run the specified test case name. Default: *"
    )
    test_parser.add_argument(
        "-t", "--exec_timeout", type=int, default=120, help="Set timeout for a single test suite. Default: 120"
    )
    test_parser.add_argument(
        "-r", "--retry_times", type=int, default=0, help="Set timeout for a single test suite. Default: 0"
    )
    test_parser.add_argument(
        "-p",
        "--print_logs",
        type=bool,
        default=True,
        help="Set whether to print test case standard output. Default: True",
    )
    test_parser.set_defaults(func=lambda func_args: tasks.run_test(ROOT_DIR, func_args))

    # 打包函数系统构建产物
    pack_parser = subparsers.add_parser("pack", help="Copy and package all compiled products of function system")
    pack_parser.add_argument(
        "-v", "--version", type=str, default="0.0.0", help="Set the version for function system package"
    )
    pack_parser.add_argument(
        "--skip_wheel", default=False, action="store_true", help="Skip packaging products as wheel"
    )
    pack_parser.add_argument(
        "--skip_metrics", default=False, action="store_true", help="Skip packaging metrics to output"
    )
    pack_parser.add_argument(
        "--skip_archive", default=False, action="store_true", help="Skip archiving folder to tar file"
    )
    pack_parser.add_argument(
        "--strip_symbols", default=True, action="store_false", help="Remove the symbols for release products"
    )
    pack_parser.set_defaults(func=lambda func_args: tasks.run_pack(ROOT_DIR, func_args))

    args = parser.parse_args()
    return parser, args


if __name__ == "__main__":
    make()