From 4bb53c6e9bf798dd7d403d9b1618d0aed960e019 Mon Sep 17 00:00:00 2001
From: dev <4052466+mattvr@users.noreply.github.com>
Date: Tue, 6 Aug 2024 18:10:39 -0700
Subject: [PATCH] add network flag to wireguard create

---
 internal/command/wireguard/root.go      | 7 +++++++
 internal/command/wireguard/wireguard.go | 3 +--
 2 files changed, 8 insertions(+), 2 deletions(-)

diff --git a/internal/command/wireguard/root.go b/internal/command/wireguard/root.go
index 4d21a46e49..54fa559366 100644
--- a/internal/command/wireguard/root.go
+++ b/internal/command/wireguard/root.go
@@ -49,6 +49,13 @@ func newWireguardCreate() *cobra.Command {
 		command.RequireSession,
 	)
 	cmd.Args = cobra.MaximumNArgs(4)
+	// Add flag for --network
+	flag.Add(cmd,
+		flag.String{
+			Name:        "network",
+			Description: "Custom network name",
+		},
+	)
 	return cmd
 }
 
diff --git a/internal/command/wireguard/wireguard.go b/internal/command/wireguard/wireguard.go
index 1f355661aa..76cb97eb8d 100644
--- a/internal/command/wireguard/wireguard.go
+++ b/internal/command/wireguard/wireguard.go
@@ -141,8 +141,7 @@ func runWireguardCreate(ctx context.Context) error {
 		name = args[2]
 	}
 
-	// TODO: allow custom network
-	network := ""
+	network := flag.GetString(ctx, "network")
 
 	state, err := wireguard.Create(apiClient, org, region, name, network, "static")
 	if err != nil {