workflow-cli/parser/whitelist.go at 03004b6ee31be5561ac630922fe8eef10e8b6ea8 · 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
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
package parser

import (
	docopt "github.com/docopt/docopt-go"
	"github.com/drycc/workflow-cli/cmd"
)

// Allowlist displays all relevant commands for `drycc allowlist`.
func Allowlist(argv []string, cmdr cmd.Commander) error {
	usage := `
Valid commands for allowlist:

allowlist:add           adds addresses to the application's allowlist
allowlist:list          list addresses in the application's allowlist
allowlist:remove        remove addresses from the application's allowlist

Use 'drycc help [command]' to learn more.
`

	switch argv[0] {
	case "allowlist:add":
		return allowlistAdd(argv, cmdr)
	case "allowlist:list":
		return allowlistList(argv, cmdr)
	case "allowlist:remove":
		return allowlistRemove(argv, cmdr)
	default:
		if printHelp(argv, usage) {
			return nil
		}

		if argv[0] == "allowlist" {
			argv[0] = "allowlist:list"
			return allowlistList(argv, cmdr)
		}

		PrintUsage(cmdr)
		return nil
	}
}

func allowlistAdd(argv []string, cmdr cmd.Commander) error {
	usage := `
Adds addresses to an application allowlist.

Usage: drycc allowlist:add <addresses> [options]

Arguments:
  <addresses>
    comma-delimited list of addresses(using IP or CIDR notation) to be allowlisted for the application, such as '1.2.3.4' or '1.2.3.4,0.0.0.0/0'.

Options:
  -a --app=<app>
    the uniquely identifiable name for the application.
`

	args, err := docopt.Parse(usage, argv, true, "", false, true)

	if err != nil {
		return err
	}

	app := safeGetValue(args, "--app")
	addresses := safeGetValue(args, "<addresses>")

	return cmdr.AllowlistAdd(app, addresses)
}

func allowlistList(argv []string, cmdr cmd.Commander) error {
	usage := `
Lists allowlisted addresses for an application.

Usage: drycc allowlist:list [options]

Options:
  -a --app=<app>
    the uniquely identifiable name for the application.
`

	args, err := docopt.Parse(usage, argv, true, "", false, true)

	if err != nil {
		return err
	}

	app := safeGetValue(args, "--app")

	return cmdr.AllowlistList(app)
}

func allowlistRemove(argv []string, cmdr cmd.Commander) error {
	usage := `
Removes addresses from an application allowlist.

Usage: drycc allowlist:remove <addresses> [options]

Arguments:
  <addresses>
    comma-delimited list of addresses(using IP or CIDR notation) to be allowlisted for the application, such as '1.2.3.4' or "1.2.3.4,0.0.0.0/0".

Options:
  -a --app=<app>
    the uniquely identifiable name for the application.
`

	args, err := docopt.Parse(usage, argv, true, "", false, true)

	if err != nil {
		return err
	}

	app := safeGetValue(args, "--app")
	addresses := safeGetValue(args, "<addresses>")

	return cmdr.AllowlistRemove(app, addresses)
}