|
|
|
@ -35,12 +35,25 @@ func (u *Kit) Generate(plugin *protogen.Plugin) error { |
|
|
|
|
return nil |
|
|
|
|
} |
|
|
|
|
u.addImports("context") |
|
|
|
|
u.addImports("git.diulo.com/mogfee/kit/middleware/jwt") |
|
|
|
|
u.addImports("git.diulo.com/mogfee/kit/transport/http") |
|
|
|
|
for _, f := range plugin.Files { |
|
|
|
|
if len(f.Services) == 0 { |
|
|
|
|
continue |
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
for _, s := range f.Services { |
|
|
|
|
addAuthKey := false |
|
|
|
|
for _, m := range s.Methods { |
|
|
|
|
autkKey := protogen2.GetAuthKey(m) |
|
|
|
|
if autkKey != "" { |
|
|
|
|
addAuthKey = true |
|
|
|
|
} |
|
|
|
|
} |
|
|
|
|
if addAuthKey { |
|
|
|
|
u.addImports("git.diulo.com/mogfee/kit/middleware/jwt") |
|
|
|
|
} |
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
fname := f.GeneratedFilenamePrefix + "_http.pb.go" |
|
|
|
|
t := plugin.NewGeneratedFile(fname, f.GoImportPath) |
|
|
|
|
t.P("package ", u.packageName(f)) |
|
|
|
|