ddd refactor: api-keys repo

apps/rowboat/app/actions/project_actions.ts

@@ -1,7 +1,7 @@
 'use server';
 import { redirect } from "next/navigation";
 import { ObjectId } from "mongodb";
-import { db, dataSourcesCollection, projectsCollection, projectMembersCollection, apiKeysCollection, dataSourceDocsCollection } from "../lib/mongodb";
+import { db, dataSourcesCollection, projectsCollection, projectMembersCollection, dataSourceDocsCollection } from "../lib/mongodb";
 import { z } from 'zod';
 import crypto from 'crypto';
 import { revalidatePath } from "next/cache";
@@ -18,12 +18,14 @@ import { IProjectActionAuthorizationPolicy } from "@/src/application/policies/pr
 import { ICreateApiKeyController } from "@/src/interface-adapters/controllers/api-keys/create-api-key.controller";
 import { IListApiKeysController } from "@/src/interface-adapters/controllers/api-keys/list-api-keys.controller";
 import { IDeleteApiKeyController } from "@/src/interface-adapters/controllers/api-keys/delete-api-key.controller";
+import { IApiKeysRepository } from "@/src/application/repositories/api-keys.repository.interface";
 const KLAVIS_API_KEY = process.env.KLAVIS_API_KEY || '';
 
 const projectActionAuthorizationPolicy = container.resolve<IProjectActionAuthorizationPolicy>('projectActionAuthorizationPolicy');
 const createApiKeyController = container.resolve<ICreateApiKeyController>('createApiKeyController');
 const listApiKeysController = container.resolve<IListApiKeysController>('listApiKeysController');
 const deleteApiKeyController = container.resolve<IDeleteApiKeyController>('deleteApiKeyController');
+const apiKeysRepository = container.resolve<IApiKeysRepository>('apiKeysRepository');
 
 export async function listTemplates() {
     const templatesArray = Object.entries(templates)
@@ -228,9 +230,7 @@ export async function deleteProject(projectId: string) {
     await projectAuthCheck(projectId);
 
     // delete api keys
-    await apiKeysCollection.deleteMany({
+    await apiKeysRepository.deleteAll(projectId);
-        projectId,
-    });
 
     // delete embeddings
     const sources = await dataSourcesCollection.find({

apps/rowboat/app/lib/mongodb.ts

@@ -19,7 +19,6 @@ export const dataSourceDocsCollection = db.collection<z.infer<typeof DataSourceD
 export const projectsCollection = db.collection<z.infer<typeof Project>>("projects");
 export const projectMembersCollection = db.collection<z.infer<typeof ProjectMember>>("project_members");
 export const agentWorkflowsCollection = db.collection<z.infer<typeof Workflow>>("agent_workflows");
-export const apiKeysCollection = db.collection<z.infer<typeof ApiKey>>("api_keys");
 export const chatsCollection = db.collection<z.infer<typeof apiV1.Chat>>("chats");
 export const chatMessagesCollection = db.collection<z.infer<typeof apiV1.ChatMessage>>("chat_messages");
 export const twilioConfigsCollection = db.collection<z.infer<typeof TwilioConfig>>("twilio_configs");

apps/rowboat/src/application/repositories/api-keys.repository.interface.ts

@@ -31,6 +31,12 @@ export interface IApiKeysRepository {
      */
     delete(projectId: string, id: string): Promise<boolean>;
 
+    /**
+     * Deletes all API keys for a given project.
+     * @param projectId - The ID of the project.
+     */
+    deleteAll(projectId: string): Promise<void>;
+
     /**
      * Checks if an API key is valid for a project and consumes it (e.g., for rate limiting or one-time use).
      * @param projectId - The ID of the project.

apps/rowboat/src/infrastructure/repositories/mongodb.api-keys.repository.ts

@@ -56,4 +56,8 @@ export class MongoDBApiKeysRepository implements IApiKeysRepository {
         const result = await this.collection.deleteOne({ projectId, _id: new ObjectId(id) });
         return result.deletedCount > 0;
     }
+
+    async deleteAll(projectId: string): Promise<void> {
+        await this.collection.deleteMany({ projectId });
+    }
 }