Feat : AI 분류 폴더 이름 리스트get api, AI 분류 링크 리스트 get api

src/app.module.ts

@@ -8,6 +8,7 @@ import { ConfigModule } from '@nestjs/config';
 import { DatabaseModule } from '@src/infrastructure';
 import { AiModule } from './infrastructure/ai/ai.module';
 import { UsersModule } from './modules/users/users.module';
+import { ClassificationModule } from './modules/classification/classification.module';
 import { AuthModule } from './modules/auth/auth.module';
 import { FoldersModule } from './modules/folders/folders.module';
 import { LinksModule } from './modules/links/links.module';
@@ -23,6 +24,7 @@ import { PostsModule } from './modules/posts/posts.module';
     DatabaseModule,
     AiModule,
     UsersModule,
+    ClassificationModule,
     AuthModule,
     FoldersModule,
     LinksModule,

src/common/decorators/getUser.decorator.ts

@@ -16,7 +16,6 @@ export const GetUser = createParamDecorator(
   ): ReqUserPayload | ReqUserPayload[keyof ReqUserPayload] => {
     // Get context Request Object
     const request = context.switchToHttp().getRequest<Request>();
-
     // Expect Request, user property as type 'ReqUserPayload'(Refer to defined in common/types/type.d.ts)
     const user: Express.User = request.user;
     return user['id'];

src/infrastructure/database/schema/base.schema.ts

@@ -1,5 +1,5 @@
 export class BaseDocument {
-  createdAt: string;
+  createdAt: Date;
 
-  updatedAt: string;
+  updatedAt: Date;
 }

src/infrastructure/database/schema/post.schema.ts

@@ -14,6 +14,7 @@ export class Post {
 
   @Prop({ required: true })
   title!: string;
+  //TODO : 사진 추가
 
   @Prop({ default: null, type: String })
   description: string | null;

src/infrastructure/database/schema/postAIClassification.schema.ts

@@ -4,13 +4,14 @@ import {
   Keyword,
   KeywordSchema,
 } from '@src/infrastructure/database/schema/keyword.schema';
+import { BaseDocument } from './base.schema';
 
 @Schema({
   collection: 'post_ai_classifications',
   timestamps: true,
   versionKey: false,
 })
-export class PostAIClassification {
+export class PostAIClassification extends BaseDocument {
   @Prop({ required: true, type: MongooseSchema.Types.ObjectId, ref: 'Folder' })
   suggestedFolderId: MongooseSchema.Types.ObjectId;
 
@@ -20,13 +21,13 @@ export class PostAIClassification {
   @Prop({ required: true })
   description: string;
 
-  @Prop({ required: true, type: [KeywordSchema] })
-  keywords: Keyword[];
+  @Prop({ required: true, type: [String] })
+  keywords: string[];
 
   @Prop({ type: Date })
   completedAt: Date;
 
-  @Prop({ type: Date })
+  @Prop({ default: null })
   deletedAt: Date;
 }
 

src/modules/classification/classification.controller.ts

@@ -0,0 +1,37 @@
+import { Controller, Get, UseGuards, Param } from '@nestjs/common';
+import { ClassificationService } from './classification.service';
+import { GetUser } from '@src/common';
+import {
+  ClassificationControllerDocs,
+  GetAIFolderNameListDocs,
+  GetAIPostListDocs,
+} from './docs';
+import { JwtGuard } from '../users/guards';
+
+import { Types } from 'mongoose';
+import { AIPostListResponse } from './dto/getAIPostList.dto';
+import { AIFolderNameListResponse } from './dto/getAIFolderNameLIst.dto';
+
+@Controller('ai')
+@UseGuards(JwtGuard)
+@ClassificationControllerDocs
+export class ClassificationController {
+  constructor(private readonly classificationService: ClassificationService) {}
+
+  @Get('/suggestions') //TODO : 정렬
+  @GetAIFolderNameListDocs
+  async getSuggestedFolderNameList(@GetUser('id') userId: Types.ObjectId) {
+    const folderNames =
+      await this.classificationService.getFolderNameList(userId);
+
+    return new AIFolderNameListResponse(folderNames);
+  }
+
+  @Get('/suggestions/:folderId')
+  @GetAIPostListDocs
+  async getSuggestedPostList(@Param('folderId') folderId: string) {
+    const posts = await this.classificationService.getPostList(folderId);
+
+    return new AIPostListResponse(posts);
+  }
+}

src/modules/classification/classification.module.ts

@@ -0,0 +1,26 @@
+import { Module } from '@nestjs/common';
+import { ClassificationService } from './classification.service';
+import { MongooseModule } from '@nestjs/mongoose';
+import {
+  Folder,
+  FolderSchema,
+  Post,
+  PostAIClassification,
+  PostAIClassificationSchema,
+  PostSchema,
+} from '@src/infrastructure/database/schema';
+
+import { ClassificationController } from './classification.controller';
+
+@Module({
+  imports: [
+    MongooseModule.forFeature([
+      { name: Folder.name, schema: FolderSchema },
+      { name: PostAIClassification.name, schema: PostAIClassificationSchema },
+      { name: Post.name, schema: PostSchema },
+    ]),
+  ],
+  controllers: [ClassificationController],
+  providers: [ClassificationService],
+})
+export class ClassificationModule {}

src/modules/classification/classification.service.spec.ts

@@ -0,0 +1,18 @@
+import { Test, TestingModule } from '@nestjs/testing';
+import { ClassificationService } from './classification.service';
+
+describe('ClassificationService', () => {
+  let service: ClassificationService;
+
+  beforeEach(async () => {
+    const module: TestingModule = await Test.createTestingModule({
+      providers: [ClassificationService],
+    }).compile();
+
+    service = module.get<ClassificationService>(ClassificationService);
+  });
+
+  it('should be defined', () => {
+    expect(service).toBeDefined();
+  });
+});

src/modules/classification/classification.service.ts

@@ -0,0 +1,64 @@
+import { Injectable } from '@nestjs/common';
+import {
+  Folder,
+  FolderDocument,
+  Post,
+  PostAIClassification,
+} from '@src/infrastructure/database/schema';
+
+import { InjectModel } from '@nestjs/mongoose';
+import { Model, Types } from 'mongoose';
+import { AIFolderNameServiceDto } from './dto/getAIFolderNameLIst.dto';
+import { AIPostServiceDto } from './dto/getAIPostList.dto';
+
+@Injectable()
+export class ClassificationService {
+  constructor(
+    @InjectModel(Folder.name) private folderModel: Model<Folder>,
+    @InjectModel(PostAIClassification.name)
+    private postAiClassificationModel: Model<PostAIClassification>,
+    @InjectModel(Post.name) private postModel: Model<Post>,
+  ) {}
+
+  async getFolderNameList(
+    userId: Types.ObjectId,
+  ): Promise<AIFolderNameServiceDto[]> {
+    const folders = await this.folderModel.find({ userId }).exec();
+    const folderIds = folders.map((folder) => folder._id);
+
+    const classifications = await this.postAiClassificationModel
+      .find({ suggestedFolderId: { $in: folderIds } })
+      .exec();
+
+    const uniqueFolderIds = [
+      ...new Set(
+        classifications.map((classification) =>
+          classification.suggestedFolderId.toString(),
+        ),
+      ),
+    ];
+
+    const matchedFolders = await this.folderModel
+      .find({ _id: { $in: uniqueFolderIds } })
+      .exec();
+
+    return matchedFolders.map((folder) => new AIFolderNameServiceDto(folder));
+  }
+
+  async getPostList(folderId: string): Promise<AIPostServiceDto[]> {
+    const posts = await this.postModel
+      .find({ folderId })
+      .populate<{
+        aiClassificationId: PostAIClassification;
+      }>({
+        path: 'aiClassificationId',
+        match: { deletedAt: null }, // deletedAt이 null인 것만 필터링
+      })
+      .sort({ createdAt: -1 })
+      .exec();
+
+    return posts
+      .filter((post) => post.aiClassificationId)
+      .map((post) => new AIPostServiceDto(post));
+  }
+}

src/modules/classification/docs/controller.docs.ts

@@ -0,0 +1,6 @@
+import { applyDecorators } from '@nestjs/common';
+import { ApiTags } from '@nestjs/swagger';
+
+export const ClassificationControllerDocs = applyDecorators(
+  ApiTags('AI classification API'),
+);

src/modules/classification/docs/getAIFolderNameList.docs.ts

@@ -0,0 +1,14 @@
+import { applyDecorators } from '@nestjs/common';
+import { ApiBearerAuth, ApiOperation, ApiResponse } from '@nestjs/swagger';
+import { AIFolderNameListResponse } from '../dto/getAIFolderNameLIst.dto';
+
+export const GetAIFolderNameListDocs = applyDecorators(
+  ApiOperation({
+    summary: '폴더 리스트',
+    description: 'AI 분류 폴더 리스트.',
+  }),
+  ApiResponse({
+    type: AIFolderNameListResponse,
+  }),
+  ApiBearerAuth(),
+);

src/modules/classification/docs/getAIPostList.docs.ts

@@ -0,0 +1,14 @@
+import { applyDecorators } from '@nestjs/common';
+import { ApiBearerAuth, ApiOperation, ApiResponse } from '@nestjs/swagger';
+import { AIPostListResponse } from '../dto/getAIPostList.dto';
+
+export const GetAIPostListDocs = applyDecorators(
+  ApiOperation({
+    summary: '폴더 안에 들어있는 Post(링크) 리스트',
+    description: 'AI 분류 추천된 링크 리스트.',
+  }),
+  ApiResponse({
+    type: AIPostListResponse,
+  }),
+  ApiBearerAuth(),
+);

src/modules/classification/docs/index.ts

@@ -0,0 +1,3 @@
+export * from './controller.docs';
+export * from './getAIFolderNameList.docs';
+export * from './getAIPostList.docs';

src/modules/classification/dto/getAIFolderNameLIst.dto.ts

@@ -0,0 +1,31 @@
+import { ApiProperty } from '@nestjs/swagger';
+import { FolderDocument } from '@src/infrastructure';
+import { IsNotEmpty, IsString } from 'class-validator';
+
+export class AIFolderNameServiceDto {
+  @ApiProperty({ description: '폴더 id' })
+  @IsNotEmpty()
+  @IsString()
+  id: string;
+
+  @ApiProperty({ description: '폴더 이름' })
+  @IsNotEmpty()
+  @IsString()
+  name: string;
+
+  constructor(data: FolderDocument) {
+    (this.id = data._id.toString()), (this.name = data.name);
+  }
+}
+
+export class AIFolderNameListResponse {
+  @ApiProperty({
+    type: AIFolderNameServiceDto,
+    isArray: true,
+  })
+  list: AIFolderNameServiceDto[];
+
+  constructor(data: AIFolderNameServiceDto[]) {
+    this.list = data;
+  }
+}

src/modules/classification/dto/getAIPostList.dto.ts

@@ -0,0 +1,65 @@
+import { ApiProperty } from '@nestjs/swagger';
+import { IsNotEmpty, IsString, IsArray, IsDate } from 'class-validator';
+import { Type } from 'class-transformer';
+import { Post, PostAIClassification, PostDocument } from '@src/infrastructure';
+import { Document } from 'mongoose';
+import { MergeType } from 'mongoose';
+import { Types } from 'mongoose';
+
+type InputType = Document<
+  unknown,
+  {},
+  MergeType<
+    Post,
+    {
+      aiClassificationId: PostAIClassification;
+    }
+  >
+> &
+  Omit<Post, 'aiClassificationId'> & {
+    aiClassificationId: PostAIClassification;
+  } & {
+    _id: Types.ObjectId;
+  };
+export class AIPostServiceDto {
+  @ApiProperty({ description: 'Title' })
+  @IsNotEmpty()
+  @IsString()
+  title: string;
+
+  @ApiProperty({ description: 'URL' })
+  @IsNotEmpty()
+  @IsString()
+  url: string;
+
+  @ApiProperty({ description: 'Description' })
+  @IsNotEmpty()
+  @IsString()
+  description: string;
+
+  @ApiProperty({ description: 'Keywords' })
+  @IsArray()
+  keywords: string[];
+
+  @ApiProperty({ description: 'Created At' })
+  @IsString()
+  createdAt: Date;
+
+  constructor(data: InputType) {
+    this.title = data.title;
+    this.url = data.url;
+    this.description = data.description;
+    this.keywords = data.aiClassificationId.keywords;
+    this.createdAt = data.aiClassificationId.createdAt;
+  }
+}
+
+export class AIPostListResponse {
+  @ApiProperty({ type: [AIPostServiceDto] })
+  @Type(() => AIPostServiceDto)
+  list: AIPostServiceDto[];
+
+  constructor(data: AIPostServiceDto[]) {
+    this.list = data;
+  }
+}

src/modules/folders/responses/folder-summary.response.ts

@@ -13,7 +13,7 @@ export class FolderSummaryResponse {
   type: FolderType;
 
   @ApiProperty()
-  createdAt: string;
+  createdAt: Date;
 
   constructor(data: FolderDocument) {
     this.id = data._id.toString();