@talkpilot/core-db 1.1.18 → 1.2.0

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
Files changed (133) hide show
  1. package/.cursor/rules/development.mdc +65 -65
  2. package/DEVELOPMENT.md +98 -98
  3. package/README.md +160 -160
  4. package/dist/talkpilot/calls/calls.getters.d.ts +2 -1
  5. package/dist/talkpilot/calls/calls.getters.d.ts.map +1 -1
  6. package/dist/talkpilot/calls/calls.getters.js +176 -0
  7. package/dist/talkpilot/calls/calls.getters.js.map +1 -1
  8. package/dist/talkpilot/calls/calls.types.d.ts +49 -2
  9. package/dist/talkpilot/calls/calls.types.d.ts.map +1 -1
  10. package/dist/talkpilot/calls/calls.types.js.map +1 -1
  11. package/dist/talkpilot/clientsConfig/clientsConfig.getters.d.ts +1 -0
  12. package/dist/talkpilot/clientsConfig/clientsConfig.getters.d.ts.map +1 -1
  13. package/dist/talkpilot/clientsConfig/clientsConfig.getters.js +13 -0
  14. package/dist/talkpilot/clientsConfig/clientsConfig.getters.js.map +1 -1
  15. package/dist/talkpilot/clientsConfig/clientsConfig.types.d.ts +2 -0
  16. package/dist/talkpilot/clientsConfig/clientsConfig.types.d.ts.map +1 -1
  17. package/jest.config.js +19 -19
  18. package/package.json +45 -45
  19. package/src/__tests__/setup.ts +20 -20
  20. package/src/connection.ts +42 -42
  21. package/src/index.ts +16 -16
  22. package/src/municipal/__tests__/validation.spec.ts +62 -62
  23. package/src/municipal/cities/cities.getters.ts +50 -50
  24. package/src/municipal/cities/cities.types.ts +11 -11
  25. package/src/municipal/cities/index.ts +2 -2
  26. package/src/municipal/departmentsSubjects/departmentsSubjects.getters.ts +282 -282
  27. package/src/municipal/departmentsSubjects/departmentsSubjects.types.ts +72 -72
  28. package/src/municipal/departmentsSubjects/index.ts +9 -9
  29. package/src/municipal/index.ts +21 -21
  30. package/src/municipal/mongodb-client.ts +61 -61
  31. package/src/municipal/streets/index.ts +2 -2
  32. package/src/municipal/streets/streets.getters.ts +125 -125
  33. package/src/municipal/streets/streets.types.ts +18 -18
  34. package/src/municipal/systemInstructions/__tests__/getters.spec.ts +113 -113
  35. package/src/municipal/systemInstructions/__tests__/setters.spec.ts +274 -274
  36. package/src/municipal/systemInstructions/index.ts +7 -7
  37. package/src/municipal/systemInstructions/instructions.getters.ts +57 -57
  38. package/src/municipal/systemInstructions/instructions.setters.ts +119 -119
  39. package/src/municipal/systemInstructions/instructions.types.ts +30 -30
  40. package/src/municipal/tickets/__tests__/tickets.getters.spec.ts +66 -66
  41. package/src/municipal/tickets/index.ts +2 -2
  42. package/src/municipal/tickets/tickets.getters.ts +261 -261
  43. package/src/municipal/tickets/tickets.types.ts +43 -43
  44. package/src/municipal/utils/types.ts +11 -11
  45. package/src/talkpilot/__tests__/db.spec.ts +38 -38
  46. package/src/talkpilot/__tests__/mongodb-client.spec.ts +18 -18
  47. package/src/talkpilot/__tests__/validation.spec.ts +68 -68
  48. package/src/talkpilot/agents/__tests__/agents.getters.spec.ts +29 -29
  49. package/src/talkpilot/agents/agents.getters.ts +34 -34
  50. package/src/talkpilot/agents/agents.types.ts +14 -14
  51. package/src/talkpilot/agents/index.ts +2 -2
  52. package/src/talkpilot/backgroundToolResults/__tests__/backgroundToolResults.getters.spec.ts +147 -147
  53. package/src/talkpilot/backgroundToolResults/backgroundToolResults.getters.ts +65 -65
  54. package/src/talkpilot/backgroundToolResults/backgroundToolResults.types.ts +23 -23
  55. package/src/talkpilot/backgroundToolResults/index.ts +2 -2
  56. package/src/talkpilot/calls/__tests__/callStats.utils.spec.ts +128 -128
  57. package/src/talkpilot/calls/__tests__/calls.spec.ts +252 -252
  58. package/src/talkpilot/calls/calls.getters.ts +446 -248
  59. package/src/talkpilot/calls/calls.types.ts +171 -116
  60. package/src/talkpilot/calls/index.ts +2 -2
  61. package/src/talkpilot/clientAudioBuffers/__tests__/clientAudioBuffer.getters.spec.ts +160 -160
  62. package/src/talkpilot/clientAudioBuffers/clientAudioBuffer.getters.ts +117 -117
  63. package/src/talkpilot/clientAudioBuffers/clientsAudioBuffers.types.ts +25 -25
  64. package/src/talkpilot/clientAudioBuffers/index.ts +2 -2
  65. package/src/talkpilot/clients/clients.getters.ts +16 -16
  66. package/src/talkpilot/clients/clients.types.ts +14 -14
  67. package/src/talkpilot/clients/index.ts +2 -2
  68. package/src/talkpilot/clientsConfig/__tests__/clientsConfig.spec.ts +106 -106
  69. package/src/talkpilot/clientsConfig/clientsConfig.getters.ts +44 -22
  70. package/src/talkpilot/clientsConfig/clientsConfig.types.ts +94 -92
  71. package/src/talkpilot/clientsConfig/index.ts +2 -2
  72. package/src/talkpilot/flows/__tests__/flows.schema.spec.ts +67 -67
  73. package/src/talkpilot/flows/flows.getter.ts +14 -14
  74. package/src/talkpilot/flows/flows.schema.ts +153 -153
  75. package/src/talkpilot/flows/flows.types.ts +184 -184
  76. package/src/talkpilot/flows/index.ts +2 -2
  77. package/src/talkpilot/groups/__tests__/groups.spec.ts +90 -90
  78. package/src/talkpilot/groups/__tests__/phone.utils.spec.ts +32 -32
  79. package/src/talkpilot/groups/groups.getters.ts +30 -30
  80. package/src/talkpilot/groups/groups.types.ts +29 -29
  81. package/src/talkpilot/groups/index.ts +3 -3
  82. package/src/talkpilot/groups/phone.utils.ts +46 -46
  83. package/src/talkpilot/index.ts +29 -29
  84. package/src/talkpilot/leads/index.ts +2 -2
  85. package/src/talkpilot/leads/leads.getter.ts +6 -6
  86. package/src/talkpilot/leads/leads.schema.ts +33 -33
  87. package/src/talkpilot/leads/leads.types.ts +20 -20
  88. package/src/talkpilot/mongodb-client.ts +78 -78
  89. package/src/talkpilot/phone_numbers/__tests__/phone_numbers.spec.ts +247 -247
  90. package/src/talkpilot/phone_numbers/index.ts +2 -2
  91. package/src/talkpilot/phone_numbers/phone_numbers.getter.ts +154 -154
  92. package/src/talkpilot/phone_numbers/phone_numbers.schema.ts +17 -17
  93. package/src/talkpilot/phone_numbers/phone_numbers.types.ts +30 -30
  94. package/src/talkpilot/plans/__tests__/plans.spec.ts +70 -70
  95. package/src/talkpilot/plans/index.ts +2 -2
  96. package/src/talkpilot/plans/plans.getters.ts +132 -132
  97. package/src/talkpilot/plans/plans.types.ts +89 -89
  98. package/src/talkpilot/results/index.ts +7 -7
  99. package/src/talkpilot/results/results.getter.ts +35 -35
  100. package/src/talkpilot/results/results.schema.ts +25 -25
  101. package/src/talkpilot/results/results.types.ts +34 -34
  102. package/src/talkpilot/retry_analyze/__tests__/retryAnalyze.getters.spec.ts +156 -156
  103. package/src/talkpilot/retry_analyze/index.ts +2 -2
  104. package/src/talkpilot/retry_analyze/retryAnalyze.getters.ts +75 -75
  105. package/src/talkpilot/retry_analyze/retryAnalyze.types.ts +13 -13
  106. package/src/talkpilot/sessions/__tests__/sessions.spec.ts +147 -147
  107. package/src/talkpilot/sessions/index.ts +2 -2
  108. package/src/talkpilot/sessions/sessions.getter.ts +92 -92
  109. package/src/talkpilot/sessions/sessions.schema.ts +34 -34
  110. package/src/talkpilot/sessions/sessions.types.ts +30 -30
  111. package/src/talkpilot/subscriptions/__tests__/subscriptions.getters.utils.spec.ts +45 -45
  112. package/src/talkpilot/subscriptions/index.ts +3 -3
  113. package/src/talkpilot/subscriptions/subscriptions.getters.ts +146 -146
  114. package/src/talkpilot/subscriptions/subscriptions.getters.utils.ts +33 -33
  115. package/src/talkpilot/subscriptions/subscriptions.types.ts +66 -66
  116. package/src/talkpilot/utils/__tests__/query.utils.spec.ts +49 -49
  117. package/src/talkpilot/utils/query.utils.ts +21 -21
  118. package/src/test-utils/db-utils.ts +24 -24
  119. package/src/test-utils/factories/index.ts +12 -12
  120. package/src/test-utils/factories/municipal/cities.ts +16 -16
  121. package/src/test-utils/factories/municipal/departmentsSubjects.ts +37 -37
  122. package/src/test-utils/factories/municipal/streets.ts +22 -22
  123. package/src/test-utils/factories/municipal/tickets.ts +39 -39
  124. package/src/test-utils/factories/talkpilot/agents.ts +19 -19
  125. package/src/test-utils/factories/talkpilot/calls.ts +37 -37
  126. package/src/test-utils/factories/talkpilot/clientAudioBuffers.ts +20 -20
  127. package/src/test-utils/factories/talkpilot/clientsConfig.ts +18 -18
  128. package/src/test-utils/factories/talkpilot/flows.ts +33 -33
  129. package/src/test-utils/factories/talkpilot/groups.ts +33 -33
  130. package/src/test-utils/factories/talkpilot/phone_numbers.ts +22 -22
  131. package/src/test-utils/factories/talkpilot/sessions.ts +35 -35
  132. package/src/utils/validation.ts +23 -23
  133. package/tsconfig.json +23 -23
@@ -1,30 +1,30 @@
1
- import { getDb, ObjectId, Group } from "../index";
2
- import { Collection, Filter } from "mongodb";
3
-
4
- export const getGroupsCollection = (): Collection<Group> => {
5
- return getDb().collection<Group>("groups");
6
- };
7
-
8
- export const findGroups = async (filter: Filter<Group>): Promise<Group[]> => {
9
- return await getGroupsCollection().find(filter).toArray();
10
- };
11
-
12
- export const createGroup = async (group: Group) => {
13
- const { insertedId } = await getGroupsCollection().insertOne(group);
14
- return insertedId;
15
- };
16
-
17
- export const updateGroup = async (
18
- filter: Filter<Group>,
19
- data: Partial<Group>,
20
- ) => {
21
- return await getGroupsCollection().findOneAndUpdate(
22
- filter,
23
- { $set: { ...data, updatedAt: new Date() } },
24
- { returnDocument: "after" },
25
- );
26
- };
27
-
28
- export const removeGroup = async (groupId: string) => {
29
- return await getGroupsCollection().deleteOne({ _id: new ObjectId(groupId) });
30
- };
1
+ import { getDb, ObjectId, Group } from "../index";
2
+ import { Collection, Filter } from "mongodb";
3
+
4
+ export const getGroupsCollection = (): Collection<Group> => {
5
+ return getDb().collection<Group>("groups");
6
+ };
7
+
8
+ export const findGroups = async (filter: Filter<Group>): Promise<Group[]> => {
9
+ return await getGroupsCollection().find(filter).toArray();
10
+ };
11
+
12
+ export const createGroup = async (group: Group) => {
13
+ const { insertedId } = await getGroupsCollection().insertOne(group);
14
+ return insertedId;
15
+ };
16
+
17
+ export const updateGroup = async (
18
+ filter: Filter<Group>,
19
+ data: Partial<Group>,
20
+ ) => {
21
+ return await getGroupsCollection().findOneAndUpdate(
22
+ filter,
23
+ { $set: { ...data, updatedAt: new Date() } },
24
+ { returnDocument: "after" },
25
+ );
26
+ };
27
+
28
+ export const removeGroup = async (groupId: string) => {
29
+ return await getGroupsCollection().deleteOne({ _id: new ObjectId(groupId) });
30
+ };
@@ -1,29 +1,29 @@
1
- import { WithId } from "mongodb";
2
- export type E164 = `+${string}`;
3
- export type CountryISO = Uppercase<string>;
4
-
5
- export type Phone = {
6
- e164: E164;
7
- region?: CountryISO;
8
- countryCallingCode?: string;
9
- national?: string;
10
- ext?: string;
11
- formatted?: string;
12
- };
13
-
14
- export type GroupContact = {
15
- name: string;
16
- phone: Phone;
17
- gender: string;
18
- };
19
-
20
- export type Group = {
21
- clientId: string;
22
- createdAt: Date;
23
- updatedAt: Date;
24
- name: string;
25
- description: string;
26
- contacts: GroupContact[];
27
- };
28
-
29
- export type GroupDoc = WithId<Group>;
1
+ import { WithId } from "mongodb";
2
+ export type E164 = `+${string}`;
3
+ export type CountryISO = Uppercase<string>;
4
+
5
+ export type Phone = {
6
+ e164: E164;
7
+ region?: CountryISO;
8
+ countryCallingCode?: string;
9
+ national?: string;
10
+ ext?: string;
11
+ formatted?: string;
12
+ };
13
+
14
+ export type GroupContact = {
15
+ name: string;
16
+ phone: Phone;
17
+ gender: string;
18
+ };
19
+
20
+ export type Group = {
21
+ clientId: string;
22
+ createdAt: Date;
23
+ updatedAt: Date;
24
+ name: string;
25
+ description: string;
26
+ contacts: GroupContact[];
27
+ };
28
+
29
+ export type GroupDoc = WithId<Group>;
@@ -1,3 +1,3 @@
1
- export * from "./groups.getters";
2
- export * from "./groups.types";
3
- export * from "./phone.utils";
1
+ export * from "./groups.getters";
2
+ export * from "./groups.types";
3
+ export * from "./phone.utils";
@@ -1,46 +1,46 @@
1
- import { PhoneNumberUtil, PhoneNumberFormat } from "google-libphonenumber";
2
- import type { Phone, E164, CountryISO } from "./groups.types";
3
-
4
- const phoneUtil = PhoneNumberUtil.getInstance();
5
-
6
- export const enrichPhoneNumber = (e164: string): Phone | null => {
7
- try {
8
- const phoneNumber = phoneUtil.parse(e164, "ZZ");
9
- if (!phoneUtil.isValidNumber(phoneNumber)) {
10
- return null;
11
- }
12
-
13
- const normalizedE164 = phoneUtil.format(
14
- phoneNumber,
15
- PhoneNumberFormat.E164,
16
- ) as E164;
17
- const region = phoneUtil.getRegionCodeForNumber(phoneNumber) as CountryISO;
18
- const countryCallingCode = phoneNumber.getCountryCode()?.toString();
19
- const national = phoneNumber.getNationalNumber()?.toString();
20
- const ext = phoneNumber.getExtension() || undefined;
21
- const formatted = phoneUtil.format(
22
- phoneNumber,
23
- PhoneNumberFormat.INTERNATIONAL,
24
- );
25
-
26
- return {
27
- e164: normalizedE164,
28
- region,
29
- countryCallingCode,
30
- national,
31
- ext,
32
- formatted,
33
- };
34
- } catch (error) {
35
- return null;
36
- }
37
- };
38
-
39
- export const isValidE164 = (e164: string): boolean => {
40
- try {
41
- const phoneNumber = phoneUtil.parse(e164, "ZZ");
42
- return phoneUtil.isValidNumber(phoneNumber);
43
- } catch (error) {
44
- return false;
45
- }
46
- };
1
+ import { PhoneNumberUtil, PhoneNumberFormat } from "google-libphonenumber";
2
+ import type { Phone, E164, CountryISO } from "./groups.types";
3
+
4
+ const phoneUtil = PhoneNumberUtil.getInstance();
5
+
6
+ export const enrichPhoneNumber = (e164: string): Phone | null => {
7
+ try {
8
+ const phoneNumber = phoneUtil.parse(e164, "ZZ");
9
+ if (!phoneUtil.isValidNumber(phoneNumber)) {
10
+ return null;
11
+ }
12
+
13
+ const normalizedE164 = phoneUtil.format(
14
+ phoneNumber,
15
+ PhoneNumberFormat.E164,
16
+ ) as E164;
17
+ const region = phoneUtil.getRegionCodeForNumber(phoneNumber) as CountryISO;
18
+ const countryCallingCode = phoneNumber.getCountryCode()?.toString();
19
+ const national = phoneNumber.getNationalNumber()?.toString();
20
+ const ext = phoneNumber.getExtension() || undefined;
21
+ const formatted = phoneUtil.format(
22
+ phoneNumber,
23
+ PhoneNumberFormat.INTERNATIONAL,
24
+ );
25
+
26
+ return {
27
+ e164: normalizedE164,
28
+ region,
29
+ countryCallingCode,
30
+ national,
31
+ ext,
32
+ formatted,
33
+ };
34
+ } catch (error) {
35
+ return null;
36
+ }
37
+ };
38
+
39
+ export const isValidE164 = (e164: string): boolean => {
40
+ try {
41
+ const phoneNumber = phoneUtil.parse(e164, "ZZ");
42
+ return phoneUtil.isValidNumber(phoneNumber);
43
+ } catch (error) {
44
+ return false;
45
+ }
46
+ };
@@ -1,29 +1,29 @@
1
- import { Db, ObjectId as MongoObjectId } from "mongodb";
2
-
3
- export * from "./flows";
4
- export * from "./results";
5
- export * from "./sessions";
6
- export * from "./phone_numbers";
7
- export * from "./leads";
8
- export * from "./clients";
9
- export * from "./agents";
10
- export * from "./plans";
11
- export * from "./subscriptions";
12
- export * from "./groups";
13
- export * from "./calls";
14
- export * from "./clientAudioBuffers";
15
- export * from "./clientsConfig";
16
- export * from "./backgroundToolResults";
17
- export * from "./retry_analyze";
18
- export { mongodbClient } from "./mongodb-client";
19
-
20
- let db: Db;
21
- export const setDb = (d: Db) => {
22
- db = d;
23
- };
24
- export const getDb = (): Db => {
25
- if (!db) throw new Error("DB not initialised");
26
- return db;
27
- };
28
-
29
- export const ObjectId = MongoObjectId;
1
+ import { Db, ObjectId as MongoObjectId } from "mongodb";
2
+
3
+ export * from "./flows";
4
+ export * from "./results";
5
+ export * from "./sessions";
6
+ export * from "./phone_numbers";
7
+ export * from "./leads";
8
+ export * from "./clients";
9
+ export * from "./agents";
10
+ export * from "./plans";
11
+ export * from "./subscriptions";
12
+ export * from "./groups";
13
+ export * from "./calls";
14
+ export * from "./clientAudioBuffers";
15
+ export * from "./clientsConfig";
16
+ export * from "./backgroundToolResults";
17
+ export * from "./retry_analyze";
18
+ export { mongodbClient } from "./mongodb-client";
19
+
20
+ let db: Db;
21
+ export const setDb = (d: Db) => {
22
+ db = d;
23
+ };
24
+ export const getDb = (): Db => {
25
+ if (!db) throw new Error("DB not initialised");
26
+ return db;
27
+ };
28
+
29
+ export const ObjectId = MongoObjectId;
@@ -1,2 +1,2 @@
1
- export { getLeadsCollection } from "./leads.getter";
2
- export type { Lead, LeadItem, LeadProperty } from "./leads.types";
1
+ export { getLeadsCollection } from "./leads.getter";
2
+ export type { Lead, LeadItem, LeadProperty } from "./leads.types";
@@ -1,6 +1,6 @@
1
- import type { Collection } from "mongodb";
2
- import { getDb } from "../index";
3
- import type { Lead } from "./leads.types";
4
-
5
- export const getLeadsCollection = (): Collection<Lead> =>
6
- getDb().collection<Lead>("leads");
1
+ import type { Collection } from "mongodb";
2
+ import { getDb } from "../index";
3
+ import type { Lead } from "./leads.types";
4
+
5
+ export const getLeadsCollection = (): Collection<Lead> =>
6
+ getDb().collection<Lead>("leads");
@@ -1,33 +1,33 @@
1
- export const leadsMongoSchema = {
2
- bsonType: "object",
3
- required: ["flowId", "clientId", "createdAt", "leads"],
4
- properties: {
5
- _id: { bsonType: "objectId" },
6
- flowId: { bsonType: "string" },
7
- clientId: { bsonType: "string" },
8
- createdAt: { bsonType: "date" },
9
- leads: {
10
- bsonType: "array",
11
- items: {
12
- bsonType: "object",
13
- required: ["callSid", "timestamp"],
14
- properties: {
15
- callSid: { bsonType: "string" },
16
- timestamp: { bsonType: "date" },
17
- properties: {
18
- bsonType: "array",
19
- items: {
20
- bsonType: "object",
21
- required: ["name", "value"],
22
- properties: {
23
- name: { bsonType: "string" },
24
- value: { bsonType: "string" },
25
- },
26
- },
27
- },
28
- },
29
- },
30
- },
31
- },
32
- additionalProperties: false,
33
- } as const;
1
+ export const leadsMongoSchema = {
2
+ bsonType: "object",
3
+ required: ["flowId", "clientId", "createdAt", "leads"],
4
+ properties: {
5
+ _id: { bsonType: "objectId" },
6
+ flowId: { bsonType: "string" },
7
+ clientId: { bsonType: "string" },
8
+ createdAt: { bsonType: "date" },
9
+ leads: {
10
+ bsonType: "array",
11
+ items: {
12
+ bsonType: "object",
13
+ required: ["callSid", "timestamp"],
14
+ properties: {
15
+ callSid: { bsonType: "string" },
16
+ timestamp: { bsonType: "date" },
17
+ properties: {
18
+ bsonType: "array",
19
+ items: {
20
+ bsonType: "object",
21
+ required: ["name", "value"],
22
+ properties: {
23
+ name: { bsonType: "string" },
24
+ value: { bsonType: "string" },
25
+ },
26
+ },
27
+ },
28
+ },
29
+ },
30
+ },
31
+ },
32
+ additionalProperties: false,
33
+ } as const;
@@ -1,20 +1,20 @@
1
- import { ObjectId } from "mongodb";
2
-
3
- export interface LeadProperty {
4
- name: string;
5
- value: string;
6
- }
7
-
8
- export interface LeadItem {
9
- callSid: string;
10
- timestamp: Date;
11
- properties?: LeadProperty[];
12
- }
13
-
14
- export interface Lead {
15
- _id?: ObjectId;
16
- flowId: ObjectId;
17
- clientId: string;
18
- createdAt: Date;
19
- leads: LeadItem[];
20
- }
1
+ import { ObjectId } from "mongodb";
2
+
3
+ export interface LeadProperty {
4
+ name: string;
5
+ value: string;
6
+ }
7
+
8
+ export interface LeadItem {
9
+ callSid: string;
10
+ timestamp: Date;
11
+ properties?: LeadProperty[];
12
+ }
13
+
14
+ export interface Lead {
15
+ _id?: ObjectId;
16
+ flowId: ObjectId;
17
+ clientId: string;
18
+ createdAt: Date;
19
+ leads: LeadItem[];
20
+ }
@@ -1,78 +1,78 @@
1
- import { MongoClient, Db } from "mongodb";
2
- import { setDb } from "./index";
3
- import { validateConfig, validateMongoUri } from "../utils/validation";
4
-
5
- class MongoDBClient {
6
- private client: MongoClient | null = null;
7
- private db: Db | null = null;
8
-
9
- private extractDatabaseName(uri: string): string | null {
10
- try {
11
- const url = new URL(uri);
12
- const dbName = url.pathname.slice(1);
13
- return dbName || null;
14
- } catch {
15
- return null;
16
- }
17
- }
18
-
19
- async connect(uri?: string, databaseName?: string): Promise<void> {
20
- if (this.client) {
21
- return;
22
- }
23
-
24
- const mongodbUri = uri || process.env.MONGO_URI || process.env.MONGODB_URI;
25
- validateConfig("MONGO_URI", mongodbUri);
26
- validateMongoUri(mongodbUri!);
27
-
28
- try {
29
- this.client = new MongoClient(mongodbUri!);
30
- await this.client.connect();
31
-
32
- const dbNameFromUri = this.extractDatabaseName(mongodbUri!);
33
- const dbName =
34
- databaseName || process.env.TALKPILOT_DB_NAME || dbNameFromUri;
35
-
36
- if (!dbName) {
37
- throw new Error(
38
- "[core-db] Database name not specified. Please pass databaseName to connect(), " +
39
- "add it to the URI, or set TALKPILOT_DB_NAME.",
40
- );
41
- }
42
-
43
- this.db = this.client.db(dbName);
44
- setDb(this.db);
45
- console.info(`[core-db] TalkPilot MongoDB connected: ${dbName}`);
46
- } catch (error) {
47
- console.error("[core-db] TalkPilot connection failed", error);
48
- throw error;
49
- }
50
- }
51
-
52
- async disconnect(): Promise<void> {
53
- if (this.client) {
54
- try {
55
- await this.client.close();
56
- this.client = null;
57
- this.db = null;
58
- console.info("MongoDB disconnected successfully");
59
- } catch (error) {
60
- console.error("[core-db] Disconnection failed", error);
61
- throw error;
62
- }
63
- }
64
- }
65
-
66
- getDb(): Db {
67
- if (!this.db) {
68
- throw new Error("Database not initialized. Call connect() first.");
69
- }
70
- return this.db;
71
- }
72
-
73
- isConnected(): boolean {
74
- return this.client !== null && this.client !== undefined;
75
- }
76
- }
77
-
78
- export const mongodbClient = new MongoDBClient();
1
+ import { MongoClient, Db } from "mongodb";
2
+ import { setDb } from "./index";
3
+ import { validateConfig, validateMongoUri } from "../utils/validation";
4
+
5
+ class MongoDBClient {
6
+ private client: MongoClient | null = null;
7
+ private db: Db | null = null;
8
+
9
+ private extractDatabaseName(uri: string): string | null {
10
+ try {
11
+ const url = new URL(uri);
12
+ const dbName = url.pathname.slice(1);
13
+ return dbName || null;
14
+ } catch {
15
+ return null;
16
+ }
17
+ }
18
+
19
+ async connect(uri?: string, databaseName?: string): Promise<void> {
20
+ if (this.client) {
21
+ return;
22
+ }
23
+
24
+ const mongodbUri = uri || process.env.MONGO_URI || process.env.MONGODB_URI;
25
+ validateConfig("MONGO_URI", mongodbUri);
26
+ validateMongoUri(mongodbUri!);
27
+
28
+ try {
29
+ this.client = new MongoClient(mongodbUri!);
30
+ await this.client.connect();
31
+
32
+ const dbNameFromUri = this.extractDatabaseName(mongodbUri!);
33
+ const dbName =
34
+ databaseName || process.env.TALKPILOT_DB_NAME || dbNameFromUri;
35
+
36
+ if (!dbName) {
37
+ throw new Error(
38
+ "[core-db] Database name not specified. Please pass databaseName to connect(), " +
39
+ "add it to the URI, or set TALKPILOT_DB_NAME.",
40
+ );
41
+ }
42
+
43
+ this.db = this.client.db(dbName);
44
+ setDb(this.db);
45
+ console.info(`[core-db] TalkPilot MongoDB connected: ${dbName}`);
46
+ } catch (error) {
47
+ console.error("[core-db] TalkPilot connection failed", error);
48
+ throw error;
49
+ }
50
+ }
51
+
52
+ async disconnect(): Promise<void> {
53
+ if (this.client) {
54
+ try {
55
+ await this.client.close();
56
+ this.client = null;
57
+ this.db = null;
58
+ console.info("MongoDB disconnected successfully");
59
+ } catch (error) {
60
+ console.error("[core-db] Disconnection failed", error);
61
+ throw error;
62
+ }
63
+ }
64
+ }
65
+
66
+ getDb(): Db {
67
+ if (!this.db) {
68
+ throw new Error("Database not initialized. Call connect() first.");
69
+ }
70
+ return this.db;
71
+ }
72
+
73
+ isConnected(): boolean {
74
+ return this.client !== null && this.client !== undefined;
75
+ }
76
+ }
77
+
78
+ export const mongodbClient = new MongoDBClient();