56 lines
1.1 KiB
Go
56 lines
1.1 KiB
Go
|
package echoswagger
|
||
|
|
|
||
|
|
import (
|
||
|
|
"bytes"
|
||
|
|
"fmt"
|
||
|
|
"html/template"
|
||
|
|
"net/http"
|
||
|
|
|
||
|
|
"github.com/labstack/echo"
|
||
|
|
)
|
||
|
|
|
||
|
|
const SwaggerVersion = "2.0"
|
||
|
|
|
||
|
|
func (r *Root) Spec(c echo.Context) error {
|
||
|
|
err := r.genSpec(c)
|
||
|
|
if err != nil {
|
||
|
|
return c.String(http.StatusInternalServerError, err.Error())
|
||
|
|
}
|
||
|
|
return c.JSON(http.StatusOK, r.spec)
|
||
|
|
}
|
||
|
|
|
||
|
|
func (r *Root) genSpec(c echo.Context) error {
|
||
|
|
r.spec.Swagger = SwaggerVersion
|
||
|
|
r.spec.Paths = make(map[string]interface{})
|
||
|
|
r.spec.Host = c.Request().Host
|
||
|
|
r.spec.Schemes = []string{c.Scheme()}
|
||
|
|
|
||
|
|
for i := range r.groups {
|
||
|
|
group := &r.groups[i]
|
||
|
|
r.spec.Tags = append(r.spec.Tags, &group.tag)
|
||
|
|
for j := range group.apis {
|
||
|
|
a := &group.apis[j]
|
||
|
|
fmt.Println("Group:", a)
|
||
|
|
// TODO
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
return nil
|
||
|
|
}
|
||
|
|
|
||
|
|
func (r *Root) docHandler(swaggerPath string) echo.HandlerFunc {
|
||
|
|
t, err := template.New("swagger").Parse(SwaggerUIContent)
|
||
|
|
if err != nil {
|
||
|
|
panic(err)
|
||
|
|
}
|
||
|
|
|
||
|
|
return func(c echo.Context) error {
|
||
|
|
buf := new(bytes.Buffer)
|
||
|
|
t.Execute(buf, map[string]interface{}{
|
||
|
|
"title": r.spec.Info.Title,
|
||
|
|
"url": c.Scheme() + "://" + c.Request().Host + swaggerPath,
|
||
|
|
})
|
||
|
|
return c.HTMLBlob(http.StatusOK, buf.Bytes())
|
||
|
|
}
|
||
|
|
}
|