"""Live ingestion daemon. Continuously feeds fresh plate reads into Sightline through the same

normalize/dedup/watchlist pipeline as everything else, so the platform shows genuine real-time

activity. Reads carry the current timestamp, hit random cameras nationwide, reuse existing

plates (so vehicle histories grow) with occasional new plates and hotlist hits.

Run detached:  python live_feed.py

Stops on SIGTERM/SIGINT.

"""

from __future__ import annotations

import random

import signal

import time

from datetime import datetime, timezone

from sqlalchemy import select

from app.db.session import SessionLocal

from app.models import LicensePlate, Source

from app.services.ingest import ingest_live_reads

from generate_data import CAMERAS

INTERVAL_SECONDS = 4

BATCH_MIN, BATCH_MAX = 2, 7

_running = True

def _stop(*_):

    global _running

    _running = False

def main():

    signal.signal(signal.SIGTERM, _stop)

    signal.signal(signal.SIGINT, _stop)

    db = SessionLocal()

    source = db.scalar(select(Source).order_by(Source.created_at))

    src_id = source.id if source else None

    pool = [(p.plate_norm, p.state) for p in

            db.scalars(select(LicensePlate).limit(400)).all()]

    if not pool:

        pool = [("LIVE001", "TN")]

    print(f"live feed started: source={source.name if source else None}, pool={len(pool)} plates, "

          f"{len(CAMERAS)} cameras, every {INTERVAL_SECONDS}s", flush=True)

    total = 0

    while _running:

        now = datetime.now(timezone.utc)

        batch = []

        for _ in range(random.randint(BATCH_MIN, BATCH_MAX)):

            cam = random.choice(CAMERAS)

            if random.random() < 0.05:

                plate, state = "BNDT447", "TN"

                hotlist = "true"

            elif random.random() < 0.15:

                plate = "".join(random.choices("ABCDEFGHJKLMNPQRSTUVWXYZ", k=3)) + \

                        "".join(random.choices("0123456789", k=4))

                state = random.choice([s for s in {c[4].split(", ")[1] for c in CAMERAS}])

                hotlist = "false"

            else:

                plate, state = random.choice(pool)

                hotlist = "false"

            batch.append({

                "plate": plate, "state": state,

                "timestamp": now.strftime("%Y-%m-%dT%H:%M:%SZ"),

                "camera": cam[0], "latitude": cam[2], "longitude": cam[3],

                "confidence": random.randint(82, 99), "hotlist": hotlist,

            })

        try:

            res = ingest_live_reads(db, src_id, batch, label=f"live-{int(now.timestamp())}")

            total += res.get("inserted", 0)

        except Exception as e:

            print(f"live feed error: {e}", flush=True)

            db.rollback()

        for _ in range(INTERVAL_SECONDS):

            if not _running:

                break

            time.sleep(1)

    db.close()

    print(f"live feed stopped after {total} reads", flush=True)

if __name__ == "__main__":

    main()