Merge pull request #96 from TeamHARA/feat/enablePushAlarm

[FEAT] 알림 활성화/비활성화 API

src/constants/responseMessage.ts

@@ -86,4 +86,10 @@ export default {
   READ_KAKAO_TOKEN_SUCCESS: "카카오 토큰 조회 성공",
   READ_KAKAO_TOKEN_FAIL: "카카오 토큰 조회 실패",
 
+  //알람
+  ALARM_ENABLE_SUCCESS: "알림 활성화 성공",
+  ALARM_ENABLE_FAIL: "알림 활성화 실패",
+  ALARM_DISABLE_SUCCESS: "알림 비활성화 성공",
+  ALARM_DISABLE_FAIL: "알림 비활성화 실패",
+
 };

src/controller/alarmController.ts

@@ -2,9 +2,11 @@ import { NextFunction, Request, Response } from "express";
 import admin from '../modules/firebaseAdmin';
 import { userService } from "../service";
 import tokenService from "../service/tokenService";
-import { alarm } from "../constants";
 import moment from "moment";
 import alarmService from "../service/alarmService";
+import { alarm, rm, sc } from "../constants";
+import { fail, success } from "../constants/response";
+import statusCode from "../constants/statusCode"
 
 const setFinishedAlarm = async(req: Request, res: Response, next: NextFunction) => {
     try{
@@ -175,10 +177,34 @@ const pushAlarmToMany = async (data: any) => {
 
 }
 
+const settingAlarm = async (req: Request, res: Response, next: NextFunction) => {
+    try {
+        const { isTrue } = req.params
+        const { userId, deviceToken } = req.body
+
+        if(+isTrue == 1){
+            await tokenService.setDeviceToken(userId, deviceToken)
+            return res.status(sc.OK).send(success(statusCode.OK, rm.ALARM_ENABLE_SUCCESS));
+
+        }
+        if(+isTrue == 0){
+            await tokenService.disableDeviceToken(userId, deviceToken)
+            return res.status(sc.OK).send(success(statusCode.OK, rm.ALARM_DISABLE_SUCCESS));
+        }
+
+
+    } catch (error) {
+        next(error);
+    }
+}
+
+
+
 export default{
     setFinishedAlarm,
     setOnDeadlineAlarm,
     setBeforeDeadlineAlarm,
     setNoDeadlineAlarm,
-    pushAlarm
+    pushAlarm,
+    settingAlarm
 }

src/repository/tokenRepository.ts

@@ -24,7 +24,7 @@ const findRefreshTokenById = async (userId: number) => {
 const findDeviceTokenById = async (userId: number) => {
     return await prisma.token.findUnique({
         select: {
-            device_token:true
+            device_token: true
         },
         where: {
             user_id: userId,
@@ -78,6 +78,28 @@ const disableRefreshTokenById = async (userId: number) => {
     });
 };
 
+const enableDeviceToken = async (userId: number, deviceToken: string) => {
+    return await prisma.token.update({
+        where: {
+            user_id: userId
+        },
+        data:{
+            device_token: deviceToken
+        }
+    });
+};
+
+const disableDeviceToken = async (userId: number) => {
+    return await prisma.token.update({
+        where: {
+            user_id: userId
+        },
+        data:{
+            device_token: ""
+        }
+    });
+};
+
 
 export default { 
     createRefreshToken,
@@ -86,6 +108,7 @@ export default {
     findIdByRefreshToken,
     findDeviceTokenListByIds,
     updateTokenById,
-    disableRefreshTokenById
-
+    disableRefreshTokenById,
+    enableDeviceToken,
+    disableDeviceToken
 }

src/router/alarmRouter.ts

@@ -0,0 +1,12 @@
+import { Router } from "express";
+import { auth } from "../middlewares";
+import alarmController from "../controller/alarmController";
+
+const router = Router();
+
+router.get("/enable/:isTrue", 
+    auth,
+    alarmController.settingAlarm
+)
+
+export default router;

src/router/index.ts

@@ -4,6 +4,7 @@ import templateRouter from "./templateRouter";
 import worryRouter from "./worryRouter";
 import reviewRouter from "./reviewRouter";
 import authRouter from "./authRouter";
+import alarmRouter from "./alarmRouter";
 
 const router: Router = Router();
 
@@ -12,6 +13,8 @@ router.use("/template",templateRouter);
 router.use("/worry",worryRouter);
 router.use("/review",reviewRouter);
 router.use("/auth",authRouter);
+router.use("/alarm",alarmRouter);
+
 
 
 

src/service/tokenService.ts

@@ -28,8 +28,33 @@ const getDeviceToken =async (userId: number) => {
 }
 
 
+const setDeviceToken =async (userId: number, deviceToken: string) => {
+
+
+    const data = await tokenRepository.enableDeviceToken(userId, deviceToken)
+
+
+    return
+}
+
+const disableDeviceToken =async (userId: number, deviceToken: string) => {
+    const isValidToken = await tokenRepository.findDeviceTokenById(userId)
+    if(!isValidToken){
+        throw new ClientException("device token not found in database");
+    }
+
+    if(isValidToken.device_token != deviceToken){
+        throw new ClientException("wrong device token")
+    }
+
+    return await tokenRepository.disableDeviceToken(userId)
+
+}
+
 
 export default{
     refreshAccessToken,
-    getDeviceToken
+    getDeviceToken,
+    setDeviceToken,
+    disableDeviceToken
 }