# -------------------------------------------------------------------------
#  This file is part of the MindStudio project.
# Copyright (c) 2025 Huawei Technologies Co.,Ltd.
#
# MindStudio is licensed under Mulan PSL v2.
# You can use this software according to the terms and conditions of the Mulan PSL v2.
# You may obtain a copy of Mulan PSL v2 at:
#
#          http://license.coscl.org.cn/MulanPSL2
#
# 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 FIT FOR A PARTICULAR PURPOSE.
# See the Mulan PSL v2 for more details.
# -------------------------------------------------------------------------

"""
this is atk bin module
"""
import logging
import os
import click
from atk import VERSION_BANNER
from atk.bin.base import DynamicCLI, AtkOption

HELP_GROUP = 'main'


@click.group(cls=DynamicCLI, invoke_without_command=True, chain=True)
@click.option("--version", cls=AtkOption, is_flag=True, help_group="Global Options")
@click.option(
    "--nocolor",
    cls=AtkOption,
    is_flag=True,
    default=False,
    help="Disable colored logs",
    help_group="Global Options",
)
@click.pass_context
def cli(ctx, version, nocolor) -> int:
    """
    atk 命令主函数入口
    :param ctx:  命令参数上下文信息
    :param version:  版本号
    :return:  执行命令结果
    """
    ctx.ensure_object(dict)
    if version:
        click.echo(VERSION_BANNER)
    logging.info(f'ATK Version: {VERSION_BANNER}')
    os.environ["NO_COLOR"] = "1" if nocolor else "0"
    return 0