feat: implement use case and controller method to delete user threads with AI assistant

Author: Artuomka

backend/src/common/data-injection.tokens.ts

@@ -169,4 +169,5 @@ export enum UseCaseType {
   ADD_MESSAGE_TO_THREAD_WITH_AI_ASSISTANT = 'ADD_MESSAGE_TO_THREAD_WITH_AI_ASSISTANT',
   GET_ALL_USER_THREADS_WITH_AI_ASSISTANT = 'GET_ALL_USER_THREADS_WITH_AI_ASSISTANT',
   GET_ALL_THREAD_MESSAGES = 'GET_ALL_THREAD_MESSAGES',
+  DELETE_THREAD_WITH_AI_ASSISTANT = 'DELETE_THREAD_WITH_AI_ASSISTANT',
 }

backend/src/entities/ai/ai-use-cases.interface.ts

@@ -1,4 +1,5 @@
 import { InTransactionEnum } from '../../enums/in-transaction.enum.js';
+import { SuccessResponse } from '../../microservices/saas-microservice/data-structures/common-responce.ds.js';
 import { AddMessageToThreadWithAssistantDS } from './application/data-structures/add-message-to-thread-with-assistant.ds.js';
 import { CreateThreadWithAssistantDS } from './application/data-structures/create-thread-with-assistant.ds.js';
 import { FindAllThreadMessagesDS } from './application/data-structures/find-all-thread-messages.ds.js';
@@ -29,3 +30,7 @@ export interface IGetAllThreadMessages {
     inTransaction: InTransactionEnum,
   ): Promise<Array<FoundUserThreadMessagesRO>>;
 }
+
+export interface IDeleteThreadWithAIAssistant {
+  execute(inputData: FindAllThreadMessagesDS, inTransaction: InTransactionEnum): Promise<SuccessResponse>;
+}

backend/src/entities/ai/ai.module.ts

@@ -12,6 +12,7 @@ import { CreateThreadWithAIAssistantUseCase } from './use-cases/create-thread-wi
 import { AddMessageToThreadWithAIAssistantUseCase } from './use-cases/add-message-to-thread-with-ai.use.case.js';
 import { FindAllUserThreadsWithAssistantUseCase } from './use-cases/find-all-user-threads-with-assistant.use.case.js';
 import { FindAllMessagesInAiThreadUseCase } from './use-cases/find-all-messages-in-ai-thread.use.case.js';
+import { DeleteThreadWithAIAssistantUseCase } from './use-cases/delete-thread-with-ai-assistant.use.case.js';
 
 @Module({
   imports: [TypeOrmModule.forFeature([UserEntity, LogOutEntity])],
@@ -40,6 +41,10 @@ import { FindAllMessagesInAiThreadUseCase } from './use-cases/find-all-messages-
       provide: UseCaseType.GET_ALL_THREAD_MESSAGES,
       useClass: FindAllMessagesInAiThreadUseCase,
     },
+    {
+      provide: UseCaseType.DELETE_THREAD_WITH_AI_ASSISTANT,
+      useClass: DeleteThreadWithAIAssistantUseCase,
+    },
   ],
   controllers: [UserAIRequestsController, UserAIThreadsController],
 })
@@ -53,6 +58,7 @@ export class AIModule implements NestModule {
         { path: '/ai/thread/message/:connectionId/:threadId', method: RequestMethod.POST },
         { path: '/ai/threads', method: RequestMethod.GET },
         { path: '/ai/thread/messages/:threadId', method: RequestMethod.GET },
+        { path: '/ai/thread/:threadId', method: RequestMethod.DELETE },
       );
   }
 }

backend/src/entities/ai/use-cases/delete-thread-with-ai-assistant.use.case.ts

@@ -0,0 +1,39 @@
+import { Inject, Injectable, NotFoundException, Scope } from '@nestjs/common';
+import AbstractUseCase from '../../../common/abstract-use.case.js';
+import { IGlobalDatabaseContext } from '../../../common/application/global-database-context.interface.js';
+import { BaseType } from '../../../common/data-injection.tokens.js';
+import { IDeleteThreadWithAIAssistant } from '../ai-use-cases.interface.js';
+import { FindAllThreadMessagesDS } from '../application/data-structures/find-all-thread-messages.ds.js';
+import { SuccessResponse } from '../../../microservices/saas-microservice/data-structures/common-responce.ds.js';
+import { Messages } from '../../../exceptions/text/messages.js';
+import { getOpenAiClient } from '../utils/get-open-ai-client.js';
+
+@Injectable({ scope: Scope.REQUEST })
+export class DeleteThreadWithAIAssistantUseCase
+  extends AbstractUseCase<FindAllThreadMessagesDS, SuccessResponse>
+  implements IDeleteThreadWithAIAssistant
+{
+  constructor(
+    @Inject(BaseType.GLOBAL_DB_CONTEXT)
+    protected _dbContext: IGlobalDatabaseContext,
+  ) {
+    super();
+  }
+
+  public async implementation(inputData: FindAllThreadMessagesDS): Promise<SuccessResponse> {
+    const { threadId, userId } = inputData;
+    const foundThread = await this._dbContext.aiUserThreadsRepository.findThreadByIdAndUserId(threadId, userId);
+    if (!foundThread) {
+      throw new NotFoundException(Messages.AI_THREAD_NOT_FOUND);
+    }
+
+    const { openai } = getOpenAiClient();
+
+    await openai.beta.threads.del(foundThread.thread_ai_id);
+    await this._dbContext.aiUserThreadsRepository.delete(foundThread.id);
+
+    return {
+      success: true,
+    };
+  }
+}

backend/src/entities/ai/user-ai-threads.controller.ts

@@ -1,10 +1,22 @@
-import { UseInterceptors, Controller, Injectable, Inject, UseGuards, Post, Body, Res, Get } from '@nestjs/common';
+import {
+  UseInterceptors,
+  Controller,
+  Injectable,
+  Inject,
+  UseGuards,
+  Post,
+  Body,
+  Res,
+  Get,
+  Delete,
+} from '@nestjs/common';
 import { ApiBearerAuth, ApiBody, ApiOperation, ApiQuery, ApiResponse, ApiTags } from '@nestjs/swagger';
 import { SentryInterceptor } from '../../interceptors/sentry.interceptor.js';
 import { UseCaseType } from '../../common/data-injection.tokens.js';
 import {
   IAddMessageToThreadWithAIAssistant,
   ICreateThreadWithAIAssistant,
+  IDeleteThreadWithAIAssistant,
   IGetAllThreadMessages,
   IGetAllUserThreadsWithAIAssistant,
 } from './ai-use-cases.interface.js';
@@ -21,6 +33,7 @@ import { AddMessageToThreadWithAssistantDS } from './application/data-structures
 import { FoundUserThreadsWithAiRO } from './application/dto/found-user-threads-with-ai.ro.js';
 import { FoundUserThreadMessagesRO } from './application/dto/found-user-thread-messages.ro.js';
 import { FindAllThreadMessagesDS } from './application/data-structures/find-all-thread-messages.ds.js';
+import { SuccessResponse } from '../../microservices/saas-microservice/data-structures/common-responce.ds.js';
 
 @UseInterceptors(SentryInterceptor)
 @Controller()
@@ -37,6 +50,8 @@ export class UserAIThreadsController {
     private readonly getAllUserThreadsWithAIAssistantUseCase: IGetAllUserThreadsWithAIAssistant,
     @Inject(UseCaseType.GET_ALL_THREAD_MESSAGES)
     private readonly getAllThreadMessagesUseCase: IGetAllThreadMessages,
+    @Inject(UseCaseType.DELETE_THREAD_WITH_AI_ASSISTANT)
+    private readonly deleteThreadWithAIAssistantUseCase: IDeleteThreadWithAIAssistant,
   ) {}
 
   @ApiOperation({ summary: 'Create new thread with ai assistant' })
@@ -127,4 +142,23 @@ export class UserAIThreadsController {
     };
     return await this.getAllThreadMessagesUseCase.execute(inputData, InTransactionEnum.OFF);
   }
+
+  @ApiOperation({ summary: 'Delete users thread with ai assistant' })
+  @ApiResponse({
+    status: 201,
+    description: 'Delete users thread.',
+    type: SuccessResponse,
+    isArray: true,
+  })
+  @Delete('/ai/thread/:threadId')
+  public async deleteThreadWithAssistant(
+    @UserId() userId: string,
+    @SlugUuid('threadId') threadId: string,
+  ): Promise<SuccessResponse> {
+    const inputData: FindAllThreadMessagesDS = {
+      threadId,
+      userId,
+    };
+    return await this.deleteThreadWithAIAssistantUseCase.execute(inputData, InTransactionEnum.ON);
+  }
 }