inject.go

package ginx

import (
	"github.com/gin-gonic/gin"
	"github.com/swaggo/swag"

	"github.com/xiaoshicae/ginx/options"
	"github.com/xiaoshicae/ginx/swagger"
	"github.com/xiaoshicae/xone"
)

func injectSwaggerInfo(swaggerInfo *swag.Spec, engine *gin.Engine, opts ...options.SwaggerOption) {
	if swaggerInfo == nil || engine == nil {
		return
	}

	engine.FuncMap[xone.SwaggerInfoFuncKey] = func() *swag.Spec { return swaggerInfo }

	dso := options.DefaultSwaggerOptions()
	for _, opt := range opts {
		opt(dso)
	}

	swaggerUrl := swagger.SwaggerUrl
	if dso.UrlPrefix != "" {
		swaggerUrl = dso.UrlPrefix + swaggerUrl
	}

	engine.GET(swaggerUrl, swagger.SwaggerHandler)
}

func injectPrintBanner(engine *gin.Engine) {
	engine.FuncMap[xone.PrintBannerFuncKey] = PrintBanner
}