#!/usr/bin/env python
# coding=utf-8
# Copyright (c) Huawei Technologies Co., Ltd. 2024-2025. All rights reserved.
# MindIE 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.

import argparse
import os
import sys
import subprocess
from setuptools import setup
from setuptools.command.build_py import build_py

os.environ["SOURCE_DATE_EPOCH"] = "0"

parser = argparse.ArgumentParser(description="MindIE LLM Setup Parameters")
parser.add_argument("--setup_cmd", type=str, default="bdist_wheel")
parser.add_argument("--version", type=str, default="1.0.RC3")

args = parser.parse_args()
sys.argv = [sys.argv[0], args.setup_cmd]
mindie_llm_version = args.version


# 定义.proto文件路径（假设在项目根目录的protos/文件夹下）
class BuildPyCommand(build_py):
    def run(self):
        # 获取当前环境变量
        env = os.environ.copy()
        # 添加protoc所需的动态库路径到LD_LIBRARY_PATH
        # 假设protoc依赖的库在./third_party/grpc/lib目录下
        grpc_lib_path = os.path.abspath("./third_party/output/grpc/lib")
        
        # 如果LD_LIBRARY_PATH已存在，追加路径；否则创建新的
        if 'LD_LIBRARY_PATH' in env:
            env['LD_LIBRARY_PATH'] = f"{grpc_lib_path}:{env['LD_LIBRARY_PATH']}"
        else:
            env['LD_LIBRARY_PATH'] = grpc_lib_path
        subprocess.run([
            "./third_party/output/grpc/bin/protoc",  # 使用相对路径调用protoc
            "--experimental_allow_proto3_optional",
            "--python_out=./mindie_llm/connector/common/",
            "--proto_path=./proto/",
            "model_execute_data.proto"
        ], env=env, check=True)

        # 继续执行默认的build_py逻辑
        super().run()

setup(
    name="mindie_llm",
    version=mindie_llm_version,
    author="",
    author_email="",
    description="MindIE LLM Project",
    long_description="",
    package_dir={'mindie_llm': 'mindie_llm'},
    install_requires=[],
    package_data={
        '': ['*.xlsx', '*.h5', '*.csv', '*.so', '*.avsc', '*.xml', '*.pkl', '*.sql', '*.ini']
    },
    zip_safe=False,
    python_requires=">=3.10",
    cmdclass={
        "build_py": BuildPyCommand,
    },
    include_package_data=True,
    entry_points={
        "console_scripts": [
            'mindie_llm_backend = mindie_llm.connector.main:main'
        ]
    }
)