From 7c31b75ddb67608b0a86ce2317608f275d44a872 Mon Sep 17 00:00:00 2001
From: Helius <wangdoubleone@gmail.com>
Date: Thu, 14 May 2020 10:56:02 +0800
Subject: [PATCH] modify swagger some config

---
 src/main/java/com/xcong/excoin/configurations/SwaggerConfig.java                 |    6 ++++--
 src/main/java/com/xcong/excoin/common/system/dto/LoginDto.java                   |    4 ++--
 src/main/java/com/xcong/excoin/configurations/security/TokenFilter.java          |    1 -
 src/main/java/com/xcong/excoin/common/system/controller/LoginController.java     |   31 +++++++++++++++++++++++--------
 src/main/java/com/xcong/excoin/configurations/properties/SecurityProperties.java |    2 ++
 src/main/resources/application.yml                                               |    1 +
 6 files changed, 32 insertions(+), 13 deletions(-)

diff --git a/src/main/java/com/xcong/excoin/common/system/controller/LoginController.java b/src/main/java/com/xcong/excoin/common/system/controller/LoginController.java
index 7d601a7..e10dc75 100644
--- a/src/main/java/com/xcong/excoin/common/system/controller/LoginController.java
+++ b/src/main/java/com/xcong/excoin/common/system/controller/LoginController.java
@@ -1,11 +1,15 @@
 package com.xcong.excoin.common.system.controller;
 
 import cn.hutool.core.util.IdUtil;
+import cn.hutool.crypto.asymmetric.KeyType;
+import cn.hutool.crypto.asymmetric.RSA;
 import com.alibaba.fastjson.JSONObject;
 import com.xcong.excoin.common.contants.AppContants;
 import com.xcong.excoin.common.response.Result;
 import com.xcong.excoin.common.system.bean.LoginUserBean;
 import com.xcong.excoin.common.system.dto.LoginDto;
+import com.xcong.excoin.configurations.properties.ApplicationProperties;
+import com.xcong.excoin.configurations.properties.SecurityProperties;
 import com.xcong.excoin.utils.RedisUtils;
 import io.swagger.annotations.Api;
 import lombok.extern.slf4j.Slf4j;
@@ -32,8 +36,11 @@
 @RequestMapping(value = "/")
 public class LoginController {
 
-    @Value("${rsa.private_key}")
-    private String privateKey;
+    @Resource
+    private ApplicationProperties applicationProperties;
+
+    @Resource
+    private SecurityProperties securityProperties;
 
     @Resource
     private AuthenticationManagerBuilder authenticationManagerBuilder;
@@ -48,12 +55,20 @@
         String token = IdUtil.simpleUUID();
         LoginUserBean loginUserBean = (LoginUserBean) authentication.getPrincipal();
         redisUtils.set(AppContants.APP_LOGIN_PREFIX + token, JSONObject.toJSONString(loginUserBean), 300000);
-        Map<String, Object> authInfo = new HashMap<String, Object>(2){
-            {
-                put("token", token);
-                put("user", loginUserBean);
-            }
-        };
+        Map<String, Object> authInfo = new HashMap<>();
+        if (applicationProperties.isDebug()) {
+            authInfo.put("token", token);
+            authInfo.put("rsaToken", generateAsaToken(token));
+            authInfo.put("user", loginUserBean);
+        } else {
+            authInfo.put("token", token);
+            authInfo.put("user", loginUserBean);
+        }
         return Result.ok("success", authInfo);
     }
+
+    public String generateAsaToken(String token) {
+        RSA rsa = new RSA(null, securityProperties.getPublicKey());
+        return rsa.encryptBase64(token + "_" +System.currentTimeMillis(), KeyType.PublicKey);
+    }
 }
diff --git a/src/main/java/com/xcong/excoin/common/system/dto/LoginDto.java b/src/main/java/com/xcong/excoin/common/system/dto/LoginDto.java
index af17f24..1721f62 100644
--- a/src/main/java/com/xcong/excoin/common/system/dto/LoginDto.java
+++ b/src/main/java/com/xcong/excoin/common/system/dto/LoginDto.java
@@ -15,11 +15,11 @@
 public class LoginDto {
 
 
-    @ApiModelProperty(value = "用户名", example = "123")
+    @ApiModelProperty(value = "用户名", example = "11111")
     @NotBlank(message = "用户名或密码错误")
     private String username;
 
-    @ApiModelProperty(value = "密码", example = "3333")
+    @ApiModelProperty(value = "密码", example = "123456")
     @NotBlank(message = "用户名或密码错误")
     private String password;
 }
diff --git a/src/main/java/com/xcong/excoin/configurations/SwaggerConfig.java b/src/main/java/com/xcong/excoin/configurations/SwaggerConfig.java
index 1d1c0aa..1b79633 100644
--- a/src/main/java/com/xcong/excoin/configurations/SwaggerConfig.java
+++ b/src/main/java/com/xcong/excoin/configurations/SwaggerConfig.java
@@ -1,5 +1,7 @@
 package com.xcong.excoin.configurations;
 
+import com.xcong.excoin.modules.member.entity.MemberEntity;
+import io.swagger.annotations.Api;
 import org.springframework.context.annotation.Bean;
 import org.springframework.context.annotation.Configuration;
 import springfox.documentation.builders.ApiInfoBuilder;
@@ -34,8 +36,8 @@
         parameterBuilder.name("Authorization").description("令牌").modelRef(new ModelRef("string")).parameterType("header")
                 .required(false).build();
         parameters.add(parameterBuilder.build());
-        return new Docket(DocumentationType.SWAGGER_2).apiInfo(apiInfo()).select().apis(RequestHandlerSelectors.any())
-                .paths(PathSelectors.any()).build().globalOperationParameters(parameters);
+        return new Docket(DocumentationType.SWAGGER_2).apiInfo(apiInfo()).select().apis(RequestHandlerSelectors.withClassAnnotation(Api.class))
+                .paths(PathSelectors.any()).build().globalOperationParameters(parameters).ignoredParameterTypes(MemberEntity.class);
     }
 
     private ApiInfo apiInfo(){
diff --git a/src/main/java/com/xcong/excoin/configurations/properties/SecurityProperties.java b/src/main/java/com/xcong/excoin/configurations/properties/SecurityProperties.java
index 0227458..abf0d06 100644
--- a/src/main/java/com/xcong/excoin/configurations/properties/SecurityProperties.java
+++ b/src/main/java/com/xcong/excoin/configurations/properties/SecurityProperties.java
@@ -14,4 +14,6 @@
 public class SecurityProperties {
 
     private String privateKey;
+
+    private String publicKey;
 }
diff --git a/src/main/java/com/xcong/excoin/configurations/security/TokenFilter.java b/src/main/java/com/xcong/excoin/configurations/security/TokenFilter.java
index 6ebf4d8..c697505 100644
--- a/src/main/java/com/xcong/excoin/configurations/security/TokenFilter.java
+++ b/src/main/java/com/xcong/excoin/configurations/security/TokenFilter.java
@@ -73,7 +73,6 @@
                 String rsaToken = bearerToken.replace(AppContants.TOKEN_START_WITH, "");
                 RSA rsa = new RSA(securityProperties.getPrivateKey(), null);
                 String[] tokens = StrUtil.split(rsa.decryptStr(rsaToken, KeyType.PrivateKey), "_");
-
                 if (verifyTokenExpired(Long.parseLong(tokens[1]))) {
                     return tokens[0];
                 }
diff --git a/src/main/resources/application.yml b/src/main/resources/application.yml
index c6e5be1..e418a9f 100644
--- a/src/main/resources/application.yml
+++ b/src/main/resources/application.yml
@@ -77,4 +77,5 @@
   debug: true
 
 rsa:
+  public_key: MIGfMA0GCSqGSIb3DQEBAQUAA4GNADCBiQKBgQCCf8UFZK54AiK4PRu7tNd+Z8qZ77o/QXCnk25DRmygVpOEu5mGNSAvfnWmKp2pEV2RljeXq3Rid/+LQkonaebMJeXKSF0yxL/VgyeT8JaQ5gNbOrdfdlc+mFkXJyzyJt8YkvApEdPRNSU2ENBn7mgRfD0BYPM4vZ6/rv+de38FJwIDAQAB
   private_key: MIICdgIBADANBgkqhkiG9w0BAQEFAASCAmAwggJcAgEAAoGBAIJ/xQVkrngCIrg9G7u0135nypnvuj9BcKeTbkNGbKBWk4S7mYY1IC9+daYqnakRXZGWN5erdGJ3/4tCSidp5swl5cpIXTLEv9WDJ5PwlpDmA1s6t192Vz6YWRcnLPIm3xiS8CkR09E1JTYQ0GfuaBF8PQFg8zi9nr+u/517fwUnAgMBAAECgYBhPt9NvpI4wbanvnndLczr2GJkxfzvSE+vwLCJF4C5FusFHVsxZINggQcg1V75bwRgCiXRMyYefreCSdrCditS43PzTOmE4RRrpxLlm8oubJc0C98LQ2qlN9AsUqL5IHpVGgbHDyWAwjc1GBID6nwXKpxq1/VodFqhahG9D5EZsQJBALnkb+5VTxQbiyQI4Uc9NIvAyVcNY1OisbvY6tvNgdBbJkADgAb78M1HWxxYjUqsvzggNHc7cWqWBHMgpnJaqm8CQQCztze4D7uAk7OC9MJHY5eE980J8Kk+GEZKxz4LahzU6V6dcb9GFac3wEtgilj/tOAn9y0/Q8sm9vvCIbMDzgzJAkEAqRYcqhF26LdVDOX25DHMBgLKISDQZFbsjA13M4/usHL4i+mjHrc0BcUOHu59NpuDI65HitzLAUSLr5zXSdUmiQJAW77wOg4GCejdXsB3IhzMsHwU97sdm26nC+vVV9xvJZ6Rx8zW+f9543NOx9U5BCmhuaVtOvvwDU9PTVcI3atmSQJAXAIJ5gGdtXx0DXiX4VvzNFHqgaqHMGvXyjNVkU2FYQbSAd2A6app4uRO+BkZu9dSjh14m+oXMnV2HzAN2rRnjA==

--
Gitblit v1.9.1