sexy/lib/db/queries/stats.ts

import { sql, eq, and, ne } from "drizzle-orm";
import { db } from "@/lib/db/client";
import { playSessions, sessionDevices, sessionEvents, devices, recordings } from "@/lib/db/schema";

export async function getSessionsSummary() {
  const [totals] = await db
    .select({
      count: sql<number>`count(*)`,
      totalDurationMs: sql<number>`coalesce(sum(${playSessions.durationMs}), 0)`,
      avgDurationMs: sql<number>`coalesce(avg(${playSessions.durationMs}), 0)`,
    })
    .from(playSessions)
    .where(eq(playSessions.status, "completed"));

  const byKind = await db
    .select({
      kind: playSessions.kind,
      count: sql<number>`count(*)`,
      totalDurationMs: sql<number>`coalesce(sum(${playSessions.durationMs}), 0)`,
    })
    .from(playSessions)
    .where(eq(playSessions.status, "completed"))
    .groupBy(playSessions.kind);

  const durationPerDevice = await db
    .select({
      deviceId: devices.id,
      displayName: devices.displayName,
      bleName: devices.bleName,
      totalActiveMs: sql<number>`coalesce(sum(coalesce(${sessionDevices.disconnectedAt}, ${sessionDevices.connectedAt}) - ${sessionDevices.connectedAt}), 0)`,
    })
    .from(sessionDevices)
    .innerJoin(devices, eq(sessionDevices.deviceId, devices.id))
    .groupBy(devices.id);

  return { ...totals, byKind, durationPerDevice };
}

export async function getSessionTimeline(playSessionId: number, bucketMs = 1000) {
  return db
    .select({
      bucket: sql<number>`(${sessionEvents.tsMs} / ${bucketMs}) * ${bucketMs}`,
      sessionDeviceId: sessionEvents.sessionDeviceId,
      slotLabel: sessionDevices.slotLabel,
      avgValue: sql<number>`avg(${sessionEvents.value})`,
      maxValue: sql<number>`max(${sessionEvents.value})`,
    })
    .from(sessionEvents)
    .innerJoin(sessionDevices, eq(sessionEvents.sessionDeviceId, sessionDevices.id))
    .where(and(eq(sessionEvents.playSessionId, playSessionId), ne(sessionEvents.commandType, "stop")))
    .groupBy(sql`bucket`, sessionEvents.sessionDeviceId)
    .orderBy(sql`bucket`);
}

export async function getDeviceUsageStats() {
  return db
    .select({
      deviceId: devices.id,
      displayName: devices.displayName,
      bleName: devices.bleName,
      sessionCount: sql<number>`count(distinct ${sessionDevices.playSessionId})`,
      totalActiveMs: sql<number>`coalesce(sum(coalesce(${sessionDevices.disconnectedAt}, ${sessionDevices.connectedAt}) - ${sessionDevices.connectedAt}), 0)`,
      lastUsedAt: sql<number | null>`max(${sessionDevices.connectedAt})`,
    })
    .from(devices)
    .leftJoin(sessionDevices, eq(sessionDevices.deviceId, devices.id))
    .groupBy(devices.id);
}

export async function getDeviceCommandCounts() {
  return db
    .select({
      deviceId: devices.id,
      commandCount: sql<number>`count(${sessionEvents.id})`,
    })
    .from(devices)
    .leftJoin(sessionDevices, eq(sessionDevices.deviceId, devices.id))
    .leftJoin(sessionEvents, eq(sessionEvents.sessionDeviceId, sessionDevices.id))
    .groupBy(devices.id);
}

export async function getRecordingLibraryStats() {
  const [totals] = await db
    .select({
      count: sql<number>`count(*)`,
      avgDurationMs: sql<number>`coalesce(avg(${recordings.durationMs}), 0)`,
      totalPlayCount: sql<number>`coalesce(sum(${recordings.playCount}), 0)`,
    })
    .from(recordings);

  const list = await db
    .select({
      id: recordings.id,
      name: recordings.name,
      durationMs: recordings.durationMs,
      playCount: recordings.playCount,
      lastPlayedAt: recordings.lastPlayedAt,
    })
    .from(recordings)
    .orderBy(sql`${recordings.playCount} desc`);

  return { ...totals, list };
}