from typing import Optional
from sqlmodel import Field, SQLModel
from datetime import datetime

class BotInstance(SQLModel, table=True):
    id: str = Field(primary_key=True)
    name: str
    workspace_dir: str = Field(unique=True)
    docker_status: str = Field(default="STOPPED", index=True)
    current_state: Optional[str] = Field(default="IDLE")
    last_action: Optional[str] = Field(default=None)
    image_tag: str = Field(default="nanobot-base:v0.1.4") # 记录该机器人使用的镜像版本
    created_at: datetime = Field(default_factory=datetime.utcnow)
    updated_at: datetime = Field(default_factory=datetime.utcnow)

class BotMessage(SQLModel, table=True):
    id: Optional[int] = Field(default=None, primary_key=True)
    bot_id: str = Field(foreign_key="botinstance.id", index=True)
    role: str = Field(index=True)  # user | assistant | system
    text: str
    media_json: Optional[str] = Field(default=None)  # JSON string list of workspace-relative file paths
    created_at: datetime = Field(default_factory=datetime.utcnow, index=True)

class NanobotImage(SQLModel, table=True):
    tag: str = Field(primary_key=True) # e.g., nanobot-base:v0.1.4
    image_id: Optional[str] = Field(default=None) # Docker 内部的 Image ID
    version: str # e.g., 0.1.4
    status: str = Field(default="READY") # READY, BUILDING, ERROR
    source_dir: Optional[str] = Field(default=None) # 本地来源标识（如 manual）
    created_at: datetime = Field(default_factory=datetime.utcnow)