@talkpilot/core-db 1.3.60 → 1.3.61

package/dist/websitalk/scans/scans.types.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"scans.types.d.ts","sourceRoot":"","sources":["../../../src/websitalk/scans/scans.types.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,MAAM,EAAE,MAAM,SAAS,CAAC;AAEjC,OAAO,KAAK,EAAE,aAAa,EAAE,MAAM,mBAAmB,CAAC;AAEvD,MAAM,MAAM,UAAU,GAAG,CAAC,OAAO,aAAa,CAAC,CAAC,MAAM,OAAO,aAAa,CAAC,CAAC;AAE5E,MAAM,MAAM,eAAe,GAAG;IAC5B,GAAG,EAAE,MAAM,CAAC;IACZ,YAAY,CAAC,EAAE,MAAM,GAAG,IAAI,CAAC;IAC7B,KAAK,CAAC,EAAE,MAAM,GAAG,IAAI,CAAC;IACtB,EAAE,CAAC,EAAE,MAAM,GAAG,IAAI,CAAC;IACnB,QAAQ,CAAC,EAAE,MAAM,GAAG,IAAI,CAAC;IACzB,UAAU,CAAC,EAAE,MAAM,GAAG,IAAI,CAAC;IAC3B,MAAM,CAAC,EAAE,MAAM,GAAG,IAAI,CAAC;IACvB,OAAO,CAAC,EAAE,MAAM,GAAG,IAAI,CAAC;IACxB,WAAW,CAAC,EAAE,MAAM,GAAG,IAAI,CAAC;IAC5B,YAAY,CAAC,EAAE,MAAM,GAAG,IAAI,CAAC;IAC7B,WAAW,CAAC,EAAE,OAAO,CAAC;IACtB,WAAW,CAAC,EAAE,MAAM,GAAG,IAAI,CAAC;IAC5B,SAAS,CAAC,EAAE,MAAM,GAAG,IAAI,CAAC;IAC1B,SAAS,CAAC,EAAE,IAAI,GAAG,IAAI,CAAC;IACxB,SAAS,CAAC,EAAE,IAAI,CAAC;IACjB,SAAS,CAAC,EAAE,IAAI,CAAC;CAClB,CAAC;AAEF,MAAM,MAAM,YAAY,GAAG;IACzB,IAAI,EAAE,MAAM,CAAC;IACb,MAAM,EAAE,MAAM,CAAC;CAChB,CAAC;AAEF,MAAM,MAAM,IAAI,GAAG;IACjB,QAAQ,EAAE,MAAM,CAAC;IACjB,YAAY,EAAE,MAAM,CAAC;IACrB,iBAAiB,CAAC,EAAE,MAAM,GAAG,IAAI,CAAC;IAClC,MAAM,EAAE,UAAU,CAAC;IACnB,gBAAgB,EAAE,MAAM,CAAC;IACzB,YAAY,EAAE,MAAM,CAAC;IACrB,YAAY,EAAE,MAAM,CAAC;IACrB,aAAa,CAAC,EAAE,MAAM,CAAC;IACvB,iBAAiB,CAAC,EAAE,MAAM,CAAC;IAC3B,OAAO,EAAE,YAAY,EAAE,CAAC;IACxB,UAAU,EAAE,eAAe,CAAC;IAC5B,SAAS,EAAE,IAAI,CAAC;IAChB,SAAS,EAAE,IAAI,CAAC;IAKhB,SAAS,CAAC,EAAE,IAAI,CAAC;IACjB,QAAQ,CAAC,EAAE,IAAI,GAAG,IAAI,CAAC;IAKvB,WAAW,CAAC,EAAE,IAAI,GAAG,IAAI,CAAC;IAC1B,cAAc,CAAC,EAAE,MAAM,CAAC;IACxB,oBAAoB,CAAC,EAAE,MAAM,CAAC;IAC9B,cAAc,CAAC,EAAE,MAAM,CAAC;CACzB,CAAC;AAEF,MAAM,MAAM,OAAO,GAAG,MAAM,CAAC,IAAI,CAAC,CAAC"}
+{"version":3,"file":"scans.types.d.ts","sourceRoot":"","sources":["../../../src/websitalk/scans/scans.types.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,MAAM,EAAE,MAAM,SAAS,CAAC;AAEjC,OAAO,KAAK,EAAE,aAAa,EAAE,MAAM,mBAAmB,CAAC;AAEvD,MAAM,MAAM,UAAU,GAAG,CAAC,OAAO,aAAa,CAAC,CAAC,MAAM,OAAO,aAAa,CAAC,CAAC;AAE5E,MAAM,MAAM,eAAe,GAAG;IAC5B,GAAG,EAAE,MAAM,CAAC;IACZ,YAAY,CAAC,EAAE,MAAM,GAAG,IAAI,CAAC;IAC7B,KAAK,CAAC,EAAE,MAAM,GAAG,IAAI,CAAC;IACtB,EAAE,CAAC,EAAE,MAAM,GAAG,IAAI,CAAC;IACnB,QAAQ,CAAC,EAAE,MAAM,GAAG,IAAI,CAAC;IACzB,UAAU,CAAC,EAAE,MAAM,GAAG,IAAI,CAAC;IAC3B,MAAM,CAAC,EAAE,MAAM,GAAG,IAAI,CAAC;IACvB,OAAO,CAAC,EAAE,MAAM,GAAG,IAAI,CAAC;IACxB,WAAW,CAAC,EAAE,MAAM,GAAG,IAAI,CAAC;IAC5B,YAAY,CAAC,EAAE,MAAM,GAAG,IAAI,CAAC;IAC7B,WAAW,CAAC,EAAE,OAAO,CAAC;IACtB,WAAW,CAAC,EAAE,MAAM,GAAG,IAAI,CAAC;IAC5B,SAAS,CAAC,EAAE,MAAM,GAAG,IAAI,CAAC;IAC1B,SAAS,CAAC,EAAE,IAAI,GAAG,IAAI,CAAC;IACxB,SAAS,CAAC,EAAE,IAAI,CAAC;IACjB,SAAS,CAAC,EAAE,IAAI,CAAC;CAClB,CAAC;AAEF,MAAM,MAAM,YAAY,GAAG;IACzB,IAAI,EAAE,MAAM,CAAC;IACb,MAAM,EAAE,MAAM,CAAC;CAChB,CAAC;AAEF,MAAM,MAAM,IAAI,GAAG;IACjB,QAAQ,EAAE,MAAM,CAAC;IACjB,YAAY,EAAE,MAAM,CAAC;IACrB,iBAAiB,CAAC,EAAE,MAAM,GAAG,IAAI,CAAC;IAClC,MAAM,EAAE,UAAU,CAAC;IACnB,gBAAgB,EAAE,MAAM,CAAC;IACzB,YAAY,EAAE,MAAM,CAAC;IACrB,YAAY,EAAE,MAAM,CAAC;IACrB,aAAa,CAAC,EAAE,MAAM,CAAC;IACvB,iBAAiB,CAAC,EAAE,MAAM,CAAC;IAC3B,OAAO,EAAE,YAAY,EAAE,CAAC;IACxB,UAAU,EAAE,eAAe,CAAC;IAC5B,SAAS,EAAE,IAAI,CAAC;IAChB,SAAS,EAAE,IAAI,CAAC;IAKhB,SAAS,CAAC,EAAE,IAAI,CAAC;IACjB,WAAW,CAAC,EAAE,IAAI,GAAG,IAAI,CAAC;IAC1B,cAAc,CAAC,EAAE,MAAM,CAAC;IACxB,oBAAoB,CAAC,EAAE,MAAM,CAAC;IAC9B,cAAc,CAAC,EAAE,MAAM,CAAC;CACzB,CAAC;AAEF,MAAM,MAAM,OAAO,GAAG,MAAM,CAAC,IAAI,CAAC,CAAC"}

package/jest.config.js

@@ -1,20 +1,20 @@
-/** @type {import('ts-jest').JestConfigWithTsJest} */
-module.exports = {
-  preset: 'ts-jest',
-  testEnvironment: 'node',
-  roots: ['<rootDir>/src'],
-  testPathIgnorePatterns: ['<rootDir>/src/__tests__/legacy/'],
-  testMatch: ['**/__tests__/**/*.spec.ts', '**/*.spec.ts'],
-  transform: {
-    '^.+\\.tsx?$': ['ts-jest', {
-      tsconfig: 'tsconfig.json',
-    }],
-  },
-  setupFilesAfterEnv: ['<rootDir>/src/__tests__/setup.ts'],
-  maxWorkers: 1,
-  verbose: true,
-  forceExit: true,
-  clearMocks: true,
-  resetMocks: true,
-  restoreMocks: true,
-};
+/** @type {import('ts-jest').JestConfigWithTsJest} */
+module.exports = {
+  preset: 'ts-jest',
+  testEnvironment: 'node',
+  roots: ['<rootDir>/src'],
+  testPathIgnorePatterns: ['<rootDir>/src/__tests__/legacy/'],
+  testMatch: ['**/__tests__/**/*.spec.ts', '**/*.spec.ts'],
+  transform: {
+    '^.+\\.tsx?$': ['ts-jest', {
+      tsconfig: 'tsconfig.json',
+    }],
+  },
+  setupFilesAfterEnv: ['<rootDir>/src/__tests__/setup.ts'],
+  maxWorkers: 1,
+  verbose: true,
+  forceExit: true,
+  clearMocks: true,
+  resetMocks: true,
+  restoreMocks: true,
+};

package/package.json

@@ -1,46 +1,46 @@
-{
-  "name": "@talkpilot/core-db",
-  "version": "1.3.60",
-  "description": "Core database package for centralized connections and ORM integration.",
-  "main": "dist/index.js",
-  "types": "dist/index.d.ts",
-  "scripts": {
-    "build": "tsc",
-    "dev": "tsc --watch",
-    "test": "jest",
-    "lint": "eslint 'src/**/*.{ts,tsx}'",
-    "format": "prettier --write \"src/**/*.{ts,tsx}\"",
-    "prepare": "npm run build"
-  },
-  "keywords": [
-    "database",
-    "orm",
-    "typescript",
-    "postgres",
-    "db-client"
-  ],
-  "author": "",
-  "license": "MIT",
-  "dependencies": {
-    "dayjs": "^1.11.21",
-    "express": "^4.18.0",
-    "google-libphonenumber": "^3.2.0",
-    "mongodb": "^6.11.0"
-  },
-  "devDependencies": {
-    "@faker-js/faker": "^8.0.0",
-    "@types/express": "^4.17.0",
-    "@types/google-libphonenumber": "^7.4.0",
-    "@types/jest": "^29.0.0",
-    "@types/node": "^20.0.0",
-    "fishery": "^2.4.0",
-    "jest": "^29.0.0",
-    "mongodb-memory-server": "^10.0.0",
-    "prettier": "^3.8.2",
-    "ts-jest": "^29.0.0",
-    "typescript": "^5.0.0"
-  },
-  "publishConfig": {
-    "access": "public"
-  }
-}
+{
+  "name": "@talkpilot/core-db",
+  "version": "1.3.61",
+  "description": "Core database package for centralized connections and ORM integration.",
+  "main": "dist/index.js",
+  "types": "dist/index.d.ts",
+  "scripts": {
+    "build": "tsc",
+    "dev": "tsc --watch",
+    "test": "jest",
+    "lint": "eslint 'src/**/*.{ts,tsx}'",
+    "format": "prettier --write \"src/**/*.{ts,tsx}\"",
+    "prepare": "npm run build"
+  },
+  "keywords": [
+    "database",
+    "orm",
+    "typescript",
+    "postgres",
+    "db-client"
+  ],
+  "author": "",
+  "license": "MIT",
+  "dependencies": {
+    "dayjs": "^1.11.21",
+    "express": "^4.18.0",
+    "google-libphonenumber": "^3.2.0",
+    "mongodb": "^6.11.0"
+  },
+  "devDependencies": {
+    "@faker-js/faker": "^8.0.0",
+    "@types/express": "^4.17.0",
+    "@types/google-libphonenumber": "^7.4.0",
+    "@types/jest": "^29.0.0",
+    "@types/node": "^20.0.0",
+    "fishery": "^2.4.0",
+    "jest": "^29.0.0",
+    "mongodb-memory-server": "^10.0.0",
+    "prettier": "^3.8.2",
+    "ts-jest": "^29.0.0",
+    "typescript": "^5.0.0"
+  },
+  "publishConfig": {
+    "access": "public"
+  }
+}

package/src/__tests__/setup.ts

@@ -1,20 +1,20 @@
-import { initTestDb, closeTestDb } from "../test-utils/db-utils";
-
-declare global {
-  // eslint-disable-next-line no-var
-  var db: import("mongodb").Db;
-}
-
-// Set up global mocks if needed
-// global.fetch = jest.fn();
-
-beforeEach(async () => {
-  global.db = await initTestDb();
-}, 30_000);
-
-afterEach(async () => {
-  await closeTestDb();
-  jest.resetAllMocks();
-});
-
-process.env.NODE_ENV = "test";
+import { initTestDb, closeTestDb } from "../test-utils/db-utils";
+
+declare global {
+  // eslint-disable-next-line no-var
+  var db: import("mongodb").Db;
+}
+
+// Set up global mocks if needed
+// global.fetch = jest.fn();
+
+beforeEach(async () => {
+  global.db = await initTestDb();
+}, 30_000);
+
+afterEach(async () => {
+  await closeTestDb();
+  jest.resetAllMocks();
+});
+
+process.env.NODE_ENV = "test";

package/src/bulkWrite/__tests__/bulkWrite.spec.ts

@@ -1,65 +1,65 @@
-import {
-  buildDeleteOp,
-  buildInsertOp,
-  buildSetOp,
-  buildUpsertOp,
-} from "../bulkWrite.ops";
-
-interface TestDoc {
-  _id: string;
-  key: string;
-  value: string;
-}
-
-describe("buildSetOp", () => {
-  it("returns a plain $set updateOne with no upsert", () => {
-    const op = buildSetOp<TestDoc>({ key: "k1" }, { value: "v2" });
-    expect(op).toEqual({
-      updateOne: { filter: { key: "k1" }, update: { $set: { value: "v2" } } },
-    });
-  });
-});
-
-describe("buildUpsertOp", () => {
-  it("returns $set + $setOnInsert + upsert:true", () => {
-    const op = buildUpsertOp<TestDoc>(
-      { key: "k1" },
-      { value: "v1" },
-      { key: "k1" },
-    );
-    expect(op).toEqual({
-      updateOne: {
-        filter: { key: "k1" },
-        update: { $set: { value: "v1" }, $setOnInsert: { key: "k1" } },
-        upsert: true,
-      },
-    });
-  });
-
-  it("defaults $setOnInsert to an empty object when omitted", () => {
-    const op = buildUpsertOp<TestDoc>({ key: "k1" }, { value: "v1" });
-    expect(op).toEqual({
-      updateOne: {
-        filter: { key: "k1" },
-        update: { $set: { value: "v1" }, $setOnInsert: {} },
-        upsert: true,
-      },
-    });
-  });
-});
-
-describe("buildInsertOp", () => {
-  it("returns an insertOne with the given document", () => {
-    const op = buildInsertOp<TestDoc>({ _id: "1", key: "k1", value: "v1" });
-    expect(op).toEqual({
-      insertOne: { document: { _id: "1", key: "k1", value: "v1" } },
-    });
-  });
-});
-
-describe("buildDeleteOp", () => {
-  it("returns a deleteOne matching filter", () => {
-    const op = buildDeleteOp<TestDoc>({ key: "k1" });
-    expect(op).toEqual({ deleteOne: { filter: { key: "k1" } } });
-  });
-});
+import {
+  buildDeleteOp,
+  buildInsertOp,
+  buildSetOp,
+  buildUpsertOp,
+} from "../bulkWrite.ops";
+
+interface TestDoc {
+  _id: string;
+  key: string;
+  value: string;
+}
+
+describe("buildSetOp", () => {
+  it("returns a plain $set updateOne with no upsert", () => {
+    const op = buildSetOp<TestDoc>({ key: "k1" }, { value: "v2" });
+    expect(op).toEqual({
+      updateOne: { filter: { key: "k1" }, update: { $set: { value: "v2" } } },
+    });
+  });
+});
+
+describe("buildUpsertOp", () => {
+  it("returns $set + $setOnInsert + upsert:true", () => {
+    const op = buildUpsertOp<TestDoc>(
+      { key: "k1" },
+      { value: "v1" },
+      { key: "k1" },
+    );
+    expect(op).toEqual({
+      updateOne: {
+        filter: { key: "k1" },
+        update: { $set: { value: "v1" }, $setOnInsert: { key: "k1" } },
+        upsert: true,
+      },
+    });
+  });
+
+  it("defaults $setOnInsert to an empty object when omitted", () => {
+    const op = buildUpsertOp<TestDoc>({ key: "k1" }, { value: "v1" });
+    expect(op).toEqual({
+      updateOne: {
+        filter: { key: "k1" },
+        update: { $set: { value: "v1" }, $setOnInsert: {} },
+        upsert: true,
+      },
+    });
+  });
+});
+
+describe("buildInsertOp", () => {
+  it("returns an insertOne with the given document", () => {
+    const op = buildInsertOp<TestDoc>({ _id: "1", key: "k1", value: "v1" });
+    expect(op).toEqual({
+      insertOne: { document: { _id: "1", key: "k1", value: "v1" } },
+    });
+  });
+});
+
+describe("buildDeleteOp", () => {
+  it("returns a deleteOne matching filter", () => {
+    const op = buildDeleteOp<TestDoc>({ key: "k1" });
+    expect(op).toEqual({ deleteOne: { filter: { key: "k1" } } });
+  });
+});

package/src/bulkWrite/bulkWrite.ops.ts

@@ -1,45 +1,45 @@
-import { Document, Filter, OptionalId } from "mongodb";
-import { BulkWriteOp } from "./bulkWrite.types";
-
-/**
- * Updates specific fields on the record matching filter.
- * Does nothing if no record matches.
- */
-export function buildSetOp<T extends Document>(
-  filter: Filter<T>,
-  setFields: Partial<T>,
-): BulkWriteOp<T> {
-  return { updateOne: { filter, update: { $set: setFields } } };
-}
-
-/**
- * Updates specific fields on the record matching the filter or creates a new record if none matches.
- * setOnInsertFields are only applied when a new record is created; they're ignored on an update.
- */
-export function buildUpsertOp<T extends Document>(
-  filter: Filter<T>,
-  setFields: Partial<T>,
-  setOnInsertFields: Partial<T> = {},
-): BulkWriteOp<T> {
-  return {
-    updateOne: {
-      filter,
-      update: { $set: setFields, $setOnInsert: setOnInsertFields },
-      upsert: true,
-    },
-  };
-}
-
-/** Creates a new record. */
-export function buildInsertOp<T extends Document>(
-  document: OptionalId<T>,
-): BulkWriteOp<T> {
-  return { insertOne: { document } };
-}
-
-/** Permanently removes the record matching the filter. */
-export function buildDeleteOp<T extends Document>(
-  filter: Filter<T>,
-): BulkWriteOp<T> {
-  return { deleteOne: { filter } };
-}
+import { Document, Filter, OptionalId } from "mongodb";
+import { BulkWriteOp } from "./bulkWrite.types";
+
+/**
+ * Updates specific fields on the record matching filter.
+ * Does nothing if no record matches.
+ */
+export function buildSetOp<T extends Document>(
+  filter: Filter<T>,
+  setFields: Partial<T>,
+): BulkWriteOp<T> {
+  return { updateOne: { filter, update: { $set: setFields } } };
+}
+
+/**
+ * Updates specific fields on the record matching the filter or creates a new record if none matches.
+ * setOnInsertFields are only applied when a new record is created; they're ignored on an update.
+ */
+export function buildUpsertOp<T extends Document>(
+  filter: Filter<T>,
+  setFields: Partial<T>,
+  setOnInsertFields: Partial<T> = {},
+): BulkWriteOp<T> {
+  return {
+    updateOne: {
+      filter,
+      update: { $set: setFields, $setOnInsert: setOnInsertFields },
+      upsert: true,
+    },
+  };
+}
+
+/** Creates a new record. */
+export function buildInsertOp<T extends Document>(
+  document: OptionalId<T>,
+): BulkWriteOp<T> {
+  return { insertOne: { document } };
+}
+
+/** Permanently removes the record matching the filter. */
+export function buildDeleteOp<T extends Document>(
+  filter: Filter<T>,
+): BulkWriteOp<T> {
+  return { deleteOne: { filter } };
+}

package/src/bulkWrite/bulkWrite.types.ts

@@ -1,4 +1,4 @@
-import { AnyBulkWriteOperation, Document } from "mongodb";
-
-/** A single MongoDB bulk-write operation, typed against the target document shape. */
-export type BulkWriteOp<T extends Document> = AnyBulkWriteOperation<T>;
+import { AnyBulkWriteOperation, Document } from "mongodb";
+
+/** A single MongoDB bulk-write operation, typed against the target document shape. */
+export type BulkWriteOp<T extends Document> = AnyBulkWriteOperation<T>;

package/src/bulkWrite/index.ts

@@ -1,2 +1,2 @@
-export * from "./bulkWrite.types";
-export * from "./bulkWrite.ops";
+export * from "./bulkWrite.types";
+export * from "./bulkWrite.ops";

package/src/config.ts

@@ -1,9 +1,9 @@
-/**
- * Centralized env var access. Add new fields here instead of reading
- * process.env directly elsewhere, so each env var has exactly one source of truth in the code.
- */
-
-export const configurationDbConfig = {
-  mongoUri: process.env.MONGO_URI || process.env.MONGODB_URI,
-  configurationDbName: process.env.CONFIGURATION_DB_NAME,
-};
+/**
+ * Centralized env var access. Add new fields here instead of reading
+ * process.env directly elsewhere, so each env var has exactly one source of truth in the code.
+ */
+
+export const configurationDbConfig = {
+  mongoUri: process.env.MONGO_URI || process.env.MONGODB_URI,
+  configurationDbName: process.env.CONFIGURATION_DB_NAME,
+};

package/src/configuration/index.ts

@@ -1,17 +1,17 @@
-import { Db, ObjectId as MongoObjectId } from "mongodb";
-
-export * from "./prompts";
-export * from "./models";
-export * from "./internalModels";
-export { configurationMongodbClient } from "./mongodb-client";
-
-let db: Db;
-export const setDb = (d: Db) => {
-  db = d;
-};
-
-export const getDb = (): Db => {
-  if (!db) throw new Error("Configuration DB not initialised");
-  return db;
-};
-export const ObjectId = MongoObjectId;
+import { Db, ObjectId as MongoObjectId } from "mongodb";
+
+export * from "./prompts";
+export * from "./models";
+export * from "./internalModels";
+export { configurationMongodbClient } from "./mongodb-client";
+
+let db: Db;
+export const setDb = (d: Db) => {
+  db = d;
+};
+
+export const getDb = (): Db => {
+  if (!db) throw new Error("Configuration DB not initialised");
+  return db;
+};
+export const ObjectId = MongoObjectId;

package/src/configuration/internalModels/index.ts

@@ -1,3 +1,3 @@
-export { INTERNAL_MODELS_COLLECTION } from "./internalModels.constants";
-export * from "./internalModels.getters";
-export type { InternalModel } from "./internalModels.types";
+export { INTERNAL_MODELS_COLLECTION } from "./internalModels.constants";
+export * from "./internalModels.getters";
+export type { InternalModel } from "./internalModels.types";

package/src/configuration/internalModels/internalModels.constants.ts

@@ -1 +1 @@
-export const INTERNAL_MODELS_COLLECTION = "internal-models";
+export const INTERNAL_MODELS_COLLECTION = "internal-models";

package/src/configuration/internalModels/internalModels.getters.ts

@@ -1,33 +1,33 @@
-import { getDb } from "../index";
-import { Collection, Filter } from "mongodb";
-import { InternalModel } from "./internalModels.types";
-import { INTERNAL_MODELS_COLLECTION } from "./internalModels.constants";
-
-export const getInternalModelsCollection = (): Collection<InternalModel> => {
-  return getDb().collection<InternalModel>(INTERNAL_MODELS_COLLECTION);
-};
-
-/**
- * Get an internal model config by its unique name
- */
-export const getInternalModelByName = async (
-  name: string
-): Promise<InternalModel | null> => {
-  return await getInternalModelsCollection().findOne({ name });
-};
-
-/**
- * Find internal models by filter.
- */
-export const findInternalModels = async (
-  filter: Filter<InternalModel> = {}
-): Promise<InternalModel[]> => {
-  return await getInternalModelsCollection().find(filter).toArray();
-};
-
-/**
- * Get all internal models.
- */
-export const getAllInternalModels = async (): Promise<InternalModel[]> => {
-  return await findInternalModels();
-};
+import { getDb } from "../index";
+import { Collection, Filter } from "mongodb";
+import { InternalModel } from "./internalModels.types";
+import { INTERNAL_MODELS_COLLECTION } from "./internalModels.constants";
+
+export const getInternalModelsCollection = (): Collection<InternalModel> => {
+  return getDb().collection<InternalModel>(INTERNAL_MODELS_COLLECTION);
+};
+
+/**
+ * Get an internal model config by its unique name
+ */
+export const getInternalModelByName = async (
+  name: string
+): Promise<InternalModel | null> => {
+  return await getInternalModelsCollection().findOne({ name });
+};
+
+/**
+ * Find internal models by filter.
+ */
+export const findInternalModels = async (
+  filter: Filter<InternalModel> = {}
+): Promise<InternalModel[]> => {
+  return await getInternalModelsCollection().find(filter).toArray();
+};
+
+/**
+ * Get all internal models.
+ */
+export const getAllInternalModels = async (): Promise<InternalModel[]> => {
+  return await findInternalModels();
+};

package/src/configuration/internalModels/internalModels.types.ts

@@ -1,12 +1,12 @@
-import { ObjectId } from "mongodb";
-
-export type InternalModel = {
-  _id: ObjectId;
-  /** Has to be unique between other names in collection */
-  name: string;
-  provider: string;
-  model: string;
-  config?: Record<string, unknown>;
-  createdAt: Date;
-  updatedAt: Date;
-};
+import { ObjectId } from "mongodb";
+
+export type InternalModel = {
+  _id: ObjectId;
+  /** Has to be unique between other names in collection */
+  name: string;
+  provider: string;
+  model: string;
+  config?: Record<string, unknown>;
+  createdAt: Date;
+  updatedAt: Date;
+};