66 lines
1.5 KiB
Go
66 lines
1.5 KiB
Go
package main
|
|
|
|
import (
|
|
"log"
|
|
"os"
|
|
|
|
"github.com/pkg/errors"
|
|
"github.com/swaggo/swag"
|
|
"github.com/swaggo/swag/gen"
|
|
"github.com/urfave/cli"
|
|
)
|
|
|
|
func main() {
|
|
app := cli.NewApp()
|
|
app.Version = swag.Version
|
|
app.Usage = "Automatically generate RESTful API documentation with Swagger 2.0 for Go."
|
|
app.Commands = []cli.Command{
|
|
{
|
|
Name: "init",
|
|
Aliases: []string{"i"},
|
|
Usage: "Create docs.go",
|
|
Action: func(c *cli.Context) error {
|
|
dir := c.String("dir")
|
|
mainAPIFile := c.String("generalInfo")
|
|
swaggerConfDir := c.String("swagger")
|
|
strategy := c.String("propertyStrategy")
|
|
switch strategy {
|
|
case swag.CamelCase, swag.SnakeCase, swag.PascalCase:
|
|
default:
|
|
return errors.Errorf("not supported %s propertyStrategy", strategy)
|
|
}
|
|
|
|
gen.New().Build(dir, mainAPIFile, swaggerConfDir, strategy)
|
|
return nil
|
|
},
|
|
Flags: []cli.Flag{
|
|
cli.StringFlag{
|
|
Name: "generalInfo, g",
|
|
Value: "main.go",
|
|
Usage: "Go file path in which 'swagger general API Info' is written",
|
|
},
|
|
cli.StringFlag{
|
|
Name: "dir, d",
|
|
Value: "./",
|
|
Usage: "Directory you want to parse",
|
|
},
|
|
cli.StringFlag{
|
|
Name: "swagger, s",
|
|
Value: "./docs/swagger",
|
|
Usage: "Output the swagger conf for json and yaml",
|
|
},
|
|
cli.StringFlag{
|
|
Name: "propertyStrategy, p",
|
|
Value: "camelcase",
|
|
Usage: "Property Naming Strategy like snakecase,camelcase,pascalcase",
|
|
},
|
|
},
|
|
},
|
|
}
|
|
|
|
err := app.Run(os.Args)
|
|
if err != nil {
|
|
log.Fatal(err)
|
|
}
|
|
}
|