From 98891d1d1bcaa414f336b9c05ad94d39fe9cb1bc Mon Sep 17 00:00:00 2001
From: Mila <Mila@pollex.com.tw>
Date: 星期一, 29 十一月 2021 15:34:57 +0800
Subject: [PATCH] Merge branch 'master' of https://192.168.0.10:8443/r/pcalife/PAM

---
 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