import enum
import uuid
from datetime import datetime, timezone
from typing import List, Optional

from sqlalchemy import DateTime, Enum as SAEnum, Float, String, Text
from sqlalchemy.dialects.postgresql import UUID
from sqlalchemy.orm import Mapped, mapped_column, relationship

from app.database import Base


class SoilType(enum.Enum):
    sandy = "sandy"
    loam = "loam"
    clay = "clay"
    silt = "silt"
    peat = "peat"
    chalky = "chalky"


class DrainageRate(enum.Enum):
    poor = "poor"
    moderate = "moderate"
    good = "good"
    excessive = "excessive"


class WaterRetention(enum.Enum):
    low = "low"
    medium = "medium"
    high = "high"


class SoilProfile(Base):
    __tablename__ = "soil_profiles"

    id: Mapped[uuid.UUID] = mapped_column(
        UUID(as_uuid=True), primary_key=True, default=uuid.uuid4
    )
    name: Mapped[str] = mapped_column(String(255), nullable=False)
    soil_type: Mapped[SoilType] = mapped_column(
        SAEnum(SoilType, name="soiltype"), nullable=False
    )
    ph_level: Mapped[float] = mapped_column(Float, nullable=False)
    organic_matter_percent: Mapped[float] = mapped_column(Float, nullable=False)
    drainage_rate: Mapped[DrainageRate] = mapped_column(
        SAEnum(DrainageRate, name="drainagerate"), nullable=False
    )
    water_retention: Mapped[WaterRetention] = mapped_column(
        SAEnum(WaterRetention, name="waterretention"), nullable=False
    )
    native_nitrogen: Mapped[float] = mapped_column(Float, nullable=False)
    native_phosphorus: Mapped[float] = mapped_column(Float, nullable=False)
    native_potassium: Mapped[float] = mapped_column(Float, nullable=False)
    salinity_level: Mapped[float] = mapped_column(Float, nullable=False)
    temperature_range_min: Mapped[float] = mapped_column(Float, nullable=False)
    temperature_range_max: Mapped[float] = mapped_column(Float, nullable=False)
    description: Mapped[Optional[str]] = mapped_column(Text, nullable=True)
    climate_zone: Mapped[str] = mapped_column(String(100), nullable=False)
    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),
    )

    scenarios: Mapped[List["Scenario"]] = relationship(  # type: ignore[name-defined]
        "Scenario", back_populates="soil_profile"
    )