Ludus FastMCP

"""Orchestration schemas for smart deployment workflows."""

from datetime import datetime
from typing import Any
from pydantic import BaseModel, Field


class DeploymentStep(BaseModel):
    """A single step in the deployment process."""

    name: str = Field(..., description="Step name")
    status: str = Field(..., description="Status: pending, in_progress, completed, failed")
    started_at: datetime | None = Field(default=None, description="When step started")
    completed_at: datetime | None = Field(default=None, description="When step completed")
    message: str = Field(default="", description="Step message or error")


class SmartDeployResult(BaseModel):
    """Result of smart deployment orchestration."""

    status: str = Field(..., description="Deployment status: started, deploying, success, failed")
    deployment_id: str | None = Field(default=None, description="Deployment identifier")
    scenario_key: str = Field(..., description="Scenario that was deployed")
    siem_type: str = Field(..., description="SIEM type")
    snapshot_id: str | None = Field(default=None, description="Pre-deployment snapshot ID if created")

    # Preview info
    vm_count: int = Field(..., description="Number of VMs")
    estimated_time: str = Field(..., description="Estimated deployment time")

    # Monitoring info
    auto_monitor: bool = Field(default=True, description="Whether auto-monitoring is enabled")
    check_interval: int = Field(default=30, description="Monitoring interval in seconds")
    next_check_message: str = Field(default="", description="Message about next check")

    # Commands for user
    monitoring_commands: dict[str, str] = Field(
        default_factory=dict,
        description="Commands for manual monitoring"
    )

    # Additional context
    message: str = Field(default="", description="Human-readable status message")


class DeploymentTimeline(BaseModel):
    """Timeline of deployment progress."""

    started_at: datetime = Field(..., description="Deployment start time")
    steps: list[DeploymentStep] = Field(default_factory=list, description="Timeline steps")
    current_step: str = Field(default="", description="Current step name")
    progress_percentage: int = Field(default=0, description="Progress percentage (0-100)")
    estimated_completion: datetime | None = Field(default=None, description="Estimated completion time")
    elapsed_minutes: int = Field(default=0, description="Elapsed time in minutes")
    remaining_minutes: int = Field(default=0, description="Estimated remaining time in minutes")


class MonitoringUpdate(BaseModel):
    """Real-time monitoring update."""

    timestamp: datetime = Field(default_factory=datetime.now, description="Update timestamp")
    range_state: str = Field(..., description="Current range state")
    vm_count: int = Field(default=0, description="Total VMs")
    vms_ready: int = Field(default=0, description="VMs ready/running")
    current_task: str = Field(default="", description="Current task being executed")
    progress_percentage: int = Field(default=0, description="Progress percentage")

    # Recent activity
    recent_activity: list[str] = Field(default_factory=list, description="Recent log entries")

    # Status indicators
    is_healthy: bool = Field(default=True, description="Whether deployment is healthy")
    issues: list[str] = Field(default_factory=list, description="Any issues detected")

    # Timing
    elapsed_minutes: int = Field(default=0, description="Time elapsed")
    eta_minutes: int = Field(default=0, description="Estimated time remaining")

    # Next steps
    next_check_in: int = Field(default=30, description="Seconds until next check")
    should_continue_monitoring: bool = Field(default=True, description="Whether to continue monitoring")


class RecoveryRecommendation(BaseModel):
    """Recovery recommendation for failed deployment."""

    action: str = Field(..., description="Recommended action: wait, fix_config, retry, destroy")
    reason: str = Field(..., description="Why this action is recommended")
    severity: str = Field(..., description="Issue severity: info, warning, error, critical")
    steps: list[str] = Field(default_factory=list, description="Step-by-step recovery instructions")
    commands: dict[str, str] = Field(default_factory=dict, description="Useful commands")
    estimated_recovery_time: str = Field(default="", description="Estimated time to recover")