llvy.ltd/bee
1
0
Fork 0
mirror of https://github.com/casbin/bee.git synced 2025-05-24 08:20:32 +08:00
Code Issues Packages Projects Releases Wiki Activity

feat: add tag support

Browse Source 
Signed-off-by: Товарищ программист <2962928213@qq.com>
This commit is contained in:
Товарищ программист 2021-12-01 23:45:14 +08:00
parent d459bbfc09
commit 0c1eb35c61
1 changed files with 40 additions and 38 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

78
generate/swaggergen/g_docs.go
View File

@ -291,31 +291,31 @@ func GenerateDocs(curpath string) {
for _, p := range params {
switch pp := p.(type) {
case *ast.CallExpr:
var controllerName string
//var controllerName string
if selname := pp.Fun.(*ast.SelectorExpr).Sel.String(); selname == "NSNamespace" {
s, params := analyseNewNamespace(pp)
for _, sp := range params {
switch pp := sp.(type) {
case *ast.CallExpr:
if pp.Fun.(*ast.SelectorExpr).Sel.String() == "NSInclude" {
controllerName = analyseNSInclude(s, pp)
if v, ok := controllerComments[controllerName]; ok {
rootapi.Tags = append(rootapi.Tags, swagger.Tag{
Name: strings.Trim(s, "/"),
Description: v,
})
}
_ = analyseNSInclude(s, pp)
// if v, ok := controllerComments[controllerName]; ok {
// rootapi.Tags = append(rootapi.Tags, swagger.Tag{
// Name: strings.Trim(s, "/"),
// Description: v,
// })
// }
}
}
}
} else if selname == "NSInclude" {
controllerName = analyseNSInclude("", pp)
if v, ok := controllerComments[controllerName]; ok {
rootapi.Tags = append(rootapi.Tags, swagger.Tag{
Name: controllerName, // if the NSInclude has no prefix, we use the controllername as the tag
Description: v,
})
}
_ = analyseNSInclude("", pp)
// if v, ok := controllerComments[controllerName]; ok {
// rootapi.Tags = append(rootapi.Tags, swagger.Tag{
// Name: controllerName, // if the NSInclude has no prefix, we use the controllername as the tag
// Description: v,
// })
// }
}
}
}
@ -387,32 +387,32 @@ func analyseNSInclude(baseurl string, ce *ast.CallExpr) string {
}
if apis, ok := controllerList[cname]; ok {
for rt, item := range apis {
tag := cname
//tag := cname
if baseurl != "" {
rt = baseurl + rt
tag = strings.Trim(baseurl, "/")
}
if item.Get != nil {
item.Get.Tags = []string{tag}
}
if item.Post != nil {
item.Post.Tags = []string{tag}
}
if item.Put != nil {
item.Put.Tags = []string{tag}
}
if item.Patch != nil {
item.Patch.Tags = []string{tag}
}
if item.Head != nil {
item.Head.Tags = []string{tag}
}
if item.Delete != nil {
item.Delete.Tags = []string{tag}
}
if item.Options != nil {
item.Options.Tags = []string{tag}
//tag = strings.Trim(baseurl, "/")
}
// if item.Get != nil {
// item.Get.Tags = []string{tag}
// }
// if item.Post != nil {
// item.Post.Tags = []string{tag}
// }
// if item.Put != nil {
// item.Put.Tags = []string{tag}
// }
// if item.Patch != nil {
// item.Patch.Tags = []string{tag}
// }
// if item.Head != nil {
// item.Head.Tags = []string{tag}
// }
// if item.Delete != nil {
// item.Delete.Tags = []string{tag}
// }
// if item.Options != nil {
// item.Options.Tags = []string{tag}
// }
if len(rootapi.Paths) == 0 {
rootapi.Paths = make(map[string]*swagger.Item)
}
@ -555,6 +555,8 @@ func parserComments(f *ast.FuncDecl, controllerName, pkgpath string) error {
opts.Description = strings.TrimSpace(t[len("@Description"):])
} else if strings.HasPrefix(t, "@Summary") {
opts.Summary = strings.TrimSpace(t[len("@Summary"):])
} else if strings.HasPrefix(t, "@Tag") {
opts.Tags = []string{strings.TrimSpace(t[len("@Tag"):])}
} else if strings.HasPrefix(t, "@Success") {
ss := strings.TrimSpace(t[len("@Success"):])
rs := swagger.Response{}