SubMiner/src/core/services/immersion-tracker/session.ts

import crypto from 'node:crypto';
import type { DatabaseSync } from './sqlite';
import { createInitialSessionState } from './reducer';
import { nowMs } from './time';
import { SESSION_STATUS_ACTIVE, SESSION_STATUS_ENDED } from './types';
import type { SessionState } from './types';
import { toDbMs, toDbTimestamp } from './query-shared';

export function startSessionRecord(
  db: DatabaseSync,
  videoId: number,
  startedAtMs = nowMs(),
): { sessionId: number; state: SessionState } {
  const sessionUuid = crypto.randomUUID();
  const createdAtMs = nowMs();
  const result = db
    .prepare(
      `
        INSERT INTO imm_sessions (
          session_uuid, video_id, started_at_ms, status,
          CREATED_DATE, LAST_UPDATE_DATE
      ) VALUES (?, ?, ?, ?, ?, ?)
    `,
    )
    .run(
      sessionUuid,
      videoId,
      toDbTimestamp(startedAtMs),
      SESSION_STATUS_ACTIVE,
      toDbTimestamp(startedAtMs),
      toDbTimestamp(createdAtMs),
    );
  const sessionId = Number(result.lastInsertRowid);
  return {
    sessionId,
    state: createInitialSessionState(sessionId, videoId, startedAtMs),
  };
}

export function finalizeSessionRecord(
  db: DatabaseSync,
  sessionState: SessionState,
  endedAtMs: number | string = nowMs(),
): void {
  db.prepare(
    `
      UPDATE imm_sessions
      SET
        ended_at_ms = ?,
        status = ?,
        ended_media_ms = ?,
        total_watched_ms = ?,
        active_watched_ms = ?,
        lines_seen = ?,
        tokens_seen = ?,
        cards_mined = ?,
        lookup_count = ?,
        lookup_hits = ?,
        yomitan_lookup_count = ?,
        pause_count = ?,
        pause_ms = ?,
        seek_forward_count = ?,
        seek_backward_count = ?,
        media_buffer_events = ?,
        LAST_UPDATE_DATE = ?
      WHERE session_id = ?
    `,
  ).run(
    toDbTimestamp(endedAtMs),
    SESSION_STATUS_ENDED,
    sessionState.lastMediaMs === null ? null : toDbMs(sessionState.lastMediaMs),
    sessionState.totalWatchedMs,
    sessionState.activeWatchedMs,
    sessionState.linesSeen,
    sessionState.tokensSeen,
    sessionState.cardsMined,
    sessionState.lookupCount,
    sessionState.lookupHits,
    sessionState.yomitanLookupCount,
    sessionState.pauseCount,
    sessionState.pauseMs,
    sessionState.seekForwardCount,
    sessionState.seekBackwardCount,
    sessionState.mediaBufferEvents,
    toDbTimestamp(nowMs()),
    sessionState.sessionId,
  );
}