@talkpilot/core-db 1.3.72 → 1.3.75

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 (296) hide show
  1. package/.cursor/rules/development.mdc +65 -65
  2. package/AGENTS.md +49 -49
  3. package/DEVELOPMENT.md +141 -141
  4. package/README.md +818 -836
  5. package/dist/index.d.ts +1 -0
  6. package/dist/index.d.ts.map +1 -1
  7. package/dist/index.js +1 -0
  8. package/dist/index.js.map +1 -1
  9. package/dist/municipal/index.d.ts +1 -0
  10. package/dist/municipal/index.d.ts.map +1 -1
  11. package/dist/municipal/index.js +1 -0
  12. package/dist/municipal/index.js.map +1 -1
  13. package/dist/municipal/muniIssues/index.d.ts +2 -2
  14. package/dist/municipal/muniIssues/index.d.ts.map +1 -1
  15. package/dist/municipal/muniIssues/index.js +1 -2
  16. package/dist/municipal/muniIssues/index.js.map +1 -1
  17. package/dist/municipal/muniIssues/muniIssues.constants.d.ts +0 -2
  18. package/dist/municipal/muniIssues/muniIssues.constants.d.ts.map +1 -1
  19. package/dist/municipal/muniIssues/muniIssues.constants.js +1 -8
  20. package/dist/municipal/muniIssues/muniIssues.constants.js.map +1 -1
  21. package/dist/municipal/muniIssues/muniIssues.getters.d.ts +1 -6
  22. package/dist/municipal/muniIssues/muniIssues.getters.d.ts.map +1 -1
  23. package/dist/municipal/muniIssues/muniIssues.getters.js +1 -31
  24. package/dist/municipal/muniIssues/muniIssues.getters.js.map +1 -1
  25. package/dist/municipal/muniIssues/muniIssues.schema.d.ts.map +1 -1
  26. package/dist/municipal/muniIssues/muniIssues.schema.js +0 -13
  27. package/dist/municipal/muniIssues/muniIssues.schema.js.map +1 -1
  28. package/dist/municipal/muniIssues/muniIssues.setters.d.ts +1 -5
  29. package/dist/municipal/muniIssues/muniIssues.setters.d.ts.map +1 -1
  30. package/dist/municipal/muniIssues/muniIssues.setters.js +2 -39
  31. package/dist/municipal/muniIssues/muniIssues.setters.js.map +1 -1
  32. package/dist/municipal/muniIssues/muniIssues.types.d.ts +2 -23
  33. package/dist/municipal/muniIssues/muniIssues.types.d.ts.map +1 -1
  34. package/dist/municipal/residents/index.d.ts +5 -0
  35. package/dist/municipal/residents/index.d.ts.map +1 -0
  36. package/dist/municipal/residents/index.js +24 -0
  37. package/dist/municipal/residents/index.js.map +1 -0
  38. package/dist/municipal/residents/residents.constants.d.ts +3 -0
  39. package/dist/municipal/residents/residents.constants.d.ts.map +1 -0
  40. package/dist/municipal/residents/residents.constants.js +6 -0
  41. package/dist/municipal/residents/residents.constants.js.map +1 -0
  42. package/dist/municipal/residents/residents.getters.d.ts +9 -0
  43. package/dist/municipal/residents/residents.getters.d.ts.map +1 -0
  44. package/dist/municipal/residents/residents.getters.js +32 -0
  45. package/dist/municipal/residents/residents.getters.js.map +1 -0
  46. package/dist/municipal/residents/residents.schema.d.ts +4 -0
  47. package/dist/municipal/residents/residents.schema.d.ts.map +1 -0
  48. package/dist/municipal/residents/residents.schema.js +29 -0
  49. package/dist/municipal/residents/residents.schema.js.map +1 -0
  50. package/dist/municipal/residents/residents.types.d.ts +15 -0
  51. package/dist/municipal/residents/residents.types.d.ts.map +1 -0
  52. package/dist/municipal/residents/residents.types.js +3 -0
  53. package/dist/municipal/residents/residents.types.js.map +1 -0
  54. package/dist/talkpilot/calls/calls.constants.d.ts +0 -6
  55. package/dist/talkpilot/calls/calls.constants.d.ts.map +1 -1
  56. package/dist/talkpilot/calls/calls.constants.js +1 -7
  57. package/dist/talkpilot/calls/calls.constants.js.map +1 -1
  58. package/dist/talkpilot/calls/calls.getters.d.ts +1 -2
  59. package/dist/talkpilot/calls/calls.getters.d.ts.map +1 -1
  60. package/dist/talkpilot/calls/calls.getters.js +1 -6
  61. package/dist/talkpilot/calls/calls.getters.js.map +1 -1
  62. package/dist/talkpilot/calls/calls.types.d.ts +2 -4
  63. package/dist/talkpilot/calls/calls.types.d.ts.map +1 -1
  64. package/dist/talkpilot/flows/flows.schema.d.ts +28 -0
  65. package/dist/talkpilot/flows/flows.schema.d.ts.map +1 -1
  66. package/dist/talkpilot/flows/flows.schema.js +16 -0
  67. package/dist/talkpilot/flows/flows.schema.js.map +1 -1
  68. package/dist/talkpilot/flows/flows.types.d.ts +8 -0
  69. package/dist/talkpilot/flows/flows.types.d.ts.map +1 -1
  70. package/dist/talkpilot/subscriptions/subscriptions.utils.d.ts +4 -0
  71. package/dist/talkpilot/subscriptions/subscriptions.utils.d.ts.map +1 -0
  72. package/dist/talkpilot/subscriptions/subscriptions.utils.js +20 -0
  73. package/dist/talkpilot/subscriptions/subscriptions.utils.js.map +1 -0
  74. package/dist/test-utils/factories/index.d.ts +1 -0
  75. package/dist/test-utils/factories/index.d.ts.map +1 -1
  76. package/dist/test-utils/factories/index.js +1 -0
  77. package/dist/test-utils/factories/index.js.map +1 -1
  78. package/dist/test-utils/factories/municipal/residents.d.ts +5 -0
  79. package/dist/test-utils/factories/municipal/residents.d.ts.map +1 -0
  80. package/dist/test-utils/factories/municipal/residents.js +22 -0
  81. package/dist/test-utils/factories/municipal/residents.js.map +1 -0
  82. package/jest.config.js +20 -20
  83. package/package.json +46 -46
  84. package/src/__tests__/setup.ts +20 -20
  85. package/src/bulkWrite/__tests__/bulkWrite.spec.ts +65 -65
  86. package/src/bulkWrite/bulkWrite.ops.ts +45 -45
  87. package/src/bulkWrite/bulkWrite.types.ts +4 -4
  88. package/src/bulkWrite/index.ts +2 -2
  89. package/src/config.ts +9 -9
  90. package/src/configuration/index.ts +17 -17
  91. package/src/configuration/internalModels/index.ts +3 -3
  92. package/src/configuration/internalModels/internalModels.constants.ts +1 -1
  93. package/src/configuration/internalModels/internalModels.getters.ts +33 -33
  94. package/src/configuration/internalModels/internalModels.types.ts +12 -12
  95. package/src/configuration/models/__tests__/models.getters.spec.ts +67 -67
  96. package/src/configuration/models/index.ts +27 -27
  97. package/src/configuration/models/models.constants.ts +20 -20
  98. package/src/configuration/models/models.getters.ts +71 -71
  99. package/src/configuration/models/models.types.ts +80 -80
  100. package/src/configuration/mongodb-client.ts +62 -62
  101. package/src/configuration/prompts/index.ts +2 -2
  102. package/src/configuration/prompts/prompts.getters.ts +49 -49
  103. package/src/configuration/prompts/prompts.types.ts +11 -11
  104. package/src/connection.ts +81 -81
  105. package/src/index.ts +48 -47
  106. package/src/municipal/__tests__/validation.spec.ts +62 -62
  107. package/src/municipal/cities/cities.getters.ts +50 -50
  108. package/src/municipal/cities/cities.types.ts +11 -11
  109. package/src/municipal/cities/index.ts +2 -2
  110. package/src/municipal/departmentsSubjects/__tests__/departmentsSubjects.spec.ts +272 -272
  111. package/src/municipal/departmentsSubjects/departmentsSubjects.getters.ts +317 -317
  112. package/src/municipal/departmentsSubjects/departmentsSubjects.types.ts +73 -73
  113. package/src/municipal/departmentsSubjects/index.ts +9 -9
  114. package/src/municipal/index.ts +23 -22
  115. package/src/municipal/mongodb-client.ts +61 -61
  116. package/src/municipal/muniIssues/__tests__/muniIssues.setters.spec.ts +94 -167
  117. package/src/municipal/muniIssues/__tests__/muniIssues.setup.spec.ts +92 -93
  118. package/src/municipal/muniIssues/index.ts +17 -22
  119. package/src/municipal/muniIssues/muniIssues.constants.ts +18 -26
  120. package/src/municipal/muniIssues/muniIssues.getters.ts +24 -68
  121. package/src/municipal/muniIssues/muniIssues.schema.ts +66 -80
  122. package/src/municipal/muniIssues/muniIssues.setters.ts +24 -80
  123. package/src/municipal/muniIssues/muniIssues.setup.ts +65 -65
  124. package/src/municipal/muniIssues/muniIssues.types.ts +37 -67
  125. package/src/municipal/residents/__tests__/residents.spec.ts +92 -0
  126. package/src/municipal/residents/index.ts +4 -0
  127. package/src/municipal/residents/residents.constants.ts +2 -0
  128. package/src/municipal/residents/residents.getters.ts +42 -0
  129. package/src/municipal/residents/residents.schema.ts +28 -0
  130. package/src/municipal/residents/residents.types.ts +17 -0
  131. package/src/municipal/streets/__tests__/streets.spec.ts +253 -253
  132. package/src/municipal/streets/index.ts +2 -2
  133. package/src/municipal/streets/streets.getters.ts +140 -140
  134. package/src/municipal/streets/streets.types.ts +19 -19
  135. package/src/municipal/systemInstructions/__tests__/getters.spec.ts +113 -113
  136. package/src/municipal/systemInstructions/__tests__/setters.spec.ts +274 -274
  137. package/src/municipal/systemInstructions/index.ts +7 -7
  138. package/src/municipal/systemInstructions/instructions.getters.ts +57 -57
  139. package/src/municipal/systemInstructions/instructions.setters.ts +119 -119
  140. package/src/municipal/systemInstructions/instructions.types.ts +30 -30
  141. package/src/municipal/tickets/__tests__/tickets.getters.spec.ts +30 -30
  142. package/src/municipal/tickets/__tests__/tickets.statistics.spec.ts +99 -99
  143. package/src/municipal/tickets/index.ts +9 -9
  144. package/src/municipal/tickets/tickets.constants.ts +8 -8
  145. package/src/municipal/tickets/tickets.getters.ts +121 -121
  146. package/src/municipal/tickets/tickets.statistics.aggregation.ts +110 -110
  147. package/src/municipal/tickets/tickets.statistics.getters.ts +145 -145
  148. package/src/municipal/tickets/tickets.types.ts +54 -54
  149. package/src/municipal/utils/types.ts +11 -11
  150. package/src/talkpilot/__tests__/db.spec.ts +38 -38
  151. package/src/talkpilot/__tests__/mongodb-client.spec.ts +18 -18
  152. package/src/talkpilot/__tests__/validation.spec.ts +68 -68
  153. package/src/talkpilot/agents/__tests__/agents.getters.spec.ts +29 -29
  154. package/src/talkpilot/agents/agents.getters.ts +34 -34
  155. package/src/talkpilot/agents/agents.types.ts +14 -14
  156. package/src/talkpilot/agents/index.ts +2 -2
  157. package/src/talkpilot/backgroundToolResults/__tests__/backgroundToolResults.getters.spec.ts +147 -147
  158. package/src/talkpilot/backgroundToolResults/backgroundToolResults.getters.ts +65 -65
  159. package/src/talkpilot/backgroundToolResults/backgroundToolResults.types.ts +23 -23
  160. package/src/talkpilot/backgroundToolResults/index.ts +2 -2
  161. package/src/talkpilot/callTokenUsage/__tests__/callTokenUsage.getters.spec.ts +57 -57
  162. package/src/talkpilot/callTokenUsage/callTokenUsage.getters.ts +42 -42
  163. package/src/talkpilot/callTokenUsage/callTokenUsage.types.ts +28 -28
  164. package/src/talkpilot/callTokenUsage/index.ts +2 -2
  165. package/src/talkpilot/calls/__tests__/callStats.utils.spec.ts +128 -128
  166. package/src/talkpilot/calls/__tests__/calls.dashboard.spec.ts +229 -229
  167. package/src/talkpilot/calls/__tests__/calls.spec.ts +471 -519
  168. package/src/talkpilot/calls/__tests__/calls.statistics.spec.ts +483 -483
  169. package/src/talkpilot/calls/calls.constants.ts +49 -56
  170. package/src/talkpilot/calls/calls.getters.ts +277 -295
  171. package/src/talkpilot/calls/calls.statistics.getters.ts +668 -668
  172. package/src/talkpilot/calls/calls.statistics.types.ts +48 -48
  173. package/src/talkpilot/calls/calls.types.ts +151 -155
  174. package/src/talkpilot/calls/dashboard/calls.dashboard.ts +295 -295
  175. package/src/talkpilot/calls/dashboard/calls.dashboard.types.ts +57 -57
  176. package/src/talkpilot/calls/index.ts +6 -6
  177. package/src/talkpilot/clientAudioBuffers/__tests__/clientAudioBuffer.getters.spec.ts +160 -160
  178. package/src/talkpilot/clientAudioBuffers/clientAudioBuffer.getters.ts +117 -117
  179. package/src/talkpilot/clientAudioBuffers/clientsAudioBuffers.types.ts +25 -25
  180. package/src/talkpilot/clientAudioBuffers/index.ts +2 -2
  181. package/src/talkpilot/clients/__tests__/clients.quota.spec.ts +123 -123
  182. package/src/talkpilot/clients/clients.getters.ts +19 -19
  183. package/src/talkpilot/clients/clients.quota.getters.ts +119 -119
  184. package/src/talkpilot/clients/clients.quota.types.ts +27 -27
  185. package/src/talkpilot/clients/clients.types.ts +18 -18
  186. package/src/talkpilot/clients/index.ts +4 -4
  187. package/src/talkpilot/clientsConfig/__tests__/clientsConfig.getters.spec.ts +423 -423
  188. package/src/talkpilot/clientsConfig/__tests__/clientsConfig.spec.ts +212 -212
  189. package/src/talkpilot/clientsConfig/clientsConfig.constants.ts +2 -2
  190. package/src/talkpilot/clientsConfig/clientsConfig.getters.ts +247 -247
  191. package/src/talkpilot/clientsConfig/clientsConfig.types.ts +137 -137
  192. package/src/talkpilot/clientsConfig/index.ts +3 -3
  193. package/src/talkpilot/contextNotes/__tests__/contextNotes.getters.spec.ts +217 -217
  194. package/src/talkpilot/contextNotes/contextNotes.getters.ts +85 -85
  195. package/src/talkpilot/contextNotes/contextNotes.types.ts +20 -20
  196. package/src/talkpilot/contextNotes/index.ts +2 -2
  197. package/src/talkpilot/flows/__tests__/flows.schema.spec.ts +152 -124
  198. package/src/talkpilot/flows/flows.getter.ts +34 -34
  199. package/src/talkpilot/flows/flows.schema.ts +196 -180
  200. package/src/talkpilot/flows/flows.types.ts +220 -211
  201. package/src/talkpilot/flows/index.ts +7 -7
  202. package/src/talkpilot/groups/__tests__/groups.spec.ts +90 -90
  203. package/src/talkpilot/groups/__tests__/phone.utils.spec.ts +32 -32
  204. package/src/talkpilot/groups/groups.getters.ts +30 -30
  205. package/src/talkpilot/groups/groups.types.ts +29 -29
  206. package/src/talkpilot/groups/index.ts +3 -3
  207. package/src/talkpilot/groups/phone.utils.ts +46 -46
  208. package/src/talkpilot/index.ts +32 -32
  209. package/src/talkpilot/leads/index.ts +2 -2
  210. package/src/talkpilot/leads/leads.getter.ts +6 -6
  211. package/src/talkpilot/leads/leads.schema.ts +33 -33
  212. package/src/talkpilot/leads/leads.types.ts +20 -20
  213. package/src/talkpilot/mongodb-client.ts +78 -78
  214. package/src/talkpilot/phone_numbers/__tests__/phone_numbers.spec.ts +530 -530
  215. package/src/talkpilot/phone_numbers/index.ts +2 -2
  216. package/src/talkpilot/phone_numbers/phone_numbers.getter.ts +240 -240
  217. package/src/talkpilot/phone_numbers/phone_numbers.schema.ts +17 -17
  218. package/src/talkpilot/phone_numbers/phone_numbers.types.ts +30 -30
  219. package/src/talkpilot/phone_numbers/phone_numbers.utils.ts +9 -9
  220. package/src/talkpilot/plans/__tests__/plans.spec.ts +70 -70
  221. package/src/talkpilot/plans/index.ts +2 -2
  222. package/src/talkpilot/plans/plans.getters.ts +132 -132
  223. package/src/talkpilot/plans/plans.types.ts +89 -89
  224. package/src/talkpilot/products/__tests__/products.getters.spec.ts +44 -44
  225. package/src/talkpilot/products/index.ts +2 -2
  226. package/src/talkpilot/products/products.getters.ts +12 -12
  227. package/src/talkpilot/products/products.types.ts +9 -9
  228. package/src/talkpilot/results/index.ts +7 -7
  229. package/src/talkpilot/results/results.getter.ts +39 -39
  230. package/src/talkpilot/results/results.schema.ts +25 -25
  231. package/src/talkpilot/results/results.types.ts +34 -34
  232. package/src/talkpilot/retry_analyze/__tests__/retryAnalyze.getters.spec.ts +156 -156
  233. package/src/talkpilot/retry_analyze/index.ts +2 -2
  234. package/src/talkpilot/retry_analyze/retryAnalyze.getters.ts +84 -84
  235. package/src/talkpilot/retry_analyze/retryAnalyze.types.ts +13 -13
  236. package/src/talkpilot/sessions/__tests__/sessions.spec.ts +147 -147
  237. package/src/talkpilot/sessions/index.ts +2 -2
  238. package/src/talkpilot/sessions/sessions.getter.ts +92 -92
  239. package/src/talkpilot/sessions/sessions.schema.ts +34 -34
  240. package/src/talkpilot/sessions/sessions.types.ts +30 -30
  241. package/src/talkpilot/subscriptions/__tests__/subscriptions.getters.utils.spec.ts +45 -45
  242. package/src/talkpilot/subscriptions/index.ts +3 -3
  243. package/src/talkpilot/subscriptions/subscriptions.getters.ts +146 -146
  244. package/src/talkpilot/subscriptions/subscriptions.getters.utils.ts +33 -33
  245. package/src/talkpilot/subscriptions/subscriptions.types.ts +66 -66
  246. package/src/talkpilot/utils/__tests__/query.utils.spec.ts +49 -49
  247. package/src/talkpilot/utils/query.utils.ts +21 -21
  248. package/src/test-utils/db-utils.ts +33 -33
  249. package/src/test-utils/factories/index.ts +19 -18
  250. package/src/test-utils/factories/municipal/cities.ts +16 -16
  251. package/src/test-utils/factories/municipal/departmentsSubjects.ts +37 -37
  252. package/src/test-utils/factories/municipal/muniIssues.ts +32 -32
  253. package/src/test-utils/factories/municipal/residents.ts +20 -0
  254. package/src/test-utils/factories/municipal/streets.ts +22 -22
  255. package/src/test-utils/factories/municipal/tickets.ts +39 -39
  256. package/src/test-utils/factories/talkpilot/agents.ts +19 -19
  257. package/src/test-utils/factories/talkpilot/calls.ts +37 -37
  258. package/src/test-utils/factories/talkpilot/clientAudioBuffers.ts +20 -20
  259. package/src/test-utils/factories/talkpilot/clientsConfig.ts +18 -18
  260. package/src/test-utils/factories/talkpilot/contextNotes.ts +40 -40
  261. package/src/test-utils/factories/talkpilot/flows.ts +33 -33
  262. package/src/test-utils/factories/talkpilot/groups.ts +33 -33
  263. package/src/test-utils/factories/talkpilot/phone_numbers.ts +22 -22
  264. package/src/test-utils/factories/talkpilot/sessions.ts +35 -35
  265. package/src/test-utils/factories/websitalk/knowledgeDocuments.ts +29 -29
  266. package/src/test-utils/factories/websitalk/scanResources.ts +26 -26
  267. package/src/test-utils/factories/websitalk/scans.ts +30 -30
  268. package/src/test-utils/factories/websitalk/websiteUrls.ts +17 -17
  269. package/src/utils/date.utils.ts +126 -126
  270. package/src/utils/shared.types.ts +4 -4
  271. package/src/utils/validation.ts +23 -23
  272. package/src/websitalk/index.ts +18 -18
  273. package/src/websitalk/knowledgeDocuments/__tests__/knowledgeDocuments.spec.ts +238 -238
  274. package/src/websitalk/knowledgeDocuments/index.ts +3 -3
  275. package/src/websitalk/knowledgeDocuments/knowledgeDocuments.constants.ts +26 -26
  276. package/src/websitalk/knowledgeDocuments/knowledgeDocuments.getters.ts +95 -95
  277. package/src/websitalk/knowledgeDocuments/knowledgeDocuments.types.ts +75 -75
  278. package/src/websitalk/mongodb-client.ts +61 -61
  279. package/src/websitalk/scanResources/__tests__/scanResources.spec.ts +267 -267
  280. package/src/websitalk/scanResources/index.ts +4 -4
  281. package/src/websitalk/scanResources/scanResources.constants.ts +60 -60
  282. package/src/websitalk/scanResources/scanResources.getters.ts +128 -128
  283. package/src/websitalk/scanResources/scanResources.types.ts +105 -105
  284. package/src/websitalk/scanResources/scanResources.utils.ts +31 -31
  285. package/src/websitalk/scans/__tests__/scans.spec.ts +362 -362
  286. package/src/websitalk/scans/index.ts +4 -4
  287. package/src/websitalk/scans/scans.constants.ts +18 -18
  288. package/src/websitalk/scans/scans.getters.ts +172 -172
  289. package/src/websitalk/scans/scans.types.ts +58 -58
  290. package/src/websitalk/scans/scans.utils.ts +5 -5
  291. package/src/websitalk/websiteUrls/__tests__/websiteUrls.spec.ts +77 -77
  292. package/src/websitalk/websiteUrls/index.ts +2 -2
  293. package/src/websitalk/websiteUrls/websiteUrls.getters.ts +65 -65
  294. package/src/websitalk/websiteUrls/websiteUrls.types.ts +13 -13
  295. package/tsconfig.json +23 -23
  296. package/src/municipal/muniIssues/__tests__/muniIssues.getters.spec.ts +0 -60
@@ -1,119 +1,119 @@
1
- import { getClientConfig } from "../clientsConfig/clientsConfig.getters";
2
- import { getClientsCollection } from "./clients.getters";
3
- import type {
4
- ClientQuotaState,
5
- QuotaAlertInput,
6
- } from "./clients.quota.types";
7
-
8
- export async function getClientQuotaState(
9
- clientId: string,
10
- ): Promise<ClientQuotaState | null> {
11
- const [config, client] = await Promise.all([
12
- getClientConfig(clientId),
13
- getClientsCollection().findOne({ clientId }),
14
- ]);
15
-
16
- const usage = config?.quota;
17
- if (usage?.totalQuota === undefined || usage?.totalQuota === null) return null;
18
-
19
- const settings = client?.quota;
20
-
21
- return {
22
- totalQuota: usage.totalQuota,
23
- usedQuota: usage.usedQuota ?? 0,
24
- expiresAt: usage.expiresAt,
25
- alerts: settings?.alerts,
26
- expiredSent: settings?.expiredSent,
27
- };
28
- }
29
-
30
- const ensureClientDoc = async (clientId: string): Promise<void> => {
31
- const now = new Date();
32
- await getClientsCollection().updateOne(
33
- { clientId },
34
- { $setOnInsert: { clientId, createdAt: now, updatedAt: now, credits: 0 } },
35
- { upsert: true },
36
- );
37
- await getClientsCollection().updateOne({ clientId }, { $set: { updatedAt: now } });
38
- };
39
-
40
- export async function updateQuotaAlertsConfig(
41
- clientId: string,
42
- alerts: QuotaAlertInput[],
43
- ): Promise<void> {
44
- const config = await getClientConfig(clientId);
45
- if (!config) {
46
- throw new Error(`No client config found for clientId: ${clientId}`);
47
- }
48
-
49
- await ensureClientDoc(clientId);
50
- const state = await getClientQuotaState(clientId);
51
- const existing = state?.alerts ?? [];
52
- const sentByThreshold = new Map(
53
- existing.filter((a) => a.thresholdSent).map((a) => [a.thresholdPct, true]),
54
- );
55
- const merged = alerts.map((alert) => ({
56
- thresholdPct: alert.thresholdPct,
57
- recipientEmails: alert.recipientEmails,
58
- ...(sentByThreshold.get(alert.thresholdPct) ? { thresholdSent: true as const } : {}),
59
- }));
60
-
61
- const result = await getClientsCollection().updateOne(
62
- { clientId },
63
- { $set: { "quota.alerts": merged, updatedAt: new Date() } },
64
- );
65
-
66
- if (result.matchedCount === 0) {
67
- throw new Error(`No client found for clientId: ${clientId}`);
68
- }
69
- }
70
-
71
- export async function markQuotaAlertSent(
72
- clientId: string,
73
- alertIndex: number,
74
- ): Promise<boolean> {
75
- await ensureClientDoc(clientId);
76
- const result = await getClientsCollection().findOneAndUpdate(
77
- {
78
- clientId,
79
- [`quota.alerts.${alertIndex}.thresholdSent`]: { $ne: true },
80
- },
81
- { $set: { [`quota.alerts.${alertIndex}.thresholdSent`]: true, updatedAt: new Date() } },
82
- { returnDocument: "after" },
83
- );
84
-
85
- return result !== null;
86
- }
87
-
88
- export async function markQuotaExpiredSent(clientId: string): Promise<boolean> {
89
- await ensureClientDoc(clientId);
90
- const result = await getClientsCollection().findOneAndUpdate(
91
- {
92
- clientId,
93
- "quota.expiredSent": { $ne: true },
94
- },
95
- { $set: { "quota.expiredSent": true, updatedAt: new Date() } },
96
- { returnDocument: "after" },
97
- );
98
-
99
- return result !== null;
100
- }
101
-
102
- export async function resetClientQuotaSettings(clientId: string): Promise<void> {
103
- const client = await getClientsCollection().findOne({ clientId });
104
- const alerts =
105
- client?.quota?.alerts?.map(({ thresholdSent: _s, ...alert }) => alert) ?? [];
106
-
107
- if (client) {
108
- await getClientsCollection().updateOne(
109
- { clientId },
110
- {
111
- $set: {
112
- ...(alerts.length ? { "quota.alerts": alerts } : {}),
113
- updatedAt: new Date(),
114
- },
115
- $unset: { "quota.expiredSent": "" },
116
- },
117
- );
118
- }
119
- }
1
+ import { getClientConfig } from "../clientsConfig/clientsConfig.getters";
2
+ import { getClientsCollection } from "./clients.getters";
3
+ import type {
4
+ ClientQuotaState,
5
+ QuotaAlertInput,
6
+ } from "./clients.quota.types";
7
+
8
+ export async function getClientQuotaState(
9
+ clientId: string,
10
+ ): Promise<ClientQuotaState | null> {
11
+ const [config, client] = await Promise.all([
12
+ getClientConfig(clientId),
13
+ getClientsCollection().findOne({ clientId }),
14
+ ]);
15
+
16
+ const usage = config?.quota;
17
+ if (usage?.totalQuota === undefined || usage?.totalQuota === null) return null;
18
+
19
+ const settings = client?.quota;
20
+
21
+ return {
22
+ totalQuota: usage.totalQuota,
23
+ usedQuota: usage.usedQuota ?? 0,
24
+ expiresAt: usage.expiresAt,
25
+ alerts: settings?.alerts,
26
+ expiredSent: settings?.expiredSent,
27
+ };
28
+ }
29
+
30
+ const ensureClientDoc = async (clientId: string): Promise<void> => {
31
+ const now = new Date();
32
+ await getClientsCollection().updateOne(
33
+ { clientId },
34
+ { $setOnInsert: { clientId, createdAt: now, updatedAt: now, credits: 0 } },
35
+ { upsert: true },
36
+ );
37
+ await getClientsCollection().updateOne({ clientId }, { $set: { updatedAt: now } });
38
+ };
39
+
40
+ export async function updateQuotaAlertsConfig(
41
+ clientId: string,
42
+ alerts: QuotaAlertInput[],
43
+ ): Promise<void> {
44
+ const config = await getClientConfig(clientId);
45
+ if (!config) {
46
+ throw new Error(`No client config found for clientId: ${clientId}`);
47
+ }
48
+
49
+ await ensureClientDoc(clientId);
50
+ const state = await getClientQuotaState(clientId);
51
+ const existing = state?.alerts ?? [];
52
+ const sentByThreshold = new Map(
53
+ existing.filter((a) => a.thresholdSent).map((a) => [a.thresholdPct, true]),
54
+ );
55
+ const merged = alerts.map((alert) => ({
56
+ thresholdPct: alert.thresholdPct,
57
+ recipientEmails: alert.recipientEmails,
58
+ ...(sentByThreshold.get(alert.thresholdPct) ? { thresholdSent: true as const } : {}),
59
+ }));
60
+
61
+ const result = await getClientsCollection().updateOne(
62
+ { clientId },
63
+ { $set: { "quota.alerts": merged, updatedAt: new Date() } },
64
+ );
65
+
66
+ if (result.matchedCount === 0) {
67
+ throw new Error(`No client found for clientId: ${clientId}`);
68
+ }
69
+ }
70
+
71
+ export async function markQuotaAlertSent(
72
+ clientId: string,
73
+ alertIndex: number,
74
+ ): Promise<boolean> {
75
+ await ensureClientDoc(clientId);
76
+ const result = await getClientsCollection().findOneAndUpdate(
77
+ {
78
+ clientId,
79
+ [`quota.alerts.${alertIndex}.thresholdSent`]: { $ne: true },
80
+ },
81
+ { $set: { [`quota.alerts.${alertIndex}.thresholdSent`]: true, updatedAt: new Date() } },
82
+ { returnDocument: "after" },
83
+ );
84
+
85
+ return result !== null;
86
+ }
87
+
88
+ export async function markQuotaExpiredSent(clientId: string): Promise<boolean> {
89
+ await ensureClientDoc(clientId);
90
+ const result = await getClientsCollection().findOneAndUpdate(
91
+ {
92
+ clientId,
93
+ "quota.expiredSent": { $ne: true },
94
+ },
95
+ { $set: { "quota.expiredSent": true, updatedAt: new Date() } },
96
+ { returnDocument: "after" },
97
+ );
98
+
99
+ return result !== null;
100
+ }
101
+
102
+ export async function resetClientQuotaSettings(clientId: string): Promise<void> {
103
+ const client = await getClientsCollection().findOne({ clientId });
104
+ const alerts =
105
+ client?.quota?.alerts?.map(({ thresholdSent: _s, ...alert }) => alert) ?? [];
106
+
107
+ if (client) {
108
+ await getClientsCollection().updateOne(
109
+ { clientId },
110
+ {
111
+ $set: {
112
+ ...(alerts.length ? { "quota.alerts": alerts } : {}),
113
+ updatedAt: new Date(),
114
+ },
115
+ $unset: { "quota.expiredSent": "" },
116
+ },
117
+ );
118
+ }
119
+ }
@@ -1,27 +1,27 @@
1
- export type QuotaThresholdAlert = {
2
- thresholdPct: number;
3
- recipientEmails: string[];
4
- thresholdSent?: boolean;
5
- };
6
-
7
- export type ClientQuotaSettings = {
8
- /** Up to 3 threshold alerts, each with up to 3 recipient emails. */
9
- alerts?: QuotaThresholdAlert[];
10
- /** Set after expired-quota email is sent successfully. */
11
- expiredSent?: boolean;
12
- };
13
-
14
- export type QuotaAlertInput = {
15
- thresholdPct: number;
16
- recipientEmails: string[];
17
- };
18
-
19
- export type ClientQuotaUsage = {
20
- totalQuota: number;
21
- usedQuota: number;
22
- expiresAt?: Date;
23
- /** Human-readable account label for quota alert emails. */
24
- clientDisplayName?: string;
25
- };
26
-
27
- export type ClientQuotaState = ClientQuotaUsage & ClientQuotaSettings;
1
+ export type QuotaThresholdAlert = {
2
+ thresholdPct: number;
3
+ recipientEmails: string[];
4
+ thresholdSent?: boolean;
5
+ };
6
+
7
+ export type ClientQuotaSettings = {
8
+ /** Up to 3 threshold alerts, each with up to 3 recipient emails. */
9
+ alerts?: QuotaThresholdAlert[];
10
+ /** Set after expired-quota email is sent successfully. */
11
+ expiredSent?: boolean;
12
+ };
13
+
14
+ export type QuotaAlertInput = {
15
+ thresholdPct: number;
16
+ recipientEmails: string[];
17
+ };
18
+
19
+ export type ClientQuotaUsage = {
20
+ totalQuota: number;
21
+ usedQuota: number;
22
+ expiresAt?: Date;
23
+ /** Human-readable account label for quota alert emails. */
24
+ clientDisplayName?: string;
25
+ };
26
+
27
+ export type ClientQuotaState = ClientQuotaUsage & ClientQuotaSettings;
@@ -1,18 +1,18 @@
1
- import { ObjectId } from "mongodb";
2
- import { ClientQuotaSettings } from "./clients.quota.types";
3
-
4
- export type { QuotaThresholdAlert, ClientQuotaSettings, QuotaAlertInput, ClientQuotaUsage, ClientQuotaState } from "./clients.quota.types";
5
-
6
- export type Client = {
7
- _id?: ObjectId;
8
- createdAt: Date;
9
- updatedAt: Date;
10
- clientId: string;
11
- isActive?: boolean | null;
12
- apiKey?: string | null;
13
- credits: number;
14
- managedBy?: string;
15
- quota?: ClientQuotaSettings;
16
- };
17
-
18
- export type ClientDoc = Client & { _id: ObjectId };
1
+ import { ObjectId } from "mongodb";
2
+ import { ClientQuotaSettings } from "./clients.quota.types";
3
+
4
+ export type { QuotaThresholdAlert, ClientQuotaSettings, QuotaAlertInput, ClientQuotaUsage, ClientQuotaState } from "./clients.quota.types";
5
+
6
+ export type Client = {
7
+ _id?: ObjectId;
8
+ createdAt: Date;
9
+ updatedAt: Date;
10
+ clientId: string;
11
+ isActive?: boolean | null;
12
+ apiKey?: string | null;
13
+ credits: number;
14
+ managedBy?: string;
15
+ quota?: ClientQuotaSettings;
16
+ };
17
+
18
+ export type ClientDoc = Client & { _id: ObjectId };
@@ -1,4 +1,4 @@
1
- export * from "./clients.getters";
2
- export * from "./clients.types";
3
- export * from "./clients.quota.getters";
4
- export * from "./clients.quota.types";
1
+ export * from "./clients.getters";
2
+ export * from "./clients.types";
3
+ export * from "./clients.quota.getters";
4
+ export * from "./clients.quota.types";