# To build the package, run `python -m build --wheel`.

from pathlib import Path
import os
import shlex
from glob import glob
import shutil

from setuptools import setup, Command
from setuptools_rust import RustBin
import tomli

try:
    # setuptools >= 70.1.0
    from setuptools.command.bdist_wheel import bdist_wheel
except ImportError:
    from wheel.bdist_wheel import bdist_wheel

def removeprefix(string, prefix):
    if string.startswith(prefix):
        return string[len(prefix):]
    else:
        return string

class BdistWheel(bdist_wheel):
    def get_tag(self):
        _, _, plat = super().get_tag()
        return "py3", "none", plat

class Clean(Command):
    user_options = []
    def initialize_options(self):
        pass
    def finalize_options(self):
        pass
    def run(self):
        # super().run()
        for d in ["build", "dist", "src/pylyzer.egg-info"]:
            shutil.rmtree(d, ignore_errors=True)

with open("README.md", encoding="utf-8", errors="ignore") as fp:
    long_description = fp.read()

with open("Cargo.toml", "rb") as fp:
    toml = tomli.load(fp)
    name = toml["package"]["name"]
    description = toml["package"]["description"]
    version = toml["workspace"]["package"]["version"]
    license = toml["workspace"]["package"]["license"]
    url = toml["workspace"]["package"]["repository"]

cargo_args = ["--no-default-features"]

home = os.path.expanduser("~")
file_and_dirs = glob(home + "/" + ".erg/lib/**", recursive=True)
paths = [Path(path) for path in file_and_dirs if os.path.isfile(path)]
files = [(removeprefix(str(path.parent), home), str(path)) for path in paths]
data_files = {}
for key, value in files:
    if key in data_files:
        data_files[key].append(value)
    else:
        data_files[key] = [value]
data_files = list(data_files.items())

setup(
    name=name,
    author="mtshiba",
    author_email="sbym1346@gmail.com",
    url=url,
    description=description,
    long_description=long_description,
    long_description_content_type="text/markdown",
    version=version,
    license=license,
    python_requires=">=3.7",
    rust_extensions=[
        RustBin("pylyzer", args=cargo_args, cargo_manifest_args=["--locked"])
    ],
    cmdclass={
        "clean": Clean,
        "bdist_wheel": BdistWheel,
    },
    classifiers=[
        "Development Status :: 2 - Pre-Alpha",
        "Operating System :: OS Independent",
        "Programming Language :: Python",
        "Programming Language :: Rust",
        "Topic :: Software Development :: Quality Assurance",
    ],
    data_files=data_files,
)