@talkpilot/core-db 1.3.39 → 1.3.40

package/src/talkpilot/clients/clients.quota.getters.ts

@@ -1,119 +1,119 @@
-import { getClientConfig } from "../clientsConfig/clientsConfig.getters";
-import { getClientsCollection } from "./clients.getters";
-import type {
-  ClientQuotaState,
-  QuotaAlertInput,
-} from "./clients.quota.types";
-
-export async function getClientQuotaState(
-  clientId: string,
-): Promise<ClientQuotaState | null> {
-  const [config, client] = await Promise.all([
-    getClientConfig(clientId),
-    getClientsCollection().findOne({ clientId }),
-  ]);
-
-  const usage = config?.quota;
-  if (usage?.totalQuota === undefined || usage?.totalQuota === null) return null;
-
-  const settings = client?.quota;
-
-  return {
-    totalQuota: usage.totalQuota,
-    usedQuota: usage.usedQuota ?? 0,
-    expiresAt: usage.expiresAt,
-    alerts: settings?.alerts,
-    expiredSent: settings?.expiredSent,
-  };
-}
-
-const ensureClientDoc = async (clientId: string): Promise<void> => {
-  const now = new Date();
-  await getClientsCollection().updateOne(
-    { clientId },
-    { $setOnInsert: { clientId, createdAt: now, updatedAt: now, credits: 0 } },
-    { upsert: true },
-  );
-  await getClientsCollection().updateOne({ clientId }, { $set: { updatedAt: now } });
-};
-
-export async function updateQuotaAlertsConfig(
-  clientId: string,
-  alerts: QuotaAlertInput[],
-): Promise<void> {
-  const config = await getClientConfig(clientId);
-  if (!config) {
-    throw new Error(`No client config found for clientId: ${clientId}`);
-  }
-
-  await ensureClientDoc(clientId);
-  const state = await getClientQuotaState(clientId);
-  const existing = state?.alerts ?? [];
-  const sentByThreshold = new Map(
-    existing.filter((a) => a.thresholdSent).map((a) => [a.thresholdPct, true]),
-  );
-  const merged = alerts.map((alert) => ({
-    thresholdPct: alert.thresholdPct,
-    recipientEmails: alert.recipientEmails,
-    ...(sentByThreshold.get(alert.thresholdPct) ? { thresholdSent: true as const } : {}),
-  }));
-
-  const result = await getClientsCollection().updateOne(
-    { clientId },
-    { $set: { "quota.alerts": merged, updatedAt: new Date() } },
-  );
-
-  if (result.matchedCount === 0) {
-    throw new Error(`No client found for clientId: ${clientId}`);
-  }
-}
-
-export async function markQuotaAlertSent(
-  clientId: string,
-  alertIndex: number,
-): Promise<boolean> {
-  await ensureClientDoc(clientId);
-  const result = await getClientsCollection().findOneAndUpdate(
-    {
-      clientId,
-      [`quota.alerts.${alertIndex}.thresholdSent`]: { $ne: true },
-    },
-    { $set: { [`quota.alerts.${alertIndex}.thresholdSent`]: true, updatedAt: new Date() } },
-    { returnDocument: "after" },
-  );
-
-  return result !== null;
-}
-
-export async function markQuotaExpiredSent(clientId: string): Promise<boolean> {
-  await ensureClientDoc(clientId);
-  const result = await getClientsCollection().findOneAndUpdate(
-    {
-      clientId,
-      "quota.expiredSent": { $ne: true },
-    },
-    { $set: { "quota.expiredSent": true, updatedAt: new Date() } },
-    { returnDocument: "after" },
-  );
-
-  return result !== null;
-}
-
-export async function resetClientQuotaSettings(clientId: string): Promise<void> {
-  const client = await getClientsCollection().findOne({ clientId });
-  const alerts =
-    client?.quota?.alerts?.map(({ thresholdSent: _s, ...alert }) => alert) ?? [];
-
-  if (client) {
-    await getClientsCollection().updateOne(
-      { clientId },
-      {
-        $set: {
-          ...(alerts.length ? { "quota.alerts": alerts } : {}),
-          updatedAt: new Date(),
-        },
-        $unset: { "quota.expiredSent": "" },
-      },
-    );
-  }
-}
+import { getClientConfig } from "../clientsConfig/clientsConfig.getters";
+import { getClientsCollection } from "./clients.getters";
+import type {
+  ClientQuotaState,
+  QuotaAlertInput,
+} from "./clients.quota.types";
+
+export async function getClientQuotaState(
+  clientId: string,
+): Promise<ClientQuotaState | null> {
+  const [config, client] = await Promise.all([
+    getClientConfig(clientId),
+    getClientsCollection().findOne({ clientId }),
+  ]);
+
+  const usage = config?.quota;
+  if (usage?.totalQuota === undefined || usage?.totalQuota === null) return null;
+
+  const settings = client?.quota;
+
+  return {
+    totalQuota: usage.totalQuota,
+    usedQuota: usage.usedQuota ?? 0,
+    expiresAt: usage.expiresAt,
+    alerts: settings?.alerts,
+    expiredSent: settings?.expiredSent,
+  };
+}
+
+const ensureClientDoc = async (clientId: string): Promise<void> => {
+  const now = new Date();
+  await getClientsCollection().updateOne(
+    { clientId },
+    { $setOnInsert: { clientId, createdAt: now, updatedAt: now, credits: 0 } },
+    { upsert: true },
+  );
+  await getClientsCollection().updateOne({ clientId }, { $set: { updatedAt: now } });
+};
+
+export async function updateQuotaAlertsConfig(
+  clientId: string,
+  alerts: QuotaAlertInput[],
+): Promise<void> {
+  const config = await getClientConfig(clientId);
+  if (!config) {
+    throw new Error(`No client config found for clientId: ${clientId}`);
+  }
+
+  await ensureClientDoc(clientId);
+  const state = await getClientQuotaState(clientId);
+  const existing = state?.alerts ?? [];
+  const sentByThreshold = new Map(
+    existing.filter((a) => a.thresholdSent).map((a) => [a.thresholdPct, true]),
+  );
+  const merged = alerts.map((alert) => ({
+    thresholdPct: alert.thresholdPct,
+    recipientEmails: alert.recipientEmails,
+    ...(sentByThreshold.get(alert.thresholdPct) ? { thresholdSent: true as const } : {}),
+  }));
+
+  const result = await getClientsCollection().updateOne(
+    { clientId },
+    { $set: { "quota.alerts": merged, updatedAt: new Date() } },
+  );
+
+  if (result.matchedCount === 0) {
+    throw new Error(`No client found for clientId: ${clientId}`);
+  }
+}
+
+export async function markQuotaAlertSent(
+  clientId: string,
+  alertIndex: number,
+): Promise<boolean> {
+  await ensureClientDoc(clientId);
+  const result = await getClientsCollection().findOneAndUpdate(
+    {
+      clientId,
+      [`quota.alerts.${alertIndex}.thresholdSent`]: { $ne: true },
+    },
+    { $set: { [`quota.alerts.${alertIndex}.thresholdSent`]: true, updatedAt: new Date() } },
+    { returnDocument: "after" },
+  );
+
+  return result !== null;
+}
+
+export async function markQuotaExpiredSent(clientId: string): Promise<boolean> {
+  await ensureClientDoc(clientId);
+  const result = await getClientsCollection().findOneAndUpdate(
+    {
+      clientId,
+      "quota.expiredSent": { $ne: true },
+    },
+    { $set: { "quota.expiredSent": true, updatedAt: new Date() } },
+    { returnDocument: "after" },
+  );
+
+  return result !== null;
+}
+
+export async function resetClientQuotaSettings(clientId: string): Promise<void> {
+  const client = await getClientsCollection().findOne({ clientId });
+  const alerts =
+    client?.quota?.alerts?.map(({ thresholdSent: _s, ...alert }) => alert) ?? [];
+
+  if (client) {
+    await getClientsCollection().updateOne(
+      { clientId },
+      {
+        $set: {
+          ...(alerts.length ? { "quota.alerts": alerts } : {}),
+          updatedAt: new Date(),
+        },
+        $unset: { "quota.expiredSent": "" },
+      },
+    );
+  }
+}

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

@@ -1,27 +1,27 @@
-export type QuotaThresholdAlert = {
-  thresholdPct: number;
-  recipientEmails: string[];
-  thresholdSent?: boolean;
-};
-
-export type ClientQuotaSettings = {
-  /** Up to 3 threshold alerts, each with up to 3 recipient emails. */
-  alerts?: QuotaThresholdAlert[];
-  /** Set after expired-quota email is sent successfully. */
-  expiredSent?: boolean;
-};
-
-export type QuotaAlertInput = {
-  thresholdPct: number;
-  recipientEmails: string[];
-};
-
-export type ClientQuotaUsage = {
-  totalQuota: number;
-  usedQuota: number;
-  expiresAt?: Date;
-  /** Human-readable account label for quota alert emails. */
-  clientDisplayName?: string;
-};
-
-export type ClientQuotaState = ClientQuotaUsage & ClientQuotaSettings;
+export type QuotaThresholdAlert = {
+  thresholdPct: number;
+  recipientEmails: string[];
+  thresholdSent?: boolean;
+};
+
+export type ClientQuotaSettings = {
+  /** Up to 3 threshold alerts, each with up to 3 recipient emails. */
+  alerts?: QuotaThresholdAlert[];
+  /** Set after expired-quota email is sent successfully. */
+  expiredSent?: boolean;
+};
+
+export type QuotaAlertInput = {
+  thresholdPct: number;
+  recipientEmails: string[];
+};
+
+export type ClientQuotaUsage = {
+  totalQuota: number;
+  usedQuota: number;
+  expiresAt?: Date;
+  /** Human-readable account label for quota alert emails. */
+  clientDisplayName?: string;
+};
+
+export type ClientQuotaState = ClientQuotaUsage & ClientQuotaSettings;

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

@@ -1,18 +1,18 @@
-import { ObjectId } from "mongodb";
-import { ClientQuotaSettings } from "./clients.quota.types";
-
-export type { QuotaThresholdAlert, ClientQuotaSettings, QuotaAlertInput, ClientQuotaUsage, ClientQuotaState } from "./clients.quota.types";
-
-export type Client = {
-  _id?: ObjectId;
-  createdAt: Date;
-  updatedAt: Date;
-  clientId: string;
-  isActive?: boolean | null;
-  apiKey?: string | null;
-  credits: number;
-  managedBy?: string;
-  quota?: ClientQuotaSettings;
-};
-
-export type ClientDoc = Client & { _id: ObjectId };
+import { ObjectId } from "mongodb";
+import { ClientQuotaSettings } from "./clients.quota.types";
+
+export type { QuotaThresholdAlert, ClientQuotaSettings, QuotaAlertInput, ClientQuotaUsage, ClientQuotaState } from "./clients.quota.types";
+
+export type Client = {
+  _id?: ObjectId;
+  createdAt: Date;
+  updatedAt: Date;
+  clientId: string;
+  isActive?: boolean | null;
+  apiKey?: string | null;
+  credits: number;
+  managedBy?: string;
+  quota?: ClientQuotaSettings;
+};
+
+export type ClientDoc = Client & { _id: ObjectId };

package/src/talkpilot/clients/index.ts

@@ -1,4 +1,4 @@
-export * from "./clients.getters";
-export * from "./clients.types";
-export * from "./clients.quota.getters";
-export * from "./clients.quota.types";
+export * from "./clients.getters";
+export * from "./clients.types";
+export * from "./clients.quota.getters";
+export * from "./clients.quota.types";