add network flag to wireguard create

superfly · Aug 7, 2024 · 4bb53c6 · 4bb53c6
1 parent 87aa565
commit 4bb53c6
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 2 deletions.
diff --git 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
@@ -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 {