Firebase setup

src/main/java/com/iemr/common/controller/firebaseNotification/FirebaseNotificationController.java

@@ -1,7 +1,33 @@
+/*
+* AMRIT – Accessible Medical Records via Integrated Technology
+* Integrated EHR (Electronic Health Records) Solution
+*
+* Copyright (C) "Piramal Swasthya Management and Research Institute"
+*
+* This file is part of AMRIT.
+*
+* This program is free software: you can redistribute it and/or modify
+* it under the terms of the GNU General Public License as published by
+* the Free Software Foundation, either version 3 of the License, or
+* (at your option) any later version.
+*
+* This program is distributed in the hope that it will be useful,
+* but WITHOUT ANY WARRANTY; without even the implied warranty of
+* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+* GNU General Public License for more details.
+*
+* You should have received a copy of the GNU General Public License
+* along with this program.  If not, see https://www.gnu.org/licenses/.
+*
+/*
+* AMRIT – Accessible Medical Records via Integrated Technology
+*/
 package com.iemr.common.controller.firebaseNotification;
 
 import com.iemr.common.model.notification.NotificationMessage;
+import com.iemr.common.model.notification.UserToken;
 import com.iemr.common.service.firebaseNotification.FirebaseNotificationService;
+import com.iemr.common.utils.exception.IEMRException;
 import org.slf4j.Logger;
 import org.slf4j.LoggerFactory;
 import org.springframework.beans.factory.annotation.Autowired;
@@ -20,5 +46,16 @@ public String sendNotificationByToken(@RequestBody NotificationMessage notificat
         return firebaseNotificationService.sendNotification(notificationMessage);
     }
 
+    @RequestMapping(value = "updateToken",method = RequestMethod.POST)
+    public String  updateToken(@RequestBody UserToken userToken){
+        return firebaseNotificationService.updateToken(userToken);
+    }
+
+    @RequestMapping(value = "getToken",method = RequestMethod.GET,headers = "Authorization")
+    public String  getUserToken() throws IEMRException {
+
+        return  firebaseNotificationService.getUserToken();
+    }
+
 
 }

src/main/java/com/iemr/common/data/userToken/UserTokenData.java

@@ -0,0 +1,46 @@
+/*
+* AMRIT – Accessible Medical Records via Integrated Technology
+* Integrated EHR (Electronic Health Records) Solution
+*
+* Copyright (C) "Piramal Swasthya Management and Research Institute"
+*
+* This file is part of AMRIT.
+*
+* This program is free software: you can redistribute it and/or modify
+* it under the terms of the GNU General Public License as published by
+* the Free Software Foundation, either version 3 of the License, or
+* (at your option) any later version.
+*
+* This program is distributed in the hope that it will be useful,
+* but WITHOUT ANY WARRANTY; without even the implied warranty of
+* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+* GNU General Public License for more details.
+*
+* You should have received a copy of the GNU General Public License
+* along with this program.  If not, see https://www.gnu.org/licenses/.
+*
+/*
+* AMRIT – Accessible Medical Records via Integrated Technology
+*/
+package com.iemr.common.data.userToken;
+
+import jakarta.persistence.Column;
+import jakarta.persistence.Entity;
+import jakarta.persistence.Id;
+import jakarta.persistence.Table;
+import lombok.Data;
+
+import java.sql.Timestamp;
+
+@Entity
+@Table(name = "user_tokens", schema = "db_iemr")
+@Data
+public class UserTokenData {
+    @Id
+    @Column(name = "user_id")
+    Integer userId;
+    @Column(name = "token")
+    String token;
+    @Column(name = "updated_at")
+    Timestamp updatedAt;
+}

src/main/java/com/iemr/common/model/notification/NotificationMessage.java

@@ -1,3 +1,27 @@
+/*
+* AMRIT – Accessible Medical Records via Integrated Technology
+* Integrated EHR (Electronic Health Records) Solution
+*
+* Copyright (C) "Piramal Swasthya Management and Research Institute"
+*
+* This file is part of AMRIT.
+*
+* This program is free software: you can redistribute it and/or modify
+* it under the terms of the GNU General Public License as published by
+* the Free Software Foundation, either version 3 of the License, or
+* (at your option) any later version.
+*
+* This program is distributed in the hope that it will be useful,
+* but WITHOUT ANY WARRANTY; without even the implied warranty of
+* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+* GNU General Public License for more details.
+*
+* You should have received a copy of the GNU General Public License
+* along with this program.  If not, see https://www.gnu.org/licenses/.
+*
+/*
+* AMRIT – Accessible Medical Records via Integrated Technology
+*/
 package com.iemr.common.model.notification;
 
 import lombok.Data;
@@ -8,7 +32,6 @@
 public class NotificationMessage {
     private String appType;
     private String token;
-    private String topic;
     private String title;
     private String body;
     private Map<String ,String> data;

src/main/java/com/iemr/common/model/notification/UserToken.java

@@ -0,0 +1,33 @@
+/*
+* AMRIT – Accessible Medical Records via Integrated Technology
+* Integrated EHR (Electronic Health Records) Solution
+*
+* Copyright (C) "Piramal Swasthya Management and Research Institute"
+*
+* This file is part of AMRIT.
+*
+* This program is free software: you can redistribute it and/or modify
+* it under the terms of the GNU General Public License as published by
+* the Free Software Foundation, either version 3 of the License, or
+* (at your option) any later version.
+*
+* This program is distributed in the hope that it will be useful,
+* but WITHOUT ANY WARRANTY; without even the implied warranty of
+* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+* GNU General Public License for more details.
+*
+* You should have received a copy of the GNU General Public License
+* along with this program.  If not, see https://www.gnu.org/licenses/.
+*
+/*
+* AMRIT – Accessible Medical Records via Integrated Technology
+*/
+package com.iemr.common.model.notification;
+
+import lombok.Data;
+
+@Data
+public class UserToken {
+    Integer userId;
+    String token;
+}

src/main/java/com/iemr/common/repo/userToken/UserTokenRepo.java

@@ -0,0 +1,31 @@
+/*
+* AMRIT – Accessible Medical Records via Integrated Technology
+* Integrated EHR (Electronic Health Records) Solution
+*
+* Copyright (C) "Piramal Swasthya Management and Research Institute"
+*
+* This file is part of AMRIT.
+*
+* This program is free software: you can redistribute it and/or modify
+* it under the terms of the GNU General Public License as published by
+* the Free Software Foundation, either version 3 of the License, or
+* (at your option) any later version.
+*
+* This program is distributed in the hope that it will be useful,
+* but WITHOUT ANY WARRANTY; without even the implied warranty of
+* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+* GNU General Public License for more details.
+*
+* You should have received a copy of the GNU General Public License
+* along with this program.  If not, see https://www.gnu.org/licenses/.
+*
+/*
+* AMRIT – Accessible Medical Records via Integrated Technology
+*/
+package com.iemr.common.repo.userToken;
+
+import com.iemr.common.data.userToken.UserTokenData;
+import org.springframework.data.jpa.repository.JpaRepository;
+
+public interface UserTokenRepo extends JpaRepository<UserTokenData,Integer> {
+}

src/main/java/com/iemr/common/service/firebaseNotification/FirebaseNotificationService.java

@@ -1,14 +1,50 @@
+/*
+* AMRIT – Accessible Medical Records via Integrated Technology
+* Integrated EHR (Electronic Health Records) Solution
+*
+* Copyright (C) "Piramal Swasthya Management and Research Institute"
+*
+* This file is part of AMRIT.
+*
+* This program is free software: you can redistribute it and/or modify
+* it under the terms of the GNU General Public License as published by
+* the Free Software Foundation, either version 3 of the License, or
+* (at your option) any later version.
+*
+* This program is distributed in the hope that it will be useful,
+* but WITHOUT ANY WARRANTY; without even the implied warranty of
+* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+* GNU General Public License for more details.
+*
+* You should have received a copy of the GNU General Public License
+* along with this program.  If not, see https://www.gnu.org/licenses/.
+*
+/*
+* AMRIT – Accessible Medical Records via Integrated Technology
+*/
 package com.iemr.common.service.firebaseNotification;
 
 import com.google.firebase.FirebaseException;
 import com.google.firebase.messaging.FirebaseMessaging;
 import com.google.firebase.messaging.Message;
 import com.google.firebase.messaging.Notification;
+import com.iemr.common.data.userToken.UserTokenData;
 import com.iemr.common.model.notification.NotificationMessage;
+import com.iemr.common.model.notification.UserToken;
+import com.iemr.common.repo.userToken.UserTokenRepo;
+import com.iemr.common.utils.CookieUtil;
+import com.iemr.common.utils.JwtUtil;
+import com.iemr.common.utils.exception.IEMRException;
+import jakarta.servlet.http.HttpServletRequest;
 import org.slf4j.Logger;
 import org.slf4j.LoggerFactory;
 import org.springframework.beans.factory.annotation.Autowired;
 import org.springframework.stereotype.Service;
+import org.springframework.web.context.request.RequestContextHolder;
+import org.springframework.web.context.request.ServletRequestAttributes;
+
+import java.sql.Timestamp;
+import java.util.Optional;
 
 @Service
 public class FirebaseNotificationService {
@@ -17,6 +53,16 @@ public class FirebaseNotificationService {
     @Autowired
     FirebaseMessaging firebaseMessaging;
 
+    @Autowired
+    private UserTokenRepo userTokenRepo;
+
+    @Autowired
+    private CookieUtil cookieUtil;
+
+    @Autowired
+    private JwtUtil jwtUtil;
+
+
     public String sendNotification(NotificationMessage notificationMessage) {
 
         Notification notification = Notification.builder().setTitle(notificationMessage.getTitle()).setBody(notificationMessage.getBody()).build();
@@ -34,4 +80,35 @@ public String sendNotification(NotificationMessage notificationMessage) {
         }
     }
 
+    public String updateToken(UserToken userToken) {
+        Optional<UserTokenData> existingTokenData = userTokenRepo.findById(userToken.getUserId());
+
+        UserTokenData userTokenData;
+
+        if (existingTokenData.isPresent()) {
+            // User token exist karta hai => update karna hai
+            userTokenData = existingTokenData.get();
+            userTokenData.setToken(userToken.getToken());
+            userTokenData.setUpdatedAt(new Timestamp(System.currentTimeMillis()));
+        } else {
+            // User token nahi mila => naya insert karna hai
+            userTokenData = new UserTokenData();
+            userTokenData.setUserId(userToken.getUserId());
+            userTokenData.setToken(userToken.getToken());
+            userTokenData.setUpdatedAt(new Timestamp(System.currentTimeMillis()));
+        }
+
+        userTokenRepo.save(userTokenData);
+        return "Save Successfully";
+    }
+
+    public String getUserToken() throws IEMRException {
+        HttpServletRequest requestHeader = ((ServletRequestAttributes) RequestContextHolder.getRequestAttributes())
+                .getRequest();
+        String jwtTokenFromCookie = cookieUtil.getJwtTokenFromCookie(requestHeader);
+        return userTokenRepo.findById(Integer.parseInt(jwtUtil.extractUserId(jwtTokenFromCookie))) // because your userId is Long in DB
+                .map(UserTokenData::getToken)  // mil gaya to token nikalo
+                .orElse(null); // nah
+    }
+
 }