From 4ef680d917e7fcf09e66bf1306133b3e66355d76 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=E7=99=BD=E8=8C=B6=E6=B8=85=E6=AC=A2?=
 <go_developer@163.com>
Date: Sun, 16 Feb 2025 17:22:18 +0800
Subject: [PATCH] =?UTF-8?q?=E4=BC=98=E5=8C=96=E4=BB=A3=E7=A0=81=E5=A4=84?=
 =?UTF-8?q?=E7=90=86?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 swagger_ui.go | 48 +++++++++++++++---------------------------------
 1 file changed, 15 insertions(+), 33 deletions(-)

diff --git a/swagger_ui.go b/swagger_ui.go
index 4f51f8f..dc00fa7 100644
--- a/swagger_ui.go
+++ b/swagger_ui.go
@@ -63,21 +63,6 @@ func (su *SwaggerUI) RegisterHandler(router *gin.Engine, baseUri string) {
 		baseUri = "/docs/swagger"
 	}
 	su.baseUri = baseUri
-	router.GET("/assets/*any", func(ctx *gin.Context) {
-		if su.uiTheme == define.SwaggerUIThemeYDocLucky {
-			fileRealPath := strings.TrimPrefix(ctx.Request.RequestURI, baseUri)
-			byteData, _ := ydocUIFiles.ReadFile(define.SwaggerUIThemeYDocLucky + fileRealPath)
-			uriArr := strings.Split(ctx.Request.RequestURI, ".")
-			contentType := "text/" + uriArr[len(uriArr)-1]
-			if strings.HasSuffix(ctx.Request.RequestURI, "png") {
-				contentType = "image/png"
-			}
-			ctx.Header("Content-Type", contentType)
-			ctx.String(http.StatusOK, string(byteData))
-			ctx.Abort()
-		}
-		ctx.AbortWithStatus(http.StatusNotFound)
-	})
 	router.GET(baseUri+"/*any", func(ctx *gin.Context) {
 		if ctx.Request.RequestURI == baseUri+"/doc.json" {
 			// 默认swagger, 通过此接口读取文档数据
@@ -95,23 +80,6 @@ func (su *SwaggerUI) RegisterHandler(router *gin.Engine, baseUri string) {
 			})
 			ctx.Abort()
 		}
-		if su.uiTheme == define.SwaggerUIThemeYDocLucky {
-			fileRealPath := strings.TrimPrefix(ctx.Request.RequestURI, baseUri)
-			byteData, _ := ydocUIFiles.ReadFile(define.SwaggerUIThemeYDocLucky + fileRealPath)
-			if strings.HasSuffix(ctx.Request.RequestURI, "html") {
-				byteData = []byte(strings.ReplaceAll(string(byteData), "{{BASE_URI}}", baseUri))
-			}
-			uriArr := strings.Split(ctx.Request.RequestURI, ".")
-			contentType := "text/" + uriArr[len(uriArr)-1]
-			if strings.HasSuffix(ctx.Request.RequestURI, "png") {
-				contentType = "image/png"
-			} else if strings.HasSuffix(ctx.Request.RequestURI, "js") {
-				contentType = "application/javascript"
-			}
-			ctx.Header("Content-Type", contentType)
-			ctx.String(http.StatusOK, string(byteData))
-			ctx.Abort()
-		}
 	}, su.Handler())
 	router.GET("/swagger-resources", func(ctx *gin.Context) { // lucky UI获取分组信息
 		ctx.Writer.Header().Set("Access-Control-Allow-Origin", "*") // 允许访问所有域
@@ -151,7 +119,21 @@ func (su *SwaggerUI) Handler() func(ctx *gin.Context) {
 func (su *SwaggerUI) HandleLuckyUI() func(ctx *gin.Context) {
 	// su.router.StaticFS(su.baseUri+"/assets", http.FS(ydocUIFiles))
 	return func(ctx *gin.Context) {
-
+		fileRealPath := strings.TrimPrefix(ctx.Request.RequestURI, su.baseUri)
+		byteData, _ := ydocUIFiles.ReadFile(define.SwaggerUIThemeYDocLucky + fileRealPath)
+		if strings.HasSuffix(ctx.Request.RequestURI, "html") {
+			byteData = []byte(strings.ReplaceAll(string(byteData), "{{BASE_URI}}", su.baseUri))
+		}
+		uriArr := strings.Split(ctx.Request.RequestURI, ".")
+		contentType := "text/" + uriArr[len(uriArr)-1]
+		if strings.HasSuffix(ctx.Request.RequestURI, "png") {
+			contentType = "image/png"
+		} else if strings.HasSuffix(ctx.Request.RequestURI, "js") {
+			contentType = "application/javascript"
+		}
+		ctx.Header("Content-Type", contentType)
+		ctx.String(http.StatusOK, string(byteData))
+		ctx.Abort()
 	}
 }