"""行为助推记录模型。

时机判断 + 策略选择 + 效果追踪。
"""

import uuid
from datetime import datetime, timezone

from sqlalchemy import DateTime, ForeignKey, String, Text
from sqlalchemy.orm import Mapped, mapped_column

from app.core.database import Base
from app.core.types import JSONBType


class NudgeRecord(Base):
    """行为助推记录。"""

    __tablename__ = "nudge_records"

    id: Mapped[str] = mapped_column(String(36), primary_key=True, default=lambda: str(uuid.uuid4()))
    company_id: Mapped[str] = mapped_column(String(36), ForeignKey("companies.id"), nullable=False, index=True)
    nudge_type: Mapped[str] = mapped_column(String(50), nullable=False, comment="anchoring/loss_aversion/social_proof/default/timing")
    context: Mapped[str | None] = mapped_column(Text, nullable=True, comment="助推上下文")
    message: Mapped[str] = mapped_column(Text, nullable=False, comment="助推内容")
    target_user_id: Mapped[str | None] = mapped_column(String(36), ForeignKey("users.id"), nullable=True)
    accepted: Mapped[bool | None] = mapped_column(nullable=True, comment="是否接受")
    effect_result: Mapped[dict | None] = mapped_column(JSONBType, nullable=True, comment="效果追踪")
    created_at: Mapped[datetime] = mapped_column(
        DateTime(timezone=True), default=lambda: datetime.now(timezone.utc)
    )