"""Idempotent seed loader. Creates an admin user, a source, a watchlist, ingests the

bundled sample exports, and stands up a demo investigation with sightings attached.

Safe to run repeatedly (uses get-or-create + content-hash import dedup)."""

from __future__ import annotations

import hashlib

import os

from sqlalchemy import select

from app.core.security import hash_password

from app.db.session import SessionLocal

from app.models import (

    Import, Investigation, Sighting, Source, User, Watchlist, WatchlistEntry, PlateRead,

)

from app.services.ingest import process_import

SAMPLE_DIR = os.path.join(os.path.dirname(__file__), "sample_data")

def get_or_create(db, model, defaults=None, **filters):

    obj = db.scalar(select(model).filter_by(**filters))

    if obj:

        return obj, False

    obj = model(**filters, **(defaults or {}))

    db.add(obj)

    db.flush()

    return obj, True

def ingest_file(db, source, path):

    with open(path, "rb") as fh:

        data = fh.read()

    sha = hashlib.sha256(data).hexdigest()

    existing = db.scalar(select(Import).where(Import.sha256 == sha))

    if existing:

        return existing.id, "already imported"

    ctype = ("text/csv" if path.endswith(".csv")

             else "application/pdf" if path.endswith(".pdf") else "application/json")

    imp = Import(source_id=source.id, filename=os.path.basename(path),

                 content_type=ctype, sha256=sha, raw_blob_ref=path, status="received")

    db.add(imp)

    db.commit()

    summary = process_import(db, imp.id)

    return imp.id, summary

def main():

    db = SessionLocal()

    try:

        admin, created = get_or_create(

            db, User, email="admin@sightline.app",

            defaults={"hashed_password": hash_password("changeme123"),

                      "full_name": "Demo Admin", "role": "admin"})

        if created:

            print("created admin: admin@sightline.app / changeme123")

        source, _ = get_or_create(

            db, Source, name="Demo Flock Export",

            defaults={"system_type": "flock", "organization": "default",

                      "jurisdiction": "Shelby County, TN", "trust_level": 0.9})

        wl, _ = get_or_create(db, Watchlist, name="Stolen Vehicles",

                              defaults={"kind": "plate", "organization": "default",

                                        "created_by": admin.id})

        if not db.scalar(select(WatchlistEntry).where(WatchlistEntry.plate_norm == "XYZ7788")):

            db.add(WatchlistEntry(watchlist_id=wl.id, plate_norm="XYZ7788",

                                  reason="reported stolen (demo)"))

        db.commit()

        for fname in ("flock_sample.csv", "flock_sample.json", "flock_report.pdf"):

            path = os.path.join(SAMPLE_DIR, fname)

            if os.path.exists(path):

                _id, summary = ingest_file(db, source, path)

                print(f"ingested {fname}: {summary}")

        inv, created = get_or_create(

            db, Investigation, title="Demo: Downtown Corridor Activity",

            defaults={"summary": "Sample investigation tracking vehicles through the "

                                  "downtown/Poplar corridor.", "priority": "high",

                      "status": "active", "organization": "default", "lead_user_id": admin.id})

        if created:

            for plate in ("ABC1234", "XYZ7788", "GHI5566"):

                reads = db.scalars(select(PlateRead).where(PlateRead.plate_norm == plate)).all()

                for r in reads:

                    s = db.scalar(select(Sighting).where(Sighting.plate_read_id == r.id))

                    if s:

                        s.investigation_id = inv.id

            db.commit()

            print(f"created investigation: {inv.title} ({inv.id})")

        total = db.scalar(select(PlateRead).limit(1))

        print("seed complete." if total else "seed complete (no reads - check sample data).")

    finally:

        db.close()

if __name__ == "__main__":

    main()