From b0500fee9c01e17c81b9293b244c8ed88dc9ca29 Mon Sep 17 00:00:00 2001
From: liuxiaolong <736321739@qq.com>
Date: 星期五, 11 十月 2019 15:12:25 +0800
Subject: [PATCH] add swagger

---
 middlewares/auth/auth.go |   43 +++++++++++++++++++++++++++----------------
 1 files changed, 27 insertions(+), 16 deletions(-)

diff --git a/middlewares/auth/auth.go b/middlewares/auth/auth.go
index b029741..98d4836 100644
--- a/middlewares/auth/auth.go
+++ b/middlewares/auth/auth.go
@@ -1,11 +1,13 @@
 package auth
 
 import (
-	"fmt"
 	"github.com/gin-gonic/gin"
 	"net/http"
 	"strings"
+	"webserver/extend/code"
 	"webserver/extend/config"
+	"basic.com/valib/logger.git"
+	"webserver/extend/util"
 )
 
 const (
@@ -28,27 +30,36 @@
 func AuthHandler() gin.HandlerFunc {
 	return func(c *gin.Context) {
 		urlPath := c.Request.URL.Path
+
 		if strings.Contains(urlPath,"/data/api-v") && !strings.Contains(urlPath,"login"){
-			//jwtDriver :=NewJwtAuthDriver()
-			//if !jwtDriver.Check(c) {
-			//	util.ResponseFormat(c,code.TokenNotFound,"灏氭湭鐧诲綍锛岃鐧诲綍")
-			//	c.Abort()
-			//}
+			jwtDriver :=NewJwtAuthDriver()
+			if !jwtDriver.Check(c) {
+				util.ResponseFormat(c,code.TokenNotFound,"灏氭湭鐧诲綍锛岃鐧诲綍")
+				c.Abort()
+			}
 			c.Next()
-		} else if strings.HasPrefix(urlPath,"/httpImage") {
-			imgUrl := config.Server.ImageUrl
-			idx := strings.Index(urlPath, ":")
-			tmpPath := urlPath
+		} else if strings.Contains(urlPath,"/httpImage") {
+			domain := config.Server.PublicDomain
+			//domainReg := regexp.MustCompile(``+domain+``)
+			//if domainReg.MatchString(host) {//鍩熷悕璁块棶
+			//   imgUrl = domain
+			//}
+
+			urlPath = strings.Replace(urlPath, "/httpImage", "", -1)
+			if strings.Contains(urlPath,domain) {
+				urlPath = strings.Replace(urlPath,"/"+domain,"",-1)
+			}
+			logger.Debug("urlPath:",urlPath)
+			idx := strings.LastIndex(urlPath, ":")
+			tmpPath := ""
 			if idx >-1 {//璺緞涓寘鍚湁绔彛,鍙栫鍙d互鍚庤矾寰�
 				tmpPath = urlPath[idx:]
-				imgIndx := strings.Index(imgUrl, ":")
-				if imgIndx > -1 {
-					imgUrl = imgUrl[:imgIndx-1]
-				}
+			} else {
+				tmpPath = ":6080"+tmpPath
 			}
-			fmt.Println("imgUrl+tmpPath:", imgUrl+tmpPath)
 			c.Header("Access-Control-Allow-Origin","*")
-			c.Redirect(http.StatusMovedPermanently, imgUrl+tmpPath)
+			logger.Debug("domain+tmpPath:",domain+tmpPath)
+			c.Redirect(http.StatusMovedPermanently, domain+tmpPath)
 			return
 		} else {
 			c.Next()

--
Gitblit v1.8.0