整合ydoc-lucky-ui主题, 代码待整理

swagger_ui.go

@@ -8,14 +8,21 @@
 package api_doc
 
 import (
+	"embed"
 	"git.zhangdeman.cn/gateway/api-doc/define"
+	"git.zhangdeman.cn/zhangdeman/consts"
 	"github.com/gin-gonic/gin"
 	knife4goFiles "github.com/go-webtools/knife4go"
 	knife4goGin "github.com/go-webtools/knife4go/gin"
 	swaggerFiles "github.com/swaggo/files"
 	ginSwagger "github.com/swaggo/gin-swagger"
+	"net/http"
+	"strings"
 )
 
+//go:embed ydoc-lucky-ui/*
+var ydocUIFiles embed.FS
+
 // NewSwaggerUI ...
 //
 // Author : go_developer@163.com<白茶清欢>
@@ -31,6 +38,8 @@ func NewSwaggerUI(info *define.Info, servers []*define.ServerItem, uiTheme strin
 type SwaggerUI struct {
 	docInstance *Generate // 文档实例
 	uiTheme     string    // 文档主题, swaggerUI / knife4go, 默认 knife4go
+	router      *gin.Engine
+	baseUri     string
 }
 
 // DocInstance 文档实例
@@ -42,6 +51,81 @@ func (su *SwaggerUI) DocInstance() *Generate {
 	return su.docInstance
 }
 
+// RegisterHandler ...
+//
+// Author : go_developer@163.com<白茶清欢>
+//
+// Date : 15:00 2025/2/16
+func (su *SwaggerUI) RegisterHandler(router *gin.Engine, baseUri string) {
+	su.router = router
+	baseUri = strings.TrimRight(baseUri, "/")
+	if len(baseUri) == 0 {
+		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, 通过此接口读取文档数据
+			ctx.JSON(http.StatusOK, su.docInstance.Doc())
+			ctx.Abort()
+		}
+		if ctx.Request.RequestURI == "/doc/swagger/openapi.json" {
+			// knife4go 文档通过此接口读取文档列表
+			ctx.JSON(http.StatusOK, []map[string]any{
+				{
+					"name":           "服务文档",
+					"url":            "doc.json",
+					"swaggerVersion": consts.SwaggerDocVersion3,
+				},
+			})
+			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", "*") // 允许访问所有域
+		ctx.JSON(http.StatusOK, []map[string]any{
+			{
+				"name":           "服务文档",
+				"url":            baseUri + "/doc.json",
+				"swaggerVersion": consts.SwaggerDocVersion3,
+			},
+		})
+		// ctx.JSON(http.StatusOK, swaggerInstance.docInstance.Data())
+	})
+}
+
 // Handler 访问文档的接口处理
 //
 // Author : go_developer@163.com<白茶清欢>
@@ -52,13 +136,25 @@ func (su *SwaggerUI) Handler() func(ctx *gin.Context) {
 	case define.SwaggerUIThemeKnife4go:
 		return su.HandleKnife4goUI()
 	case define.SwaggerUIThemeYDocLucky:
-		// TODO : YDoc-Lucky-UI 主题处理
-		return su.HandleSwaggerUI()
+		// YDoc-Lucky-UI 主题处理
+		return su.HandleLuckyUI()
 	default:
 		return su.HandleSwaggerUI()
 	}
 }
 
+// HandleLuckyUI ...
+//
+// Author : go_developer@163.com<白茶清欢>
+//
+// Date : 16:04 2025/2/16
+func (su *SwaggerUI) HandleLuckyUI() func(ctx *gin.Context) {
+	// su.router.StaticFS(su.baseUri+"/assets", http.FS(ydocUIFiles))
+	return func(ctx *gin.Context) {
+
+	}
+}
+
 // HandleKnife4goUI ...
 //
 // Author : go_developer@163.com<白茶清欢>