workflow-cli/internal/parser/timeouts.go at 0c86ccbd9cb5c98cb9ed4023227a227bc13b4485 · drycc/workflow-cli · GitHub

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
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
package parser

import (
	"github.com/drycc/workflow-cli/internal/commands"
	"github.com/drycc/workflow-cli/internal/completion"
	"github.com/drycc/workflow-cli/internal/template"
	"github.com/drycc/workflow-cli/pkg/i18n"
	"github.com/spf13/cobra"
)

// NewTimeoutsCommand creates the timeouts command
func NewTimeoutsCommand(cmdr *commands.DryccCmd) *cobra.Command {
	cmd := &cobra.Command{
		Use:   "timeouts",
		Short: i18n.T("Manage pods termination grace period"),
		RunE: func(_ *cobra.Command, _ []string) error {
			return cmdr.TimeoutsList(app, version)
		},
	}

	cmd.PersistentFlags().StringVarP(&app, "app", "a", "", i18n.T("The uniquely identifiable name for the application"))
	cmd.Flags().IntVarP(&version, "version", "v", 0, i18n.T("The version for which the timeout needs to be listed"))

	appCompletion := completion.AppCompletion{ArgsLen: -1, ConfigFile: &cmdr.ConfigFile}
	cmd.RegisterFlagCompletionFunc("app", appCompletion.CompletionFunc)
	releaseCompletion := completion.ReleaseCompletion{ConfigFile: &cmdr.ConfigFile, AppID: &app}
	cmd.RegisterFlagCompletionFunc("version", releaseCompletion.CompletionFunc)

	cmd.AddCommand(timeoutListCommand(cmdr))
	cmd.AddCommand(timeoutSetCommand(cmdr))
	cmd.AddCommand(timeoutUnsetCommand(cmdr))
	return cmd
}

func timeoutListCommand(cmdr *commands.DryccCmd) *cobra.Command {
	cmd := &cobra.Command{
		Use:   "list",
		Short: i18n.T("List resource timeouts for an app"),
		RunE: func(_ *cobra.Command, _ []string) error {
			return cmdr.TimeoutsList(app, version)
		},
	}

	cmd.Flags().IntVarP(&version, "version", "v", 0, i18n.T("The version for which the timeout needs to be listed"))

	releaseCompletion := completion.ReleaseCompletion{ConfigFile: &cmdr.ConfigFile, AppID: &app}
	cmd.RegisterFlagCompletionFunc("version", releaseCompletion.CompletionFunc)
	return cmd
}

func timeoutSetCommand(cmdr *commands.DryccCmd) *cobra.Command {
	ptsSetArgsCompletion := completion.PtsSetArgsCompletion{
		PtsCompletion: &completion.PtsCompletion{AppID: &app, ArgsLen: -1, ConfigFile: &cmdr.ConfigFile},
	}
	cmd := &cobra.Command{
		Use: "set <ptype>=<value>...",
		Example: template.CustomExample(
			"drycc timeouts set web=30 worker=60",
			map[string]string{
				"<ptype>": i18n.T("The process type as defined in your Procfile"),
				"<value>": i18n.T("The value to apply to the process type in seconds"),
			},
		),
		Short:             i18n.T("Set resource timeouts for an app"),
		Long:              i18n.T("Sets termination grace period for an application"),
		Args:              cobra.MinimumNArgs(1),
		ValidArgsFunction: ptsSetArgsCompletion.CompletionFunc,
		RunE: func(_ *cobra.Command, args []string) error {
			timeouts := args
			return cmdr.TimeoutsSet(app, timeouts)
		},
	}

	return cmd
}

func timeoutUnsetCommand(cmdr *commands.DryccCmd) *cobra.Command {
	ptsArgsCompletion := completion.PtsArgsCompletion{
		PtsCompletion: &completion.PtsCompletion{AppID: &app, ArgsLen: -1, ConfigFile: &cmdr.ConfigFile},
	}
	cmd := &cobra.Command{
		Use: "unset <ptype>...",
		Example: template.CustomExample(
			"drycc timeouts unset web worker",
			map[string]string{
				"<ptype>": i18n.T("The process type as defined in your Procfile"),
			},
		),
		Short:             i18n.T("Unset resource timeouts for an app"),
		Long:              i18n.T("Unsets timeouts for an application. Default value (30s) or set by drycc controller"),
		Args:              cobra.MinimumNArgs(1),
		ValidArgsFunction: ptsArgsCompletion.CompletionFunc,
		RunE: func(_ *cobra.Command, args []string) error {
			ptypes := args
			return cmdr.TimeoutsUnset(app, ptypes)
		},
	}

	return cmd
}