@talkpilot/core-db 1.3.6 → 1.3.7

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 (167) hide show
  1. package/.cursor/rules/development.mdc +65 -65
  2. package/DEVELOPMENT.md +141 -141
  3. package/README.md +190 -190
  4. package/README_OLD.md +160 -160
  5. package/dist/talkpilot/flows/flows.schema.d.ts +3 -0
  6. package/dist/talkpilot/flows/flows.schema.d.ts.map +1 -1
  7. package/dist/talkpilot/flows/flows.schema.js +1 -0
  8. package/dist/talkpilot/flows/flows.schema.js.map +1 -1
  9. package/dist/talkpilot/flows/flows.types.d.ts +1 -0
  10. package/dist/talkpilot/flows/flows.types.d.ts.map +1 -1
  11. package/jest.config.js +20 -20
  12. package/package.json +46 -46
  13. package/src/__tests__/setup.ts +20 -20
  14. package/src/connection.ts +54 -54
  15. package/src/index.ts +26 -26
  16. package/src/municipal/__tests__/validation.spec.ts +62 -62
  17. package/src/municipal/cities/cities.getters.ts +50 -50
  18. package/src/municipal/cities/cities.types.ts +11 -11
  19. package/src/municipal/cities/index.ts +2 -2
  20. package/src/municipal/departmentsSubjects/departmentsSubjects.getters.ts +282 -282
  21. package/src/municipal/departmentsSubjects/departmentsSubjects.types.ts +72 -72
  22. package/src/municipal/departmentsSubjects/index.ts +9 -9
  23. package/src/municipal/index.ts +22 -22
  24. package/src/municipal/mongodb-client.ts +61 -61
  25. package/src/municipal/muniIssues/__tests__/muniIssues.setters.spec.ts +92 -92
  26. package/src/municipal/muniIssues/__tests__/muniIssues.setup.spec.ts +92 -92
  27. package/src/municipal/muniIssues/index.ts +17 -17
  28. package/src/municipal/muniIssues/muniIssues.constants.ts +18 -18
  29. package/src/municipal/muniIssues/muniIssues.getters.ts +24 -24
  30. package/src/municipal/muniIssues/muniIssues.schema.ts +66 -66
  31. package/src/municipal/muniIssues/muniIssues.setters.ts +24 -24
  32. package/src/municipal/muniIssues/muniIssues.setup.ts +65 -65
  33. package/src/municipal/muniIssues/muniIssues.types.ts +37 -37
  34. package/src/municipal/streets/index.ts +2 -2
  35. package/src/municipal/streets/streets.getters.ts +125 -125
  36. package/src/municipal/streets/streets.types.ts +18 -18
  37. package/src/municipal/systemInstructions/__tests__/getters.spec.ts +113 -113
  38. package/src/municipal/systemInstructions/__tests__/setters.spec.ts +274 -274
  39. package/src/municipal/systemInstructions/index.ts +7 -7
  40. package/src/municipal/systemInstructions/instructions.getters.ts +57 -57
  41. package/src/municipal/systemInstructions/instructions.setters.ts +119 -119
  42. package/src/municipal/systemInstructions/instructions.types.ts +30 -30
  43. package/src/municipal/tickets/__tests__/tickets.getters.spec.ts +30 -30
  44. package/src/municipal/tickets/__tests__/tickets.statistics.spec.ts +51 -51
  45. package/src/municipal/tickets/index.ts +3 -3
  46. package/src/municipal/tickets/tickets.constants.ts +8 -8
  47. package/src/municipal/tickets/tickets.getters.ts +121 -121
  48. package/src/municipal/tickets/tickets.statistics.aggregation.ts +96 -96
  49. package/src/municipal/tickets/tickets.statistics.getters.ts +71 -71
  50. package/src/municipal/tickets/tickets.types.ts +53 -53
  51. package/src/municipal/utils/types.ts +11 -11
  52. package/src/talkpilot/__tests__/db.spec.ts +38 -38
  53. package/src/talkpilot/__tests__/mongodb-client.spec.ts +18 -18
  54. package/src/talkpilot/__tests__/validation.spec.ts +68 -68
  55. package/src/talkpilot/agents/__tests__/agents.getters.spec.ts +29 -29
  56. package/src/talkpilot/agents/agents.getters.ts +34 -34
  57. package/src/talkpilot/agents/agents.types.ts +14 -14
  58. package/src/talkpilot/agents/index.ts +2 -2
  59. package/src/talkpilot/backgroundToolResults/__tests__/backgroundToolResults.getters.spec.ts +147 -147
  60. package/src/talkpilot/backgroundToolResults/backgroundToolResults.getters.ts +65 -65
  61. package/src/talkpilot/backgroundToolResults/backgroundToolResults.types.ts +23 -23
  62. package/src/talkpilot/backgroundToolResults/index.ts +2 -2
  63. package/src/talkpilot/calls/__tests__/callStats.utils.spec.ts +128 -128
  64. package/src/talkpilot/calls/__tests__/calls.dashboard.spec.ts +149 -149
  65. package/src/talkpilot/calls/__tests__/calls.spec.ts +270 -270
  66. package/src/talkpilot/calls/__tests__/calls.statistics.spec.ts +344 -344
  67. package/src/talkpilot/calls/calls.constants.ts +20 -20
  68. package/src/talkpilot/calls/calls.getters.ts +248 -248
  69. package/src/talkpilot/calls/calls.statistics.getters.ts +587 -587
  70. package/src/talkpilot/calls/calls.statistics.types.ts +44 -44
  71. package/src/talkpilot/calls/calls.types.ts +119 -119
  72. package/src/talkpilot/calls/dashboard/calls.dashboard.ts +292 -292
  73. package/src/talkpilot/calls/dashboard/calls.dashboard.types.ts +57 -57
  74. package/src/talkpilot/calls/index.ts +6 -6
  75. package/src/talkpilot/clientAudioBuffers/__tests__/clientAudioBuffer.getters.spec.ts +160 -160
  76. package/src/talkpilot/clientAudioBuffers/clientAudioBuffer.getters.ts +117 -117
  77. package/src/talkpilot/clientAudioBuffers/clientsAudioBuffers.types.ts +25 -25
  78. package/src/talkpilot/clientAudioBuffers/index.ts +2 -2
  79. package/src/talkpilot/clients/clients.getters.ts +16 -16
  80. package/src/talkpilot/clients/clients.types.ts +14 -14
  81. package/src/talkpilot/clients/index.ts +2 -2
  82. package/src/talkpilot/clientsConfig/__tests__/clientsConfig.getters.spec.ts +53 -53
  83. package/src/talkpilot/clientsConfig/__tests__/clientsConfig.spec.ts +190 -190
  84. package/src/talkpilot/clientsConfig/clientsConfig.getters.ts +55 -55
  85. package/src/talkpilot/clientsConfig/clientsConfig.types.ts +127 -127
  86. package/src/talkpilot/clientsConfig/index.ts +2 -2
  87. package/src/talkpilot/contextNotes/__tests__/contextNotes.getters.spec.ts +176 -176
  88. package/src/talkpilot/contextNotes/contextNotes.getters.ts +86 -86
  89. package/src/talkpilot/contextNotes/contextNotes.types.ts +20 -20
  90. package/src/talkpilot/contextNotes/index.ts +2 -2
  91. package/src/talkpilot/flows/__tests__/flows.schema.spec.ts +96 -71
  92. package/src/talkpilot/flows/flows.getter.ts +14 -14
  93. package/src/talkpilot/flows/flows.schema.ts +154 -153
  94. package/src/talkpilot/flows/flows.types.ts +187 -186
  95. package/src/talkpilot/flows/index.ts +2 -2
  96. package/src/talkpilot/groups/__tests__/groups.spec.ts +90 -90
  97. package/src/talkpilot/groups/__tests__/phone.utils.spec.ts +32 -32
  98. package/src/talkpilot/groups/groups.getters.ts +30 -30
  99. package/src/talkpilot/groups/groups.types.ts +29 -29
  100. package/src/talkpilot/groups/index.ts +3 -3
  101. package/src/talkpilot/groups/phone.utils.ts +46 -46
  102. package/src/talkpilot/index.ts +31 -31
  103. package/src/talkpilot/leads/index.ts +2 -2
  104. package/src/talkpilot/leads/leads.getter.ts +6 -6
  105. package/src/talkpilot/leads/leads.schema.ts +33 -33
  106. package/src/talkpilot/leads/leads.types.ts +20 -20
  107. package/src/talkpilot/mongodb-client.ts +78 -78
  108. package/src/talkpilot/phone_numbers/__tests__/phone_numbers.spec.ts +252 -252
  109. package/src/talkpilot/phone_numbers/index.ts +2 -2
  110. package/src/talkpilot/phone_numbers/phone_numbers.getter.ts +158 -158
  111. package/src/talkpilot/phone_numbers/phone_numbers.schema.ts +17 -17
  112. package/src/talkpilot/phone_numbers/phone_numbers.types.ts +30 -30
  113. package/src/talkpilot/plans/__tests__/plans.spec.ts +70 -70
  114. package/src/talkpilot/plans/index.ts +2 -2
  115. package/src/talkpilot/plans/plans.getters.ts +132 -132
  116. package/src/talkpilot/plans/plans.types.ts +89 -89
  117. package/src/talkpilot/products/__tests__/products.getters.spec.ts +45 -45
  118. package/src/talkpilot/products/index.ts +2 -2
  119. package/src/talkpilot/products/products.getters.ts +12 -12
  120. package/src/talkpilot/products/products.types.ts +9 -9
  121. package/src/talkpilot/results/index.ts +7 -7
  122. package/src/talkpilot/results/results.getter.ts +39 -39
  123. package/src/talkpilot/results/results.schema.ts +25 -25
  124. package/src/talkpilot/results/results.types.ts +34 -34
  125. package/src/talkpilot/retry_analyze/__tests__/retryAnalyze.getters.spec.ts +156 -156
  126. package/src/talkpilot/retry_analyze/index.ts +2 -2
  127. package/src/talkpilot/retry_analyze/retryAnalyze.getters.ts +84 -84
  128. package/src/talkpilot/retry_analyze/retryAnalyze.types.ts +13 -13
  129. package/src/talkpilot/sessions/__tests__/sessions.spec.ts +147 -147
  130. package/src/talkpilot/sessions/index.ts +2 -2
  131. package/src/talkpilot/sessions/sessions.getter.ts +92 -92
  132. package/src/talkpilot/sessions/sessions.schema.ts +34 -34
  133. package/src/talkpilot/sessions/sessions.types.ts +30 -30
  134. package/src/talkpilot/subscriptions/__tests__/subscriptions.getters.utils.spec.ts +45 -45
  135. package/src/talkpilot/subscriptions/index.ts +3 -3
  136. package/src/talkpilot/subscriptions/subscriptions.getters.ts +146 -146
  137. package/src/talkpilot/subscriptions/subscriptions.getters.utils.ts +33 -33
  138. package/src/talkpilot/subscriptions/subscriptions.types.ts +66 -66
  139. package/src/talkpilot/utils/__tests__/query.utils.spec.ts +49 -49
  140. package/src/talkpilot/utils/query.utils.ts +21 -21
  141. package/src/test-utils/db-utils.ts +26 -26
  142. package/src/test-utils/factories/index.ts +15 -15
  143. package/src/test-utils/factories/municipal/cities.ts +16 -16
  144. package/src/test-utils/factories/municipal/departmentsSubjects.ts +37 -37
  145. package/src/test-utils/factories/municipal/muniIssues.ts +32 -32
  146. package/src/test-utils/factories/municipal/streets.ts +22 -22
  147. package/src/test-utils/factories/municipal/tickets.ts +39 -39
  148. package/src/test-utils/factories/talkpilot/agents.ts +19 -19
  149. package/src/test-utils/factories/talkpilot/calls.ts +37 -37
  150. package/src/test-utils/factories/talkpilot/clientAudioBuffers.ts +20 -20
  151. package/src/test-utils/factories/talkpilot/clientsConfig.ts +18 -18
  152. package/src/test-utils/factories/talkpilot/contextNotes.ts +33 -33
  153. package/src/test-utils/factories/talkpilot/flows.ts +33 -33
  154. package/src/test-utils/factories/talkpilot/groups.ts +33 -33
  155. package/src/test-utils/factories/talkpilot/phone_numbers.ts +22 -22
  156. package/src/test-utils/factories/talkpilot/sessions.ts +35 -35
  157. package/src/test-utils/factories/websitalk/scans.ts +23 -23
  158. package/src/utils/date.utils.ts +116 -116
  159. package/src/utils/shared.types.ts +4 -4
  160. package/src/utils/validation.ts +23 -23
  161. package/src/websitalk/index.ts +15 -15
  162. package/src/websitalk/mongodb-client.ts +61 -61
  163. package/src/websitalk/scans/__tests__/scans.spec.ts +218 -218
  164. package/src/websitalk/scans/index.ts +2 -2
  165. package/src/websitalk/scans/scans.getters.ts +113 -113
  166. package/src/websitalk/scans/scans.types.ts +53 -53
  167. package/tsconfig.json +23 -23
package/src/connection.ts CHANGED
@@ -1,54 +1,54 @@
1
- import { mongodbClient } from "./talkpilot";
2
- import { municipalDataMongodbClient } from "./municipal";
3
- import { websitalkMongodbClient } from "./websitalk";
4
-
5
- let isConnected = false;
6
-
7
- /**
8
- * Ensures the database clients are initialized.
9
- * This function reads from MONGO_URI.
10
- */
11
- export async function ensureDbConnected() {
12
- if (isConnected) {
13
- return;
14
- }
15
-
16
- const mongoUri = process.env.MONGO_URI || process.env.MONGODB_URI;
17
- const talkpilotUri = mongoUri;
18
- const municipalUri = mongoUri;
19
- const websiteTalkUri = mongoUri;
20
-
21
- if (!talkpilotUri) {
22
- console.warn("[core-db] MONGO_URI is not set");
23
- } else {
24
- const dbNameFromUri = talkpilotUri.split("/").pop()?.split("?")[0];
25
- const dbName = dbNameFromUri || process.env.TALKPILOT_DB_NAME;
26
- await mongodbClient.connect(talkpilotUri, dbName);
27
- console.info(
28
- `[core-db] Connected to TalkPilot MongoDB: ${dbName || "via env"}`,
29
- );
30
- }
31
-
32
- if (!municipalUri) {
33
- console.warn("[core-db] MONGO_URI is not set");
34
- } else {
35
- const dbNameFromUri = municipalUri.split("/").pop()?.split("?")[0];
36
- const dbName = dbNameFromUri || process.env.MUNICIPAL_DB_NAME;
37
- await municipalDataMongodbClient.connect(municipalUri, dbName);
38
- console.info(
39
- `[core-db] Connected to Municipal Data MongoDB: ${dbName || "via env"}`,
40
- );
41
- }
42
-
43
- if (!websiteTalkUri) {
44
- console.warn("[core-db] MONGO_URI is not set");
45
- } else {
46
- const dbName = process.env.WEBSITALK_DB_NAME;
47
- await websitalkMongodbClient.connect(websiteTalkUri, dbName);
48
- console.info(
49
- `[core-db] Connected to Website Talk MongoDB: ${dbName || "website-talk (default)"}`,
50
- );
51
- }
52
-
53
- isConnected = true;
54
- }
1
+ import { mongodbClient } from "./talkpilot";
2
+ import { municipalDataMongodbClient } from "./municipal";
3
+ import { websitalkMongodbClient } from "./websitalk";
4
+
5
+ let isConnected = false;
6
+
7
+ /**
8
+ * Ensures the database clients are initialized.
9
+ * This function reads from MONGO_URI.
10
+ */
11
+ export async function ensureDbConnected() {
12
+ if (isConnected) {
13
+ return;
14
+ }
15
+
16
+ const mongoUri = process.env.MONGO_URI || process.env.MONGODB_URI;
17
+ const talkpilotUri = mongoUri;
18
+ const municipalUri = mongoUri;
19
+ const websiteTalkUri = mongoUri;
20
+
21
+ if (!talkpilotUri) {
22
+ console.warn("[core-db] MONGO_URI is not set");
23
+ } else {
24
+ const dbNameFromUri = talkpilotUri.split("/").pop()?.split("?")[0];
25
+ const dbName = dbNameFromUri || process.env.TALKPILOT_DB_NAME;
26
+ await mongodbClient.connect(talkpilotUri, dbName);
27
+ console.info(
28
+ `[core-db] Connected to TalkPilot MongoDB: ${dbName || "via env"}`,
29
+ );
30
+ }
31
+
32
+ if (!municipalUri) {
33
+ console.warn("[core-db] MONGO_URI is not set");
34
+ } else {
35
+ const dbNameFromUri = municipalUri.split("/").pop()?.split("?")[0];
36
+ const dbName = dbNameFromUri || process.env.MUNICIPAL_DB_NAME;
37
+ await municipalDataMongodbClient.connect(municipalUri, dbName);
38
+ console.info(
39
+ `[core-db] Connected to Municipal Data MongoDB: ${dbName || "via env"}`,
40
+ );
41
+ }
42
+
43
+ if (!websiteTalkUri) {
44
+ console.warn("[core-db] MONGO_URI is not set");
45
+ } else {
46
+ const dbName = process.env.WEBSITALK_DB_NAME;
47
+ await websitalkMongodbClient.connect(websiteTalkUri, dbName);
48
+ console.info(
49
+ `[core-db] Connected to Website Talk MongoDB: ${dbName || "website-talk (default)"}`,
50
+ );
51
+ }
52
+
53
+ isConnected = true;
54
+ }
package/src/index.ts CHANGED
@@ -1,26 +1,26 @@
1
- // TalkPilot DB exports
2
- export * from "./talkpilot";
3
- export * from "./connection";
4
-
5
- export {
6
- municipalDataMongodbClient,
7
- getDb as getMunicipalDataDb,
8
- setDb as setMunicipalDataDb,
9
- ObjectId as MunicipalDataObjectId,
10
- } from "./municipal";
11
- export * from "./municipal/cities";
12
- export * from "./municipal/streets";
13
- export * from "./municipal/departmentsSubjects";
14
- export * from "./municipal/tickets";
15
- export * from "./municipal/systemInstructions";
16
- export * from "./municipal/muniIssues";
17
- export * from "./municipal/utils/types";
18
-
19
- export {
20
- websitalkMongodbClient,
21
- getDb as getWebsitalkDb,
22
- setDb as setWebsitalkDb,
23
- ObjectId as WebsitalkObjectId,
24
- } from "./websitalk";
25
- export * from "./websitalk/scans/scans.getters";
26
- export type * from "./websitalk/scans/scans.types";
1
+ // TalkPilot DB exports
2
+ export * from "./talkpilot";
3
+ export * from "./connection";
4
+
5
+ export {
6
+ municipalDataMongodbClient,
7
+ getDb as getMunicipalDataDb,
8
+ setDb as setMunicipalDataDb,
9
+ ObjectId as MunicipalDataObjectId,
10
+ } from "./municipal";
11
+ export * from "./municipal/cities";
12
+ export * from "./municipal/streets";
13
+ export * from "./municipal/departmentsSubjects";
14
+ export * from "./municipal/tickets";
15
+ export * from "./municipal/systemInstructions";
16
+ export * from "./municipal/muniIssues";
17
+ export * from "./municipal/utils/types";
18
+
19
+ export {
20
+ websitalkMongodbClient,
21
+ getDb as getWebsitalkDb,
22
+ setDb as setWebsitalkDb,
23
+ ObjectId as WebsitalkObjectId,
24
+ } from "./websitalk";
25
+ export * from "./websitalk/scans/scans.getters";
26
+ export type * from "./websitalk/scans/scans.types";
@@ -1,62 +1,62 @@
1
- import { municipalDataMongodbClient } from "../mongodb-client";
2
- import { MongoClient } from "mongodb";
3
-
4
- describe("MunicipalDataMongoDBClient Validation", () => {
5
- const originalEnv = process.env;
6
-
7
- beforeEach(() => {
8
- process.env = { ...originalEnv };
9
- delete process.env.MONGO_URI;
10
- delete process.env.MONGODB_URI;
11
-
12
- // Force reset singleton state
13
- (municipalDataMongodbClient as any).client = null;
14
- (municipalDataMongodbClient as any).db = null;
15
- });
16
-
17
- afterAll(async () => {
18
- process.env = originalEnv;
19
- await municipalDataMongodbClient.disconnect();
20
- });
21
-
22
- it("should succeed with default database name if only URI is provided", async () => {
23
- const mockDb = { databaseName: "municipal-data" };
24
- const connectSpy = jest
25
- .spyOn(MongoClient.prototype, "connect")
26
- .mockResolvedValue({} as any);
27
- const dbSpy = jest
28
- .spyOn(MongoClient.prototype, "db")
29
- .mockReturnValue(mockDb as any);
30
-
31
- const validUri = "mongodb://127.0.0.1:27017";
32
- await municipalDataMongodbClient.connect(validUri);
33
-
34
- expect(municipalDataMongodbClient.getDb()).toBeDefined();
35
- expect(municipalDataMongodbClient.getDb().databaseName).toBe(
36
- "municipal-data",
37
- );
38
-
39
- connectSpy.mockRestore();
40
- dbSpy.mockRestore();
41
- });
42
-
43
- it("should allow overriding the default database name", async () => {
44
- const mockDb = { databaseName: "custom-municipal-db" };
45
- const connectSpy = jest
46
- .spyOn(MongoClient.prototype, "connect")
47
- .mockResolvedValue({} as any);
48
- const dbSpy = jest
49
- .spyOn(MongoClient.prototype, "db")
50
- .mockReturnValue(mockDb as any);
51
-
52
- const validUri = "mongodb://127.0.0.1:27017";
53
- await municipalDataMongodbClient.connect(validUri, "custom-municipal-db");
54
-
55
- expect(municipalDataMongodbClient.getDb().databaseName).toBe(
56
- "custom-municipal-db",
57
- );
58
-
59
- connectSpy.mockRestore();
60
- dbSpy.mockRestore();
61
- });
62
- });
1
+ import { municipalDataMongodbClient } from "../mongodb-client";
2
+ import { MongoClient } from "mongodb";
3
+
4
+ describe("MunicipalDataMongoDBClient Validation", () => {
5
+ const originalEnv = process.env;
6
+
7
+ beforeEach(() => {
8
+ process.env = { ...originalEnv };
9
+ delete process.env.MONGO_URI;
10
+ delete process.env.MONGODB_URI;
11
+
12
+ // Force reset singleton state
13
+ (municipalDataMongodbClient as any).client = null;
14
+ (municipalDataMongodbClient as any).db = null;
15
+ });
16
+
17
+ afterAll(async () => {
18
+ process.env = originalEnv;
19
+ await municipalDataMongodbClient.disconnect();
20
+ });
21
+
22
+ it("should succeed with default database name if only URI is provided", async () => {
23
+ const mockDb = { databaseName: "municipal-data" };
24
+ const connectSpy = jest
25
+ .spyOn(MongoClient.prototype, "connect")
26
+ .mockResolvedValue({} as any);
27
+ const dbSpy = jest
28
+ .spyOn(MongoClient.prototype, "db")
29
+ .mockReturnValue(mockDb as any);
30
+
31
+ const validUri = "mongodb://127.0.0.1:27017";
32
+ await municipalDataMongodbClient.connect(validUri);
33
+
34
+ expect(municipalDataMongodbClient.getDb()).toBeDefined();
35
+ expect(municipalDataMongodbClient.getDb().databaseName).toBe(
36
+ "municipal-data",
37
+ );
38
+
39
+ connectSpy.mockRestore();
40
+ dbSpy.mockRestore();
41
+ });
42
+
43
+ it("should allow overriding the default database name", async () => {
44
+ const mockDb = { databaseName: "custom-municipal-db" };
45
+ const connectSpy = jest
46
+ .spyOn(MongoClient.prototype, "connect")
47
+ .mockResolvedValue({} as any);
48
+ const dbSpy = jest
49
+ .spyOn(MongoClient.prototype, "db")
50
+ .mockReturnValue(mockDb as any);
51
+
52
+ const validUri = "mongodb://127.0.0.1:27017";
53
+ await municipalDataMongodbClient.connect(validUri, "custom-municipal-db");
54
+
55
+ expect(municipalDataMongodbClient.getDb().databaseName).toBe(
56
+ "custom-municipal-db",
57
+ );
58
+
59
+ connectSpy.mockRestore();
60
+ dbSpy.mockRestore();
61
+ });
62
+ });
@@ -1,50 +1,50 @@
1
- import { getDb, ObjectId, City } from "../index";
2
- import { Collection, Filter, ObjectId as MongoObjectId } from "mongodb";
3
-
4
- export const getCitiesCollection = (): Collection<City> => {
5
- return getDb().collection<City>("cities");
6
- };
7
-
8
- export const findCities = async (
9
- filter: Filter<City> = {},
10
- ): Promise<City[]> => {
11
- return await getCitiesCollection().find(filter).toArray();
12
- };
13
-
14
- export const getCityById = async (cityId: string): Promise<City | null> => {
15
- const city = await getCitiesCollection().findOne({
16
- _id: new ObjectId(cityId),
17
- });
18
- return city ? city : null;
19
- };
20
-
21
- export const createCity = async (
22
- cityData: Omit<City, "_id" | "createdAt" | "updatedAt">,
23
- ): Promise<MongoObjectId> => {
24
- const city: Omit<City, "_id"> = {
25
- ...cityData,
26
- createdAt: new Date(),
27
- updatedAt: new Date(),
28
- };
29
- const { insertedId } = await getCitiesCollection().insertOne(city as City);
30
- return insertedId;
31
- };
32
-
33
- export const updateCity = async (
34
- cityId: string,
35
- data: Partial<Omit<City, "_id" | "createdAt" | "updatedAt">>,
36
- ): Promise<City | null> => {
37
- const result = await getCitiesCollection().findOneAndUpdate(
38
- { _id: new ObjectId(cityId) },
39
- { $set: { ...data, updatedAt: new Date() } },
40
- { returnDocument: "after" },
41
- );
42
- return result || null;
43
- };
44
-
45
- export const deleteCity = async (cityId: string): Promise<boolean> => {
46
- const result = await getCitiesCollection().deleteOne({
47
- _id: new ObjectId(cityId),
48
- });
49
- return result.deletedCount > 0;
50
- };
1
+ import { getDb, ObjectId, City } from "../index";
2
+ import { Collection, Filter, ObjectId as MongoObjectId } from "mongodb";
3
+
4
+ export const getCitiesCollection = (): Collection<City> => {
5
+ return getDb().collection<City>("cities");
6
+ };
7
+
8
+ export const findCities = async (
9
+ filter: Filter<City> = {},
10
+ ): Promise<City[]> => {
11
+ return await getCitiesCollection().find(filter).toArray();
12
+ };
13
+
14
+ export const getCityById = async (cityId: string): Promise<City | null> => {
15
+ const city = await getCitiesCollection().findOne({
16
+ _id: new ObjectId(cityId),
17
+ });
18
+ return city ? city : null;
19
+ };
20
+
21
+ export const createCity = async (
22
+ cityData: Omit<City, "_id" | "createdAt" | "updatedAt">,
23
+ ): Promise<MongoObjectId> => {
24
+ const city: Omit<City, "_id"> = {
25
+ ...cityData,
26
+ createdAt: new Date(),
27
+ updatedAt: new Date(),
28
+ };
29
+ const { insertedId } = await getCitiesCollection().insertOne(city as City);
30
+ return insertedId;
31
+ };
32
+
33
+ export const updateCity = async (
34
+ cityId: string,
35
+ data: Partial<Omit<City, "_id" | "createdAt" | "updatedAt">>,
36
+ ): Promise<City | null> => {
37
+ const result = await getCitiesCollection().findOneAndUpdate(
38
+ { _id: new ObjectId(cityId) },
39
+ { $set: { ...data, updatedAt: new Date() } },
40
+ { returnDocument: "after" },
41
+ );
42
+ return result || null;
43
+ };
44
+
45
+ export const deleteCity = async (cityId: string): Promise<boolean> => {
46
+ const result = await getCitiesCollection().deleteOne({
47
+ _id: new ObjectId(cityId),
48
+ });
49
+ return result.deletedCount > 0;
50
+ };
@@ -1,11 +1,11 @@
1
- import { ObjectId, WithId } from "mongodb";
2
-
3
- export type City = {
4
- _id: ObjectId;
5
- createdAt: Date;
6
- updatedAt: Date;
7
- name: string;
8
- displayName: string;
9
- };
10
-
11
- export type CityDoc = WithId<City>;
1
+ import { ObjectId, WithId } from "mongodb";
2
+
3
+ export type City = {
4
+ _id: ObjectId;
5
+ createdAt: Date;
6
+ updatedAt: Date;
7
+ name: string;
8
+ displayName: string;
9
+ };
10
+
11
+ export type CityDoc = WithId<City>;
@@ -1,2 +1,2 @@
1
- export * from "./cities.getters";
2
- export type { City, CityDoc } from "./cities.types";
1
+ export * from "./cities.getters";
2
+ export type { City, CityDoc } from "./cities.types";