[#18] 게시판 스크랩 기능

Author: sejin5

src/app.module.ts

@@ -8,6 +8,7 @@ import { AuthModule } from './auth/auth.module';
 import { ormConfig } from 'orm.config';
 import { KakaoMapModule } from './kakaomap/map.module';
 import { BoardsModule } from './boards/boards.module';
+import { ScrapModule } from './scrap/scrap.module';
 
 @Module({
   imports: [
@@ -27,7 +28,8 @@ import { BoardsModule } from './boards/boards.module';
     UsersModule,
     AuthModule,
     KakaoMapModule,
-    BoardsModule
+    BoardsModule,
+    ScrapModule
   ],
   controllers: [AppController],
   providers: [AppService]

src/boards/boards.module.ts

@@ -5,7 +5,10 @@ import { TypeOrmModule } from '@nestjs/typeorm';
 import { BoardsEntity } from './entities/boards.entity';
 
 @Module({
-  imports: [TypeOrmModule.forFeature([BoardsEntity])],
+  imports: [
+    TypeOrmModule.forFeature([BoardsEntity]),
+    BoardsModule],
+  exports: [BoardsService],
   controllers: [BoardsController],
   providers: [BoardsService]
 })

src/boards/entities/boards.entity.ts

@@ -1,4 +1,5 @@
-import { Entity, Column, PrimaryGeneratedColumn, CreateDateColumn, UpdateDateColumn, DeleteDateColumn } from "typeorm";
+import { Entity, Column, PrimaryGeneratedColumn, CreateDateColumn, UpdateDateColumn, DeleteDateColumn, OneToMany } from "typeorm";
+import { ScrapEntity } from "src/scrap/scrap.entity";
 
 @Entity('boards')
 export class BoardsEntity {
@@ -20,6 +21,9 @@ export class BoardsEntity {
     @DeleteDateColumn()
     deletedAt: Date;
 
+    @OneToMany(() => ScrapEntity, scrap => scrap.board)
+    bridges: ScrapEntity[];
+
     public static validate(title: string, contents: string): boolean {
         const isTitleValid = title;
         const isContentsValid = contents;

src/scrap/scrap.controller.spec.ts

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

src/scrap/scrap.controller.ts

@@ -0,0 +1,24 @@
+import { Controller, Post, UseGuards, Param, ParseIntPipe, Delete } from '@nestjs/common';
+import { ScrapService } from './scrap.service';
+import { ScrapEntity } from './scrap.entity';
+import { User } from 'src/users/user.decorator';
+import { UserEntity } from 'src/users/entities/user.entity';
+import { JwtAuthGuard } from 'src/auth/security/passport.jwt';
+
+@Controller('scrap')
+export class ScrapController {
+    constructor(private readonly scrapService: ScrapService) { }
+
+    @UseGuards(JwtAuthGuard)
+    @Post(':boardId')
+    async createScrap(@User() user: UserEntity,
+        @Param('boardId', ParseIntPipe) boardId: number) {
+        return this.scrapService.createScrap(user, boardId);
+    }
+
+    @Delete(':boardId')
+    async deleteScrap(@User() user: UserEntity,
+        @Param('boardId', ParseIntPipe) boardId: number) {
+        return this.scrapService.deleteScrap(user, boardId);
+    }
+}

src/scrap/scrap.entity.ts

@@ -0,0 +1,23 @@
+import { Entity, PrimaryColumn, ManyToOne, CreateDateColumn, JoinColumn } from "typeorm";
+import { UserEntity } from "src/users/entities/user.entity";
+import { BoardsEntity } from "src/boards/entities/boards.entity";
+
+@Entity('scrap')
+export class ScrapEntity {
+    @PrimaryColumn()
+    userNo: number;
+
+    @PrimaryColumn()
+    boardId: number;
+
+    @CreateDateColumn()
+    createdAt: Date;
+
+    @ManyToOne(() => UserEntity, (user) => user.bridges)
+    @JoinColumn({ name: 'userNo', referencedColumnName: 'userNo' })
+    user: UserEntity;
+
+    @ManyToOne(() => BoardsEntity, (board) => board.bridges)
+    @JoinColumn({ name: 'boardId', referencedColumnName: 'boardId' })
+    board: BoardsEntity;
+}

src/scrap/scrap.module.ts

@@ -0,0 +1,17 @@
+import { Module } from '@nestjs/common';
+import { ScrapController } from './scrap.controller';
+import { ScrapService } from './scrap.service';
+import { UsersModule } from 'src/users/users.module';
+import { BoardsModule } from 'src/boards/boards.module';
+import { TypeOrmModule } from '@nestjs/typeorm';
+import { ScrapEntity } from './scrap.entity';
+
+@Module({
+  imports: [
+    TypeOrmModule.forFeature([ScrapEntity]),
+    UsersModule,
+    BoardsModule],
+  controllers: [ScrapController],
+  providers: [ScrapService]
+})
+export class ScrapModule { }

src/scrap/scrap.service.spec.ts

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

src/scrap/scrap.service.ts

@@ -0,0 +1,46 @@
+import { Injectable } from '@nestjs/common';
+import { ScrapEntity } from './scrap.entity';
+import { UsersService } from 'src/users/users.service';
+import { BoardsService } from 'src/boards/boards.service';
+import { DataSource, Repository } from 'typeorm';
+import { UserEntity } from 'src/users/entities/user.entity';
+import { BoardsEntity } from 'src/boards/entities/boards.entity';
+import { InjectRepository } from '@nestjs/typeorm';
+
+@Injectable()
+export class ScrapService {
+    constructor(private readonly dataSource: DataSource,
+        @InjectRepository(ScrapEntity)
+        private scrapRepository: Repository<ScrapEntity>) { }
+
+    async createScrap(user, postId) {
+        const userNo = user.payload.userNo;
+        const userForSave = new UserEntity;
+        userForSave.userNo = userNo;
+
+        await this.dataSource.manager.save(userForSave);
+
+        const board = new BoardsEntity;
+        board.boardId = postId;
+
+        await this.dataSource.manager.save(board);
+
+        const scrap = new ScrapEntity;
+        scrap.userNo = userNo;
+        scrap.boardId = postId;
+
+        await this.dataSource.manager.save(scrap);
+
+        return '스크랩 되었습니다.';
+    }
+
+    async deleteScrap(user, boardId) {
+        const scrap = new ScrapEntity;
+        scrap.userNo = user.payload.userNo;
+        scrap.boardId = boardId;
+
+        await this.scrapRepository.remove(scrap);
+    }
+
+    //TODO scrap read
+}

src/users/entities/user.entity.ts

@@ -1,5 +1,6 @@
-import { PrimaryGeneratedColumn, Column, BeforeInsert, Entity} from 'typeorm';
+import { PrimaryGeneratedColumn, Column, BeforeInsert, Entity, OneToMany } from 'typeorm';
 import * as bcrypt from 'bcrypt';
+import { ScrapEntity } from 'src/scrap/scrap.entity';
 
 @Entity('tb_user')
 export class UserEntity {
@@ -19,8 +20,11 @@ export class UserEntity {
     password: string;
 
     @BeforeInsert()
-    private beforeInsert(){
+    private beforeInsert() {
         this.password = bcrypt.hashSync(this.password, 5);
     }
 
+    @OneToMany(() => ScrapEntity, scrap => scrap.user)
+    bridges: ScrapEntity[];
+
 }