From 073f149c57f79d61bf605c9d38e332259b49d96d Mon Sep 17 00:00:00 2001
From: wayne <wayne8692wayne8692@gmail.com>
Date: 星期六, 22 一月 2022 14:56:05 +0800
Subject: [PATCH] Merge branch '未處理預約單API' into Phase3

---
 pamapi/src/main/java/com/pollex/pam/security/SecurityUtils.java |   23 +++++++++++++++++++++++
 1 files changed, 23 insertions(+), 0 deletions(-)

diff --git a/pamapi/src/main/java/com/pollex/pam/security/SecurityUtils.java b/pamapi/src/main/java/com/pollex/pam/security/SecurityUtils.java
index 2e170c5..787c1f3 100644
--- a/pamapi/src/main/java/com/pollex/pam/security/SecurityUtils.java
+++ b/pamapi/src/main/java/com/pollex/pam/security/SecurityUtils.java
@@ -1,8 +1,12 @@
 package com.pollex.pam.security;
 
 import java.util.Arrays;
+import java.util.Map;
 import java.util.Optional;
 import java.util.stream.Stream;
+
+import com.pollex.pam.enums.ConsultantDetailEnum;
+import com.pollex.pam.enums.CustomerDetailEnum;
 import org.springframework.security.core.Authentication;
 import org.springframework.security.core.GrantedAuthority;
 import org.springframework.security.core.context.SecurityContext;
@@ -97,4 +101,23 @@
     private static Stream<String> getAuthorities(Authentication authentication) {
         return authentication.getAuthorities().stream().map(GrantedAuthority::getAuthority);
     }
+
+    public static String getAgentNo() {
+        Map<String, String> userDetails = getCurrentUserDetails();
+        return userDetails.get(ConsultantDetailEnum.AGENT_NO.getValue());
+    }
+
+    public static Long getCustomerDBId() {
+    	Map<String, String> userDetails = getCurrentUserDetails();
+    	return Long.parseLong(userDetails.get(CustomerDetailEnum.DB_ID.getValue()));
+    }
+
+    public static Map<String, String> getCurrentUserDetails() {
+   	 SecurityContext securityContext = SecurityContextHolder.getContext();
+   	 if(securityContext.getAuthentication()==null ||
+   			 securityContext.getAuthentication().getDetails() ==null) {
+   		 return null;
+   	 }
+   	 return (Map<String, String>) securityContext.getAuthentication().getDetails();
+   }
 }

--
Gitblit v1.8.0