Merge pull request #20 from DevKor-github/infra/logging

Infra/logging 오류 수정

Author: whsung0330

src/log/log.controller.ts

@@ -1,24 +1,39 @@
-import { Controller, Get, Res, StreamableFile } from '@nestjs/common';
+import {
+  Controller,
+  Get,
+  Param,
+  Query,
+  Res,
+  StreamableFile,
+} from '@nestjs/common';
 import { LogService } from './log.service';
 import { ApiOperation, ApiTags } from '@nestjs/swagger';
 import { Public } from 'src/auth/auth.guard';
+import { string } from 'joi';
 
 @ApiTags('Log')
 @Controller('log')
 export class LogController {
   constructor(private readonly logService: LogService) {}
 
+  @Public()
+  @Get('/test')
+  getTest() {
+    console.log('test');
+    return 'test';
+  }
+
   @ApiOperation({ summary: '로그 파일 확인' })
   @Public()
   @Get()
-  getLog(): StreamableFile {
-    return this.logService.getLogs();
+  getLog(@Query('pw') pw: string): StreamableFile {
+    return this.logService.getLogs(pw);
   }
 
   @ApiOperation({ summary: '에러 로그 확인' })
   @Public()
   @Get('/error')
-  getErrorLog(): StreamableFile {
-    return this.logService.getErrorLogs();
+  getErrorLog(@Query('pw') pw: string): StreamableFile {
+    return this.logService.getErrorLogs(pw);
   }
 }

src/log/log.service.ts

@@ -1,4 +1,9 @@
-import { Injectable, NotFoundException, StreamableFile } from '@nestjs/common';
+import {
+  Injectable,
+  NotFoundException,
+  StreamableFile,
+  UnauthorizedException,
+} from '@nestjs/common';
 import { ConfigService } from '@nestjs/config';
 import { createReadStream, existsSync } from 'fs';
 
@@ -10,15 +15,22 @@ export class LogService {
     this.logFilePath = this.configService.get('LOG_FILE_PATH');
     this.errorLogFilePath = this.configService.get('ERROR_LOG_FILE_PATH');
   }
-  getLogs(): StreamableFile {
+  getLogs(pw: string): StreamableFile {
+    if (pw !== this.configService.get('LOG_PW')) {
+      throw new UnauthorizedException('로그 파일을 확인할 권한이 없습니다.');
+    }
     if (!existsSync(this.logFilePath)) {
       throw new NotFoundException('로그 파일이 존재하지 않습니다.');
     }
     return new StreamableFile(createReadStream(this.logFilePath), {
       type: 'text/plain',
     });
   }
-  getErrorLogs(): StreamableFile {
+
+  getErrorLogs(pw: string): StreamableFile {
+    if (pw != this.configService.get('LOG_PW')) {
+      throw new UnauthorizedException('로그 파일을 확인할 권한이 없습니다.');
+    }
     if (!existsSync(this.logFilePath)) {
       throw new NotFoundException('로그 파일이 존재하지 않습니다.');
     }