import uuid
from typing import TYPE_CHECKING
from sqlalchemy import String, Text, ForeignKey
from sqlalchemy.orm import Mapped, mapped_column, relationship
from sqlalchemy.dialects.postgresql import UUID, JSONB
from app.database import Base

if TYPE_CHECKING:
    from app.models.mini_business_log import MiniBusinessLog


class MiniBusiness(Base):
    __tablename__ = "mini_business"

    id: Mapped[uuid.UUID] = mapped_column(UUID(as_uuid=True), primary_key=True, default=uuid.uuid4)
    customer_id: Mapped[uuid.UUID] = mapped_column(UUID(as_uuid=True), ForeignKey("customers.id"), index=True)
    product_type: Mapped[str] = mapped_column(String(200), default="")
    amount: Mapped[str] = mapped_column(String(100), default="")
    follow_up_detail: Mapped[str] = mapped_column(Text, default="")
    status: Mapped[str] = mapped_column(String(50), default="跟进中")
    manager_id: Mapped[uuid.UUID] = mapped_column(UUID(as_uuid=True), ForeignKey("users.id"), index=True)
    expected_revenue_date: Mapped[str] = mapped_column(String(50), default="")
    edit_log: Mapped[list] = mapped_column(JSONB, default=list)

    logs: Mapped[list["MiniBusinessLog"]] = relationship(back_populates="business", cascade="all, delete-orphan")