forked from cloudfoundry/cli
-
Notifications
You must be signed in to change notification settings - Fork 0
/
add_network_policy_command.go
97 lines (80 loc) · 3.69 KB
/
add_network_policy_command.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
84
85
86
87
88
89
90
91
92
93
94
95
96
97
package v3
import (
"net/http"
"code.cloudfoundry.org/cli/actor/cfnetworkingaction"
"code.cloudfoundry.org/cli/actor/sharedaction"
"code.cloudfoundry.org/cli/actor/v3action"
"code.cloudfoundry.org/cli/api/cloudcontroller/ccerror"
"code.cloudfoundry.org/cli/command"
"code.cloudfoundry.org/cli/command/flag"
"code.cloudfoundry.org/cli/command/translatableerror"
"code.cloudfoundry.org/cli/command/v3/shared"
)
//go:generate counterfeiter . AddNetworkPolicyActor
type AddNetworkPolicyActor interface {
AddNetworkPolicy(spaceGUID string, srcAppName string, destAppName string, protocol string, startPort int, endPort int) (cfnetworkingaction.Warnings, error)
}
type AddNetworkPolicyCommand struct {
RequiredArgs flag.AddNetworkPolicyArgs `positional-args:"yes"`
DestinationApp string `long:"destination-app" required:"true" description:"Name of app to connect to"`
Port flag.NetworkPort `long:"port" description:"Port or range of ports for connection to destination app (Default: 8080)"`
Protocol flag.NetworkProtocol `long:"protocol" description:"Protocol to connect apps with (Default: tcp)"`
usage interface{} `usage:"CF_NAME add-network-policy SOURCE_APP --destination-app DESTINATION_APP [(--protocol (tcp | udp) --port RANGE)]\n\nEXAMPLES:\n CF_NAME add-network-policy frontend --destination-app backend --protocol tcp --port 8081\n CF_NAME add-network-policy frontend --destination-app backend --protocol tcp --port 8080-8090"`
relatedCommands interface{} `related_commands:"apps, network-policies"`
UI command.UI
Config command.Config
SharedActor command.SharedActor
Actor AddNetworkPolicyActor
}
func (cmd *AddNetworkPolicyCommand) Setup(config command.Config, ui command.UI) error {
cmd.UI = ui
cmd.Config = config
cmd.SharedActor = sharedaction.NewActor(config)
client, uaa, err := shared.NewClients(config, ui, true)
if err != nil {
if v3Err, ok := err.(ccerror.V3UnexpectedResponseError); ok && v3Err.ResponseCode == http.StatusNotFound {
return translatableerror.CFNetworkingEndpointNotFoundError{}
}
return err
}
v3Actor := v3action.NewActor(client, config, nil, nil)
networkingClient, err := shared.NewNetworkingClient(client.NetworkPolicyV1(), config, uaa, ui)
if err != nil {
return err
}
cmd.Actor = cfnetworkingaction.NewActor(networkingClient, v3Actor)
return nil
}
func (cmd AddNetworkPolicyCommand) Execute(args []string) error {
switch {
case cmd.Protocol.Protocol != "" && cmd.Port.StartPort == 0 && cmd.Port.EndPort == 0:
return translatableerror.NetworkPolicyProtocolOrPortNotProvidedError{}
case cmd.Protocol.Protocol == "" && (cmd.Port.StartPort != 0 || cmd.Port.EndPort != 0):
return translatableerror.NetworkPolicyProtocolOrPortNotProvidedError{}
case cmd.Protocol.Protocol == "" && cmd.Port.StartPort == 0 && cmd.Port.EndPort == 0:
cmd.Protocol.Protocol = "tcp"
cmd.Port.StartPort = 8080
cmd.Port.EndPort = 8080
}
err := cmd.SharedActor.CheckTarget(true, true)
if err != nil {
return err
}
user, err := cmd.Config.CurrentUser()
if err != nil {
return err
}
cmd.UI.DisplayTextWithFlavor("Adding network policy to app {{.SrcAppName}} in org {{.Org}} / space {{.Space}} as {{.User}}...", map[string]interface{}{
"SrcAppName": cmd.RequiredArgs.SourceApp,
"Org": cmd.Config.TargetedOrganization().Name,
"Space": cmd.Config.TargetedSpace().Name,
"User": user.Name,
})
warnings, err := cmd.Actor.AddNetworkPolicy(cmd.Config.TargetedSpace().GUID, cmd.RequiredArgs.SourceApp, cmd.DestinationApp, cmd.Protocol.Protocol, cmd.Port.StartPort, cmd.Port.EndPort)
cmd.UI.DisplayWarnings(warnings)
if err != nil {
return err
}
cmd.UI.DisplayOK()
return nil
}