comfygit-core 0.2.0__py3-none-any.whl

comfygit_core/models/environment.py

@@ -0,0 +1,293 @@
+"""models/environment.py - Environment models for ComfyDock."""
+
+from dataclasses import dataclass, field
+from enum import Enum
+from pathlib import Path
+from typing import List, TYPE_CHECKING
+
+from .workflow import DetailedWorkflowStatus
+
+if TYPE_CHECKING:
+    from .manifest import ManifestModel
+
+
+@dataclass
+class PackageSyncStatus:
+    """Status of package synchronization."""
+
+    in_sync: bool
+    message: str
+    details: str | None = None
+
+
+@dataclass
+class GitStatus:
+    """Encapsulated git status information."""
+
+    has_changes: bool
+    current_branch: str | None = None  # None = detached HEAD
+    has_other_changes: bool = False  # Changes beyond workflows/pyproject
+    # diff: str
+    workflow_changes: dict[str, str] = field(default_factory=dict)
+
+    # Git change details (populated by parser if needed)
+    nodes_added: list[dict] = field(default_factory=list)  # {"name": str, "is_development": bool}
+    nodes_removed: list[dict] = field(default_factory=list)  # {"name": str, "is_development": bool}
+    dependencies_added: list[dict] = field(default_factory=list)
+    dependencies_removed: list[dict] = field(default_factory=list)
+    dependencies_updated: list[dict] = field(default_factory=list)
+    constraints_added: list[str] = field(default_factory=list)
+    constraints_removed: list[str] = field(default_factory=list)
+
+@dataclass
+class GitInfo:
+    commit: str | None = None
+    branch: str | None = None
+    tag: str | None = None
+    is_dirty: bool = False
+    remote_url: str | None = None
+    github_owner: str | None = None
+    github_repo: str | None = None
+
+@dataclass
+class EnvironmentComparison:
+    """Comparison between current and expected environment states."""
+
+    missing_nodes: list[str] = field(default_factory=list)
+    extra_nodes: list[str] = field(default_factory=list)
+    version_mismatches: list[dict] = field(
+        default_factory=list
+    )  # {name, expected, actual}
+    packages_in_sync: bool = True
+    package_sync_message: str = ""
+    potential_dev_rename: bool = False
+
+    @property
+    def is_synced(self) -> bool:
+        """Check if environment is fully synced."""
+        return (
+            not self.missing_nodes
+            and not self.extra_nodes
+            and not self.version_mismatches
+            and self.packages_in_sync
+        )
+
+@dataclass
+class NodeState:
+    """State of an installed custom node."""
+
+    name: str
+    path: Path
+    disabled: bool = False
+    git_commit: str | None = None
+    git_branch: str | None = None
+    version: str | None = None  # From git tag or pyproject
+    is_dirty: bool = False
+    source: str | None = None  # 'registry', 'git', 'development', etc.
+
+
+@dataclass
+class EnvironmentState:
+    """Current state of an environment."""
+
+    custom_nodes: dict[str, NodeState]  # name -> state
+    packages: dict[str, str] | None  # name -> version
+    python_version: str | None
+
+
+@dataclass
+class MissingModelInfo:
+    """Information about a model that's in pyproject but not in local index."""
+    model: "ManifestModel"  # From global models table
+    workflow_names: list[str]  # Which workflows need it
+    criticality: str  # "required", "flexible", "optional" (worst case across workflows)
+    can_download: bool  # Has sources available
+
+    @property
+    def is_required(self) -> bool:
+        return self.criticality == "required"
+
+
+# === Semantic Value Objects ===
+
+
+class UserAction(Enum):
+    """Recommended user actions."""
+
+    SYNC_REQUIRED = "sync"
+    COMMIT_REQUIRED = "commit"
+    NO_ACTION_NEEDED = "none"
+
+
+@dataclass
+class ChangesSummary:
+    """Summary of changes with semantic meaning."""
+
+    primary_changes: List[str] = field(default_factory=list)
+    secondary_changes: List[str] = field(default_factory=list)
+    has_breaking_changes: bool = False
+
+    def get_headline(self) -> str:
+        """Get a headline summary of changes."""
+        if not self.primary_changes and not self.secondary_changes:
+            return "No changes"
+
+        if self.has_breaking_changes:
+            return "Breaking changes detected"
+
+        if len(self.primary_changes) == 1 and not self.secondary_changes:
+            return self.primary_changes[0]
+
+        total = len(self.primary_changes) + len(self.secondary_changes)
+        return f"{total} changes"
+
+    def get_commit_message(self) -> str:
+        """Generate a commit message from changes."""
+        parts = self.primary_changes + self.secondary_changes
+        if not parts:
+            return "Update environment configuration"
+        return "; ".join(parts)
+
+
+
+
+@dataclass
+class EnvironmentStatus:
+    """Complete environment status including comparison and git/workflow state."""
+
+    comparison: EnvironmentComparison
+    git: GitStatus
+    workflow: DetailedWorkflowStatus
+    missing_models: list[MissingModelInfo] = field(default_factory=list)
+
+    @classmethod
+    def create(
+        cls,
+        comparison: EnvironmentComparison,
+        git_status: GitStatus,
+        workflow_status: DetailedWorkflowStatus,
+        missing_models: list[MissingModelInfo] | None = None,
+    ) -> "EnvironmentStatus":
+        """Factory method to create EnvironmentStatus from components."""
+        return cls(
+            comparison=comparison,
+            git=git_status,
+            workflow=workflow_status,
+            missing_models=missing_models or []
+        )
+
+    @property
+    def is_synced(self) -> bool:
+        """Check if environment is fully synced (nodes, packages, workflows, and models)."""
+        return (
+            self.comparison.is_synced and
+            self.workflow.sync_status.is_synced and
+            not self.missing_models
+        )
+
+    # === Semantic Methods ===
+
+    def get_changes_summary(self) -> ChangesSummary:
+        """Analyze and categorize all changes."""
+        primary_changes = []
+        secondary_changes = []
+
+        # Node changes (most specific)
+        if self.git.nodes_added and self.git.nodes_removed:
+            primary_changes.append(
+                f"Update nodes: +{len(self.git.nodes_added)}, -{len(self.git.nodes_removed)}"
+            )
+        elif self.git.nodes_added:
+            if len(self.git.nodes_added) == 1:
+                primary_changes.append(f"Add {self.git.nodes_added[0]['name']}")
+            else:
+                primary_changes.append(f"Add {len(self.git.nodes_added)} nodes")
+        elif self.git.nodes_removed:
+            if len(self.git.nodes_removed) == 1:
+                primary_changes.append(f"Remove {self.git.nodes_removed[0]['name']}")
+            else:
+                primary_changes.append(f"Remove {len(self.git.nodes_removed)} nodes")
+
+        # Dependency changes
+        if (
+            self.git.dependencies_added
+            or self.git.dependencies_removed
+            or self.git.dependencies_updated
+        ):
+            dep_count = (
+                len(self.git.dependencies_added)
+                + len(self.git.dependencies_removed)
+                + len(self.git.dependencies_updated)
+            )
+            secondary_changes.append(f"Update {dep_count} dependencies")
+
+        # Constraint changes
+        if self.git.constraints_added or self.git.constraints_removed:
+            secondary_changes.append("Update constraints")
+
+        # No more workflow tracking changes - all workflows are automatically managed
+
+        # Workflow file changes
+        if self.git.workflow_changes:
+            workflow_count = len(self.git.workflow_changes)
+            if workflow_count == 1:
+                workflow_name, workflow_status = list(
+                    self.git.workflow_changes.items()
+                )[0]
+                if workflow_status == "modified":
+                    primary_changes.append(f"Update {workflow_name}")
+                elif workflow_status == "added":
+                    primary_changes.append(f"Add {workflow_name}")
+                elif workflow_status == "deleted":
+                    primary_changes.append(f"Remove {workflow_name}")
+            else:
+                primary_changes.append(f"Update {workflow_count} workflows")
+
+        # Detect breaking changes
+        has_breaking = bool(
+            self.git.nodes_removed
+            or self.git.dependencies_removed
+            or self.git.constraints_removed
+        )
+
+        return ChangesSummary(
+            primary_changes=primary_changes,
+            secondary_changes=secondary_changes,
+            has_breaking_changes=has_breaking,
+        )
+
+    def get_recommended_action(self) -> UserAction:
+        """Determine what the user should do next."""
+        if not self.is_synced:
+            return UserAction.SYNC_REQUIRED
+        elif self.git.has_changes:
+            return UserAction.COMMIT_REQUIRED
+        else:
+            return UserAction.NO_ACTION_NEEDED
+
+    def generate_commit_message(self) -> str:
+        """Generate a semantic commit message."""
+        summary = self.get_changes_summary()
+        return summary.get_commit_message()
+
+    def get_sync_preview(self) -> dict:
+        """Get preview of what sync operation will do.
+
+        Note: WorkflowSyncStatus is from ComfyUI's perspective:
+        - new: in ComfyUI but not .cec → will be REMOVED
+        - deleted: in .cec but not ComfyUI → will be ADDED
+        - modified: differs between ComfyUI and .cec → will be UPDATED
+        """
+        return {
+            'nodes_to_install': self.comparison.missing_nodes,
+            'nodes_to_remove': self.comparison.extra_nodes,
+            'nodes_to_update': self.comparison.version_mismatches,
+            'packages_to_sync': not self.comparison.packages_in_sync,
+            'workflows_to_add': self.workflow.sync_status.deleted,  # Deleted from ComfyUI, will restore
+            'workflows_to_update': self.workflow.sync_status.modified,  # Modified, will sync
+            'workflows_to_remove': self.workflow.sync_status.new,  # New in ComfyUI, will remove
+            'models_missing': self.missing_models,
+            'models_downloadable': [m for m in self.missing_models if m.can_download],
+            'models_unavailable': [m for m in self.missing_models if not m.can_download],
+            'models_required': [m for m in self.missing_models if m.is_required],
+        }

comfygit_core/models/exceptions.py

@@ -0,0 +1,378 @@
+# models/exceptions.py
+
+from dataclasses import dataclass, field
+from typing import Literal
+
+
+class ComfyDockError(Exception):
+    """Base exception for ComfyDock errors."""
+    pass
+
+# ====================================================
+# Workspace exceptions
+# ====================================================
+
+class CDWorkspaceNotFoundError(ComfyDockError):
+    """Workspace doesn't exist."""
+    pass
+
+class CDWorkspaceExistsError(ComfyDockError):
+    """Workspace already exists."""
+    pass
+
+class CDWorkspaceError(ComfyDockError):
+    """Workspace-related errors."""
+    pass
+
+# ===================================================
+# Environment exceptions
+# ===================================================
+
+class CDEnvironmentError(ComfyDockError):
+    """Environment-related errors."""
+    pass
+
+class CDEnvironmentNotFoundError(ComfyDockError):
+    """Environment doesn't exist."""
+    pass
+
+class CDEnvironmentExistsError(ComfyDockError):
+    """Environment already exists."""
+    pass
+
+# ===================================================
+# Resolution exceptions
+# ==================================================
+
+class CDResolutionError(ComfyDockError):
+    """Resolution errors."""
+    pass
+
+# ===================================================
+# Node exceptions
+# ===================================================
+
+class CDNodeNotFoundError(ComfyDockError):
+    """Raised when Node not found."""
+    pass
+
+@dataclass
+class NodeAction:
+    """Represents a possible action to resolve an error."""
+    action_type: Literal[
+        'remove_node',
+        'add_node_dev',
+        'add_node_force',
+        'add_node_version',
+        'rename_directory',
+        'update_node',
+        'add_constraint',
+        'skip_node'
+    ]
+
+    # Parameters needed for the action
+    node_identifier: str | None = None
+    node_name: str | None = None
+    directory_name: str | None = None
+    new_name: str | None = None
+    package_name: str | None = None  # For add_constraint action
+
+    # Human-readable description (client-agnostic)
+    description: str = ""
+
+
+@dataclass
+class NodeConflictContext:
+    """Context about what conflicted and why."""
+    conflict_type: Literal[
+        'already_tracked',
+        'directory_exists_non_git',
+        'directory_exists_no_remote',
+        'same_repo_exists',
+        'different_repo_exists',
+        'dev_node_replacement',
+        'user_cancelled'
+    ]
+
+    node_name: str
+    identifier: str | None = None
+    existing_identifier: str | None = None
+    filesystem_path: str | None = None
+    local_remote_url: str | None = None
+    expected_remote_url: str | None = None
+    is_development: bool = False
+
+    # Suggested actions
+    suggested_actions: list[NodeAction] = field(default_factory=list)
+
+
+class CDNodeConflictError(ComfyDockError):
+    """Raised when Node has conflicts with enhanced context."""
+
+    def __init__(self, message: str, context: NodeConflictContext | None = None):
+        super().__init__(message)
+        self.context = context
+
+    def get_actions(self) -> list[NodeAction]:
+        """Get suggested actions for resolving this conflict."""
+        return self.context.suggested_actions if self.context else []
+
+
+@dataclass
+class DependencyConflictContext:
+    """Context about dependency conflicts during node installation."""
+
+    # The node being added
+    node_name: str
+
+    # Parsed conflict information
+    conflicting_packages: list[tuple[str, str]]  # Package pairs that conflict
+    conflict_descriptions: list[str]  # Simplified conflict messages
+
+    # Raw UV stderr for verbose mode
+    raw_stderr: str = ""
+
+    # Suggested resolutions
+    suggested_actions: list[NodeAction] = field(default_factory=list)
+
+
+class CDDependencyConflictError(ComfyDockError):
+    """Raised when dependency resolution fails during node installation."""
+
+    def __init__(self, message: str, context: DependencyConflictContext | None = None):
+        super().__init__(message)
+        self.context = context
+
+    def get_actions(self) -> list[NodeAction]:
+        """Get suggested actions for resolving this conflict."""
+        return self.context.suggested_actions if self.context else []
+
+# ===================================================
+# Registry exceptions
+# ===================================================
+
+class CDRegistryError(ComfyDockError):
+    """Base class for registry errors."""
+    pass
+
+class CDRegistryAuthError(CDRegistryError):
+    """Authentication/authorization errors with registry."""
+    pass
+
+class CDRegistryServerError(CDRegistryError):
+    """Registry server errors (5xx)."""
+    pass
+
+class CDRegistryConnectionError(CDRegistryError):
+    """Network/connection errors to registry."""
+    pass
+
+class CDRegistryDataError(ComfyDockError):
+    """Registry data is not available or cannot be loaded.
+
+    This error indicates that registry node mappings are missing or corrupted.
+    Recovery typically involves downloading or updating the registry data.
+    """
+
+    def __init__(
+        self,
+        message: str,
+        cache_path: str | None = None,
+        can_retry: bool = True
+    ):
+        super().__init__(message)
+        self.cache_path = cache_path
+        self.can_retry = can_retry
+
+# ===================================================
+# Pyproject exceptions
+# ===================================================
+
+class CDPyprojectError(ComfyDockError):
+    """Errors related to pyproject.toml operations."""
+    pass
+
+class CDPyprojectNotFoundError(CDPyprojectError):
+    """pyproject.toml file not found."""
+    pass
+
+class CDPyprojectInvalidError(CDPyprojectError):
+    """pyproject.toml file is invalid or corrupted."""
+    pass
+
+# ===================================================
+# Dependency exceptions
+# ===================================================
+
+class CDDependencyError(ComfyDockError):
+    """Dependency-related errors."""
+    pass
+
+class CDPackageSyncError(CDDependencyError):
+    """Package synchronization errors."""
+    pass
+
+# ===================================================
+# Index exceptions
+# ===================================================
+
+class CDIndexError(ComfyDockError):
+    """Index configuration errors."""
+    pass
+
+# ===================================================
+# Process/Command exceptions
+# ===================================================
+
+class CDProcessError(ComfyDockError):
+    """Raised when subprocess command execution fails."""
+
+    def __init__(
+        self,
+        message: str,
+        command: list[str] | None = None,
+        stderr: str | None = None,
+        stdout: str | None = None,
+        returncode: int | None = None,
+    ):
+        super().__init__(message)
+        self.command = command
+        self.stderr = stderr
+        self.stdout = stdout
+        self.returncode = returncode
+
+
+# ===================================================
+# UV exceptions
+# ==================================================
+
+class UVNotInstalledError(ComfyDockError):
+    """Raised when UV is not installed."""
+    pass
+
+
+class UVCommandError(ComfyDockError):
+    """Raised when UV command execution fails."""
+
+    def __init__(
+        self,
+        message: str,
+        command: list[str] | None = None,
+        stderr: str | None = None,
+        stdout: str | None = None,
+        returncode: int | None = None,
+    ):
+        super().__init__(message)
+        self.command = command
+        self.stderr = stderr
+        self.stdout = stdout
+        self.returncode = returncode
+
+    def __str__(self) -> str:
+        """Include stderr/stdout in string representation for better error messages."""
+        parts = [super().__str__()]
+
+        if self.stderr and self.stderr.strip():
+            parts.append(f"\nStderr: {self.stderr.strip()}")
+
+        if self.stdout and self.stdout.strip():
+            parts.append(f"\nStdout: {self.stdout.strip()}")
+
+        return "".join(parts)
+
+
+# ===================================================
+# Export/Import exceptions
+# ===================================================
+
+@dataclass
+class ExportErrorContext:
+    """Context about an export failure."""
+    uncommitted_workflows: list[str] = field(default_factory=list)
+    uncommitted_git_changes: bool = False
+    has_unresolved_issues: bool = False
+
+
+class CDExportError(ComfyDockError):
+    """Export operation failed with detailed context."""
+
+    def __init__(self, message: str, context: ExportErrorContext | None = None):
+        super().__init__(message)
+        self.context = context
+
+    @property
+    def uncommitted_workflows(self) -> list[str] | None:
+        """Get list of uncommitted workflows for backward compatibility."""
+        return self.context.uncommitted_workflows if self.context else None
+
+
+# ===================================================
+# Model Download exceptions
+# ===================================================
+
+@dataclass
+class DownloadErrorContext:
+    """Detailed context about a download failure."""
+    provider: str  # 'civitai', 'huggingface', 'custom'
+    error_category: str  # 'auth_missing', 'auth_invalid', 'forbidden', 'not_found', 'network', 'server', 'unknown'
+    http_status: int | None
+    url: str
+    has_configured_auth: bool  # Was auth configured (even if invalid)?
+    raw_error: str  # Original error message for debugging
+
+    def get_user_message(self) -> str:
+        """Generate user-friendly error message."""
+        if self.provider == "civitai":
+            if self.error_category == "auth_missing":
+                return (
+                    f"CivitAI model requires authentication (HTTP {self.http_status}). "
+                    "No API key found. Get your key from https://civitai.com/user/account "
+                    "and add it with: comfygit config --civitai-key <your-key>"
+                )
+            elif self.error_category == "auth_invalid":
+                return (
+                    f"CivitAI authentication failed (HTTP {self.http_status}). "
+                    "Your API key may be invalid or expired. "
+                    "Update it with: comfygit config --civitai-key <your-key>"
+                )
+            elif self.error_category == "forbidden":
+                return (
+                    f"CivitAI access forbidden (HTTP {self.http_status}). "
+                    "This model may require special permissions or may not be publicly available."
+                )
+            elif self.error_category == "not_found":
+                return f"CivitAI model not found (HTTP {self.http_status}). The URL may be incorrect or the model was removed."
+
+        elif self.provider == "huggingface":
+            if self.error_category in ("auth_missing", "auth_invalid"):
+                return (
+                    f"HuggingFace model requires authentication (HTTP {self.http_status}). "
+                    "Set the HF_TOKEN environment variable with your HuggingFace token. "
+                    "Get your token from: https://huggingface.co/settings/tokens"
+                )
+            elif self.error_category == "not_found":
+                return f"HuggingFace model not found (HTTP {self.http_status}). Check the URL is correct."
+
+        # Generic provider or fallback
+        if self.error_category == "network":
+            return f"Network error downloading from {self.provider}: {self.raw_error}"
+        elif self.error_category == "server":
+            return f"Server error from {self.provider} (HTTP {self.http_status}). Try again later."
+        elif self.http_status:
+            return f"Download failed from {self.provider} (HTTP {self.http_status}): {self.raw_error}"
+        else:
+            return f"Download failed from {self.provider}: {self.raw_error}"
+
+
+class CDModelDownloadError(ComfyDockError):
+    """Model download error with provider-specific context."""
+
+    def __init__(self, message: str, context: DownloadErrorContext | None = None):
+        super().__init__(message)
+        self.context = context
+
+    def get_user_message(self) -> str:
+        """Get user-friendly error message."""
+        if self.context:
+            return self.context.get_user_message()
+        return str(self)