forked from swaggo/swag
/
gen.go
68 lines (53 loc) · 1.23 KB
/
gen.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
package gen
import (
"encoding/json"
"log"
"os"
"path"
"text/template"
"time"
"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.
func (g *Gen) Build(searchDir, mainApiFile string) error {
log.Println("Generate swagger docs....")
p := swag.New()
p.ParseApi(searchDir, mainApiFile)
swagger := p.GetSwagger()
b, _ := json.MarshalIndent(swagger, "", " ")
os.MkdirAll(path.Join(searchDir, "docs"), os.ModePerm)
docs, _ := os.Create(path.Join(searchDir, "docs", "docs.go"))
defer docs.Close()
packageTemplate.Execute(docs, struct {
Timestamp time.Time
Doc string
}{
Timestamp: time.Now(),
Doc: "`" + string(b) + "`",
})
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 (
"github.com/swaggo/swag"
)
var doc = {{.Doc}}
type s struct{}
func (s *s) ReadDoc() string {
return doc
}
func init() {
swag.Register(swag.Name, &s{})
}
`))