@talkpilot/core-db 1.3.6 → 1.3.8

package/src/municipal/tickets/tickets.getters.ts

@@ -1,121 +1,121 @@
-import { CityName, getDb, ObjectId, Ticket } from "../index";
-import { Collection, Filter, ObjectId as MongoObjectId } from "mongodb";
-
-/**
- * Generate a new ticket ID as a string
- * @returns A new ObjectId as a string
- */
-export const generateTicketId = (): string => {
-  return new ObjectId().toString();
-};
-
-export const getTicketsCollection = (): Collection<Ticket> => {
-  return getDb().collection<Ticket>("tickets");
-};
-
-export const findTickets = async (
-  filter: Filter<Ticket> = {},
-): Promise<Ticket[]> => {
-  return await getTicketsCollection().find(filter).toArray();
-};
-
-export const findTicketsByCallSid = async (
-  callSid: string,
-): Promise<Ticket[]> => {
-  return await getTicketsCollection().find({ callSid }).toArray();
-};
-
-export const getTicketById = async (
-  ticketId: string,
-): Promise<Ticket | null> => {
-  const ticket = await getTicketsCollection().findOne({
-    _id: new ObjectId(ticketId),
-  });
-  return ticket ? ticket : null;
-};
-
-export const createTicket = async (
-  ticketData: Omit<Ticket, "_id" | "createdAt" | "updatedAt">,
-  ticketId?: string,
-): Promise<MongoObjectId> => {
-  const ticket: Ticket = {
-    _id: ticketId ? new ObjectId(ticketId) : new ObjectId(),
-    ...ticketData,
-    createdAt: new Date(),
-    updatedAt: new Date(),
-  };
-  const { insertedId } = await getTicketsCollection().insertOne(ticket);
-  return insertedId;
-};
-
-export const updateTicket = async (
-  ticketId: string,
-  data: Partial<Omit<Ticket, "_id" | "createdAt" | "updatedAt">>,
-): Promise<Ticket | null> => {
-  const result = await getTicketsCollection().findOneAndUpdate(
-    { _id: new ObjectId(ticketId) },
-    { $set: { ...data, updatedAt: new Date() } },
-    { returnDocument: "after" },
-  );
-  return result || null;
-};
-
-export const deleteTicket = async (ticketId: string): Promise<boolean> => {
-  const result = await getTicketsCollection().deleteOne({
-    _id: new ObjectId(ticketId),
-  });
-  return result.deletedCount > 0;
-};
-
-export const findTicketByQuery = async (query: Partial<Ticket>) => {
-  return await getTicketsCollection().findOne(query);
-};
-/**
- * Update ticket files by external call number
- */
-export const updateTicketFilesByCallNumber = async (params: {
-  cityName: CityName;
-  callNumber: string;
-  image1?: string;
-  image2?: string;
-  image3?: string;
-}): Promise<boolean> => {
-  const { cityName, callNumber, image1, image2, image3 } = params;
-
-  const update: any = {};
-  if (image1) update["externalCallFields.image1"] = image1;
-  if (image2) update["externalCallFields.image2"] = image2;
-  if (image3) update["externalCallFields.image3"] = image3;
-
-  const result = await getTicketsCollection().updateOne(
-    {
-      cityName: cityName,
-      "externalCallFields.call_number": callNumber,
-    },
-    {
-      $set: {
-        ...update,
-        updatedAt: new Date(),
-      },
-    },
-  );
-
-  return result.matchedCount > 0;
-};
-
-/**
- * Remove guestJwt from ticket by external call number
- */
-export const clearGuestJwtByCallNumber = async (params: {
-  cityName: CityName;
-  callNumber: string;
-}): Promise<boolean> => {
-  const { cityName, callNumber } = params;
-
-  const result = await getTicketsCollection().updateOne(
-    { cityName, "externalCallFields.call_number": callNumber },
-    { $unset: { guestJwt: "" }, $set: { updatedAt: new Date() } },
-  );
-
-  return result.matchedCount > 0;
-};
+import { CityName, getDb, ObjectId, Ticket } from "../index";
+import { Collection, Filter, ObjectId as MongoObjectId } from "mongodb";
+
+/**
+ * Generate a new ticket ID as a string
+ * @returns A new ObjectId as a string
+ */
+export const generateTicketId = (): string => {
+  return new ObjectId().toString();
+};
+
+export const getTicketsCollection = (): Collection<Ticket> => {
+  return getDb().collection<Ticket>("tickets");
+};
+
+export const findTickets = async (
+  filter: Filter<Ticket> = {},
+): Promise<Ticket[]> => {
+  return await getTicketsCollection().find(filter).toArray();
+};
+
+export const findTicketsByCallSid = async (
+  callSid: string,
+): Promise<Ticket[]> => {
+  return await getTicketsCollection().find({ callSid }).toArray();
+};
+
+export const getTicketById = async (
+  ticketId: string,
+): Promise<Ticket | null> => {
+  const ticket = await getTicketsCollection().findOne({
+    _id: new ObjectId(ticketId),
+  });
+  return ticket ? ticket : null;
+};
+
+export const createTicket = async (
+  ticketData: Omit<Ticket, "_id" | "createdAt" | "updatedAt">,
+  ticketId?: string,
+): Promise<MongoObjectId> => {
+  const ticket: Ticket = {
+    _id: ticketId ? new ObjectId(ticketId) : new ObjectId(),
+    ...ticketData,
+    createdAt: new Date(),
+    updatedAt: new Date(),
+  };
+  const { insertedId } = await getTicketsCollection().insertOne(ticket);
+  return insertedId;
+};
+
+export const updateTicket = async (
+  ticketId: string,
+  data: Partial<Omit<Ticket, "_id" | "createdAt" | "updatedAt">>,
+): Promise<Ticket | null> => {
+  const result = await getTicketsCollection().findOneAndUpdate(
+    { _id: new ObjectId(ticketId) },
+    { $set: { ...data, updatedAt: new Date() } },
+    { returnDocument: "after" },
+  );
+  return result || null;
+};
+
+export const deleteTicket = async (ticketId: string): Promise<boolean> => {
+  const result = await getTicketsCollection().deleteOne({
+    _id: new ObjectId(ticketId),
+  });
+  return result.deletedCount > 0;
+};
+
+export const findTicketByQuery = async (query: Partial<Ticket>) => {
+  return await getTicketsCollection().findOne(query);
+};
+/**
+ * Update ticket files by external call number
+ */
+export const updateTicketFilesByCallNumber = async (params: {
+  cityName: CityName;
+  callNumber: string;
+  image1?: string;
+  image2?: string;
+  image3?: string;
+}): Promise<boolean> => {
+  const { cityName, callNumber, image1, image2, image3 } = params;
+
+  const update: any = {};
+  if (image1) update["externalCallFields.image1"] = image1;
+  if (image2) update["externalCallFields.image2"] = image2;
+  if (image3) update["externalCallFields.image3"] = image3;
+
+  const result = await getTicketsCollection().updateOne(
+    {
+      cityName: cityName,
+      "externalCallFields.call_number": callNumber,
+    },
+    {
+      $set: {
+        ...update,
+        updatedAt: new Date(),
+      },
+    },
+  );
+
+  return result.matchedCount > 0;
+};
+
+/**
+ * Remove guestJwt from ticket by external call number
+ */
+export const clearGuestJwtByCallNumber = async (params: {
+  cityName: CityName;
+  callNumber: string;
+}): Promise<boolean> => {
+  const { cityName, callNumber } = params;
+
+  const result = await getTicketsCollection().updateOne(
+    { cityName, "externalCallFields.call_number": callNumber },
+    { $unset: { guestJwt: "" }, $set: { updatedAt: new Date() } },
+  );
+
+  return result.matchedCount > 0;
+};

package/src/municipal/tickets/tickets.statistics.aggregation.ts

@@ -1,96 +1,96 @@
-import { CityName } from "../utils/types";
-import { STATS_MAX_TIME_MS, UNCLASSIFIED } from "./tickets.constants";
-import { getTicketsCollection } from "./tickets.getters";
-import type { TicketStatsDateRange } from "./tickets.types";
-
-const effectiveSubjectIdExpr = {
-  $ifNull: [
-    "$externalCallFields.event_subject_id",
-    {
-      $ifNull: [
-        "$externalCallFields.event_sub_subject_id",
-        "$externalCallFields.event_sub_subject_id2",
-      ],
-    },
-  ],
-};
-
-const departmentsSubjectsLookup = (cityName: CityName) => ({
-  $lookup: {
-    from: "departmentsSubjects",
-    let: { sid: "$effectiveSubjectId", c: cityName },
-    pipeline: [
-      {
-        $match: {
-          $expr: {
-            $and: [
-              { $eq: ["$cityName", "$$c"] },
-              {
-                $or: [
-                  { $eq: ["$subject_id", { $ifNull: ["$$sid", ""] }] },
-                  { $eq: ["$sub_subject_id", { $ifNull: ["$$sid", ""] }] },
-                  { $eq: ["$sub_subject_id2", { $ifNull: ["$$sid", ""] }] },
-                ],
-              },
-            ],
-          },
-        },
-      },
-      { $limit: 1 },
-    ],
-    as: "subj",
-  },
-});
-
-const subjectLabelExpr = {
-  $cond: {
-    if: { $gt: [{ $size: "$subj" }, 0] },
-    then: {
-      $ifNull: [{ $arrayElemAt: ["$subj.subjectName", 0] }, UNCLASSIFIED],
-    },
-    else: UNCLASSIFIED,
-  },
-};
-
-const ticketDateMatch = (dateRange: TicketStatsDateRange) => ({
-  createdAt: { $gte: dateRange.from, $lte: dateRange.to },
-});
-
-export const isDraftSubjectIdExpr = {
-  $or: [
-    { $eq: [{ $type: "$externalCallFields.event_subject_id" }, "missing"] },
-    { $eq: ["$externalCallFields.event_subject_id", null] },
-    {
-      $regexMatch: {
-        input: { $ifNull: ["$externalCallFields.event_subject_id", ""] },
-        regex: /^\s*$/,
-      },
-    },
-  ],
-};
-
-export const ticketsWithCallSidMatch = (
-  cityName: CityName,
-  dateRange: TicketStatsDateRange,
-) => ({
-  cityName,
-  callSid: { $exists: true, $nin: [null, ""] },
-  ...ticketDateMatch(dateRange),
-});
-
-export const subjectResolutionStages = (
-  cityName: CityName,
-): Record<string, unknown>[] => [
-  { $addFields: { effectiveSubjectId: effectiveSubjectIdExpr } },
-  departmentsSubjectsLookup(cityName),
-  { $addFields: { subject: subjectLabelExpr } },
-];
-
-export const runTicketStatsAggregate = async <T>(
-  pipeline: Record<string, unknown>[],
-): Promise<T[]> => {
-  const rows = await getTicketsCollection()
-    .aggregate(pipeline, { maxTimeMS: STATS_MAX_TIME_MS })
-    .toArray();
-  return rows as T[];
-};
+import { CityName } from "../utils/types";
+import { STATS_MAX_TIME_MS, UNCLASSIFIED } from "./tickets.constants";
+import { getTicketsCollection } from "./tickets.getters";
+import type { TicketStatsDateRange } from "./tickets.types";
+
+const effectiveSubjectIdExpr = {
+  $ifNull: [
+    "$externalCallFields.event_subject_id",
+    {
+      $ifNull: [
+        "$externalCallFields.event_sub_subject_id",
+        "$externalCallFields.event_sub_subject_id2",
+      ],
+    },
+  ],
+};
+
+const departmentsSubjectsLookup = (cityName: CityName) => ({
+  $lookup: {
+    from: "departmentsSubjects",
+    let: { sid: "$effectiveSubjectId", c: cityName },
+    pipeline: [
+      {
+        $match: {
+          $expr: {
+            $and: [
+              { $eq: ["$cityName", "$$c"] },
+              {
+                $or: [
+                  { $eq: ["$subject_id", { $ifNull: ["$$sid", ""] }] },
+                  { $eq: ["$sub_subject_id", { $ifNull: ["$$sid", ""] }] },
+                  { $eq: ["$sub_subject_id2", { $ifNull: ["$$sid", ""] }] },
+                ],
+              },
+            ],
+          },
+        },
+      },
+      { $limit: 1 },
+    ],
+    as: "subj",
+  },
+});
+
+const subjectLabelExpr = {
+  $cond: {
+    if: { $gt: [{ $size: "$subj" }, 0] },
+    then: {
+      $ifNull: [{ $arrayElemAt: ["$subj.subjectName", 0] }, UNCLASSIFIED],
+    },
+    else: UNCLASSIFIED,
+  },
+};
+
+const ticketDateMatch = (dateRange: TicketStatsDateRange) => ({
+  createdAt: { $gte: dateRange.from, $lte: dateRange.to },
+});
+
+export const isDraftSubjectIdExpr = {
+  $or: [
+    { $eq: [{ $type: "$externalCallFields.event_subject_id" }, "missing"] },
+    { $eq: ["$externalCallFields.event_subject_id", null] },
+    {
+      $regexMatch: {
+        input: { $ifNull: ["$externalCallFields.event_subject_id", ""] },
+        regex: /^\s*$/,
+      },
+    },
+  ],
+};
+
+export const ticketsWithCallSidMatch = (
+  cityName: CityName,
+  dateRange: TicketStatsDateRange,
+) => ({
+  cityName,
+  callSid: { $exists: true, $nin: [null, ""] },
+  ...ticketDateMatch(dateRange),
+});
+
+export const subjectResolutionStages = (
+  cityName: CityName,
+): Record<string, unknown>[] => [
+  { $addFields: { effectiveSubjectId: effectiveSubjectIdExpr } },
+  departmentsSubjectsLookup(cityName),
+  { $addFields: { subject: subjectLabelExpr } },
+];
+
+export const runTicketStatsAggregate = async <T>(
+  pipeline: Record<string, unknown>[],
+): Promise<T[]> => {
+  const rows = await getTicketsCollection()
+    .aggregate(pipeline, { maxTimeMS: STATS_MAX_TIME_MS })
+    .toArray();
+  return rows as T[];
+};

package/src/municipal/tickets/tickets.statistics.getters.ts

@@ -1,71 +1,109 @@
-import { CityName } from "../utils/types";
-import {
-  DAY_MS,
-  endOfCalendarDayInTz,
-  startOfCalendarDayInTz,
-} from "../../utils/date.utils";
-import { DEFAULT_LOOKBACK_DAYS } from "./tickets.constants";
-import {
-  isDraftSubjectIdExpr,
-  runTicketStatsAggregate,
-  subjectResolutionStages,
-  ticketsWithCallSidMatch,
-} from "./tickets.statistics.aggregation";
-import type {
-  TicketStatsDateRange,
-  TicketStatsDateScope,
-  TicketSubjectRow,
-} from "./tickets.types";
-
-export const resolveTicketStatsDateRange = (
-  dateScope?: TicketStatsDateScope,
-): TicketStatsDateRange => {
-  const to = dateScope?.to ?? new Date();
-  const from =
-    dateScope?.from ?? new Date(to.getTime() - DEFAULT_LOOKBACK_DAYS * DAY_MS);
-  return { from, to };
-};
-
-export const ticketStatsDateScopeFromYmd = (
-  startStr: string,
-  endStr: string,
-  timezone: string,
-): TicketStatsDateScope => ({
-  from: startOfCalendarDayInTz(startStr, timezone),
-  to: endOfCalendarDayInTz(endStr, timezone),
-});
-
-export const findCallSidsWithTicketsByCity = async (
-  cityName: CityName,
-  dateScope?: TicketStatsDateScope,
-): Promise<string[]> => {
-  const rows = await runTicketStatsAggregate<{ _id: unknown }>([
-    { $match: ticketsWithCallSidMatch(cityName, resolveTicketStatsDateRange(dateScope)) },
-    { $group: { _id: "$callSid" } },
-  ]);
-  return rows.map((r) => String(r._id)).filter(Boolean);
-};
-
-export const findCallSidsWithDraftTicketsByCity = async (
-  cityName: CityName,
-  dateScope?: TicketStatsDateScope,
-): Promise<string[]> => {
-  const rows = await runTicketStatsAggregate<{ _id: unknown }>([
-    { $match: ticketsWithCallSidMatch(cityName, resolveTicketStatsDateRange(dateScope)) },
-    { $match: { $expr: isDraftSubjectIdExpr } },
-    { $group: { _id: "$callSid" } },
-  ]);
-  return rows.map((r) => String(r._id)).filter(Boolean);
-};
-
-export const findTicketSubjectRows = async (
-  cityName: CityName,
-  dateScope?: TicketStatsDateScope,
-): Promise<TicketSubjectRow[]> => {
-  const rows = await runTicketStatsAggregate<{ callSid: unknown; subject: unknown }>([
-    { $match: ticketsWithCallSidMatch(cityName, resolveTicketStatsDateRange(dateScope)) },
-    ...subjectResolutionStages(cityName),
-    { $project: { _id: 0, callSid: 1, subject: 1 } },
-  ]);
-  return rows.map((r) => ({ callSid: String(r.callSid), subject: String(r.subject) }));
-};
+import { CityName } from "../utils/types";
+import {
+  DAY_MS,
+  endOfCalendarDayInTz,
+  startOfCalendarDayInTz,
+} from "../../utils/date.utils";
+import { DEFAULT_LOOKBACK_DAYS } from "./tickets.constants";
+import {
+  isDraftSubjectIdExpr,
+  runTicketStatsAggregate,
+  subjectResolutionStages,
+  ticketsWithCallSidMatch,
+} from "./tickets.statistics.aggregation";
+import type {
+  TicketStatsDateRange,
+  TicketStatsDateScope,
+  TicketSubjectRow,
+} from "./tickets.types";
+
+export const resolveTicketStatsDateRange = (
+  dateScope?: TicketStatsDateScope,
+): TicketStatsDateRange => {
+  const to = dateScope?.to ?? new Date();
+  const from =
+    dateScope?.from ?? new Date(to.getTime() - DEFAULT_LOOKBACK_DAYS * DAY_MS);
+  return { from, to };
+};
+
+export const ticketStatsDateScopeFromYmd = (
+  startStr: string,
+  endStr: string,
+  timezone: string,
+): TicketStatsDateScope => ({
+  from: startOfCalendarDayInTz(startStr, timezone),
+  to: endOfCalendarDayInTz(endStr, timezone),
+});
+
+/** @deprecated Use findCallSidTicketCountsByCity instead. */
+export const findCallSidsWithTicketsByCity = async (
+  cityName: CityName,
+  dateScope?: TicketStatsDateScope,
+): Promise<string[]> => {
+  const rows = await runTicketStatsAggregate<{ _id: unknown }>([
+    { $match: ticketsWithCallSidMatch(cityName, resolveTicketStatsDateRange(dateScope)) },
+    { $group: { _id: "$callSid" } },
+  ]);
+  return rows
+    .filter((r) => r._id != null && r._id !== "")
+    .map((r) => String(r._id));
+};
+
+export const findCallSidTicketCountsByCity = async (
+  cityName: CityName,
+  dateScope?: TicketStatsDateScope,
+): Promise<Map<string, number>> => {
+  const rows = await runTicketStatsAggregate<{ _id: unknown; ticketCount: number }>([
+    { $match: ticketsWithCallSidMatch(cityName, resolveTicketStatsDateRange(dateScope)) },
+    { $group: { _id: "$callSid", ticketCount: { $sum: 1 } } },
+  ]);
+  return new Map(
+    rows
+      .filter((r) => r._id != null && r._id !== "")
+      .map((r) => [String(r._id), r.ticketCount])
+  );
+};
+
+
+/** @deprecated Use findCallSidDraftTicketCountsByCity instead. */
+export const findCallSidsWithDraftTicketsByCity = async (
+  cityName: CityName,
+  dateScope?: TicketStatsDateScope,
+): Promise<string[]> => {
+  const rows = await runTicketStatsAggregate<{ _id: unknown }>([
+    { $match: ticketsWithCallSidMatch(cityName, resolveTicketStatsDateRange(dateScope)) },
+    { $match: { $expr: isDraftSubjectIdExpr } },
+    { $group: { _id: "$callSid" } },
+  ]);
+  return rows
+    .filter((r) => r._id != null && r._id !== "")
+    .map((r) => String(r._id));
+};
+
+export const findCallSidDraftTicketCountsByCity = async (
+  cityName: CityName,
+  dateScope?: TicketStatsDateScope,
+): Promise<Map<string, number>> => {
+  const rows = await runTicketStatsAggregate<{ _id: unknown; ticketCount: number }>([
+    { $match: ticketsWithCallSidMatch(cityName, resolveTicketStatsDateRange(dateScope)) },
+    { $match: { $expr: isDraftSubjectIdExpr } },
+    { $group: { _id: "$callSid", ticketCount: { $sum: 1 } } },
+  ]);
+  return new Map(
+    rows
+      .filter((r) => r._id != null && r._id !== "")
+      .map((r) => [String(r._id), r.ticketCount])
+  );
+};
+
+export const findTicketSubjectRows = async (
+  cityName: CityName,
+  dateScope?: TicketStatsDateScope,
+): Promise<TicketSubjectRow[]> => {
+  const rows = await runTicketStatsAggregate<{ callSid: unknown; subject: unknown }>([
+    { $match: ticketsWithCallSidMatch(cityName, resolveTicketStatsDateRange(dateScope)) },
+    ...subjectResolutionStages(cityName),
+    { $project: { _id: 0, callSid: 1, subject: 1 } },
+  ]);
+  return rows.map((r) => ({ callSid: String(r.callSid), subject: String(r.subject) }));
+};