xiaoxia-saas/apps/api/app/services/script_service.py

"""ScriptService — Issue #1795 口播文案库 CRUD.

纯 Service 层封装，routes 直接调用。
"""

from __future__ import annotations

import uuid
from datetime import UTC, datetime
from typing import Optional

from sqlalchemy.orm import Session

from packages.adapters.sqlalchemy_impl.models import ScriptModel


class ScriptNotFoundError(Exception):
    """文案不存在或不属于当前用户."""


class ScriptService:
    """口播文案 CRUD."""

    def __init__(self, db: Session) -> None:
        self.db = db

    # ── list ──────────────────────────────────────────────────────────────

    def list_scripts(
        self,
        user_id: str,
        skip: int = 0,
        limit: int = 50,
        tag: Optional[str] = None,
    ) -> tuple[list[ScriptModel], int]:
        """返回 (items, total)."""
        q = self.db.query(ScriptModel).filter(ScriptModel.user_id == user_id)
        if tag:
            # JSON 数组包含查询
            q = q.filter(ScriptModel.tags.contains([tag]))
        total = q.count()
        items = q.order_by(ScriptModel.created_at.desc()).offset(skip).limit(limit).all()
        return items, total

    # ── create ────────────────────────────────────────────────────────────

    def create_script(
        self,
        user_id: str,
        title: str,
        content: str = "",
        segments: list | None = None,
        tags: list | None = None,
        title_text: str = "",
        title_category: str = "",
        title_config: dict | None = None,
    ) -> ScriptModel:
        script = ScriptModel(
            id=str(uuid.uuid4()),
            user_id=user_id,
            title=title,
            content=content,
            segments=segments if segments is not None else [],
            tags=tags if tags is not None else [],
            title_text=title_text or "",
            title_category=title_category or "",
            title_config=title_config if title_config is not None else {},
        )
        self.db.add(script)
        self.db.commit()
        self.db.refresh(script)
        return script

    # ── get ───────────────────────────────────────────────────────────────

    def get_script(self, script_id: str, user_id: str) -> ScriptModel:
        script = self.db.query(ScriptModel).filter(ScriptModel.id == script_id, ScriptModel.user_id == user_id).first()
        if script is None:
            raise ScriptNotFoundError(f"Script {script_id} not found")
        return script

    # ── update ────────────────────────────────────────────────────────────

    def update_script(
        self,
        script_id: str,
        user_id: str,
        title: Optional[str] = None,
        content: Optional[str] = None,
        segments: Optional[list] = None,
        tags: Optional[list] = None,
        title_text: Optional[str] = None,
        title_category: Optional[str] = None,
        title_config: Optional[dict] = None,
    ) -> ScriptModel:
        script = self.get_script(script_id, user_id)
        if title is not None:
            script.title = title
        if content is not None:
            script.content = content
        if segments is not None:
            script.segments = segments
        if tags is not None:
            script.tags = tags
        if title_text is not None:
            script.title_text = title_text
        if title_category is not None:
            script.title_category = title_category
        if title_config is not None:
            script.title_config = title_config
        script.updated_at = datetime.now(UTC)
        self.db.commit()
        self.db.refresh(script)
        return script

    # ── title config ─────────────────────────────────────────────────────

    def get_title_config_for_script(self, script_id: str, user_id: str) -> dict:
        """从 script 读取标题配置，返回可直接用于渲染的 title_config dict."""
        script = self.get_script(script_id, user_id)
        config = dict(script.title_config or {})
        if not config.get("text") and script.title_text:
            config["text"] = script.title_text
        return config

    # ── delete ────────────────────────────────────────────────────────────

    def delete_script(self, script_id: str, user_id: str) -> bool:
        script = self.db.query(ScriptModel).filter(ScriptModel.id == script_id, ScriptModel.user_id == user_id).first()
        if script is None:
            return False
        self.db.delete(script)
        self.db.commit()
        return True