import uuid
from datetime import datetime, timezone
from typing import Any, Dict, List, Optional

from sqlalchemy import DateTime, Float, ForeignKey, JSON, String, Text
from sqlalchemy.dialects.postgresql import UUID
from sqlalchemy.orm import Mapped, mapped_column, relationship

from app.database import Base


class GraftCombination(Base):
    __tablename__ = "graft_combinations"

    id: Mapped[uuid.UUID] = mapped_column(
        UUID(as_uuid=True), primary_key=True, default=uuid.uuid4
    )
    rootstock_id: Mapped[uuid.UUID] = mapped_column(
        UUID(as_uuid=True), ForeignKey("plants.id"), nullable=False
    )
    scion_id: Mapped[uuid.UUID] = mapped_column(
        UUID(as_uuid=True), ForeignKey("plants.id"), nullable=False
    )
    compatibility_score: Mapped[float] = mapped_column(Float, nullable=False)
    vigor_boost: Mapped[float] = mapped_column(Float, default=1.0)
    yield_modifier: Mapped[float] = mapped_column(Float, default=1.0)
    stress_tolerance_override: Mapped[Dict[str, Any]] = mapped_column(JSON, default=dict)
    notes: Mapped[Optional[str]] = mapped_column(Text, nullable=True)
    source_reference: Mapped[Optional[str]] = mapped_column(String(500), nullable=True)
    created_at: Mapped[datetime] = mapped_column(
        DateTime(timezone=True), default=lambda: datetime.now(timezone.utc)
    )
    updated_at: Mapped[datetime] = mapped_column(
        DateTime(timezone=True),
        default=lambda: datetime.now(timezone.utc),
        onupdate=lambda: datetime.now(timezone.utc),
    )

    rootstock: Mapped["Plant"] = relationship(  # type: ignore[name-defined]
        "Plant", foreign_keys=[rootstock_id], back_populates="rootstock_combinations"
    )
    scion: Mapped["Plant"] = relationship(  # type: ignore[name-defined]
        "Plant", foreign_keys=[scion_id], back_populates="scion_combinations"
    )
    scenarios: Mapped[List["Scenario"]] = relationship(  # type: ignore[name-defined]
        "Scenario", back_populates="graft_combination"
    )