"""信号源注册表。"""

from __future__ import annotations

from typing import Optional

from sqlalchemy import BigInteger, ForeignKey, Integer, Text, UniqueConstraint, VARCHAR, TIMESTAMP
from sqlalchemy.orm import Mapped, mapped_column

from .base import Base
from .types import JSONB


class SignalSourceRegistry(Base):
    __tablename__ = "signal_source_registries"
    __table_args__ = (
        UniqueConstraint(
            "tenant_id",
            "project_id",
            "source_kind",
            "source_identifier",
            name="uq_signal_source_registry_key",
        ),
    )

    id: Mapped[int] = mapped_column(Integer, primary_key=True, autoincrement=True)
    project_id: Mapped[int] = mapped_column(
        BigInteger, ForeignKey("projects.id"), nullable=False, index=True
    )
    signal_id: Mapped[Optional[int]] = mapped_column(
        BigInteger, ForeignKey("signals.id"), nullable=True, index=True
    )
    created_by_user_id: Mapped[Optional[int]] = mapped_column(
        BigInteger, ForeignKey("users.id"), nullable=True, index=True
    )
    source_kind: Mapped[str] = mapped_column(VARCHAR(40), nullable=False, index=True)
    source_name: Mapped[str] = mapped_column(VARCHAR(120), nullable=False)
    source_identifier: Mapped[str] = mapped_column(VARCHAR(180), nullable=False)
    description: Mapped[Optional[str]] = mapped_column(Text, nullable=True)
    requested_via: Mapped[str] = mapped_column(VARCHAR(40), nullable=False, default="system", index=True)
    status: Mapped[str] = mapped_column(VARCHAR(20), nullable=False, default="active", index=True)
    onboarding_status: Mapped[str] = mapped_column(VARCHAR(20), nullable=False, default="pending", index=True)
    config: Mapped[Optional[dict]] = mapped_column(JSONB, nullable=True)
    raw_context: Mapped[Optional[dict]] = mapped_column(JSONB, nullable=True)
    last_signal_at: Mapped[Optional[TIMESTAMP]] = mapped_column(TIMESTAMP, nullable=True)
    last_worker_run_at: Mapped[Optional[TIMESTAMP]] = mapped_column(TIMESTAMP, nullable=True)