Fixed: Zip the correct files in the custom node directory. (#73)

* Improve error messages for Publishing nodes.

* Fixed: Zip the correct nodes.

Author: robinjhuang

comfy_cli/command/custom_nodes/command.py

@@ -796,3 +796,19 @@ def registry_install(node_id: str, version: Optional[str] = None):
     logging.info(
         f"Node {node_id} version {node_version.version} has been successfully installed."
     )
+
+
+@app.command(
+    "pack",
+    help="Pack the current node into a tar.gz file. Ignorining .gitignore files.",
+)
+@tracking.track_command("pack")
+def pack():
+    typer.echo("Validating node configuration...")
+    config = extract_node_configuration()
+    if not config:
+        raise typer.Exit(code=1)
+
+    zip_filename = "node.tar.gz"
+    zip_files(zip_filename)
+    logging.info("Node has been packed successfully.")

comfy_cli/file_utils.py

@@ -48,7 +48,6 @@ def parse_json(input_data):
 def download_file(
     url: str, local_filepath: pathlib.Path, headers: Optional[dict] = None
 ):
-
     """Helper function to download a file."""
 
     import httpx
@@ -98,10 +97,9 @@ def zip_files(zip_filename):
                 dirs.remove(".git")
             for file in files:
                 file_path = os.path.join(root, file)
-                if not spec.match_file(file_path):
-                    zipf.write(
-                        file_path, os.path.relpath(file_path, os.path.join(root, ".."))
-                    )
+                relative_path = os.path.relpath(file_path, start=".")
+                if not spec.match_file(relative_path):
+                    zipf.write(file_path, relative_path)
 
 
 def upload_file_to_signed_url(signed_url: str, file_path: str):

comfy_cli/registry/config_parser.py

@@ -1,5 +1,5 @@
 import os
-
+from comfy_cli import ui
 import tomlkit.exceptions
 from comfy_cli.registry.types import (
     PyProjectConfig,
@@ -8,6 +8,7 @@
     Model,
     ComfyConfig,
 )
+from typing import Optional
 import tomlkit
 import subprocess
 
@@ -119,9 +120,15 @@ def initialize_project_config():
 
 def extract_node_configuration(
     path: str = os.path.join(os.getcwd(), "pyproject.toml"),
-) -> PyProjectConfig:
+) -> Optional[PyProjectConfig]:
     import tomlkit
 
+    if not os.path.isfile(path):
+        ui.display_error_message(
+            "No pyproject.toml file found in the current directory."
+        )
+        return None
+
     with open(path, "r") as file:
         data = tomlkit.load(file)