1
import { BadRequestException, Module } from '@nestjs/common'
2
import * as multer from 'multer'
3
import { v4 as uuid } from 'uuid'
4

5
import e from 'express'
6
import { CommonService } from './common.service'
7
import { CommonController } from './common.controller'
8
import { MulterModule } from '@nestjs/platform-express'
9
import { TEMP_FOLDER_PATH } from './const/path.const'
10
import { extname } from 'path'
11
import { AuthModule } from 'src/auth/auth.module'
12
import { UsersModule } from 'src/users/users.module'
13

14
@Module({
15
  imports: [
16
    AuthModule,
17
    UsersModule,
18
    MulterModule.register({
19
      limits: {
20
        // byte 단위로 입력 (10000000byte -> 10MB가 넘는 파일은 에러)
21
        fieldSize: 10 * 1024 * 1024,
22
      },
23
      /*** cb(에러, boolean)
24
       * 첫번쨰 파라미터에는 에러가 있을 경우 에러 정보를 넣어준다.
25
       * 두번쨰 파라미터는 파일을 받을지 말지 boolean을 넣어준다.
26
       */
27
      fileFilter: (req, file, cb) => {
28
        // xxx.jpg -> .jpg같이 확장자만 가져와줌
29
        const ext = extname(file.originalname)
30
        if (ext !== '.jpg' && ext !== '.jpeg' && ext !== '.png') {
31
          return cb(
32
            new BadRequestException('jpg/jpeg/png 파일만 업로드 가능합니다!'), //
33
            false,
34
          )
35
        }
36
        return cb(null, true)
37
      },
38
      storage: multer.diskStorage({
39
        // 파일을 어디에 보낼지 정의
40
        destination: (req, res, cb) => {
41
          cb(null, TEMP_FOLDER_PATH)
42
        },
43
        filename: (
44
          req: e.Request,
45
          file: Express.Multer.File,
46
          callback: (error: Error | null, filename: string) => void,
47
        ) => {
48
          callback(null, `${uuid()}${extname(file.originalname)}`)
49
        },
50
      }),
51
    }),
52
  ],
53
  controllers: [CommonController],
54
  providers: [CommonService],
55
  exports: [CommonService],
56
})
57
export class CommonModule {}

1
import { join } from 'path'
2
import * as process from 'process'
3

4
// 서버 프로젝트의 루트 폴더
5
export const PROJECT_ROOT_PATH = process.cwd()
6
// 외부에서 접근 가능한 파일들을 모아둔 폴더 이름
7
export const PUBLIC_FOLDER_NAME = 'public'
8
// 포스트 이미지들을 저장할 폴더 이름
9
export const POSTS_FOLDER_NAME = 'posts'
10
// 임시 폴더 이름
11
export const TEMP_FOLDER_NAME = 'temp'
12

13
// 실제 공개폴더의 절대 경로
14
// /{프로젝트의 위치}/public
15
export const PUBLIC_FOLDER_PATH = join(PROJECT_ROOT_PATH, PUBLIC_FOLDER_NAME)
16

17
// 포스트 이미지를 저장할 폴더
18
// /{프로젝트의 위치}/public/posts
19
export const POST_IMAGE_PATH = join(PUBLIC_FOLDER_PATH, POSTS_FOLDER_NAME)
20

21
// 절대경로 x
22
// public/posts/xxx.jpg
23
export const POST_PUBLIC_IMAGE_PATH = join(PUBLIC_FOLDER_NAME, POSTS_FOLDER_NAME)
24

25
// 임시 파일들을 저장할 폴더
26
// {프로젝트 경로}/temp
27
export const TEMP_FOLDER_PATH = join(PUBLIC_FOLDER_PATH, TEMP_FOLDER_NAME)

1
import { Controller, Post, UploadedFile, UseGuards, UseInterceptors } from '@nestjs/common'
2
import { CommonService } from './common.service'
3
import { FileInterceptor } from '@nestjs/platform-express'
4
import { AccessTokenGuard } from 'src/auth/guard/bearer-token.guard'
5

6
@Controller('common')
7
export class CommonController {
8
  constructor(private readonly commonService: CommonService) {}
9

10
  @Post('image')
11
  @UseInterceptors(FileInterceptor('image'))
12
  @UseGuards(AccessTokenGuard)
13
  postImage(@UploadedFile() file: Express.Multer.File) {
14
    return {
15
      fileName: file.filename,
16
    }
17
  }
18
}

1
// posts.controller.ts 생략
2
/*** 3) POST /posts
3
 * post를 생성한다
4
 */
5
@Post()
6
@UseGuards(AccessTokenGuard)
7
postPosts(
8
  @User('id') userId: number,
9
  @Body() body: CreatePostDto,
10
  // @Body('title') title: string,
11
  // @Body('content') content: string,
12
  // 기본값을 true로 설정하는 파이프
13
  // @Body('isPublic', new DefaultValuePipe(true)) isPublic: boolean,
14
) {
15
  return this.postsService.createPost(userId, body)
16
}

1
// posts.service.ts 생략
2
/**
3
 * 1) create : 저장할 객체를 생성
4
 * 2) save   : 객체를 저장 (create 메서드에서 생성한 객체로)
5
 */
6
async createPost(authorId: number, postDto: CreatePostDto) {
7
  const post = this.postsRepository.create({
8
    author: {
9
      id: authorId,
10
    },
11
    ...postDto,
12
    likeCount: 0,
13
    commentCount: 0,
14
  })
15
  const newPost = await this.postsRepository.save(post)
16
  return newPost
17
}

1
import { IsOptional, IsString } from 'class-validator'
2

3
import { PickType } from '@nestjs/mapped-types'
4
import { PostsModel } from '../entities/posts.entity'
5

6
/***
7
 * Pick, Omit, Partial -> Type을 반환
8
 * PickType, OmitType, PartialType -> 값을 반환
9
 */
10
export class CreatePostDto extends PickType(PostsModel, ['title', 'content']) {
11
  @IsString()
12
  @IsOptional()
13
  image?: string
14
}

1
// posts.service.ts 생략
2
async createPostImage(dto: CreatePostDto) {
3
  // dto의 이미지 이름을 기바으로 파일 경로를 생성한다
4
  const tempFilePath = join(TEMP_FOLDER_PATH, dto.image)
5

6
  try {
7
    /*** promises의 fs 모듈을 import
8
     * 파일이 존재하는지 확인
9
     * 만약에 존재하지 않는다면 에러를 던짐
10
     */
11
    await promises.access(tempFilePath)
12
  } catch (e) {
13
    throw new BadRequestException('존재하지 않는 임시 파일입니다!')
14
  }
15

16
  /*** 파일의 이름만 가져오기
17
   * /USers/aaa/bbb/ccc/asdf.jpg -> asdf.jpg
18
   */
19
  const fileName = basename(tempFilePath)
20

21
  /*** 새로 이동할 포스트 폴더의 경로 + 이미지의 이름
22
   * {프로젝트경로}/public/posts/asdf.jpg
23
   */
24
  const publicFilePath = join(POST_IMAGE_PATH, fileName)
25

26
  // 파일 옮기기
27
  await promises.rename(tempFilePath, publicFilePath)
28

29
  return true
30
}

1
// posts.controller.ts 생략
2
@Post()
3
@UseGuards(AccessTokenGuard)
4
async postPosts(
5
  @User('id') userId: number,
6
  @Body() body: CreatePostDto,
7
  // @Body('title') title: string,
8
  // @Body('content') content: string,
9
  // 기본값을 true로 설정하는 파이프
10
  // @Body('isPublic', new DefaultValuePipe(true)) isPublic: boolean,
11
) {
12
  await this.postsService.createPostImage(body)
13
  return this.postsService.createPost(userId, body)
14
}