package main
import (
"bytes"
"html/template"
"net/http"
"github.com/zeromicro/go-zero/rest/httpx"
)
type Opts func(*swaggerConfig)
// SwaggerOpts configures the Doc gmiddlewares.
type swaggerConfig struct {
// SpecURL the url to find the spec for
SpecURL string
// SwaggerHost for the js that generates the swagger ui site, defaults to: http://petstore.swagger.io/
SwaggerHost string
}
func Doc(basePath, env string, opts ...Opts) http.HandlerFunc {
config := &swaggerConfig{
SpecURL: basePath + "-json",
SwaggerHost: "https://petstore.swagger.io",
}
for _, opt := range opts {
opt(config)
}
// swagger html
tmpl := template.Must(template.New("swaggerdoc").Parse(swaggerTemplateV2))
buf := bytes.NewBuffer(nil)
err := tmpl.Execute(buf, config)
uiHTML := buf.Bytes()
// permission
needPermission := false
if env == "prod" {
needPermission = true
}
return func(rw http.ResponseWriter, r *http.Request) {
if err != nil {
httpx.Error(rw, err)
return
}
if r.URL.Path == basePath {
if needPermission {
rw.WriteHeader(http.StatusOK)
rw.Header().Set("Content-Type", "text/plain")
_, err = rw.Write([]byte("Swagger not open on prod"))
if err != nil {
httpx.Error(rw, err)
}
return
}
rw.Header().Set("Content-Type", "text/html; charset=utf-8")
_, err = rw.Write(uiHTML)
if err != nil {
httpx.Error(rw, err)
return
}
rw.WriteHeader(http.StatusOK)
return
}
}
}
const swaggerTemplateV2 = `
API documentation
`