@talkpilot/core-db 1.3.84 → 1.3.85

package/src/websitalk/knowledgeDocuments/knowledgeDocuments.getters.ts

@@ -1,95 +1,95 @@
-import { Collection, ObjectId as MongoObjectId } from "mongodb";
-import { getDb, ObjectId } from "../index";
-import {
-  KNOWLEDGE_DOCUMENTS_COLLECTION,
-  KNOWLEDGE_DOCUMENT_READY_PROJECTION,
-  KNOWLEDGE_DOCUMENT_STATUSES,
-} from "./knowledgeDocuments.constants";
-import type {
-  CreateKnowledgeDocumentInput,
-  KnowledgeDocument,
-  KnowledgeDocumentDoc,
-  KnowledgeDocumentReadyProjection,
-  UpdateKnowledgeDocumentInput,
-} from "./knowledgeDocuments.types";
-
-export const getKnowledgeDocumentsCollection =
-  (): Collection<KnowledgeDocument> => {
-    return getDb().collection<KnowledgeDocument>(
-      KNOWLEDGE_DOCUMENTS_COLLECTION,
-    );
-  };
-
-export const createKnowledgeDocument = async (
-  input: CreateKnowledgeDocumentInput,
-): Promise<MongoObjectId> => {
-  const now = new Date();
-  const { status, progressPercent, _id, ...rest } = input;
-  const { insertedId } = await getKnowledgeDocumentsCollection().insertOne({
-    ...(_id ? { _id } : {}),
-    ...rest,
-    status: status ?? KNOWLEDGE_DOCUMENT_STATUSES.UPLOADING,
-    progressPercent: progressPercent ?? 0,
-    createdAt: now,
-    updatedAt: now,
-  });
-  return insertedId;
-};
-
-export const getKnowledgeDocumentById = async (
-  id: string,
-): Promise<KnowledgeDocumentDoc | null> => {
-  return getKnowledgeDocumentsCollection().findOne({ _id: new ObjectId(id) });
-};
-
-export const listKnowledgeDocumentsByClientId = async (
-  clientId: string,
-): Promise<KnowledgeDocumentDoc[]> => {
-  return getKnowledgeDocumentsCollection()
-    .find({ clientId })
-    .sort({ createdAt: -1 })
-    .toArray();
-};
-
-export const listReadyKnowledgeDocumentIdsByClientId = async (
-  clientId: string,
-): Promise<string[]> => {
-  const docs = await getKnowledgeDocumentsCollection()
-    .find(
-      { clientId, status: KNOWLEDGE_DOCUMENT_STATUSES.READY },
-      { projection: { _id: 1 } },
-    )
-    .sort({ createdAt: -1 })
-    .toArray();
-  return docs.map((doc) => doc._id.toString());
-};
-
-export const listReadyKnowledgeDocumentsByClientId = async (
-  clientId: string,
-): Promise<KnowledgeDocumentReadyProjection[]> => {
-  return getKnowledgeDocumentsCollection()
-    .find(
-      { clientId, status: KNOWLEDGE_DOCUMENT_STATUSES.READY },
-      { projection: KNOWLEDGE_DOCUMENT_READY_PROJECTION },
-    )
-    .sort({ createdAt: -1 })
-    .toArray() as Promise<KnowledgeDocumentReadyProjection[]>;
-};
-
-export const updateKnowledgeDocument = async (
-  id: string,
-  updates: UpdateKnowledgeDocumentInput,
-): Promise<KnowledgeDocumentDoc | null> => {
-  return getKnowledgeDocumentsCollection().findOneAndUpdate(
-    { _id: new ObjectId(id) },
-    { $set: { ...updates, updatedAt: new Date() } },
-    { returnDocument: "after" },
-  );
-};
-
-export const deleteKnowledgeDocument = async (id: string): Promise<boolean> => {
-  const result = await getKnowledgeDocumentsCollection().deleteOne({
-    _id: new ObjectId(id),
-  });
-  return result.deletedCount === 1;
-};
+import { Collection, ObjectId as MongoObjectId } from "mongodb";
+import { getDb, ObjectId } from "../index";
+import {
+  KNOWLEDGE_DOCUMENTS_COLLECTION,
+  KNOWLEDGE_DOCUMENT_READY_PROJECTION,
+  KNOWLEDGE_DOCUMENT_STATUSES,
+} from "./knowledgeDocuments.constants";
+import type {
+  CreateKnowledgeDocumentInput,
+  KnowledgeDocument,
+  KnowledgeDocumentDoc,
+  KnowledgeDocumentReadyProjection,
+  UpdateKnowledgeDocumentInput,
+} from "./knowledgeDocuments.types";
+
+export const getKnowledgeDocumentsCollection =
+  (): Collection<KnowledgeDocument> => {
+    return getDb().collection<KnowledgeDocument>(
+      KNOWLEDGE_DOCUMENTS_COLLECTION,
+    );
+  };
+
+export const createKnowledgeDocument = async (
+  input: CreateKnowledgeDocumentInput,
+): Promise<MongoObjectId> => {
+  const now = new Date();
+  const { status, progressPercent, _id, ...rest } = input;
+  const { insertedId } = await getKnowledgeDocumentsCollection().insertOne({
+    ...(_id ? { _id } : {}),
+    ...rest,
+    status: status ?? KNOWLEDGE_DOCUMENT_STATUSES.UPLOADING,
+    progressPercent: progressPercent ?? 0,
+    createdAt: now,
+    updatedAt: now,
+  });
+  return insertedId;
+};
+
+export const getKnowledgeDocumentById = async (
+  id: string,
+): Promise<KnowledgeDocumentDoc | null> => {
+  return getKnowledgeDocumentsCollection().findOne({ _id: new ObjectId(id) });
+};
+
+export const listKnowledgeDocumentsByClientId = async (
+  clientId: string,
+): Promise<KnowledgeDocumentDoc[]> => {
+  return getKnowledgeDocumentsCollection()
+    .find({ clientId })
+    .sort({ createdAt: -1 })
+    .toArray();
+};
+
+export const listReadyKnowledgeDocumentIdsByClientId = async (
+  clientId: string,
+): Promise<string[]> => {
+  const docs = await getKnowledgeDocumentsCollection()
+    .find(
+      { clientId, status: KNOWLEDGE_DOCUMENT_STATUSES.READY },
+      { projection: { _id: 1 } },
+    )
+    .sort({ createdAt: -1 })
+    .toArray();
+  return docs.map((doc) => doc._id.toString());
+};
+
+export const listReadyKnowledgeDocumentsByClientId = async (
+  clientId: string,
+): Promise<KnowledgeDocumentReadyProjection[]> => {
+  return getKnowledgeDocumentsCollection()
+    .find(
+      { clientId, status: KNOWLEDGE_DOCUMENT_STATUSES.READY },
+      { projection: KNOWLEDGE_DOCUMENT_READY_PROJECTION },
+    )
+    .sort({ createdAt: -1 })
+    .toArray() as Promise<KnowledgeDocumentReadyProjection[]>;
+};
+
+export const updateKnowledgeDocument = async (
+  id: string,
+  updates: UpdateKnowledgeDocumentInput,
+): Promise<KnowledgeDocumentDoc | null> => {
+  return getKnowledgeDocumentsCollection().findOneAndUpdate(
+    { _id: new ObjectId(id) },
+    { $set: { ...updates, updatedAt: new Date() } },
+    { returnDocument: "after" },
+  );
+};
+
+export const deleteKnowledgeDocument = async (id: string): Promise<boolean> => {
+  const result = await getKnowledgeDocumentsCollection().deleteOne({
+    _id: new ObjectId(id),
+  });
+  return result.deletedCount === 1;
+};

package/src/websitalk/knowledgeDocuments/knowledgeDocuments.types.ts

@@ -1,75 +1,75 @@
-import { WithId, ObjectId as MongoObjectId } from "mongodb";
-
-import type {
-  KNOWLEDGE_CONTENT_TYPES,
-  KNOWLEDGE_DOCUMENT_STATUSES,
-  KNOWLEDGE_FILE_TYPES,
-} from "./knowledgeDocuments.constants";
-
-export type KnowledgeDocumentStatus =
-  (typeof KNOWLEDGE_DOCUMENT_STATUSES)[keyof typeof KNOWLEDGE_DOCUMENT_STATUSES];
-
-export type KnowledgeFileType =
-  (typeof KNOWLEDGE_FILE_TYPES)[keyof typeof KNOWLEDGE_FILE_TYPES];
-
-export type KnowledgeContentType =
-  (typeof KNOWLEDGE_CONTENT_TYPES)[keyof typeof KNOWLEDGE_CONTENT_TYPES];
-
-export type KnowledgeDocument = {
-  clientId: string;
-  filename: string;
-  contentType: KnowledgeContentType;
-  fileType: KnowledgeFileType;
-  storagePath: string;
-  markdownStoragePath?: string;
-  byteSize: number;
-  pageCount?: number;
-  pagesExtracted?: number;
-  charCount?: number;
-  totalTokens?: number;
-  contentHash?: string;
-  status: KnowledgeDocumentStatus;
-  errorMessage?: string;
-  progressPercent: number;
-  createdAt: Date;
-  updatedAt: Date;
-};
-
-export type KnowledgeDocumentDoc = WithId<KnowledgeDocument>;
-
-export type CreateKnowledgeDocumentInput = Omit<
-  KnowledgeDocument,
-  "createdAt" | "updatedAt" | "progressPercent" | "status"
-> & {
-  status?: KnowledgeDocumentStatus;
-  progressPercent?: number;
-  _id?: MongoObjectId;
-};
-
-export type UpdateKnowledgeDocumentInput = Partial<
-  Pick<
-    KnowledgeDocument,
-    | "storagePath"
-    | "markdownStoragePath"
-    | "byteSize"
-    | "pageCount"
-    | "pagesExtracted"
-    | "charCount"
-    | "totalTokens"
-    | "contentHash"
-    | "status"
-    | "errorMessage"
-    | "progressPercent"
-  >
->;
-
-export type KnowledgeDocumentReadyProjection = Pick<
-  KnowledgeDocumentDoc,
-  | "_id"
-  | "clientId"
-  | "filename"
-  | "markdownStoragePath"
-  | "charCount"
-  | "totalTokens"
-  | "status"
->;
+import { WithId, ObjectId as MongoObjectId } from "mongodb";
+
+import type {
+  KNOWLEDGE_CONTENT_TYPES,
+  KNOWLEDGE_DOCUMENT_STATUSES,
+  KNOWLEDGE_FILE_TYPES,
+} from "./knowledgeDocuments.constants";
+
+export type KnowledgeDocumentStatus =
+  (typeof KNOWLEDGE_DOCUMENT_STATUSES)[keyof typeof KNOWLEDGE_DOCUMENT_STATUSES];
+
+export type KnowledgeFileType =
+  (typeof KNOWLEDGE_FILE_TYPES)[keyof typeof KNOWLEDGE_FILE_TYPES];
+
+export type KnowledgeContentType =
+  (typeof KNOWLEDGE_CONTENT_TYPES)[keyof typeof KNOWLEDGE_CONTENT_TYPES];
+
+export type KnowledgeDocument = {
+  clientId: string;
+  filename: string;
+  contentType: KnowledgeContentType;
+  fileType: KnowledgeFileType;
+  storagePath: string;
+  markdownStoragePath?: string;
+  byteSize: number;
+  pageCount?: number;
+  pagesExtracted?: number;
+  charCount?: number;
+  totalTokens?: number;
+  contentHash?: string;
+  status: KnowledgeDocumentStatus;
+  errorMessage?: string;
+  progressPercent: number;
+  createdAt: Date;
+  updatedAt: Date;
+};
+
+export type KnowledgeDocumentDoc = WithId<KnowledgeDocument>;
+
+export type CreateKnowledgeDocumentInput = Omit<
+  KnowledgeDocument,
+  "createdAt" | "updatedAt" | "progressPercent" | "status"
+> & {
+  status?: KnowledgeDocumentStatus;
+  progressPercent?: number;
+  _id?: MongoObjectId;
+};
+
+export type UpdateKnowledgeDocumentInput = Partial<
+  Pick<
+    KnowledgeDocument,
+    | "storagePath"
+    | "markdownStoragePath"
+    | "byteSize"
+    | "pageCount"
+    | "pagesExtracted"
+    | "charCount"
+    | "totalTokens"
+    | "contentHash"
+    | "status"
+    | "errorMessage"
+    | "progressPercent"
+  >
+>;
+
+export type KnowledgeDocumentReadyProjection = Pick<
+  KnowledgeDocumentDoc,
+  | "_id"
+  | "clientId"
+  | "filename"
+  | "markdownStoragePath"
+  | "charCount"
+  | "totalTokens"
+  | "status"
+>;

package/src/websitalk/mongodb-client.ts

@@ -1,61 +1,61 @@
-import { MongoClient, Db } from "mongodb";
-import { setDb } from "./index";
-import { validateConfig, validateMongoUri } from "../utils/validation";
-
-class WebsitalkMongoDBClient {
-  private client: MongoClient | null = null;
-  private db: Db | null = null;
-  private readonly defaultDbName = "website-talk";
-
-  async connect(uri?: string, dbName?: string): Promise<void> {
-    if (this.client) {
-      return;
-    }
-
-    const mongodbUri = uri || process.env.MONGO_URI || process.env.MONGODB_URI;
-    validateConfig("MONGO_URI", mongodbUri);
-    validateMongoUri(mongodbUri!);
-
-    try {
-      this.client = new MongoClient(mongodbUri!);
-      await this.client.connect();
-      const targetDbName =
-        dbName || process.env.WEBSITALK_DB_NAME || this.defaultDbName;
-      this.db = this.client.db(targetDbName);
-      setDb(this.db);
-      console.info(`[core-db] Website Talk MongoDB connected: ${targetDbName}`);
-    } catch (error) {
-      console.error("[core-db] Website Talk connection failed", error);
-      throw error;
-    }
-  }
-
-  async disconnect(): Promise<void> {
-    if (this.client) {
-      try {
-        await this.client.close();
-        this.client = null;
-        this.db = null;
-        console.info("Website Talk MongoDB disconnected successfully");
-      } catch (error) {
-        console.error("[core-db] Website Talk disconnection failed", error);
-        throw error;
-      }
-    }
-  }
-
-  getDb(): Db {
-    if (!this.db) {
-      throw new Error(
-        "Website Talk database not initialized. Call connect() first.",
-      );
-    }
-    return this.db;
-  }
-
-  isConnected(): boolean {
-    return this.client !== null && this.client !== undefined;
-  }
-}
-
-export const websitalkMongodbClient = new WebsitalkMongoDBClient();
+import { MongoClient, Db } from "mongodb";
+import { setDb } from "./index";
+import { validateConfig, validateMongoUri } from "../utils/validation";
+
+class WebsitalkMongoDBClient {
+  private client: MongoClient | null = null;
+  private db: Db | null = null;
+  private readonly defaultDbName = "website-talk";
+
+  async connect(uri?: string, dbName?: string): Promise<void> {
+    if (this.client) {
+      return;
+    }
+
+    const mongodbUri = uri || process.env.MONGO_URI || process.env.MONGODB_URI;
+    validateConfig("MONGO_URI", mongodbUri);
+    validateMongoUri(mongodbUri!);
+
+    try {
+      this.client = new MongoClient(mongodbUri!);
+      await this.client.connect();
+      const targetDbName =
+        dbName || process.env.WEBSITALK_DB_NAME || this.defaultDbName;
+      this.db = this.client.db(targetDbName);
+      setDb(this.db);
+      console.info(`[core-db] Website Talk MongoDB connected: ${targetDbName}`);
+    } catch (error) {
+      console.error("[core-db] Website Talk connection failed", error);
+      throw error;
+    }
+  }
+
+  async disconnect(): Promise<void> {
+    if (this.client) {
+      try {
+        await this.client.close();
+        this.client = null;
+        this.db = null;
+        console.info("Website Talk MongoDB disconnected successfully");
+      } catch (error) {
+        console.error("[core-db] Website Talk disconnection failed", error);
+        throw error;
+      }
+    }
+  }
+
+  getDb(): Db {
+    if (!this.db) {
+      throw new Error(
+        "Website Talk database not initialized. Call connect() first.",
+      );
+    }
+    return this.db;
+  }
+
+  isConnected(): boolean {
+    return this.client !== null && this.client !== undefined;
+  }
+}
+
+export const websitalkMongodbClient = new WebsitalkMongoDBClient();