feat: harden analytics dashboard

lib/query-shared.ts

@@ -0,0 +1,54 @@
+import { query } from "./db";
+import { queryCache } from "./query-cache";
+
+export const cached = queryCache.cached;
+
+export function cacheKey(...parts: unknown[]): string {
+  return parts.map((p) => String(p ?? "")).join(":");
+}
+
+export const REAL_MODEL = `COALESCE(
+  CASE WHEN other IS NOT NULL AND other != '' AND other::jsonb ? 'upstream_model_name'
+       THEN other::jsonb->>'upstream_model_name' END,
+  model_name)`;
+
+export const CACHE_CREATION = `COALESCE(
+  CASE WHEN other IS NOT NULL AND other != '' AND other::jsonb ? 'cache_creation_tokens'
+       THEN (other::jsonb->>'cache_creation_tokens')::bigint END,
+  0)`;
+
+export const CACHE_READ = `COALESCE(
+  CASE WHEN other IS NOT NULL AND other != '' AND other::jsonb ? 'cache_tokens'
+       THEN (other::jsonb->>'cache_tokens')::bigint END,
+  0)`;
+
+export const TOKEN_NAME = `COALESCE(NULLIF(BTRIM(token_name), ''), '')`;
+
+export function timeWhere(
+  params: (string | number | boolean | null)[],
+  startTs?: number,
+  endTs?: number
+): string {
+  let where = "type = 2";
+  if (startTs) {
+    params.push(startTs);
+    where += ` AND created_at >= $${params.length}`;
+  }
+  if (endTs) {
+    params.push(endTs);
+    where += ` AND created_at <= $${params.length}`;
+  }
+  return where;
+}
+
+export async function getDisplayNames(): Promise<Record<number, string>> {
+  const rows = await query(
+    "SELECT id, display_name FROM users WHERE display_name IS NOT NULL AND display_name != ''"
+  );
+  return Object.fromEntries(rows.map((r) => [r.id, r.display_name]));
+}
+
+export async function getChannelNames(): Promise<Record<number, string>> {
+  const rows = await query("SELECT id, name FROM channels");
+  return Object.fromEntries(rows.map((r) => [r.id, r.name]));
+}