From cfeba2dea02c1de54cde6c95f2a3f46c121d34aa Mon Sep 17 00:00:00 2001
From: liuxiaolong <736321739@qq.com>
Date: 星期五, 16 八月 2019 15:16:14 +0800
Subject: [PATCH] dev add publicDomain
---
middlewares/auth/auth.go | 73 ++++++++++++++++++++++++++++++++++++
1 files changed, 73 insertions(+), 0 deletions(-)
diff --git a/middlewares/auth/auth.go b/middlewares/auth/auth.go
index 8832b06..831ba8a 100644
--- a/middlewares/auth/auth.go
+++ b/middlewares/auth/auth.go
@@ -1 +1,74 @@
package auth
+
+import (
+ "github.com/gin-gonic/gin"
+ "net/http"
+ "regexp"
+ "strings"
+ "webserver/extend/config"
+ "webserver/extend/logger"
+)
+
+const (
+ TokenKey = "abc123456789"
+)
+
+type Auth interface {
+ Check(c *gin.Context)bool
+ User(c *gin.Context)interface{}
+ Login(http *http.Request,w http.ResponseWriter,user map[string]interface{})interface{}
+ Logout(http *http.Request,w http.ResponseWriter) bool
+}
+
+func GenerateAuthDriver() *Auth {
+ var authDriver Auth
+ authDriver = NewJwtAuthDriver()
+ return &authDriver
+}
+
+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()
+ //}
+ c.Next()
+ } else if strings.HasPrefix(urlPath,"/httpImage") {
+ host := c.Request.Host
+ logger.Debug("host:", host)
+
+ imgUrl := config.Server.ImageUrl
+
+ domain := config.Server.PublicDomain
+ domainReg := regexp.MustCompile(``+domain+``)
+ if domainReg.MatchString(host) {//鍩熷悕璁块棶
+ imgUrl = domain
+ }
+
+ urlPath = strings.Replace(urlPath, "/httpImage", "", -1)
+ idx := strings.Index(urlPath, ":")
+ tmpPath := urlPath
+ if idx >-1 {//璺緞涓寘鍚湁绔彛,鍙栫鍙ｄ互鍚庤矾寰�
+ tmpPath = urlPath[idx:]
+ imgIndx := strings.LastIndex(imgUrl, ":")
+ if imgIndx > -1 {
+ imgUrl = imgUrl[:imgIndx]
+ }
+ }
+ logger.Debug("imgUrl+tmpPath:", imgUrl+tmpPath)
+ c.Header("Access-Control-Allow-Origin","*")
+ c.Redirect(http.StatusMovedPermanently, imgUrl+tmpPath)
+ return
+ } else {
+ c.Next()
+ }
+ }
+}
+
+func GetCurUser(c *gin.Context)map[string]interface{}{
+ return (*GenerateAuthDriver()).User(c).(map[string]interface{})
+}
\ No newline at end of file
--
Gitblit v1.8.0