From d45d0da48844fa5a42609e11fa2cbbfd1b1f2169 Mon Sep 17 00:00:00 2001
From: zhangqinbin <181961702@qq.com>
Date: Sat, 23 Dec 2023 00:00:27 +0800
Subject: [PATCH] =?UTF-8?q?=E6=94=BF=E4=BC=81WPS?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 .../extend/wps/controller/WpsController.java     | 16 +++++++++++-----
 1 file changed, 11 insertions(+), 5 deletions(-)

diff --git a/src/main/java/com/chinaunicom/mall/ebtp/extend/wps/controller/WpsController.java b/src/main/java/com/chinaunicom/mall/ebtp/extend/wps/controller/WpsController.java
index 6063813..b12aef9 100644
--- a/src/main/java/com/chinaunicom/mall/ebtp/extend/wps/controller/WpsController.java
+++ b/src/main/java/com/chinaunicom/mall/ebtp/extend/wps/controller/WpsController.java
@@ -157,12 +157,18 @@ public class WpsController {
 
 		if (cookie.isPresent()) {
 			log.info(String.valueOf(request.getRequestURL()));
-			String referer = request.getHeader("Referer");
-			String r1 = referer.substring(0,referer.indexOf("//")+2);
-			String r11 = referer.substring(r1.length(),referer.length());
-			String r2 = r11.substring(0,r11.indexOf("/")>=0?r11.indexOf("/"):r11.length());
-			String host = r1+r2;
+//			String referer = request.getHeader("Referer");
+//			String r1 = referer.substring(0,referer.indexOf("//")+2);
+//			String r11 = referer.substring(r1.length(),referer.length());
+//			String r2 = r11.substring(0,r11.indexOf("/")>=0?r11.indexOf("/"):r11.length());
+//			String host = r1+r2;
+			String host = request.getHeader("Host");
 			log.info("getWpsUrlBy host:"+host);
+			if(host.indexOf("18022")>=0){
+				host = "http://"+host;
+			}else if(host.indexOf("18023")>=0){
+				host = "https://"+host;
+			}
 
 			String authToken = cookie.get().getValue();
 			WpsClientInVo in = this.wpsService.getKSWpsUrlDecrypt(key,fileName);