@talkpilot/core-db 1.2.0 → 1.2.2

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

@@ -1,160 +1,160 @@
-import {
-  getClientAudioBuffersCollection,
-  insertClientAudioBuffer,
-  findClientAudioBuffersByCallSidAndRange,
-  getMergedAudioByCallSid,
-  getAudioChunksByCallSid,
-} from "../clientAudioBuffer.getters";
-import { ObjectId } from "../../index";
-import { createAudioBufferInput } from "../../../test-utils/factories";
-
-describe("clientAudioBuffers getters", () => {
-  it('should return the "clientAudioBuffers" collection', () => {
-    const collection = getClientAudioBuffersCollection();
-    expect(collection.collectionName).toBe("clientAudioBuffers");
-  });
-
-  it("should insert a client audio buffer and return insertedId", async () => {
-    const inputToCheck = createAudioBufferInput();
-    const result = await insertClientAudioBuffer(inputToCheck);
-    expect(result.insertedId).toBeInstanceOf(ObjectId);
-
-    const savedDoc = await getClientAudioBuffersCollection().findOne({
-      _id: result.insertedId,
-    });
-    expect(savedDoc).toBeDefined();
-    expect(savedDoc?.createdAt).toBeInstanceOf(Date);
-    expect(savedDoc?.expiresAt).toBeInstanceOf(Date);
-  });
-
-  it("should return only buffers in the specified bucketStartMs range", async () => {
-    const callSid = "CA1234567890abcdef1234567890abcdef";
-    await insertClientAudioBuffer(
-      createAudioBufferInput({ callSid, bucketStartMs: 500 }),
-    );
-    await insertClientAudioBuffer(
-      createAudioBufferInput({ callSid, bucketStartMs: 1000 }),
-    );
-    await insertClientAudioBuffer(
-      createAudioBufferInput({ callSid, bucketStartMs: 3000 }),
-    );
-    await insertClientAudioBuffer(
-      createAudioBufferInput({ callSid, bucketStartMs: 6000 }),
-    );
-
-    const result = await findClientAudioBuffersByCallSidAndRange(
-      callSid,
-      500,
-      3000,
-    );
-    expect(result).toHaveLength(3);
-    const bucketStartValues = result.map((doc) => doc.bucketStartMs);
-    const bucketStartsSorted = bucketStartValues.sort((a, b) => a - b);
-    expect(bucketStartsSorted).toEqual([500, 1000, 3000]);
-  });
-
-  it("should return all buffers when no range is provided", async () => {
-    const callSid = "CA1234567890abcdef1234567890abcdeg";
-    await insertClientAudioBuffer(
-      createAudioBufferInput({ callSid, bucketStartMs: 500 }),
-    );
-    await insertClientAudioBuffer(
-      createAudioBufferInput({ callSid, bucketStartMs: 1000 }),
-    );
-    await insertClientAudioBuffer(
-      createAudioBufferInput({ callSid, bucketStartMs: 3000 }),
-    );
-
-    const result = await findClientAudioBuffersByCallSidAndRange(callSid);
-    expect(result).toHaveLength(3);
-    const bucketStartValues = result.map((doc) => doc.bucketStartMs);
-    const bucketStartsSorted = bucketStartValues.sort((a, b) => a - b);
-    expect(bucketStartsSorted).toEqual([500, 1000, 3000]);
-  });
-
-  it("should merge audio buffers by callSid in chronological order", async () => {
-    const callSid = "CA1234567890abcdef1234567890abcdeh";
-
-    const firstBuffer = Buffer.from([1, 2]);
-    const secondBuffer = Buffer.from([3]);
-    const thirdBuffer = Buffer.from([4, 5]);
-
-    await insertClientAudioBuffer(
-      createAudioBufferInput({
-        callSid,
-        bucketStartMs: 1000,
-        data: secondBuffer,
-      }),
-    );
-    await insertClientAudioBuffer(
-      createAudioBufferInput({
-        callSid,
-        bucketStartMs: 500,
-        data: firstBuffer,
-      }),
-    );
-    await insertClientAudioBuffer(
-      createAudioBufferInput({
-        callSid,
-        bucketStartMs: 3000,
-        data: thirdBuffer,
-      }),
-    );
-
-    const merged = await getMergedAudioByCallSid(callSid, 10);
-    const expected = Buffer.concat([firstBuffer, secondBuffer, thirdBuffer]);
-
-    expect(merged).toBeInstanceOf(Buffer);
-    expect(merged.length).toBe(expected.length);
-    expect(merged.equals(expected)).toBe(true);
-  });
-
-  it("should return all audio chunks when less than or equal to limit buffers exist", async () => {
-    const callSid = "CA1234567890abcdef1234567890abclt";
-
-    const limit = 30;
-    const buffers: Buffer[] = [];
-    for (let i = 0; i < limit; i++) {
-      const buf = Buffer.from([i]);
-      buffers.push(buf);
-      await insertClientAudioBuffer(
-        createAudioBufferInput({
-          callSid,
-          bucketStartMs: 1000 + i * 1000,
-          data: buf,
-        }),
-      );
-    }
-
-    const chunks = await getAudioChunksByCallSid(callSid, limit);
-
-    expect(chunks).toHaveLength(buffers.length);
-    expect(chunks.every((chunk) => chunk instanceof Buffer)).toBe(true);
-    expect(Buffer.concat(chunks).equals(Buffer.concat(buffers))).toBe(true);
-  });
-
-  it("should return only the last N audio chunks when more than limit buffers exist", async () => {
-    const callSid = "CA1234567890abcdef1234567890abclm";
-
-    const limit = 30;
-    const buffers: Buffer[] = [];
-    for (let i = 0; i < 35; i++) {
-      const buf = Buffer.from([i]);
-      buffers.push(buf);
-      await insertClientAudioBuffer(
-        createAudioBufferInput({
-          callSid,
-          bucketStartMs: 1000 + i * 1000,
-          data: buf,
-        }),
-      );
-    }
-
-    const chunks = await getAudioChunksByCallSid(callSid, limit);
-    const expectedBuffers = buffers.slice(-limit);
-    expect(chunks).toHaveLength(expectedBuffers.length);
-    expect(Buffer.concat(chunks).equals(Buffer.concat(expectedBuffers))).toBe(
-      true,
-    );
-  });
-});
+import {
+  getClientAudioBuffersCollection,
+  insertClientAudioBuffer,
+  findClientAudioBuffersByCallSidAndRange,
+  getMergedAudioByCallSid,
+  getAudioChunksByCallSid,
+} from "../clientAudioBuffer.getters";
+import { ObjectId } from "../../index";
+import { createAudioBufferInput } from "../../../test-utils/factories";
+
+describe("clientAudioBuffers getters", () => {
+  it('should return the "clientAudioBuffers" collection', () => {
+    const collection = getClientAudioBuffersCollection();
+    expect(collection.collectionName).toBe("clientAudioBuffers");
+  });
+
+  it("should insert a client audio buffer and return insertedId", async () => {
+    const inputToCheck = createAudioBufferInput();
+    const result = await insertClientAudioBuffer(inputToCheck);
+    expect(result.insertedId).toBeInstanceOf(ObjectId);
+
+    const savedDoc = await getClientAudioBuffersCollection().findOne({
+      _id: result.insertedId,
+    });
+    expect(savedDoc).toBeDefined();
+    expect(savedDoc?.createdAt).toBeInstanceOf(Date);
+    expect(savedDoc?.expiresAt).toBeInstanceOf(Date);
+  });
+
+  it("should return only buffers in the specified bucketStartMs range", async () => {
+    const callSid = "CA1234567890abcdef1234567890abcdef";
+    await insertClientAudioBuffer(
+      createAudioBufferInput({ callSid, bucketStartMs: 500 }),
+    );
+    await insertClientAudioBuffer(
+      createAudioBufferInput({ callSid, bucketStartMs: 1000 }),
+    );
+    await insertClientAudioBuffer(
+      createAudioBufferInput({ callSid, bucketStartMs: 3000 }),
+    );
+    await insertClientAudioBuffer(
+      createAudioBufferInput({ callSid, bucketStartMs: 6000 }),
+    );
+
+    const result = await findClientAudioBuffersByCallSidAndRange(
+      callSid,
+      500,
+      3000,
+    );
+    expect(result).toHaveLength(3);
+    const bucketStartValues = result.map((doc) => doc.bucketStartMs);
+    const bucketStartsSorted = bucketStartValues.sort((a, b) => a - b);
+    expect(bucketStartsSorted).toEqual([500, 1000, 3000]);
+  });
+
+  it("should return all buffers when no range is provided", async () => {
+    const callSid = "CA1234567890abcdef1234567890abcdeg";
+    await insertClientAudioBuffer(
+      createAudioBufferInput({ callSid, bucketStartMs: 500 }),
+    );
+    await insertClientAudioBuffer(
+      createAudioBufferInput({ callSid, bucketStartMs: 1000 }),
+    );
+    await insertClientAudioBuffer(
+      createAudioBufferInput({ callSid, bucketStartMs: 3000 }),
+    );
+
+    const result = await findClientAudioBuffersByCallSidAndRange(callSid);
+    expect(result).toHaveLength(3);
+    const bucketStartValues = result.map((doc) => doc.bucketStartMs);
+    const bucketStartsSorted = bucketStartValues.sort((a, b) => a - b);
+    expect(bucketStartsSorted).toEqual([500, 1000, 3000]);
+  });
+
+  it("should merge audio buffers by callSid in chronological order", async () => {
+    const callSid = "CA1234567890abcdef1234567890abcdeh";
+
+    const firstBuffer = Buffer.from([1, 2]);
+    const secondBuffer = Buffer.from([3]);
+    const thirdBuffer = Buffer.from([4, 5]);
+
+    await insertClientAudioBuffer(
+      createAudioBufferInput({
+        callSid,
+        bucketStartMs: 1000,
+        data: secondBuffer,
+      }),
+    );
+    await insertClientAudioBuffer(
+      createAudioBufferInput({
+        callSid,
+        bucketStartMs: 500,
+        data: firstBuffer,
+      }),
+    );
+    await insertClientAudioBuffer(
+      createAudioBufferInput({
+        callSid,
+        bucketStartMs: 3000,
+        data: thirdBuffer,
+      }),
+    );
+
+    const merged = await getMergedAudioByCallSid(callSid, 10);
+    const expected = Buffer.concat([firstBuffer, secondBuffer, thirdBuffer]);
+
+    expect(merged).toBeInstanceOf(Buffer);
+    expect(merged.length).toBe(expected.length);
+    expect(merged.equals(expected)).toBe(true);
+  });
+
+  it("should return all audio chunks when less than or equal to limit buffers exist", async () => {
+    const callSid = "CA1234567890abcdef1234567890abclt";
+
+    const limit = 30;
+    const buffers: Buffer[] = [];
+    for (let i = 0; i < limit; i++) {
+      const buf = Buffer.from([i]);
+      buffers.push(buf);
+      await insertClientAudioBuffer(
+        createAudioBufferInput({
+          callSid,
+          bucketStartMs: 1000 + i * 1000,
+          data: buf,
+        }),
+      );
+    }
+
+    const chunks = await getAudioChunksByCallSid(callSid, limit);
+
+    expect(chunks).toHaveLength(buffers.length);
+    expect(chunks.every((chunk) => chunk instanceof Buffer)).toBe(true);
+    expect(Buffer.concat(chunks).equals(Buffer.concat(buffers))).toBe(true);
+  });
+
+  it("should return only the last N audio chunks when more than limit buffers exist", async () => {
+    const callSid = "CA1234567890abcdef1234567890abclm";
+
+    const limit = 30;
+    const buffers: Buffer[] = [];
+    for (let i = 0; i < 35; i++) {
+      const buf = Buffer.from([i]);
+      buffers.push(buf);
+      await insertClientAudioBuffer(
+        createAudioBufferInput({
+          callSid,
+          bucketStartMs: 1000 + i * 1000,
+          data: buf,
+        }),
+      );
+    }
+
+    const chunks = await getAudioChunksByCallSid(callSid, limit);
+    const expectedBuffers = buffers.slice(-limit);
+    expect(chunks).toHaveLength(expectedBuffers.length);
+    expect(Buffer.concat(chunks).equals(Buffer.concat(expectedBuffers))).toBe(
+      true,
+    );
+  });
+});

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/clients.getters.ts

@@ -1,16 +1,16 @@
-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;
-};
+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;
+};

package/src/talkpilot/clients/clients.types.ts

@@ -1,14 +1,14 @@
-import { ObjectId } from "mongodb";
-
-export type Client = {
-  _id?: ObjectId;
-  createdAt: Date;
-  updatedAt: Date;
-  clientId: string;
-  isActive?: boolean | null;
-  apiKey?: string | null;
-  credits: number;
-  managedBy?: string;
-};
-
-export type ClientDoc = Client & { _id: ObjectId };
+import { ObjectId } from "mongodb";
+
+export type Client = {
+  _id?: ObjectId;
+  createdAt: Date;
+  updatedAt: Date;
+  clientId: string;
+  isActive?: boolean | null;
+  apiKey?: string | null;
+  credits: number;
+  managedBy?: string;
+};
+
+export type ClientDoc = Client & { _id: ObjectId };