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/postAIClassification.schema.ts → src/infrastructure/database/schema/AIClassification.schema.ts

@@ -1,16 +1,14 @@
 import { Prop, Schema, SchemaFactory } from '@nestjs/mongoose';
 import { HydratedDocument, Schema as MongooseSchema } from 'mongoose';
-import {
-  Keyword,
-  KeywordSchema,
-} from '@src/infrastructure/database/schema/keyword.schema';
+
+import { BaseDocument } from './base.schema';
 
 @Schema({
-  collection: 'post_ai_classifications',
+  collection: 'ai_classifications',
   timestamps: true,
   versionKey: false,
 })
-export class PostAIClassification {
+export class AIClassification extends BaseDocument {
   @Prop({ required: true, type: MongooseSchema.Types.ObjectId, ref: 'Folder' })
   suggestedFolderId: MongooseSchema.Types.ObjectId;
 
@@ -20,17 +18,16 @@ 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;
 }
 
-export type PostAIClassificationDocument =
-  HydratedDocument<PostAIClassification>;
+export type PostAIClassificationDocument = HydratedDocument<AIClassification>;
 export const PostAIClassificationSchema =
-  SchemaFactory.createForClass(PostAIClassification);
+  SchemaFactory.createForClass(AIClassification);

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/index.ts

@@ -2,4 +2,4 @@ export * from '@src/infrastructure/database/schema/user.schema';
 export * from '@src/infrastructure/database/schema/folder.schema';
 export * from '@src/infrastructure/database/schema/keyword.schema';
 export * from '@src/infrastructure/database/schema/post.schema';
-export * from '@src/infrastructure/database/schema/postAIClassification.schema';
+export * from '@src/infrastructure/database/schema/AIClassification.schema';

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

@@ -1,5 +1,6 @@
 import { Prop, Schema, SchemaFactory } from '@nestjs/mongoose';
 import { HydratedDocument, Schema as MongooseSchema } from 'mongoose';
+import { AIClassification } from './AIClassification.schema';
 
 @Schema({ collection: 'posts', timestamps: true, versionKey: false })
 export class Post {
@@ -24,9 +25,9 @@ export class Post {
   @Prop({
     required: false,
     type: MongooseSchema.Types.ObjectId,
-    ref: 'PostAIClassification',
+    ref: 'AIClassification',
   })
-  aiClassificationId?: MongooseSchema.Types.ObjectId;
+  aiClassificationId?: MongooseSchema.Types.ObjectId | AIClassification;
 }
 
 export type PostDocument = HydratedDocument<Post>;

src/modules/classification/classification.controller.ts

@@ -0,0 +1,41 @@
+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 { AIFolderNameListResponse } from './response/ai-folder-list.dto';
+import { AIPostListResponse } from './response/ai-post-list.dto';
+
+@Controller('classification')
+@UseGuards(JwtGuard)
+@ClassificationControllerDocs
+export class ClassificationController {
+  constructor(private readonly classificationService: ClassificationService) {}
+
+  @Get('/suggestions') //TODO : 정렬
+  @GetAIFolderNameListDocs
+  async getSuggestedFolderNameList(@GetUser() userId: String) {
+    const folderNames =
+      await this.classificationService.getFolderNameList(userId);
+
+    return new AIFolderNameListResponse(folderNames);
+  }
+
+  @Get('/suggestions/:folderId')
+  @GetAIPostListDocs
+  async getSuggestedPostList(
+    @GetUser() userId: string,
+    @Param('folderId') folderId: string,
+  ) {
+    const posts = await this.classificationService.getPostList(
+      userId,
+      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,
+  AIClassification,
+  PostAIClassificationSchema,
+  PostSchema,
+} from '@src/infrastructure/database/schema';
+
+import { ClassificationController } from './classification.controller';
+
+@Module({
+  imports: [
+    MongooseModule.forFeature([
+      { name: Folder.name, schema: FolderSchema },
+      { name: AIClassification.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,58 @@
+import { Injectable } from '@nestjs/common';
+import {
+  Folder,
+  Post,
+  AIClassification,
+} 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(AIClassification.name)
+    private postAiClassificationModel: Model<AIClassification>,
+    @InjectModel(Post.name) private postModel: Model<Post>,
+  ) {}
+
+  async getFolderNameList(userId: String): Promise<AIFolderNameServiceDto[]> {
+    const folders = await this.folderModel.find({ userId }).exec();
+    const folderIds = folders.map((folder) => folder._id);
+
+    const classificationIds = await this.postAiClassificationModel
+      .distinct('suggestedFolderId')
+      .where('suggestedFolderId')
+      .in(folderIds)
+      .exec();
+
+    const matchedFolders = await this.folderModel
+      .find({ _id: { $in: classificationIds } })
+      .exec();
+
+    return matchedFolders.map((folder) => new AIFolderNameServiceDto(folder));
+  }
+
+  async getPostList(
+    userId: string,
+    folderId: string,
+  ): Promise<AIPostServiceDto[]> {
+    const posts = await this.postModel
+      .find({ userId: userId })
+      .populate<{
+        aiClassificationId: AIClassification;
+      }>({
+        path: 'aiClassificationId',
+        match: { deletedAt: null, suggestedFolderId: folderId },
+      })
+      .sort({ createdAt: -1 })
+      .exec();
+
+    return posts
+      .filter((post) => post.aiClassificationId)
+      .map((post) => new AIPostServiceDto(post, post.aiClassificationId));
+  }
+}

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 '../response/ai-folder-list.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 '../response/ai-post-list.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,14 @@
+import { ApiProperty } from '@nestjs/swagger';
+import { FolderDocument } from '@src/infrastructure';
+import { IsNotEmpty, IsString } from 'class-validator';
+
+export class AIFolderNameServiceDto {
+  id: string;
+
+  name: string;
+
+  constructor(data: FolderDocument) {
+    this.id = data._id.toString();
+    this.name = data.name;
+  }
+}

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

@@ -0,0 +1,30 @@
+import { ApiProperty } from '@nestjs/swagger';
+import { IsNotEmpty, IsString, IsArray, IsDate } from 'class-validator';
+import { Type } from 'class-transformer';
+import { Post, AIClassification, PostDocument } from '@src/infrastructure';
+import { Document } from 'mongoose';
+import { MergeType } from 'mongoose';
+import { Types } from 'mongoose';
+
+export class AIPostServiceDto {
+  id: string;
+
+  title: string;
+
+  url: string;
+
+  description: string;
+
+  keywords: string[];
+
+  createdAt: Date;
+
+  constructor(post: PostDocument, aiClassificationId: AIClassification) {
+    this.id = post.id;
+    this.title = post.title;
+    this.url = post.url;
+    this.description = post.description;
+    this.keywords = aiClassificationId.keywords;
+    this.createdAt = aiClassificationId.createdAt;
+  }
+}

src/modules/classification/response/ai-folder-list.dto.ts

@@ -0,0 +1,14 @@
+import { ApiProperty } from '@nestjs/swagger';
+import { AIFolderNameServiceDto } from '../dto/getAIFolderNameLIst.dto';
+
+export class AIFolderNameListResponse {
+  @ApiProperty({
+    type: AIFolderNameServiceDto,
+    isArray: true,
+  })
+  list: AIFolderNameServiceDto[];
+
+  constructor(data: AIFolderNameServiceDto[]) {
+    this.list = data;
+  }
+}

src/modules/classification/response/ai-post-list.dto.ts

@@ -0,0 +1,13 @@
+import { ApiProperty } from '@nestjs/swagger';
+import { Type } from 'class-transformer';
+import { AIPostServiceDto } from '../dto/getAIPostList.dto';
+
+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();