84 lines
1.9 KiB
Go
84 lines
1.9 KiB
Go
|
package cmd // import "udico.de/uditaren/opier/cmd"
|
||
|
|
import (
|
||
|
|
log "github.com/sirupsen/logrus"
|
||
|
|
"github.com/spf13/cobra"
|
||
|
|
"io/ioutil"
|
||
|
|
"os"
|
||
|
|
"path"
|
||
|
|
"strings"
|
||
|
|
"udico.de/opier/encode"
|
||
|
|
"udico.de/opier/openapi"
|
||
|
|
)
|
||
|
|
|
||
|
|
var cGenerate = &cobra.Command{
|
||
|
|
Use: "generate <input>",
|
||
|
|
Short: "Generate Go code from a given input file",
|
||
|
|
Long: `Generate Go code from a given input file, where <input> can either be a local
|
||
|
|
or a remote file (when starting with http(s)://)`,
|
||
|
|
|
||
|
|
Args: cobra.ExactArgs(1),
|
||
|
|
}
|
||
|
|
|
||
|
|
var packageName = "api"
|
||
|
|
|
||
|
|
func init() {
|
||
|
|
cGenerate.RunE = executeGenerate
|
||
|
|
cGenerate.PersistentFlags().StringVarP(&packageName, "package", "p","api", "The name of the generated package")
|
||
|
|
|
||
|
|
cRoot.AddCommand(cGenerate)
|
||
|
|
}
|
||
|
|
|
||
|
|
func executeGenerate(cmd *cobra.Command, args []string) error {
|
||
|
|
log.WithField("input", args[0]).Debugf("Generate Go code")
|
||
|
|
|
||
|
|
api, err := openapi.Load(args[0])
|
||
|
|
if err != nil {
|
||
|
|
log.WithError(err).Fatal("cannot load input data")
|
||
|
|
}
|
||
|
|
|
||
|
|
log.Infof("API: %v, %v", api.Info.Title, api.Info.Version)
|
||
|
|
log.Info("Paths:")
|
||
|
|
for k, _ := range api.Paths {
|
||
|
|
log.Infof(" - %v", k)
|
||
|
|
for m, n := range api.Paths[k].Operations() {
|
||
|
|
log.Infof( " [%v] %v: %v", m, n.OperationId, n.Summary)
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
os.Mkdir(packageName, 0755)
|
||
|
|
os.Mkdir(path.Join(packageName, "schema"), 0755)
|
||
|
|
tEnc := encode.NewEncoder(packageName)
|
||
|
|
|
||
|
|
log.Info("Types:")
|
||
|
|
for k, v := range api.Components.Schemas {
|
||
|
|
log.Infof("File: %v.go", strings.ToLower(k))
|
||
|
|
tData := ""
|
||
|
|
switch v.Type {
|
||
|
|
case "array":
|
||
|
|
//log.Infof(" [%v]", v.Items.TypeOrReference())
|
||
|
|
tData = tEnc.Array(k, v)
|
||
|
|
|
||
|
|
case "object", "":
|
||
|
|
tData = tEnc.Object(k, v)
|
||
|
|
|
||
|
|
default:
|
||
|
|
if len(v.Enum) > 0 {
|
||
|
|
tData = tEnc.Enum(k, v)
|
||
|
|
} else {
|
||
|
|
log.Warnf("UNKNOWN TYPE!!! %v , %v", k, v.Type)
|
||
|
|
}
|
||
|
|
}
|
||
|
|
if tData == "" {
|
||
|
|
log.Warnf("no data in %v", k)
|
||
|
|
continue
|
||
|
|
}
|
||
|
|
err = ioutil.WriteFile(path.Join(packageName, k + ".go"), []byte(tData[:]) ,0644)
|
||
|
|
if err != nil {
|
||
|
|
log.WithError(err).Fatal("cannot create file")
|
||
|
|
}
|
||
|
|
|
||
|
|
}
|
||
|
|
|
||
|
|
return nil
|
||
|
|
}
|