youclonedl/backend/routers/stats.py

from fastapi import APIRouter, Depends
from sqlalchemy.orm import Session
from sqlalchemy import text

from ..auth_utils import get_current_user
from ..database import get_db
from ..models import User

router = APIRouter()


@router.get("")
def get_stats(
    db: Session = Depends(get_db),
    current_user: User = Depends(get_current_user),
):
    uid = current_user.id

    totals = db.execute(
        text("""
            SELECT
                COUNT(*) AS total_watched,
                SUM(uv.watch_progress_seconds) AS total_watch_seconds
            FROM user_videos uv
            WHERE uv.user_id = :uid AND uv.watched = 1
        """),
        {"uid": uid},
    ).mappings().first()

    top_channels = db.execute(
        text("""
            SELECT c.id, c.name,
                   COUNT(*) AS watch_count,
                   SUM(uv.watch_progress_seconds) AS watch_seconds
            FROM user_videos uv
            JOIN videos v ON uv.video_id = v.id
            JOIN channels c ON v.channel_id = c.id
            WHERE uv.user_id = :uid AND uv.watched = 1
            GROUP BY c.id, c.name
            ORDER BY watch_seconds DESC
            LIMIT 10
        """),
        {"uid": uid},
    ).mappings().all()

    daily = db.execute(
        text("""
            SELECT date(uv.last_watched_at) AS date,
                   COUNT(*) AS count,
                   SUM(uv.watch_progress_seconds) AS seconds
            FROM user_videos uv
            WHERE uv.user_id = :uid
              AND uv.watched = 1
              AND uv.last_watched_at >= datetime('now', '-30 days')
            GROUP BY date(uv.last_watched_at)
            ORDER BY date ASC
        """),
        {"uid": uid},
    ).mappings().all()

    this_week = db.execute(
        text("""
            SELECT COUNT(*) AS count, SUM(uv.watch_progress_seconds) AS seconds
            FROM user_videos uv
            WHERE uv.user_id = :uid AND uv.watched = 1
              AND uv.last_watched_at >= datetime('now', '-7 days')
        """),
        {"uid": uid},
    ).mappings().first()

    this_month = db.execute(
        text("""
            SELECT COUNT(*) AS count, SUM(uv.watch_progress_seconds) AS seconds
            FROM user_videos uv
            WHERE uv.user_id = :uid AND uv.watched = 1
              AND uv.last_watched_at >= datetime('now', '-30 days')
        """),
        {"uid": uid},
    ).mappings().first()

    avg_completion = db.execute(
        text("""
            SELECT AVG(uv.completion_percent) AS avg_pct,
                   COUNT(CASE WHEN uv.completion_percent >= 90 THEN 1 END) AS finished_count,
                   COUNT(CASE WHEN uv.completion_percent < 20 AND uv.completion_percent IS NOT NULL THEN 1 END) AS bailed_count,
                   SUM(uv.rewatch_count) AS total_rewatches,
                   COUNT(CASE WHEN uv.rewatch_count > 0 THEN 1 END) AS rewatched_videos
            FROM user_videos uv
            WHERE uv.user_id = :uid AND uv.watched = 1
        """),
        {"uid": uid},
    ).mappings().first()

    top_categories = db.execute(
        text("""
            SELECT v.category, COUNT(*) AS watch_count,
                   AVG(uv.completion_percent) AS avg_completion
            FROM user_videos uv
            JOIN videos v ON uv.video_id = v.id
            WHERE uv.user_id = :uid AND uv.watched = 1 AND v.category IS NOT NULL
            GROUP BY v.category
            ORDER BY watch_count DESC
            LIMIT 8
        """),
        {"uid": uid},
    ).mappings().all()

    taste_profile = db.execute(
        text("""
            SELECT tag, score FROM user_tag_affinity
            WHERE user_id = :uid AND score > 0
            ORDER BY score DESC
            LIMIT 20
        """),
        {"uid": uid},
    ).mappings().all()

    liked_count = db.execute(
        text("SELECT COUNT(*) AS n FROM user_videos WHERE user_id = :uid AND liked = 1"),
        {"uid": uid},
    ).mappings().first()

    return {
        "total_watched": totals["total_watched"] or 0,
        "total_watch_seconds": totals["total_watch_seconds"] or 0,
        "top_channels": [dict(r) for r in top_channels],
        "daily": [dict(r) for r in daily],
        "this_week": {
            "count": this_week["count"] or 0,
            "seconds": this_week["seconds"] or 0,
        },
        "this_month": {
            "count": this_month["count"] or 0,
            "seconds": this_month["seconds"] or 0,
        },
        "avg_completion_percent": round(avg_completion["avg_pct"] or 0, 1),
        "finished_count": avg_completion["finished_count"] or 0,
        "bailed_count": avg_completion["bailed_count"] or 0,
        "total_rewatches": avg_completion["total_rewatches"] or 0,
        "rewatched_videos": avg_completion["rewatched_videos"] or 0,
        "total_liked": liked_count["n"] or 0,
        "top_categories": [dict(r) for r in top_categories],
        "taste_profile": [dict(r) for r in taste_profile],
    }