chore: install package

Author: bojiang

nodes/api.py

@@ -39,6 +39,7 @@ async def _write_snapshot(zf: zipfile.ZipFile) -> None:
     stdout, _ = await proc.communicate()
     with zf.open("snapshot.json", "w") as f:
         data = {
+            "python": f"{sys.version_info.major}.{sys.version_info.minor}.{sys.version_info.micro}",
             "comfyui": stdout.decode().strip(),
             "models": await _get_models(),
             "custom_nodes": await _get_custom_nodes(),

src/comfyui_idl/package.py

@@ -0,0 +1,137 @@
+from pathlib import Path
+import shutil
+import tempfile
+import json
+from typing import Union
+import os
+import subprocess
+
+
+COMFYUI_REPO = "https://github.com/comfyanonymous/ComfyUI.git"
+
+
+def _clone_commit(
+    url: str,
+    commit: str,
+    dir: Path,
+):
+    if not dir.exists():
+        dir.mkdir(parents=True)
+    script = f"""cd {dir}
+    git init
+    git remote add origin {url}
+    git fetch --depth 1 origin {commit}
+    git checkout FETCH_HEAD
+    """
+    subprocess.check_call(
+        script,
+        shell=True,
+    )
+
+
+def install_comfyui(snapshot, workspace: Path):
+    comfyui_commit = snapshot["comfyui"]
+
+    _clone_commit(
+        COMFYUI_REPO,
+        comfyui_commit,
+        workspace,
+    )
+
+
+def install_custom_modules(snapshot, workspace: Path):
+    for module in snapshot["custom_nodes"]:
+        url = module["url"]
+        directory = url.split("/")[-1].split(".")[0]
+        module_dir = workspace / "custom_nodes" / directory
+
+        commit_hash = module["commit_hash"]
+        disabled = module.get("disabled", False)
+        # if disabled:
+        #     continue
+        _clone_commit(
+            url,
+            commit_hash,
+            module_dir,
+        )
+
+
+def install_dependencies(snapshot: dict, req_file: str, workspace: Path):
+    python_version = snapshot["python"]
+    subprocess.check_call(
+        [
+            "uv",
+            "python",
+            "install",
+            python_version,
+        ],
+        cwd=workspace,
+    )
+    venv = workspace / ".venv"
+    if (venv / "DONE").exists():
+        return
+    venv_py = (
+        venv / "Scripts" / "python.exe" if os.name == "nt" else venv / "bin" / "python"
+    )
+    subprocess.check_call(
+        [
+            "uv",
+            "venv",
+            "--python",
+            python_version,
+            venv,
+        ],
+    )
+    subprocess.check_call(
+        [
+            "uv",
+            "pip",
+            "install",
+            "-p",
+            str(venv_py),
+            "pip",
+        ],
+    )
+    subprocess.check_call(
+        [
+            "uv",
+            "pip",
+            "install",
+            "-p",
+            str(venv_py),
+            "-r",
+            req_file,
+            "--no-deps",
+        ],
+    )
+    with open(venv / "DONE", "w") as f:
+        f.write("DONE")
+
+
+def install(cpack: Union[str, Path]):
+    workspace = Path("workspace")
+    with tempfile.TemporaryDirectory() as temp_dir:
+        pack_dir = Path(temp_dir) / ".cpack"
+        subprocess.check_call(
+            [
+                "unzip",
+                "-o",
+                cpack,
+                "-d",
+                pack_dir,
+            ]
+        )
+        snapshot = json.loads((pack_dir / "snapshot.json").read_text())
+        req_txt_file = pack_dir / "requirements.txt"
+
+        install_comfyui(snapshot, workspace)
+        install_custom_modules(snapshot, workspace)
+        install_dependencies(snapshot, str(req_txt_file), workspace)
+        for f in (pack_dir / "inputs").glob("*"):
+            shutil.copy(f, workspace / "input" / f.name)
+
+
+if __name__ == "__main__":
+    import sys
+
+    install(sys.argv[1])