feat(vpc-gw): add --wait vpcGwGatewayNetworkCreate (#3651)

scaleway · Feb 8, 2024 · 5edcd4e · 5edcd4e
1 parent 4f269b9
commit 5edcd4e
Show file tree

Hide file tree

Showing 3 changed files with 20 additions and 0 deletions.
diff --git a/cmd/scw/testdata/test-all-usage-vpc-gw-gateway-network-create-usage.golden b/cmd/scw/testdata/test-all-usage-vpc-gw-gateway-network-create-usage.golden
@@ -18,6 +18,7 @@ ARGS:
 
 FLAGS:
   -h, --help   help for create
+  -w, --wait   wait until the gateway-network is ready
 
 GLOBAL FLAGS:
   -c, --config string    The path to the config file

diff --git a/internal/namespaces/vpcgw/v1/custom.go b/internal/namespaces/vpcgw/v1/custom.go
@@ -14,6 +14,7 @@ func GetCommands() *core.Commands {
 
 	cmds.MustFind("vpc-gw", "gateway-type", "list").Override(vpcgwGatewayTypeListBuilder)
 	cmds.MustFind("vpc-gw", "gateway", "create").Override(gatewayCreateBuilder)
+	cmds.MustFind("vpc-gw", "gateway-network", "create").Override(gatewayNetworkCreateBuilder)
 
 	return cmds
 }
diff --git a/internal/namespaces/vpcgw/v1/custom_gateway_network.go b/internal/namespaces/vpcgw/v1/custom_gateway_network.go
@@ -1,9 +1,13 @@
 package vpcgw
 
 import (
+	"context"
+
 	"github.com/fatih/color"
+	"github.com/scaleway/scaleway-cli/v2/internal/core"
 	"github.com/scaleway/scaleway-cli/v2/internal/human"
 	"github.com/scaleway/scaleway-sdk-go/api/vpcgw/v1"
+	"github.com/scaleway/scaleway-sdk-go/scw"
 )
 
 var (
@@ -17,3 +21,17 @@ var (
 		vpcgw.GatewayNetworkStatusUnknown:     &human.EnumMarshalSpec{Attribute: color.Faint},
 	}
 )
+
+func gatewayNetworkCreateBuilder(c *core.Command) *core.Command {
+	c.WaitFunc = func(ctx context.Context, argsI, respI interface{}) (interface{}, error) {
+		getResp := respI.(*vpcgw.GatewayNetwork)
+		api := vpcgw.NewAPI(core.ExtractClient(ctx))
+		return api.WaitForGatewayNetwork(&vpcgw.WaitForGatewayNetworkRequest{
+			GatewayNetworkID: getResp.ID,
+			Zone:             getResp.Zone,
+			Timeout:          scw.TimeDurationPtr(gatewayActionTimeout),
+			RetryInterval:    core.DefaultRetryInterval,
+		})
+	}
+	return c
+}