import { Db, ObjectId as MongoObjectId } from "mongodb";

export * from "./scans";
export * from "./scanResources";
export * from "./websiteUrls";
export * from "./knowledgeDocuments";
export { websitalkMongodbClient } from "./mongodb-client";

let db: Db;
export const setDb = (d: Db) => {
  db = d;
};

export const getDb = (): Db => {
  if (!db) throw new Error("Website Talk DB not initialised");
  return db;
};
export const ObjectId = MongoObjectId;