"""Installation script for the 'engineai_lab' python package."""

import os
import toml

from setuptools import setup, find_packages

# Obtain the extension data from the extension.toml file
EXTENSION_PATH = os.path.dirname(os.path.realpath(__file__))
# Read the extension.toml file
EXTENSION_TOML_DATA = toml.load(os.path.join(EXTENSION_PATH, "config", "extension.toml"))

# Minimum dependencies required prior to installation
INSTALL_REQUIRES = [
    "psutil",
    "onnxscript",
    "wandb>=0.19",
]

# Installation operation
setup(
    name="engineai_lab",
    package_dir={"": "source"},
    packages=find_packages(where="source"),
    author=EXTENSION_TOML_DATA["package"]["author"],
    maintainer=EXTENSION_TOML_DATA["package"]["maintainer"],
    url=EXTENSION_TOML_DATA["package"]["repository"],
    version=EXTENSION_TOML_DATA["package"]["version"],
    description=EXTENSION_TOML_DATA["package"]["description"],
    keywords=EXTENSION_TOML_DATA["package"]["keywords"],
    install_requires=INSTALL_REQUIRES,
    license="MIT",
    # 运行时资产显式列出，避免 setuptools 把 mesh/urdf 目录误判为 namespace package。
    include_package_data=False,
    package_data={
        "engineai_lab.assets": [
            "pm01/meshes/*",
            "pm01/urdf/*",
            "gen2/README.md",
            "gen2/urdf/*",
            "gen2/meshes/*",
            "gen2/meshes/collision_simplified/*",
        ]
    },
    python_requires=">=3.10",
    classifiers=[
        "Natural Language :: English",
        "Programming Language :: Python :: 3.10",
        "Isaac Sim :: 2023.1.1",
        "Isaac Sim :: 4.0.0",
    ],
    zip_safe=False,
)