Merge pull request #87 from Together42/82-rotation-bot

seo-wo · web-flow · commit 76dd518b369d · 2024-02-02T17:39:02.000+09:00
Feat: 다음 날 사서 로테이션 알림 슬랙봇 추가
diff --git a/src/rotation/rotations.controller.ts b/src/rotation/rotations.controller.ts
@@ -53,7 +53,6 @@ export class RotationsController {
    * 당일 사서 조회 (달력)
    * 구글 시트를 위한 API
    * Auth : None
-   * 먼가 잘 안되는 것 같기도...
    */
   @Get('/today')
   @ApiOperation({
diff --git a/src/rotation/rotations.service.ts b/src/rotation/rotations.service.ts
@@ -7,27 +7,37 @@ import {
   forwardRef,
 } from '@nestjs/common';
 import { Cron } from '@nestjs/schedule';
+import { ConfigService } from '@nestjs/config';
 import { CreateRegistrationDto } from './dto/create-registration.dto';
 import { CreateRotationDto } from './dto/create-rotation.dto';
 import { UpdateRotationDto } from './dto/update-rotation.dto';
 import { RotationEntity } from './entity/rotation.entity';
 import { RotationAttendeeEntity } from './entity/rotation-attendee.entity';
 import { UserService } from 'src/user/user.service';
 import { RotationRepository } from './repository/rotations.repository';
-import { getFourthWeekdaysOfMonth, getNextYearAndMonth, getTodayDate } from './utils/date';
+import {
+  getFourthFridayOfMonth,
+  getFourthMondayOfMonth,
+  getFourthWeekdaysOfMonth,
+  getNextYearAndMonth,
+  getTodayDay,
+  getTomorrowDate,
+} from './utils/date';
 import { RotationAttendeeRepository } from './repository/rotation-attendees.repository';
 import { DayObject, RotationAttendeeInfo } from './utils/types';
 import { HolidayService } from 'src/holiday/holiday.service';
 import { createRotation } from './utils/rotation';
+import { SlackService } from 'nestjs-slack';
+import { Message } from 'slack-block-builder';
 import { FindTodayRotationDto } from './dto/find-today-rotation.dto';
 import { FindRegistrationDto } from './dto/find-registration.dto';
 import { FindAllRotationDto } from './dto/find-all-rotation.dto';
 
 function getRotationCronTime() {
   if (process.env.NODE_ENV === 'production') {
-    return '59 23 * * 5';
+    return '42 4 27 * *';
   }
-  return '0 0 * * 5';
+  return '0 0 27 * *';
 }
 
 @Injectable()
@@ -39,8 +49,114 @@ export class RotationsService {
     private rotationAttendeeRepository: RotationAttendeeRepository,
     @Inject(forwardRef(() => UserService)) private userService: UserService,
     private holidayService: HolidayService,
+    private slackService: SlackService,
+    private configService: ConfigService,
   ) {}
 
+  /*
+   * 매일 9시 42분에 내일 사서에게 메세지를 전송하는 cron job
+   */
+  @Cron('42 9 * * *', {
+    name: 'sendMessageTomorrowLibrarian',
+    timeZone: 'Asia/Seoul',
+  })
+  async sendMessageTomorrowLibrarian(): Promise<void> {
+    const tomorrow = getTomorrowDate();
+    const tomorrowYear = tomorrow.getFullYear();
+    const tomorrowMonth = tomorrow.getMonth() + 1;
+    const tomorrowDay = tomorrow.getDate();
+
+    const tomorrowLibrarian = await this.rotationRepository.find({
+      where: {
+        year: tomorrowYear,
+        month: tomorrowMonth,
+        day: tomorrowDay,
+      },
+      relations: ['user'],
+    });
+
+    if (tomorrowLibrarian.length === 0) {
+      return;
+    }
+
+    for (const item of tomorrowLibrarian) {
+      if (!item.user) {
+        this.logger.warn('Failed to get tomorrow librarian information. User not found.');
+        return;
+      }
+
+      const message = `[알림] 안녕하세요 ${item.user.nickname}님! 내일 사서 업무가 있습니다.`;
+
+      try {
+        await this.slackService.postMessage(
+          Message({
+            text: message,
+            channel: item.user.slackMemberId,
+          }).buildToObject(),
+        );
+      } catch (error) {
+        this.logger.error(`Error sending message to ${item.user.nickname}: ` + error);
+      }
+    }
+  }
+
+  /*
+   * 매월 23일, 집현전 슬랙 채널에 메세지를 보내는 cron job
+   */
+  @Cron('42 15 23 * *', {
+    name: 'sendMessageRotationDeadlineFirst',
+    timeZone: 'Asia/Seoul',
+  })
+  async sendMessageRotationDeadlineFirst(): Promise<void> {
+    const message =
+      '[알림] 4일 후 로테이션 신청이 마감됩니다.\n[신청하러 가기]: https://together.42jip.net/';
+
+    await this.slackService.postMessage(
+      Message({
+        text: message,
+        channel: this.configService.get('slack.jiphyeonjeonChannel'),
+      }).buildToObject(),
+    );
+  }
+
+  /*
+   * 매월 26일, 집현전 슬랙 채널에 메세지를 보내는 cron job
+   */
+  @Cron('42 15 26 * *', {
+    name: 'sendMessageRotationDeadlineLast',
+    timeZone: 'Asia/Seoul',
+  })
+  async sendMessageRotationDeadlineLast(): Promise<void> {
+    const message =
+      '[알림] 로테이션 신청 기간이 내일 마감됩니다. 오늘까지 신청해주세요!\n[신청하러 가기]: https://together.42jip.net/';
+
+    await this.slackService.postMessage(
+      Message({
+        text: message,
+        channel: this.configService.get('slack.jiphyeonjeonChannel'),
+      }).buildToObject(),
+    );
+  }
+
+  /*
+   * 매월 27일 사서 로테이션 설정이 완료된 후, 집현전 슬랙 채널에 메세지를 보내는 cron job
+   */
+  @Cron('42 15 27 * *', {
+    name: 'sendMessageRotationFinished',
+    timeZone: 'Asia/Seoul',
+  })
+  async sendMessageRotationFinished(): Promise<void> {
+    const message =
+      '[알림] 다음 달 로테이션이 확정되었습니다! 친바 사이트에서 확인해주세요!\n[확인하러 가기]: https://together.42jip.net/';
+
+    await this.slackService.postMessage(
+      Message({
+        text: message,
+        channel: this.configService.get('slack.jiphyeonjeonChannel'),
+      }).buildToObject(),
+    );
+  }
+
   /*
    * 4주차 월요일에 유저를 모두 DB에 담아놓는 작업 필요
    * [update 20231219] - 매 달 1일에 유저를 모두 DB에 담아놓는 작업으로 변경
@@ -72,101 +188,98 @@ export class RotationsService {
    * 매주 금요일을 체크하여, 만약 4주차 금요일인 경우,
    * 23시 59분에 로테이션을 돌린다.
    * 다음 달 로테이션 참석자를 바탕으로 로테이션 결과 반환
+   * [update 20240202] - 매월 4주차 금요일이 아닌, 매월 27일 새벽에 로테이션을 돌린다.
    */
   @Cron(`${getRotationCronTime()}`, {
     name: 'setRotation',
     timeZone: 'Asia/Seoul',
   })
   async setRotation(): Promise<void> {
-    if (getFourthWeekdaysOfMonth().indexOf(getTodayDate()) > 0) {
-      this.logger.log('Setting rotation...');
+    this.logger.log('Setting rotation...');
 
-      const { year, month } = getNextYearAndMonth();
-      const attendeeArray: Partial<RotationAttendeeEntity>[] = await this.getAllRegistration();
-      const monthArrayInfo: DayObject[][] = await this.getInitMonthArray(year, month);
+    const { year, month } = getNextYearAndMonth();
+    const attendeeArray: Partial<RotationAttendeeEntity>[] = await this.getAllRegistration();
+    const monthArrayInfo: DayObject[][] = await this.getInitMonthArray(year, month);
 
-      if (!attendeeArray || attendeeArray.length === 0) {
-        this.logger.warn('No attendees participated in the rotation');
-        return;
-      }
+    if (!attendeeArray || attendeeArray.length === 0) {
+      this.logger.warn('No attendees participated in the rotation');
+      return;
+    }
 
-      const rotationAttendeeInfo: RotationAttendeeInfo[] = attendeeArray.map((attendee) => {
-        const parsedAttendLimit: number[] = Array.isArray(attendee.attendLimit)
-          ? JSON.parse(JSON.stringify(attendee.attendLimit))
-          : [];
-        return {
-          userId: attendee.userId,
-          year: attendee.year,
-          month: attendee.month,
-          attendLimit: parsedAttendLimit,
-          attended: 0,
-        };
-      });
+    const rotationAttendeeInfo: RotationAttendeeInfo[] = attendeeArray.map((attendee) => {
+      const parsedAttendLimit: number[] = Array.isArray(attendee.attendLimit)
+        ? JSON.parse(JSON.stringify(attendee.attendLimit))
+        : [];
+      return {
+        userId: attendee.userId,
+        year: attendee.year,
+        month: attendee.month,
+        attendLimit: parsedAttendLimit,
+        attended: 0,
+      };
+    });
+
+    // 만약 year & month에 해당하는 로테이션 정보가 이미 존재한다면,
+    // 해당 로테이션 정보를 삭제하고 다시 생성한다.
+    const hasInfo = await this.rotationRepository.find({
+      where: {
+        year: year,
+        month: month,
+      },
+    });
+
+    if (hasInfo.length > 0) {
+      this.logger.log('Rotation info already exists. Deleting...');
+      await this.rotationRepository.softRemove(hasInfo);
+    }
 
-      // 만약 year & month에 해당하는 로테이션 정보가 이미 존재한다면,
-      // 해당 로테이션 정보를 삭제하고 다시 생성한다.
-      const hasInfo = await this.rotationRepository.find({
+    const rotationResultArray: DayObject[] = createRotation(rotationAttendeeInfo, monthArrayInfo);
+
+    for (const item of rotationResultArray) {
+      const [userId1, userId2] = item.arr;
+
+      const attendeeOneExist = await this.rotationRepository.findOne({
         where: {
+          userId: userId1,
           year: year,
           month: month,
+          day: item.day,
         },
       });
 
-      if (hasInfo.length > 0) {
-        this.logger.log('Rotation info already exists. Deleting...');
-        await this.rotationRepository.softRemove(hasInfo);
-      }
-
-      const rotationResultArray: DayObject[] = createRotation(rotationAttendeeInfo, monthArrayInfo);
-
-      for (const item of rotationResultArray) {
-        const [userId1, userId2] = item.arr;
+      if (!attendeeOneExist) {
+        const rotation1 = new RotationEntity();
+        rotation1.userId = userId1;
+        rotation1.updateUserId = userId1;
+        rotation1.year = year;
+        rotation1.month = month;
+        rotation1.day = item.day;
 
-        const attendeeOneExist = await this.rotationRepository.findOne({
-          where: {
-            userId: userId1,
-            year: year,
-            month: month,
-            day: item.day,
-          },
-        });
-
-        if (!attendeeOneExist) {
-          const rotation1 = new RotationEntity();
-          rotation1.userId = userId1;
-          rotation1.updateUserId = userId1;
-          rotation1.year = year;
-          rotation1.month = month;
-          rotation1.day = item.day;
-
-          await this.rotationRepository.save(rotation1);
-        }
+        await this.rotationRepository.save(rotation1);
+      }
 
-        const attendeeTwoExist = await this.rotationRepository.findOne({
-          where: {
-            userId: userId2,
-            year: year,
-            month: month,
-            day: item.day,
-          },
-        });
+      const attendeeTwoExist = await this.rotationRepository.findOne({
+        where: {
+          userId: userId2,
+          year: year,
+          month: month,
+          day: item.day,
+        },
+      });
 
-        if (!attendeeTwoExist) {
-          const rotation2 = new RotationEntity();
-          rotation2.userId = userId2;
-          rotation2.updateUserId = userId2;
-          rotation2.year = year;
-          rotation2.month = month;
-          rotation2.day = item.day;
+      if (!attendeeTwoExist) {
+        const rotation2 = new RotationEntity();
+        rotation2.userId = userId2;
+        rotation2.updateUserId = userId2;
+        rotation2.year = year;
+        rotation2.month = month;
+        rotation2.day = item.day;
 
-          await this.rotationRepository.save(rotation2);
-        }
+        await this.rotationRepository.save(rotation2);
       }
-
-      this.logger.log('Successfully set rotation!');
-    } else {
-      // skipped...
     }
+
+    this.logger.log('Successfully set rotation!');
   }
 
   /*
@@ -264,7 +377,7 @@ export class RotationsService {
     const { year, month } = getNextYearAndMonth();
 
     /* 4주차인지 확인 */
-    // if (getFourthWeekdaysOfMonth().indexOf(getTodayDate()) < 0) {
+    // if (getFourthWeekdaysOfMonth().indexOfDay()) < 0) {
     //   throw new BadRequestException(
     //     'Invalid date: Today is not a fourth weekday of the month.',
     //   );
diff --git a/src/rotation/utils/date.ts b/src/rotation/utils/date.ts
@@ -11,7 +11,11 @@ export function getNextYearAndMonth(): { year: number; month: number } {
   return { year, month };
 }
 
-export function getTodayDate(): number {
+export function getTomorrowDate(): Date {
+  return new Date(new Date().getTime() + 24 * 60 * 60 * 1000);
+}
+
+export function getTodayDay(): number {
   return new Date().getDate();
 }
 
@@ -33,8 +37,7 @@ const getFourthWeekPeriod = (date = new Date()): number[] => {
     dateOfThursdayOnFirstWeek = 1 + DAY_IN_WEEK + DAY_OF_THURSDAY - firstDay;
   }
 
-  const dateOfThursdayOfFourthWeek =
-    dateOfThursdayOnFirstWeek + 3 * DAY_IN_WEEK;
+  const dateOfThursdayOfFourthWeek = dateOfThursdayOnFirstWeek + 3 * DAY_IN_WEEK;
   const dateOfMondayOnFourthWeek = dateOfThursdayOfFourthWeek - 3;
   const dateOfSundayOnFourthWeek = dateOfThursdayOfFourthWeek + 3;
 
@@ -43,8 +46,7 @@ const getFourthWeekPeriod = (date = new Date()): number[] => {
 
 export const getFourthWeekdaysOfMonth = (date = new Date()): number[] => {
   // eslint-disable-next-line @typescript-eslint/no-unused-vars
-  const [dateOfMondayOnFourthWeek, dateOfSundayOnFourthWeek] =
-    getFourthWeekPeriod(date);
+  const [dateOfMondayOnFourthWeek, dateOfSundayOnFourthWeek] = getFourthWeekPeriod(date);
   const fourthWeekdays: number[] = [];
 
   for (let i = 0; i < 5; i++) {
@@ -55,10 +57,15 @@ export const getFourthWeekdaysOfMonth = (date = new Date()): number[] => {
   return fourthWeekdays;
 };
 
+export const getFourthMondayOfMonth = (date = new Date()): number => {
+  // eslint-disable-next-line @typescript-eslint/no-unused-vars
+  const [dateOfMondayOnFourthWeek] = getFourthWeekPeriod(date);
+  return dateOfMondayOnFourthWeek;
+};
+
 export const getFourthFridayOfMonth = (date = new Date()): number => {
   // eslint-disable-next-line @typescript-eslint/no-unused-vars
-  const [dateOfMondayOnFourthWeek, dateOfSundayOnFourthWeek] =
-    getFourthWeekPeriod(date);
+  const [dateOfMondayOnFourthWeek, dateOfSundayOnFourthWeek] = getFourthWeekPeriod(date);
   const dateOfFridayOnFourthWeek = dateOfSundayOnFourthWeek - 2;
 
   return dateOfFridayOnFourthWeek;
