@talkpilot/core-db 1.3.66 → 1.3.68

package/src/talkpilot/clientAudioBuffers/clientAudioBuffer.getters.ts

@@ -1,117 +1,117 @@
-import { getDb } from "../index";
-import {
-  ClientAudioBufferDoc,
-  ClientAudioBufferType,
-  CreateAudioBufferInput,
-} from "./clientsAudioBuffers.types";
-import { Collection, InsertOneResult, DeleteResult } from "mongodb";
-
-const getClientAudioBuffersCollection =
-  (): Collection<ClientAudioBufferType> => {
-    return getDb().collection<ClientAudioBufferType>("clientAudioBuffers");
-  };
-
-/**
- * Deletes all audio buffers associated with a specific callSid.
- * Useful for clearing context between different tickets in the same call.
- */
-const deleteAudioBuffersByCallSid = async (
-  callSid: string,
-): Promise<DeleteResult> => {
-  return getClientAudioBuffersCollection().deleteMany({ callSid });
-};
-
-const insertClientAudioBuffer = async (
-  audioBufferInput: CreateAudioBufferInput,
-): Promise<InsertOneResult<ClientAudioBufferType>> => {
-  const collection = getClientAudioBuffersCollection();
-  const realtimeDate = new Date();
-  const ttlMs = audioBufferInput.ttlMs ?? 3600_000;
-  const expiresAt = new Date(realtimeDate.getTime() + ttlMs);
-
-  const inputFieldsForDocument = {
-    callSid: audioBufferInput.callSid,
-    clientId: audioBufferInput.clientId,
-    bucketStartMs: audioBufferInput.bucketStartMs,
-    bucketDurationMs: audioBufferInput.bucketDurationMs,
-    data: audioBufferInput.data,
-  };
-  const clientAudioBufferMongoDocument: ClientAudioBufferType = {
-    ...inputFieldsForDocument,
-    codec: "mulaw",
-    sampleRateHz: 8000,
-    channels: 1,
-    createdAt: realtimeDate,
-    expiresAt: expiresAt,
-  };
-  return collection.insertOne(clientAudioBufferMongoDocument);
-};
-
-const findClientAudioBuffersByCallSidAndRange = async (
-  callSid: string,
-  startMs?: number,
-  endMs?: number,
-): Promise<ClientAudioBufferDoc[]> => {
-  const collection = getClientAudioBuffersCollection();
-  if (startMs === undefined) {
-    return collection.find({ callSid }).toArray();
-  }
-  const endMsToUse = endMs ?? Date.now();
-  return collection
-    .find({
-      callSid,
-      bucketStartMs: {
-        $gte: startMs,
-        $lte: endMsToUse,
-      },
-    })
-    .toArray();
-};
-
-const getAudioChunksByCallSid = async (
-  callSid: string,
-  limit?: number,
-): Promise<Buffer[]> => {
-  const query = getClientAudioBuffersCollection().find({ callSid });
-  if (limit) {
-    query.sort({ bucketStartMs: -1 }).limit(limit);
-  } else {
-    query.sort({ bucketStartMs: 1 });
-  }
-
-  const docs = await query.toArray();
-
-  if (!docs.length) {
-    return [];
-  }
-  const orderedDocs = limit ? docs.reverse() : docs;
-
-  return orderedDocs.map((chunk) => {
-    const data: any = chunk.data;
-    if (Buffer.isBuffer(data)) return data;
-    if (data?.buffer) return Buffer.from(data.buffer);
-    return Buffer.from(data);
-  });
-};
-
-const getMergedAudioByCallSid = async (
-  callSid: string,
-  limit: number,
-): Promise<Buffer> => {
-  const buffers = await getAudioChunksByCallSid(callSid, limit);
-
-  if (!buffers.length) {
-    return Buffer.alloc(0);
-  }
-
-  return Buffer.concat(buffers);
-};
-
-export {
-  insertClientAudioBuffer,
-  getClientAudioBuffersCollection,
-  findClientAudioBuffersByCallSidAndRange,
-  getAudioChunksByCallSid,
-  getMergedAudioByCallSid,
-  deleteAudioBuffersByCallSid,
-};
+import { getDb } from "../index";
+import {
+  ClientAudioBufferDoc,
+  ClientAudioBufferType,
+  CreateAudioBufferInput,
+} from "./clientsAudioBuffers.types";
+import { Collection, InsertOneResult, DeleteResult } from "mongodb";
+
+const getClientAudioBuffersCollection =
+  (): Collection<ClientAudioBufferType> => {
+    return getDb().collection<ClientAudioBufferType>("clientAudioBuffers");
+  };
+
+/**
+ * Deletes all audio buffers associated with a specific callSid.
+ * Useful for clearing context between different tickets in the same call.
+ */
+const deleteAudioBuffersByCallSid = async (
+  callSid: string,
+): Promise<DeleteResult> => {
+  return getClientAudioBuffersCollection().deleteMany({ callSid });
+};
+
+const insertClientAudioBuffer = async (
+  audioBufferInput: CreateAudioBufferInput,
+): Promise<InsertOneResult<ClientAudioBufferType>> => {
+  const collection = getClientAudioBuffersCollection();
+  const realtimeDate = new Date();
+  const ttlMs = audioBufferInput.ttlMs ?? 3600_000;
+  const expiresAt = new Date(realtimeDate.getTime() + ttlMs);
+
+  const inputFieldsForDocument = {
+    callSid: audioBufferInput.callSid,
+    clientId: audioBufferInput.clientId,
+    bucketStartMs: audioBufferInput.bucketStartMs,
+    bucketDurationMs: audioBufferInput.bucketDurationMs,
+    data: audioBufferInput.data,
+  };
+  const clientAudioBufferMongoDocument: ClientAudioBufferType = {
+    ...inputFieldsForDocument,
+    codec: "mulaw",
+    sampleRateHz: 8000,
+    channels: 1,
+    createdAt: realtimeDate,
+    expiresAt: expiresAt,
+  };
+  return collection.insertOne(clientAudioBufferMongoDocument);
+};
+
+const findClientAudioBuffersByCallSidAndRange = async (
+  callSid: string,
+  startMs?: number,
+  endMs?: number,
+): Promise<ClientAudioBufferDoc[]> => {
+  const collection = getClientAudioBuffersCollection();
+  if (startMs === undefined) {
+    return collection.find({ callSid }).toArray();
+  }
+  const endMsToUse = endMs ?? Date.now();
+  return collection
+    .find({
+      callSid,
+      bucketStartMs: {
+        $gte: startMs,
+        $lte: endMsToUse,
+      },
+    })
+    .toArray();
+};
+
+const getAudioChunksByCallSid = async (
+  callSid: string,
+  limit?: number,
+): Promise<Buffer[]> => {
+  const query = getClientAudioBuffersCollection().find({ callSid });
+  if (limit) {
+    query.sort({ bucketStartMs: -1 }).limit(limit);
+  } else {
+    query.sort({ bucketStartMs: 1 });
+  }
+
+  const docs = await query.toArray();
+
+  if (!docs.length) {
+    return [];
+  }
+  const orderedDocs = limit ? docs.reverse() : docs;
+
+  return orderedDocs.map((chunk) => {
+    const data: any = chunk.data;
+    if (Buffer.isBuffer(data)) return data;
+    if (data?.buffer) return Buffer.from(data.buffer);
+    return Buffer.from(data);
+  });
+};
+
+const getMergedAudioByCallSid = async (
+  callSid: string,
+  limit: number,
+): Promise<Buffer> => {
+  const buffers = await getAudioChunksByCallSid(callSid, limit);
+
+  if (!buffers.length) {
+    return Buffer.alloc(0);
+  }
+
+  return Buffer.concat(buffers);
+};
+
+export {
+  insertClientAudioBuffer,
+  getClientAudioBuffersCollection,
+  findClientAudioBuffersByCallSidAndRange,
+  getAudioChunksByCallSid,
+  getMergedAudioByCallSid,
+  deleteAudioBuffersByCallSid,
+};

package/src/talkpilot/clientAudioBuffers/clientsAudioBuffers.types.ts

@@ -1,25 +1,25 @@
-import { WithId } from "mongodb";
-
-export type ClientAudioBufferType = {
-  callSid: string;
-  clientId: string;
-  bucketStartMs: number;
-  bucketDurationMs: number;
-  codec: "mulaw";
-  sampleRateHz: 8000;
-  channels: 1;
-  data: Buffer;
-  expiresAt: Date;
-  createdAt: Date;
-};
-
-export type CreateAudioBufferInput = {
-  callSid: string;
-  clientId: string;
-  bucketStartMs: number;
-  bucketDurationMs: number;
-  data: Buffer;
-  ttlMs?: number;
-};
-
-export type ClientAudioBufferDoc = WithId<ClientAudioBufferType>;
+import { WithId } from "mongodb";
+
+export type ClientAudioBufferType = {
+  callSid: string;
+  clientId: string;
+  bucketStartMs: number;
+  bucketDurationMs: number;
+  codec: "mulaw";
+  sampleRateHz: 8000;
+  channels: 1;
+  data: Buffer;
+  expiresAt: Date;
+  createdAt: Date;
+};
+
+export type CreateAudioBufferInput = {
+  callSid: string;
+  clientId: string;
+  bucketStartMs: number;
+  bucketDurationMs: number;
+  data: Buffer;
+  ttlMs?: number;
+};
+
+export type ClientAudioBufferDoc = WithId<ClientAudioBufferType>;

package/src/talkpilot/clientAudioBuffers/index.ts

@@ -1,2 +1,2 @@
-export * from "./clientAudioBuffer.getters";
-export * from "./clientsAudioBuffers.types";
+export * from "./clientAudioBuffer.getters";
+export * from "./clientsAudioBuffers.types";

package/src/talkpilot/clients/__tests__/clients.quota.spec.ts

@@ -1,123 +1,123 @@
-import {
-  getClientsConfigCollection,
-  incrementClientQuotaUsage,
-  listClientIdsWithConfiguredQuota,
-  resetClientQuotaUsage,
-} from "../../clientsConfig/clientsConfig.getters";
-import { getClientsCollection } from "../clients.getters";
-import {
-  markQuotaAlertSent,
-  markQuotaExpiredSent,
-  updateQuotaAlertsConfig,
-} from "../clients.quota.getters";
-
-describe("quota getters", () => {
-  beforeEach(async () => {
-    await getClientsConfigCollection().deleteMany({});
-    await getClientsCollection().deleteMany({});
-  });
-
-  const seed = async (clientId: string, configQuota: object, clientQuota?: object) => {
-    await getClientsConfigCollection().insertOne({
-      clientId,
-      products: {},
-      quota: configQuota,
-    } as any);
-    if (clientQuota !== undefined) {
-      await getClientsCollection().insertOne({
-        clientId,
-        createdAt: new Date(),
-        updatedAt: new Date(),
-        credits: 0,
-        quota: clientQuota,
-      } as any);
-    }
-  };
-
-  it("updateQuotaAlertsConfig stores alerts on clients", async () => {
-    const clientId = "quota-alerts";
-    await seed(clientId, { totalQuota: 100, usedQuota: 5 });
-
-    await updateQuotaAlertsConfig(clientId, [
-      { thresholdPct: 50, recipientEmails: ["a@example.com"] },
-      { thresholdPct: 80, recipientEmails: ["b@example.com", "c@example.com"] },
-    ]);
-
-    const client = await getClientsCollection().findOne({ clientId });
-    expect(client?.quota?.alerts).toEqual([
-      { thresholdPct: 50, recipientEmails: ["a@example.com"] },
-      { thresholdPct: 80, recipientEmails: ["b@example.com", "c@example.com"] },
-    ]);
-
-    const config = await getClientsConfigCollection().findOne({ clientId });
-    expect(config?.quota).toEqual({ totalQuota: 100, usedQuota: 5 });
-  });
-
-  it("markQuotaAlertSent returns true only on first transition", async () => {
-    const clientId = "quota-alert-sent";
-    await seed(
-      clientId,
-      { totalQuota: 100, usedQuota: 0 },
-      { alerts: [{ thresholdPct: 50, recipientEmails: ["a@b.com"] }] },
-    );
-
-    expect(await markQuotaAlertSent(clientId, 0)).toBe(true);
-    expect(await markQuotaAlertSent(clientId, 0)).toBe(false);
-
-    const client = await getClientsCollection().findOne({ clientId });
-    expect(client?.quota?.alerts?.[0]?.thresholdSent).toBe(true);
-  });
-
-  it("markQuotaExpiredSent returns true only on first transition", async () => {
-    const clientId = "quota-notify";
-    await seed(clientId, { totalQuota: 10, usedQuota: 10 });
-
-    expect(await markQuotaExpiredSent(clientId)).toBe(true);
-    expect(await markQuotaExpiredSent(clientId)).toBe(false);
-
-    const client = await getClientsCollection().findOne({ clientId });
-    expect(client?.quota?.expiredSent).toBe(true);
-  });
-
-  it("listClientIdsWithConfiguredQuota returns clientIds with totalQuota set", async () => {
-    await seed("quota-a", { totalQuota: 100, usedQuota: 0 });
-    await seed("quota-b", { totalQuota: 50, usedQuota: 10 });
-    await getClientsConfigCollection().insertOne({
-      clientId: "no-quota",
-      products: {},
-    } as any);
-
-    const ids = await listClientIdsWithConfiguredQuota();
-    expect(ids.sort()).toEqual(["quota-a", "quota-b"]);
-  });
-
-  it("incrementClientQuotaUsage increments usedQuota on clientsConfig", async () => {
-    const clientId = "quota-inc";
-    await seed(clientId, { totalQuota: 100, usedQuota: 3 });
-
-    await incrementClientQuotaUsage(clientId, 2);
-    const config = await getClientsConfigCollection().findOne({ clientId });
-    expect(config?.quota?.usedQuota).toBe(5);
-  });
-
-  it("resetClientQuotaUsage zeros usedQuota and clears client quota flags", async () => {
-    const clientId = "quota-reset";
-    await seed(
-      clientId,
-      { totalQuota: 100, usedQuota: 50 },
-      {
-        expiredSent: true,
-        alerts: [{ thresholdPct: 50, recipientEmails: ["a@b.com"], thresholdSent: true }],
-      },
-    );
-
-    await resetClientQuotaUsage(clientId);
-
-    const config = await getClientsConfigCollection().findOne({ clientId });
-    expect(config?.quota?.usedQuota).toBe(0);
-
-    const client = await getClientsCollection().findOne({ clientId });
-    expect(client?.quota?.expiredSent).toBeUndefined();
-    expect(client?.quota?.alerts?.[0]?.thresholdSent).toBeUndefined();
-  });
-});
+import {
+  getClientsConfigCollection,
+  incrementClientQuotaUsage,
+  listClientIdsWithConfiguredQuota,
+  resetClientQuotaUsage,
+} from "../../clientsConfig/clientsConfig.getters";
+import { getClientsCollection } from "../clients.getters";
+import {
+  markQuotaAlertSent,
+  markQuotaExpiredSent,
+  updateQuotaAlertsConfig,
+} from "../clients.quota.getters";
+
+describe("quota getters", () => {
+  beforeEach(async () => {
+    await getClientsConfigCollection().deleteMany({});
+    await getClientsCollection().deleteMany({});
+  });
+
+  const seed = async (clientId: string, configQuota: object, clientQuota?: object) => {
+    await getClientsConfigCollection().insertOne({
+      clientId,
+      products: {},
+      quota: configQuota,
+    } as any);
+    if (clientQuota !== undefined) {
+      await getClientsCollection().insertOne({
+        clientId,
+        createdAt: new Date(),
+        updatedAt: new Date(),
+        credits: 0,
+        quota: clientQuota,
+      } as any);
+    }
+  };
+
+  it("updateQuotaAlertsConfig stores alerts on clients", async () => {
+    const clientId = "quota-alerts";
+    await seed(clientId, { totalQuota: 100, usedQuota: 5 });
+
+    await updateQuotaAlertsConfig(clientId, [
+      { thresholdPct: 50, recipientEmails: ["a@example.com"] },
+      { thresholdPct: 80, recipientEmails: ["b@example.com", "c@example.com"] },
+    ]);
+
+    const client = await getClientsCollection().findOne({ clientId });
+    expect(client?.quota?.alerts).toEqual([
+      { thresholdPct: 50, recipientEmails: ["a@example.com"] },
+      { thresholdPct: 80, recipientEmails: ["b@example.com", "c@example.com"] },
+    ]);
+
+    const config = await getClientsConfigCollection().findOne({ clientId });
+    expect(config?.quota).toEqual({ totalQuota: 100, usedQuota: 5 });
+  });
+
+  it("markQuotaAlertSent returns true only on first transition", async () => {
+    const clientId = "quota-alert-sent";
+    await seed(
+      clientId,
+      { totalQuota: 100, usedQuota: 0 },
+      { alerts: [{ thresholdPct: 50, recipientEmails: ["a@b.com"] }] },
+    );
+
+    expect(await markQuotaAlertSent(clientId, 0)).toBe(true);
+    expect(await markQuotaAlertSent(clientId, 0)).toBe(false);
+
+    const client = await getClientsCollection().findOne({ clientId });
+    expect(client?.quota?.alerts?.[0]?.thresholdSent).toBe(true);
+  });
+
+  it("markQuotaExpiredSent returns true only on first transition", async () => {
+    const clientId = "quota-notify";
+    await seed(clientId, { totalQuota: 10, usedQuota: 10 });
+
+    expect(await markQuotaExpiredSent(clientId)).toBe(true);
+    expect(await markQuotaExpiredSent(clientId)).toBe(false);
+
+    const client = await getClientsCollection().findOne({ clientId });
+    expect(client?.quota?.expiredSent).toBe(true);
+  });
+
+  it("listClientIdsWithConfiguredQuota returns clientIds with totalQuota set", async () => {
+    await seed("quota-a", { totalQuota: 100, usedQuota: 0 });
+    await seed("quota-b", { totalQuota: 50, usedQuota: 10 });
+    await getClientsConfigCollection().insertOne({
+      clientId: "no-quota",
+      products: {},
+    } as any);
+
+    const ids = await listClientIdsWithConfiguredQuota();
+    expect(ids.sort()).toEqual(["quota-a", "quota-b"]);
+  });
+
+  it("incrementClientQuotaUsage increments usedQuota on clientsConfig", async () => {
+    const clientId = "quota-inc";
+    await seed(clientId, { totalQuota: 100, usedQuota: 3 });
+
+    await incrementClientQuotaUsage(clientId, 2);
+    const config = await getClientsConfigCollection().findOne({ clientId });
+    expect(config?.quota?.usedQuota).toBe(5);
+  });
+
+  it("resetClientQuotaUsage zeros usedQuota and clears client quota flags", async () => {
+    const clientId = "quota-reset";
+    await seed(
+      clientId,
+      { totalQuota: 100, usedQuota: 50 },
+      {
+        expiredSent: true,
+        alerts: [{ thresholdPct: 50, recipientEmails: ["a@b.com"], thresholdSent: true }],
+      },
+    );
+
+    await resetClientQuotaUsage(clientId);
+
+    const config = await getClientsConfigCollection().findOne({ clientId });
+    expect(config?.quota?.usedQuota).toBe(0);
+
+    const client = await getClientsCollection().findOne({ clientId });
+    expect(client?.quota?.expiredSent).toBeUndefined();
+    expect(client?.quota?.alerts?.[0]?.thresholdSent).toBeUndefined();
+  });
+});

package/src/talkpilot/clients/clients.getters.ts

@@ -1,32 +1,19 @@
-import { getDb, Client } from "../index";
-import { Collection } from "mongodb";
-
-export const getClientsCollection = (): Collection<Client> => {
-  return getDb().collection<Client>("clients");
-};
-
-export const getClientByAPIKey = async (
-  apiKey: string,
-): Promise<Client | null> => {
-  const client = await getClientsCollection().findOne({
-    apiKey,
-    isActive: true,
-  });
-  return client ? client : null;
-};
-
-export const getAllClientCredits = async (): Promise<Map<string, number>> => {
-  const docs = await getClientsCollection()
-    .find({}, { projection: { clientId: 1, credits: 1 } })
-    .toArray();
-  return new Map(docs.map((c) => [c.clientId, c.credits ?? 0]));
-};
-
-export const getAllClientIds = async (): Promise<string[]> => {
-  const docs = await getClientsCollection()
-    .find({}, { projection: { clientId: 1 } })
-    .toArray();
-  return docs.map((c) => c.clientId);
-};
-export const getClientByClientId = (clientId: string): Promise<Client | null> =>
-  getClientsCollection().findOne({ clientId });
+import { getDb, Client } from "../index";
+import { Collection } from "mongodb";
+
+export const getClientsCollection = (): Collection<Client> => {
+  return getDb().collection<Client>("clients");
+};
+
+export const getClientByAPIKey = async (
+  apiKey: string,
+): Promise<Client | null> => {
+  const client = await getClientsCollection().findOne({
+    apiKey,
+    isActive: true,
+  });
+  return client ? client : null;
+};
+
+export const getClientByClientId = (clientId: string): Promise<Client | null> =>
+  getClientsCollection().findOne({ clientId });