From 75fd8a02c058f6a51e3c5ea36d997c157c070793 Mon Sep 17 00:00:00 2001
From: wayne <wayne8692wayne8692@gmail.com>
Date: 星期三, 24 十一月 2021 16:02:32 +0800
Subject: [PATCH] Merge remote-tracking branch 'origin/master'

---
 pamapi/src/main/java/com/pollex/pam/web/rest/OtpResource.java |   56 ++++++++++++++++++++++++++++++++++++++++----------------
 1 files changed, 40 insertions(+), 16 deletions(-)

diff --git a/pamapi/src/main/java/com/pollex/pam/web/rest/OtpResource.java b/pamapi/src/main/java/com/pollex/pam/web/rest/OtpResource.java
index 65abd26..cd6fb2b 100644
--- a/pamapi/src/main/java/com/pollex/pam/web/rest/OtpResource.java
+++ b/pamapi/src/main/java/com/pollex/pam/web/rest/OtpResource.java
@@ -5,7 +5,11 @@
 import com.pollex.pam.security.jwt.JWTFilter;
 import com.pollex.pam.security.jwt.TokenProvider;
 import com.pollex.pam.security.token.OtpAuthenticationToken;
+import com.pollex.pam.service.CustomerAuthService;
+import com.pollex.pam.service.CustomerService;
+import com.pollex.pam.service.OtpTmpService;
 import com.pollex.pam.service.OtpWebService;
+import com.pollex.pam.service.dto.CustomerRegisterDTO;
 import com.pollex.pam.service.dto.OtpResponseDTO;
 import com.pollex.pam.web.rest.vm.*;
 import org.slf4j.Logger;
@@ -41,15 +45,35 @@
 
     @Autowired
     TokenProvider tokenProvider;
+    
+    @Autowired
+    CustomerAuthService customerAuthService;
+    
+    @Autowired
+    OtpTmpService otpTmpService;
+    
+    @Autowired
+    CustomerService customerService;
 
     @PostMapping("/sendOtp")
     public ResponseEntity<Object> sendOtp(@RequestBody OtpLoginVM login) {
-        if(applicationProperty.isMockLogin()) {
-            return new ResponseEntity<>(getMockSendOtpResponse(), HttpStatus.OK);
-        }
-
-        if(login.getLoginType() == OtpLoginTypeEnum.SMS) {
-            return new ResponseEntity<>(otpWebService.sendByPhone(login.getAccount()), HttpStatus.OK);
+    	OtpResponseDTO otpResponse;
+    	try {
+            if(applicationProperty.isMockLogin()) {
+            	otpResponse = getMockSendOtpResponse();
+            }else if(login.getLoginType() == OtpLoginTypeEnum.SMS) {
+            	otpResponse = otpWebService.sendByPhone(login.getAccount());
+            }
+            else if(login.getLoginType() == OtpLoginTypeEnum.EMAIL) {
+            	otpResponse = otpWebService.sendByEmail(login.getAccount());
+            }else {
+            	return ResponseEntity.status(HttpStatus.BAD_REQUEST).body("can not support this login type, loginType = " + login.getLoginType().name());
+            }
+            otpTmpService.createOtpTmp(login.getAccount(), otpResponse.getIndexKey());
+            return new ResponseEntity<>(otpResponse, HttpStatus.OK);
+            
+        } catch (ServiceException | RemoteException e) {
+            return ResponseEntity.status(HttpStatus.INTERNAL_SERVER_ERROR).body("connecting otp web service error");
         }
         else if(login.getLoginType() == OtpLoginTypeEnum.EMAIL) {
             return new ResponseEntity<>(otpWebService.sendByEmail(login.getAccount()), HttpStatus.OK);
@@ -60,22 +84,22 @@
 
     @PostMapping("/verify")
     public ResponseEntity<UserJWTController.JWTToken> verifyOtp(@RequestBody VerifyOtpVM verifyOtpParam) {
-        OtpAccount otpAccount = new OtpAccount(verifyOtpParam.getAccount(), verifyOtpParam.getIndexKey());
-        OtpAuthenticationToken authenticationToken = new OtpAuthenticationToken(
-            otpAccount,
-            verifyOtpParam.getOtpCode()
-        );
-
-        Authentication authentication = authenticationManagerBuilder.getObject().authenticate(authenticationToken);
-        SecurityContextHolder.getContext().setAuthentication(authenticationToken);
-        String jwt = tokenProvider.createToken(authentication, false);
+        String jwt = customerAuthService.authorize(verifyOtpParam.getAccount(), verifyOtpParam.getIndexKey(), verifyOtpParam.getOtpCode());
         HttpHeaders httpHeaders = new HttpHeaders();
         httpHeaders.add(JWTFilter.AUTHORIZATION_HEADER, "Bearer" + jwt);
         return new ResponseEntity<>(new UserJWTController.JWTToken(jwt), httpHeaders, HttpStatus.OK);
     }
-
+ 
     private OtpResponseDTO getMockSendOtpResponse() {
         String indexKey = UUID.randomUUID().toString().substring(0, 8);
         return new OtpResponseDTO(Arrays.asList(indexKey, "0", "", ""));
     }
+    
+    @PostMapping("/register")
+    public ResponseEntity<UserJWTController.JWTToken> registerAccount(@RequestBody CustomerRegisterDTO registDTO) {
+    	String jwt = customerService.registerCustomer(registDTO);
+    	HttpHeaders httpHeaders = new HttpHeaders();
+        httpHeaders.add(JWTFilter.AUTHORIZATION_HEADER, "Bearer" + jwt);
+        return new ResponseEntity<>(new UserJWTController.JWTToken(jwt), httpHeaders, HttpStatus.OK);
+    }
 }

--
Gitblit v1.8.0