"""回滚操作 Pydantic Schemas — 用于回滚 SQL 生成和预览。"""
from typing import Optional
from pydantic import BaseModel, Field


class RollbackPreviewRequest(BaseModel):
    """预览回滚 SQL 请求 Schema。"""
    instance_id: int = Field(..., description="数据源实例ID")
    database: str = Field(..., min_length=1, max_length=100, description="数据库名称")
    sql: str = Field(..., min_length=1, description="原始 DML 语句")


class RollbackPreviewResponse(BaseModel):
    """预览回滚 SQL 响应 Schema。"""
    original_sql: str = Field(..., description="原始 SQL")
    rollback_sql: str = Field(..., description="生成的回滚 SQL")
    sql_type: Optional[str] = Field(default=None, description="SQL 类型 (INSERT/UPDATE/DELETE)")
    warning: Optional[str] = Field(default=None, description="警告信息")


class RollbackGenerateRequest(BaseModel):
    """为工单生成回滚 SQL 请求 Schema。"""
    auto_save: bool = Field(default=True, description="是否自动保存到工单")


class RollbackGenerateResponse(BaseModel):
    """生成回滚 SQL 响应 Schema。"""
    ticket_id: int = Field(..., description="工单ID")
    original_sql: str = Field(..., description="原始 SQL")
    rollback_sql: str = Field(..., description="生成的回滚 SQL")
    sql_type: Optional[str] = Field(default=None, description="SQL 类型")
    saved: bool = Field(default=False, description="是否已保存到工单")
    warning: Optional[str] = Field(default=None, description="警告信息")