package httpSwagger
import (
"html/template"
"net/http"
"regexp"
swaggerFiles "github.com/swaggo/files"
"github.com/swaggo/swag"
)
// WrapHandler wraps swaggerFiles.Handler and returns http.HandlerFunc
var WrapHandler = Handler()
// Config stores httpSwagger configuration variables.
type Config struct {
//The url pointing to API definition (normally swagger.json or swagger.yaml). Default is `doc.json`.
URL string
DeepLinking bool
DocExpansion string
DomID string
}
// URL presents the url pointing to API definition (normally swagger.json or swagger.yaml).
func URL(url string) func(c *Config) {
return func(c *Config) {
c.URL = url
}
}
// DeepLinking true, false
func DeepLinking(deepLinking bool) func(c *Config) {
return func(c *Config) {
c.DeepLinking = deepLinking
}
}
// DocExpansion list, full, none
func DocExpansion(docExpansion string) func(c *Config) {
return func(c *Config) {
c.DocExpansion = docExpansion
}
}
// DomID #swagger-ui
func DomID(domID string) func(c *Config) {
return func(c *Config) {
c.DomID = domID
}
}
// Handler wraps `http.Handler` into `http.HandlerFunc`.
func Handler(configFns ...func(*Config)) http.HandlerFunc {
config := &Config{
URL: "doc.json",
DeepLinking: true,
DocExpansion: "list",
DomID: "#swagger-ui",
}
for _, configFn := range configFns {
configFn(config)
}
//create a template with name
t := template.New("swagger_index.html")
index, _ := t.Parse(indexTempl)
var re = regexp.MustCompile(`^(.*/)([^?].*)?[?|.]*$`)
return func(w http.ResponseWriter, r *http.Request) {
matches := re.FindStringSubmatch(r.RequestURI)
path := matches[2]
prefix := matches[1]
h := swaggerFiles.Handler
h.Prefix = prefix
switch path {
case "index.html":
_ = index.Execute(w, config)
case "doc.json":
w.Header().Set("Content-Type", "application/json; charset=utf-8")
doc, err := swag.ReadDoc()
if err != nil {
panic(err)
}
_, _ = w.Write([]byte(doc))
case "":
http.Redirect(w, r, prefix+"index.html", 301)
default:
h.ServeHTTP(w, r)
}
return
}
}
const indexTempl = `
Swagger UI
`