siftlode/scripts/e2e_seed.py

"""Idempotent seed for the E2E characterization (golden-master) account.

Run INSIDE the api container — it needs the app's models + the argon2 hasher:

    docker exec -i siftlode-api-1 python - < scripts/e2e_seed.py

Resets a dedicated test account to a known baseline every run, so the Playwright
golden-master net is deterministic and never touches a real account. Self-healing: a crashed
spec that left the account dirty is cleaned on the next run. Prints one JSON line describing
what it seeded (the wrapper / global-setup reads it).

Credentials come from env: E2E_EMAIL (default e2e@siftlode.local) and E2E_PASSWORD (REQUIRED, no
default — the ops wrapper exports it from the secret store); the account
lives only in the localdev DB.
"""

import json
import os
import sys
from datetime import datetime, timezone

sys.path.insert(0, "/app")

from sqlalchemy import delete, func, select

from app.db import SessionLocal
from app.catalog.models import Channel, Subscription, Video
from app.models import AppState, User
from app.reader.models import Playlist, PlaylistItem, VideoState
from app.security import hash_password

EMAIL = os.environ.get("E2E_EMAIL", "e2e@siftlode.local")
PASSWORD = os.environ.get("E2E_PASSWORD", "")
if not PASSWORD:
    sys.exit(
        "e2e_seed: E2E_PASSWORD is not set (no default by design) — run it via `siftlode e2e-seed`"
    )
N_SUBS = 3


def main() -> None:
    db = SessionLocal()
    try:
        # 0. Mark the instance configured so the first-run setup wizard is skipped. A prod→dev
        #    reseed (siftlode seed-dev-from-prod) TRUNCATEs app_state — it's excluded from the
        #    data allow-list, being local/operational state — which would otherwise drop the box
        #    back into "Setup needed" and block login/UAT. Since e2e_seed always leaves a loginable
        #    admin below, the instance is legitimately set up. Idempotent (mirrors state._row +
        #    state.mark_configured). The running api process caches `configured` one-way from False,
        #    so it re-reads the DB and promotes on the next request — no restart needed.
        app_state = db.get(AppState, 1)
        if app_state is None:
            app_state = AppState(id=1, sync_paused=False)
            db.add(app_state)
        app_state.configured = True
        app_state.setup_token_hash = None
        db.commit()

        # 1. Upsert the account and always reset it to a clean, loginable baseline
        #    (email_verified + is_active + not suspended + password set).
        user = db.scalar(select(User).where(User.email == EMAIL))
        if user is None:
            user = User(email=EMAIL)
            db.add(user)
        user.password_hash = hash_password(PASSWORD)
        user.email_verified = True
        user.is_active = True
        user.is_suspended = False
        user.google_sub = None
        # Admin so the E2E can reach the admin-only pages (scheduler / config / users / audit) and
        # thus the render ternary's admin branches — the primary real user (npeter83) is admin too.
        user.role = "admin"
        user.display_name = "E2E Tester"
        user.preferences = {"language": "en"}
        db.commit()

        # 2. Subscriptions: the top-N organic channels by catalog size — deterministic, and
        #    >1 so a channel filter visibly narrows the feed. Reset each run.
        top = db.execute(
            select(Video.channel_id, func.count().label("n"))
            .join(Channel, Channel.id == Video.channel_id)
            .where(Channel.from_search.is_(False), Channel.from_explore.is_(False))
            .group_by(Video.channel_id)
            .order_by(func.count().desc())
            .limit(N_SUBS)
        ).all()
        channel_ids = [row[0] for row in top]

        db.execute(delete(Subscription).where(Subscription.user_id == user.id))
        for cid in channel_ids:
            db.add(
                Subscription(
                    user_id=user.id, channel_id=cid, subscribed_at=datetime.now(timezone.utc)
                )
            )
        db.commit()

        # 3. Deterministic watch state (reset each run):
        #    - one in-progress video (drives the resume bar + "in progress" filter) from the
        #      first channel — status stays "new" so it still counts as unwatched;
        #    - one WATCHED video from the second channel, so the "watched" show-filter is a
        #      real, non-empty narrowing of the feed (and the "all" vs "unwatched" split exists).
        db.execute(delete(VideoState).where(VideoState.user_id == user.id))

        def _newest(channel_id: str, exclude: set[str]) -> Video | None:
            conds = [
                Video.channel_id == channel_id,
                Video.duration_seconds.is_not(None),
                Video.live_status == "none",
            ]
            if exclude:
                conds.append(Video.id.not_in(exclude))
            return db.scalar(
                select(Video).where(*conds).order_by(Video.published_at.desc()).limit(1)
            )

        used: set[str] = set()
        in_progress = None
        watched = None
        if channel_ids:
            vid = _newest(channel_ids[0], used)
            if vid is not None:
                db.add(
                    VideoState(
                        user_id=user.id,
                        video_id=vid.id,
                        status="new",
                        position_seconds=42,
                        progress_updated_at=datetime.now(timezone.utc),
                    )
                )
                in_progress = vid.id
                used.add(vid.id)
        if len(channel_ids) > 1:
            wv = _newest(channel_ids[1], used)
            if wv is not None:
                db.add(
                    VideoState(
                        user_id=user.id,
                        video_id=wv.id,
                        status="watched",
                        watched_at=datetime.now(timezone.utc),
                    )
                )
                watched = wv.id
                used.add(wv.id)

        # 4. One "saved" video (Watch-later membership drives the saved marker) from the last
        #    subscribed channel. Reset the built-in playlist's items each run.
        wl = db.scalar(
            select(Playlist).where(Playlist.user_id == user.id, Playlist.kind == "watch_later")
        )
        if wl is None:
            wl = Playlist(user_id=user.id, name="Watch later", kind="watch_later")
            db.add(wl)
            db.commit()
        db.execute(delete(PlaylistItem).where(PlaylistItem.playlist_id == wl.id))
        saved = None
        if channel_ids:
            svid = _newest(channel_ids[-1], used)
            if svid is not None:
                db.add(PlaylistItem(playlist_id=wl.id, video_id=svid.id, position=0))
                saved = svid.id
        db.commit()

        # T201: this script's stdout IS its interface — the e2e runner parses this JSON.
        print(  # noqa: T201
            json.dumps(
                {
                    "ok": True,
                    "user_id": user.id,
                    "email": EMAIL,
                    "channels": channel_ids,
                    "in_progress": in_progress,
                    "watched": watched,
                    "saved": saved,
                }
            )
        )
    finally:
        db.close()


if __name__ == "__main__":
    main()