from datetime import datetime, timedelta

from app.infrastructure.database import Note, Reservation, SessionLocal, Task, init_db
from app.memory.memory_manager import MemoryManager


def seed() -> None:
    init_db()
    session = SessionLocal()
    try:
        if session.query(Task).count() == 0:
            session.add_all(
                [
                    Task(title="Review venue shortlist", assignee="Maria", status="pending"),
                    Task(title="Confirm catering tasting", assignee="Nikos", status="in_progress"),
                ]
            )
        if session.query(Note).count() == 0:
            session.add_all(
                [
                    Note(content="Customer prefers white flowers.", category="preference"),
                    Note(content="Budget conversation should stay under 20,000 EUR.", category="budget"),
                ]
            )
        if session.query(Reservation).count() == 0:
            session.add(
                Reservation(
                    vendor="Aegean Garden Venue",
                    reservation_date=datetime.utcnow() + timedelta(days=45),
                    status="pending",
                )
            )
        session.commit()
        memory = MemoryManager()
        memory.seed_demo()
    finally:
        session.close()


if __name__ == "__main__":
    seed()