state-hub/api/schemas/managed_repo.py

import uuid
from datetime import datetime
from typing import Any

from pydantic import BaseModel, ConfigDict


class RepoCreate(BaseModel):
    domain_slug: str
    slug: str
    name: str
    local_path: str | None = None
    remote_url: str | None = None
    description: str | None = None
    topic_id: uuid.UUID | None = None


class RepoUpdate(BaseModel):
    name: str | None = None
    local_path: str | None = None
    remote_url: str | None = None
    description: str | None = None
    topic_id: uuid.UUID | None = None
    last_state_synced_at: datetime | None = None


class RepoPathRegister(BaseModel):
    """Register a machine-local path for a repo on a specific host."""
    host: str
    path: str


class RepoRead(BaseModel):
    model_config = ConfigDict(from_attributes=True)
    id: uuid.UUID
    domain_id: uuid.UUID
    domain_slug: str  # derived from domain relationship
    slug: str
    name: str
    local_path: str | None = None
    host_paths: dict = {}
    remote_url: str | None = None
    description: str | None = None
    status: str
    topic_id: uuid.UUID | None = None
    sbom_source: str | None = None
    last_sbom_at: datetime | None = None
    last_state_synced_at: datetime | None = None
    created_at: datetime
    updated_at: datetime


class DispatchTask(BaseModel):
    id: uuid.UUID
    title: str
    priority: str
    status: str
    needs_human: bool


class DispatchWorkstream(BaseModel):
    id: uuid.UUID
    title: str
    status: str
    pending_tasks: list[DispatchTask]


class RepoDispatch(BaseModel):
    repo_slug: str
    active_goal: dict[str, Any] | None
    active_workstreams: list[DispatchWorkstream]
    human_interventions: list[DispatchTask]
    last_state_synced_at: datetime | None