-
Notifications
You must be signed in to change notification settings - Fork 0
/
main.go
executable file
·83 lines (70 loc) · 2.04 KB
/
main.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
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
package main
import (
"context"
"github.com/pkg/errors"
"github.com/spf13/cobra"
"github.com/izumin5210/grapi/pkg/gencmd"
"github.com/izumin5210/grapi/pkg/svcgen"
)
func main() {
buildCommand(svcgen.NewApp).MustExecute()
}
func buildCommand(createAppFunc svcgen.CreateAppFunc, opts ...gencmd.Option) gencmd.Executor {
return gencmd.New(
"service",
newGenerateCommand(createAppFunc),
newDestroyCommand(createAppFunc),
opts...,
)
}
func newGenerateCommand(createApp svcgen.CreateAppFunc) *gencmd.Command {
var (
skipTest bool
resName string
app *svcgen.App
)
cmd := &gencmd.Command{
Use: "generate NAME [flags] [METHODS...]",
Short: "Generate a new service",
Args: cobra.MinimumNArgs(1),
ShouldInsideApp: true,
PreRun: func(c *gencmd.Command, args []string) error {
var err error
app, err = createApp(c)
return errors.WithStack(err)
},
BuildParams: func(c *gencmd.Command, args []string) (interface{}, error) {
svcName := args[0]
methods := args[1:]
params, err := app.ParamsBuilder.Build(svcName, resName, methods)
return params, errors.WithStack(err)
},
PostRun: func(c *gencmd.Command, args []string) error {
return errors.WithStack(app.ProtocWrapper.Exec(context.TODO()))
},
}
cmd.Flags().BoolVarP(&skipTest, "skip-test", "T", false, "Skip test files")
cmd.Flags().StringVar(&resName, "resource-name", "", "ResourceName to be used")
return cmd
}
func newDestroyCommand(createApp svcgen.CreateAppFunc) *gencmd.Command {
var (
app *svcgen.App
)
cmd := &gencmd.Command{
Use: "destroy NAME",
Short: "Destroy an existing service",
Args: cobra.MinimumNArgs(1),
ShouldInsideApp: true,
PreRun: func(c *gencmd.Command, args []string) error {
var err error
app, err = createApp(c)
return errors.WithStack(err)
},
BuildParams: func(c *gencmd.Command, args []string) (interface{}, error) {
params, err := app.ParamsBuilder.Build(args[0], "", nil)
return params, errors.WithStack(err)
},
}
return cmd
}