@talkpilot/core-db 1.2.0 → 1.2.1

package/src/talkpilot/mongodb-client.ts

@@ -1,78 +1,78 @@
-import { MongoClient, Db } from "mongodb";
-import { setDb } from "./index";
-import { validateConfig, validateMongoUri } from "../utils/validation";
-
-class MongoDBClient {
-  private client: MongoClient | null = null;
-  private db: Db | null = null;
-
-  private extractDatabaseName(uri: string): string | null {
-    try {
-      const url = new URL(uri);
-      const dbName = url.pathname.slice(1);
-      return dbName || null;
-    } catch {
-      return null;
-    }
-  }
-
-  async connect(uri?: string, databaseName?: string): Promise<void> {
-    if (this.client) {
-      return;
-    }
-
-    const mongodbUri = uri || process.env.MONGO_URI || process.env.MONGODB_URI;
-    validateConfig("MONGO_URI", mongodbUri);
-    validateMongoUri(mongodbUri!);
-
-    try {
-      this.client = new MongoClient(mongodbUri!);
-      await this.client.connect();
-
-      const dbNameFromUri = this.extractDatabaseName(mongodbUri!);
-      const dbName =
-        databaseName || process.env.TALKPILOT_DB_NAME || dbNameFromUri;
-
-      if (!dbName) {
-        throw new Error(
-          "[core-db] Database name not specified. Please pass databaseName to connect(), " +
-            "add it to the URI, or set TALKPILOT_DB_NAME.",
-        );
-      }
-
-      this.db = this.client.db(dbName);
-      setDb(this.db);
-      console.info(`[core-db] TalkPilot MongoDB connected: ${dbName}`);
-    } catch (error) {
-      console.error("[core-db] TalkPilot connection failed", error);
-      throw error;
-    }
-  }
-
-  async disconnect(): Promise<void> {
-    if (this.client) {
-      try {
-        await this.client.close();
-        this.client = null;
-        this.db = null;
-        console.info("MongoDB disconnected successfully");
-      } catch (error) {
-        console.error("[core-db] Disconnection failed", error);
-        throw error;
-      }
-    }
-  }
-
-  getDb(): Db {
-    if (!this.db) {
-      throw new Error("Database not initialized. Call connect() first.");
-    }
-    return this.db;
-  }
-
-  isConnected(): boolean {
-    return this.client !== null && this.client !== undefined;
-  }
-}
-
-export const mongodbClient = new MongoDBClient();
+import { MongoClient, Db } from "mongodb";
+import { setDb } from "./index";
+import { validateConfig, validateMongoUri } from "../utils/validation";
+
+class MongoDBClient {
+  private client: MongoClient | null = null;
+  private db: Db | null = null;
+
+  private extractDatabaseName(uri: string): string | null {
+    try {
+      const url = new URL(uri);
+      const dbName = url.pathname.slice(1);
+      return dbName || null;
+    } catch {
+      return null;
+    }
+  }
+
+  async connect(uri?: string, databaseName?: string): Promise<void> {
+    if (this.client) {
+      return;
+    }
+
+    const mongodbUri = uri || process.env.MONGO_URI || process.env.MONGODB_URI;
+    validateConfig("MONGO_URI", mongodbUri);
+    validateMongoUri(mongodbUri!);
+
+    try {
+      this.client = new MongoClient(mongodbUri!);
+      await this.client.connect();
+
+      const dbNameFromUri = this.extractDatabaseName(mongodbUri!);
+      const dbName =
+        databaseName || process.env.TALKPILOT_DB_NAME || dbNameFromUri;
+
+      if (!dbName) {
+        throw new Error(
+          "[core-db] Database name not specified. Please pass databaseName to connect(), " +
+            "add it to the URI, or set TALKPILOT_DB_NAME.",
+        );
+      }
+
+      this.db = this.client.db(dbName);
+      setDb(this.db);
+      console.info(`[core-db] TalkPilot MongoDB connected: ${dbName}`);
+    } catch (error) {
+      console.error("[core-db] TalkPilot connection failed", error);
+      throw error;
+    }
+  }
+
+  async disconnect(): Promise<void> {
+    if (this.client) {
+      try {
+        await this.client.close();
+        this.client = null;
+        this.db = null;
+        console.info("MongoDB disconnected successfully");
+      } catch (error) {
+        console.error("[core-db] Disconnection failed", error);
+        throw error;
+      }
+    }
+  }
+
+  getDb(): Db {
+    if (!this.db) {
+      throw new Error("Database not initialized. Call connect() first.");
+    }
+    return this.db;
+  }
+
+  isConnected(): boolean {
+    return this.client !== null && this.client !== undefined;
+  }
+}
+
+export const mongodbClient = new MongoDBClient();

package/src/talkpilot/phone_numbers/__tests__/phone_numbers.spec.ts

@@ -1,247 +1,247 @@
-import {
-  getPhoneDataByPhoneNumber,
-  getPhoneNumbersCollection,
-  getPhoneNumbersForFlows,
-  getClientPhoneNumber,
-  getClientPhoneData,
-  createPhoneNumberEntity,
-  createPurchasedPhoneNumber,
-} from '../phone_numbers.getter';
-import { getFlowsCollection } from '../../flows/flows.getter';
-import { createFlow, createPhoneNumber } from '../../../test-utils/factories';
-import { ObjectId } from 'mongodb';
-
-describe('db.phoneNumbers', () => {
-  describe('getPhoneDataByPhoneNumber', () => {
-    it('return phone number data with flow', async () => {
-      const flow = createFlow({
-        clientId: "test-client-id",
-        conversationSettings: {
-          interruptions: {
-            enableInterruptionDetection: true,
-            interruptionWindowSeconds: 60,
-            interruptionThresholdSeconds: 10,
-            interruptionInstruction:
-              "I am hearing background noise that makes it hard for me to focus on what you are saying. Please try to move to a quieter place or reduce the noise so I can better assist you.",
-          },
-          silence: {
-            enableSilenceDetection: true,
-            firstWarningSilenceSeconds: 5,
-            firstWarningInstruction:
-              "I have not heard you for a few seconds. If you are still on the line, please say something so we can continue.",
-            secondWarningSilenceSeconds: 10,
-            secondWarningInstruction:
-              "I still have not heard anything from you. If you do not respond in the next few seconds, I will have to end this call.",
-            disconnectSilenceSeconds: 30,
-          },
-        },
-      });
-      const phoneData = createPhoneNumber({
-        flow_id: flow._id,
-        client_id: "test-client-id",
-      });
-
-      await getFlowsCollection().insertOne(flow);
-      await getPhoneNumbersCollection().insertOne(phoneData);
-
-      const result = await getPhoneDataByPhoneNumber(phoneData.phone_number);
-
-      expect(result).toBeDefined();
-      expect(result).toMatchObject({
-        client_id: phoneData.client_id,
-        phone_number: phoneData.phone_number,
-        flow_id: flow._id,
-      });
-      // Specifically check if the flow was joined correctly
-      expect(result?.flow.flowName).toBe(flow.flowName);
-      expect(
-        result?.flow.conversationSettings?.silence?.firstWarningSilenceSeconds,
-      ).toBe(5);
-    });
-  });
-
-  describe("getClientPhoneNumber", () => {
-    it("return phone number by client id", async () => {
-      const flow = createFlow();
-      const phoneData = createPhoneNumber({
-        flow_id: flow._id,
-        is_primary: true,
-      });
-
-      await getFlowsCollection().insertOne(flow);
-      await getPhoneNumbersCollection().insertOne(phoneData);
-
-      const result = await getClientPhoneNumber(phoneData.client_id);
-      expect(result).toBe(phoneData.phone_number);
-    });
-  });
-
-  describe("getClientPhoneData", () => {
-    it("returns primary phone data by default", async () => {
-      const clientId = "multiPhoneClient";
-      const phone1 = createPhoneNumber({
-        client_id: clientId,
-        is_primary: false,
-        phone_number: "1",
-        createdAt: new Date("2023-01-01"),
-      });
-      const phone2 = createPhoneNumber({
-        client_id: clientId,
-        is_primary: true,
-        phone_number: "2",
-        createdAt: new Date("2023-01-02"),
-      });
-
-      await getPhoneNumbersCollection().insertOne(phone1);
-      await getPhoneNumbersCollection().insertOne(phone2);
-
-      const result = await getClientPhoneData(clientId);
-      expect(result?.phone_number).toBe("2");
-    });
-
-    it("returns most recent non-primary phone data if isPrimary is false", async () => {
-      const clientId = "multiPhoneClient2";
-      const phone1 = createPhoneNumber({
-        client_id: clientId,
-        is_primary: false,
-        phone_number: "1",
-        createdAt: new Date("2023-01-01"),
-      });
-      const phone2 = createPhoneNumber({
-        client_id: clientId,
-        is_primary: false,
-        phone_number: "2",
-        createdAt: new Date("2023-01-02"),
-      });
-      const phone3 = createPhoneNumber({
-        client_id: clientId,
-        is_primary: true,
-        phone_number: "3",
-        createdAt: new Date("2023-01-03"),
-      });
-
-      await getPhoneNumbersCollection().insertOne(phone1);
-      await getPhoneNumbersCollection().insertOne(phone2);
-      await getPhoneNumbersCollection().insertOne(phone3);
-
-      const result = await getClientPhoneData(clientId, false);
-      expect(result?.phone_number).toBe("2");
-    });
-  });
-
-  describe('getPhoneNumbersForFlows', () => {
-    it('returns all phone numbers for the client with flow ObjectIds, newest first', async () => {
-      const clientId = 'flowsClient';
-      const flow1 = createFlow({ clientId });
-      const flow2 = createFlow({ clientId });
-      await getFlowsCollection().insertMany([flow1, flow2]);
-
-      const older = createPhoneNumber({
-        client_id: clientId,
-        flow_id: flow1._id,
-        phone_number: '+100',
-        is_primary: true,
-        createdAt: new Date('2023-01-01'),
-      });
-      const newer = createPhoneNumber({
-        client_id: clientId,
-        flow_id: flow2._id,
-        phone_number: '+200',
-        is_primary: false,
-        createdAt: new Date('2023-06-01'),
-      });
-
-      await getPhoneNumbersCollection().insertMany([older, newer]);
-
-      const result = await getPhoneNumbersForFlows(clientId);
-
-      expect(result).toHaveLength(2);
-      expect(result[0]).toEqual({
-        flowId: String(flow2._id),
-        phoneNumber: '+200',
-        isPrimary: false,
-      });
-      expect(result[1]).toEqual({
-        flowId: String(flow1._id),
-        phoneNumber: '+100',
-        isPrimary: true,
-      });
-    });
-
-    it('returns an empty array when the client has no phone numbers', async () => {
-      const result = await getPhoneNumbersForFlows('noSuchClient');
-      expect(result).toEqual([]);
-    });
-
-    it('does not include phone numbers for other clients', async () => {
-      const clientA = 'clientA';
-      const clientB = 'clientB';
-      const flow = createFlow();
-      await getFlowsCollection().insertOne(flow);
-      const phone = createPhoneNumber({ client_id: clientA, flow_id: flow._id });
-      await getPhoneNumbersCollection().insertOne(phone);
-
-      const result = await getPhoneNumbersForFlows(clientB);
-      expect(result).toEqual([]);
-    });
-  });
-
-  describe('createPhoneNumberEntity', () => {
-    it('creates first phone number as primary', async () => {
-      const clientId = 'newClient';
-      const flowId = new ObjectId().toHexString();
-      const phoneNumber = "+123456789";
-
-      const result = await createPhoneNumberEntity(
-        phoneNumber,
-        flowId,
-        clientId,
-      );
-      expect(result.phone_number).toBe(phoneNumber);
-      expect(result.is_primary).toBe(true);
-    });
-
-    it("creates subsequent phone number as non-primary", async () => {
-      const clientId = "existingClient";
-      const flowId = new ObjectId().toHexString();
-
-      // Create first one
-      await createPhoneNumberEntity("+111", flowId, clientId);
-
-      // Create second one
-      const result = await createPhoneNumberEntity("+222", flowId, clientId);
-      expect(result.phone_number).toBe("+222");
-      expect(result.is_primary).toBe(false);
-    });
-  });
-
-  describe('createPurchasedPhoneNumber', () => {
-    it('persists a phone_numbers document with provider payload and flow', async () => {
-      const clientId = 'purchasedClientDb';
-      const flowId = new ObjectId().toHexString();
-      const phoneNumber = '+15551234567';
-      const providerPayload = {
-        provider: 'twilio' as const,
-        provider_sid: 'PNxxxxxxxx',
-      };
-
-      await createPurchasedPhoneNumber(
-        phoneNumber,
-        providerPayload,
-        clientId,
-        flowId,
-      );
-
-      // Re-read from Mongo to confirm insert succeeded (not only the function return value).
-      const stored = await getPhoneNumbersCollection().findOne({ phone_number: phoneNumber });
-      expect(stored).not.toBeNull();
-      expect(stored).toMatchObject({
-        phone_number: phoneNumber,
-        client_id: clientId,
-        provider: 'twilio',
-        provider_sid: 'PNxxxxxxxx',
-        flow_id: new ObjectId(flowId),
-      });
-    });
-  });
-});
+import {
+  getPhoneDataByPhoneNumber,
+  getPhoneNumbersCollection,
+  getPhoneNumbersForFlows,
+  getClientPhoneNumber,
+  getClientPhoneData,
+  createPhoneNumberEntity,
+  createPurchasedPhoneNumber,
+} from '../phone_numbers.getter';
+import { getFlowsCollection } from '../../flows/flows.getter';
+import { createFlow, createPhoneNumber } from '../../../test-utils/factories';
+import { ObjectId } from 'mongodb';
+
+describe('db.phoneNumbers', () => {
+  describe('getPhoneDataByPhoneNumber', () => {
+    it('return phone number data with flow', async () => {
+      const flow = createFlow({
+        clientId: "test-client-id",
+        conversationSettings: {
+          interruptions: {
+            enableInterruptionDetection: true,
+            interruptionWindowSeconds: 60,
+            interruptionThresholdSeconds: 10,
+            interruptionInstruction:
+              "I am hearing background noise that makes it hard for me to focus on what you are saying. Please try to move to a quieter place or reduce the noise so I can better assist you.",
+          },
+          silence: {
+            enableSilenceDetection: true,
+            firstWarningSilenceSeconds: 5,
+            firstWarningInstruction:
+              "I have not heard you for a few seconds. If you are still on the line, please say something so we can continue.",
+            secondWarningSilenceSeconds: 10,
+            secondWarningInstruction:
+              "I still have not heard anything from you. If you do not respond in the next few seconds, I will have to end this call.",
+            disconnectSilenceSeconds: 30,
+          },
+        },
+      });
+      const phoneData = createPhoneNumber({
+        flow_id: flow._id,
+        client_id: "test-client-id",
+      });
+
+      await getFlowsCollection().insertOne(flow);
+      await getPhoneNumbersCollection().insertOne(phoneData);
+
+      const result = await getPhoneDataByPhoneNumber(phoneData.phone_number);
+
+      expect(result).toBeDefined();
+      expect(result).toMatchObject({
+        client_id: phoneData.client_id,
+        phone_number: phoneData.phone_number,
+        flow_id: flow._id,
+      });
+      // Specifically check if the flow was joined correctly
+      expect(result?.flow.flowName).toBe(flow.flowName);
+      expect(
+        result?.flow.conversationSettings?.silence?.firstWarningSilenceSeconds,
+      ).toBe(5);
+    });
+  });
+
+  describe("getClientPhoneNumber", () => {
+    it("return phone number by client id", async () => {
+      const flow = createFlow();
+      const phoneData = createPhoneNumber({
+        flow_id: flow._id,
+        is_primary: true,
+      });
+
+      await getFlowsCollection().insertOne(flow);
+      await getPhoneNumbersCollection().insertOne(phoneData);
+
+      const result = await getClientPhoneNumber(phoneData.client_id);
+      expect(result).toBe(phoneData.phone_number);
+    });
+  });
+
+  describe("getClientPhoneData", () => {
+    it("returns primary phone data by default", async () => {
+      const clientId = "multiPhoneClient";
+      const phone1 = createPhoneNumber({
+        client_id: clientId,
+        is_primary: false,
+        phone_number: "1",
+        createdAt: new Date("2023-01-01"),
+      });
+      const phone2 = createPhoneNumber({
+        client_id: clientId,
+        is_primary: true,
+        phone_number: "2",
+        createdAt: new Date("2023-01-02"),
+      });
+
+      await getPhoneNumbersCollection().insertOne(phone1);
+      await getPhoneNumbersCollection().insertOne(phone2);
+
+      const result = await getClientPhoneData(clientId);
+      expect(result?.phone_number).toBe("2");
+    });
+
+    it("returns most recent non-primary phone data if isPrimary is false", async () => {
+      const clientId = "multiPhoneClient2";
+      const phone1 = createPhoneNumber({
+        client_id: clientId,
+        is_primary: false,
+        phone_number: "1",
+        createdAt: new Date("2023-01-01"),
+      });
+      const phone2 = createPhoneNumber({
+        client_id: clientId,
+        is_primary: false,
+        phone_number: "2",
+        createdAt: new Date("2023-01-02"),
+      });
+      const phone3 = createPhoneNumber({
+        client_id: clientId,
+        is_primary: true,
+        phone_number: "3",
+        createdAt: new Date("2023-01-03"),
+      });
+
+      await getPhoneNumbersCollection().insertOne(phone1);
+      await getPhoneNumbersCollection().insertOne(phone2);
+      await getPhoneNumbersCollection().insertOne(phone3);
+
+      const result = await getClientPhoneData(clientId, false);
+      expect(result?.phone_number).toBe("2");
+    });
+  });
+
+  describe('getPhoneNumbersForFlows', () => {
+    it('returns all phone numbers for the client with flow ObjectIds, newest first', async () => {
+      const clientId = 'flowsClient';
+      const flow1 = createFlow({ clientId });
+      const flow2 = createFlow({ clientId });
+      await getFlowsCollection().insertMany([flow1, flow2]);
+
+      const older = createPhoneNumber({
+        client_id: clientId,
+        flow_id: flow1._id,
+        phone_number: '+100',
+        is_primary: true,
+        createdAt: new Date('2023-01-01'),
+      });
+      const newer = createPhoneNumber({
+        client_id: clientId,
+        flow_id: flow2._id,
+        phone_number: '+200',
+        is_primary: false,
+        createdAt: new Date('2023-06-01'),
+      });
+
+      await getPhoneNumbersCollection().insertMany([older, newer]);
+
+      const result = await getPhoneNumbersForFlows(clientId);
+
+      expect(result).toHaveLength(2);
+      expect(result[0]).toEqual({
+        flowId: String(flow2._id),
+        phoneNumber: '+200',
+        isPrimary: false,
+      });
+      expect(result[1]).toEqual({
+        flowId: String(flow1._id),
+        phoneNumber: '+100',
+        isPrimary: true,
+      });
+    });
+
+    it('returns an empty array when the client has no phone numbers', async () => {
+      const result = await getPhoneNumbersForFlows('noSuchClient');
+      expect(result).toEqual([]);
+    });
+
+    it('does not include phone numbers for other clients', async () => {
+      const clientA = 'clientA';
+      const clientB = 'clientB';
+      const flow = createFlow();
+      await getFlowsCollection().insertOne(flow);
+      const phone = createPhoneNumber({ client_id: clientA, flow_id: flow._id });
+      await getPhoneNumbersCollection().insertOne(phone);
+
+      const result = await getPhoneNumbersForFlows(clientB);
+      expect(result).toEqual([]);
+    });
+  });
+
+  describe('createPhoneNumberEntity', () => {
+    it('creates first phone number as primary', async () => {
+      const clientId = 'newClient';
+      const flowId = new ObjectId().toHexString();
+      const phoneNumber = "+123456789";
+
+      const result = await createPhoneNumberEntity(
+        phoneNumber,
+        flowId,
+        clientId,
+      );
+      expect(result.phone_number).toBe(phoneNumber);
+      expect(result.is_primary).toBe(true);
+    });
+
+    it("creates subsequent phone number as non-primary", async () => {
+      const clientId = "existingClient";
+      const flowId = new ObjectId().toHexString();
+
+      // Create first one
+      await createPhoneNumberEntity("+111", flowId, clientId);
+
+      // Create second one
+      const result = await createPhoneNumberEntity("+222", flowId, clientId);
+      expect(result.phone_number).toBe("+222");
+      expect(result.is_primary).toBe(false);
+    });
+  });
+
+  describe('createPurchasedPhoneNumber', () => {
+    it('persists a phone_numbers document with provider payload and flow', async () => {
+      const clientId = 'purchasedClientDb';
+      const flowId = new ObjectId().toHexString();
+      const phoneNumber = '+15551234567';
+      const providerPayload = {
+        provider: 'twilio' as const,
+        provider_sid: 'PNxxxxxxxx',
+      };
+
+      await createPurchasedPhoneNumber(
+        phoneNumber,
+        providerPayload,
+        clientId,
+        flowId,
+      );
+
+      // Re-read from Mongo to confirm insert succeeded (not only the function return value).
+      const stored = await getPhoneNumbersCollection().findOne({ phone_number: phoneNumber });
+      expect(stored).not.toBeNull();
+      expect(stored).toMatchObject({
+        phone_number: phoneNumber,
+        client_id: clientId,
+        provider: 'twilio',
+        provider_sid: 'PNxxxxxxxx',
+        flow_id: new ObjectId(flowId),
+      });
+    });
+  });
+});

package/src/talkpilot/phone_numbers/index.ts

@@ -1,2 +1,2 @@
-export * from './phone_numbers.getter';
-export type * from './phone_numbers.types';
+export * from './phone_numbers.getter';
+export type * from './phone_numbers.types';