from __future__ import annotations

import shutil
import pathlib
from typing import Any

from hatchling.builders.hooks.plugin.interface import BuildHookInterface


def copy_js_code(root: str | pathlib.Path):
    NOT_COMPONENT = [
        "app",
        "node_modules",
        "storybook",
        "playwright-report",
        "workbench",
        "tooltils",
    ]
    for entry in (pathlib.Path(root) / "js").iterdir():
        if (
            entry.is_dir()
            and not str(entry.name).startswith("_")
            and not str(entry.name) in NOT_COMPONENT
        ):

            def ignore(s, names):
                ignored = []
                for n in names:
                    if (
                        n.startswith("CHANGELOG")
                        or n.startswith("README.md")
                        or n.startswith("node_modules")
                        or ".test." in n
                        or ".stories." in n
                        or ".spec." in n
                    ):
                        ignored.append(n)
                return ignored
            shutil.copytree(
                str(entry),
                str(pathlib.Path("gradio") / "_frontend_code" / entry.name),
                ignore=ignore,
                dirs_exist_ok=True,
            )
    shutil.copytree(
        str(pathlib.Path(root) / "client" / "js"),
        str(pathlib.Path("gradio") / "_frontend_code" / "client"),
        ignore=lambda d, names: ["node_modules"],
        dirs_exist_ok=True,
    )


class BuildHook(BuildHookInterface):
    def initialize(self, version: str, build_data: dict[str, Any]) -> None:
        copy_js_code(self.root)


if __name__ == "__main__":
    copy_js_code(pathlib.Path("..").resolve())