Migrate to new swagger docs generation (#18)

vendor/github.com/swaggo/swag/gen/gen.go

@@ -0,0 +1,126 @@
+package gen
+
+import (
+	"encoding/json"
+	"log"
+	"os"
+	"path"
+	"text/template"
+	"time"
+
+	"github.com/ghodss/yaml"
+	"github.com/pkg/errors"
+	"github.com/swaggo/swag"
+)
+
+// Gen presents a generate tool for swag.
+type Gen struct {
+}
+
+// New creates a new Gen.
+func New() *Gen {
+	return &Gen{}
+}
+
+// Build builds swagger json file  for gived searchDir and mainAPIFile. Returns json
+func (g *Gen) Build(searchDir, mainAPIFile, swaggerConfDir, propNamingStrategy string) error {
+	log.Println("Generate swagger docs....")
+	p := swag.New()
+	p.PropNamingStrategy = propNamingStrategy
+	p.ParseAPI(searchDir, mainAPIFile)
+	swagger := p.GetSwagger()
+
+	b, err := json.MarshalIndent(swagger, "", "    ")
+	if err != nil {
+		return err
+	}
+
+	os.MkdirAll(path.Join(searchDir, "docs"), os.ModePerm)
+	docs, err := os.Create(path.Join(searchDir, "docs", "docs.go"))
+	if err != nil {
+		return err
+	}
+	defer docs.Close()
+
+	os.Mkdir(swaggerConfDir, os.ModePerm)
+	swaggerJSON, err := os.Create(path.Join(swaggerConfDir, "swagger.json"))
+	if err != nil {
+		return err
+	}
+
+	defer swaggerJSON.Close()
+	swaggerJSON.Write(b)
+
+	swaggerYAML, err := os.Create(path.Join(swaggerConfDir, "swagger.yaml"))
+	if err != nil {
+		return err
+	}
+
+	defer swaggerYAML.Close()
+	y, err := yaml.JSONToYAML(b)
+	if err != nil {
+		return errors.Wrap(err, "cannot covert json to yaml")
+	}
+
+	swaggerYAML.Write(y)
+
+	if err := packageTemplate.Execute(docs, struct {
+		Timestamp time.Time
+		Doc       string
+	}{
+		Timestamp: time.Now(),
+		Doc:       "`" + string(b) + "`",
+	}); err != nil {
+		return err
+	}
+
+	log.Printf("create docs.go at  %+v", docs.Name())
+	return nil
+}
+
+var packageTemplate = template.Must(template.New("").Parse(`// GENERATED BY THE COMMAND ABOVE; DO NOT EDIT
+// This file was generated by swaggo/swag at
+// {{ .Timestamp }}
+
+package docs
+
+import (
+	"bytes"
+
+	"github.com/alecthomas/template"
+	"github.com/swaggo/swag"
+)
+
+var doc = {{.Doc}}
+
+type swaggerInfo struct {
+	Version     string
+	Host        string
+	BasePath    string
+	Title       string
+	Description string
+}
+
+// SwaggerInfo holds exported Swagger Info so clients can modify it
+var SwaggerInfo swaggerInfo
+
+type s struct{}
+
+func (s *s) ReadDoc() string {
+	t, err := template.New("swagger_info").Parse(doc)
+	if err != nil {
+		return doc
+	}
+
+	var tpl bytes.Buffer
+	if err := t.Execute(&tpl, SwaggerInfo); err != nil {
+		return doc
+	}
+
+	return tpl.String()
+}
+
+func init() {
+	swag.Register(swag.Name, &s{})
+}
+`))