"""Account model: companies/organizations that the CRM tracks."""

from __future__ import annotations

from enum import Enum
from typing import TYPE_CHECKING, Any

from sqlalchemy import JSON, ForeignKey, String
from sqlalchemy.orm import Mapped, mapped_column, relationship

from app.models.base import Base, OrgScopedMixin, SoftDeleteMixin, TimestampMixin

if TYPE_CHECKING:
    from app.models.activity import Activity
    from app.models.contact import Contact
    from app.models.deal import Deal
    from app.models.note import Note
    from app.models.tag_link import TagLink
    from app.models.user import User


class Industry(str, Enum):
    """Industry classification for accounts."""

    startup = "startup"
    sme = "sme"
    midmarket = "midmarket"
    enterprise = "enterprise"
    other = "other"


class AccountSize(str, Enum):
    """Company size category."""

    startup = "startup"
    sme = "sme"
    midmarket = "midmarket"
    enterprise = "enterprise"


class Account(Base, TimestampMixin, SoftDeleteMixin, OrgScopedMixin):
    __tablename__ = "accounts"

    id: Mapped[int] = mapped_column(primary_key=True, autoincrement=True)
    name: Mapped[str] = mapped_column(String(255), nullable=False, index=True)
    website: Mapped[str | None] = mapped_column(String(512), nullable=True)
    industry: Mapped[Industry | None] = mapped_column(
        String(32), nullable=True, index=True
    )
    size: Mapped[AccountSize | None] = mapped_column(String(32), nullable=True)
    address: Mapped[dict[str, Any] | None] = mapped_column(JSON, nullable=True)
    owner_id: Mapped[int] = mapped_column(
        ForeignKey("users.id"), nullable=False, index=True
    )

    # Relationships
    owner: Mapped[User] = relationship(
        "User", foreign_keys=[owner_id], lazy="joined"
    )
    contacts: Mapped[list[Contact]] = relationship(
        "Contact", back_populates="account", lazy="selectin"
    )
    deals: Mapped[list[Deal]] = relationship(
        "Deal", back_populates="account", lazy="selectin"
    )
    activities: Mapped[list[Activity]] = relationship(
        "Activity", back_populates="account", lazy="selectin"
    )
    notes: Mapped[list[Note]] = relationship(
        "Note",
        primaryjoin="and_(Account.id==foreign(Note.parent_id), Note.parent_type=='account')",
        viewonly=True,
        lazy="selectin",
    )
    tags: Mapped[list[TagLink]] = relationship(
        "TagLink",
        primaryjoin="and_(Account.id==foreign(TagLink.parent_id), TagLink.parent_type=='account')",
        viewonly=True,
        lazy="selectin",
    )

    def __repr__(self) -> str:
        return f"<Account id={self.id} name={self.name!r}>"


__all__ = ["Account", "AccountSize", "Industry"]