Merge pull request #431 from civo/fix/network-create

Add default firewall flag for network create command

cmd/network/network.go

@@ -30,6 +30,7 @@ func init() {
 
 	networkCreateCmd.Flags().StringVarP(&cidrV4, "cidr-v4", "", "", "Custom IPv4 CIDR")
 	networkCreateCmd.Flags().StringSliceVarP(&nameserversV4, "nameservers-v4", "", nil, "Custom list of IPv4 nameservers (up to three, comma-separated)")
+	networkCreateCmd.Flags().BoolVarP(&createDefaultFirewall, "create-default-firewall", "", false, "Create a default firewall for the network")
 
 	// Flag binding for networkConnectCmd
 	networkConnectCmd.Flags().IntVar(&vlanID, "vlan-id", 0, "VLAN ID to connect")

cmd/network/network_create.go

@@ -2,24 +2,26 @@ package network
 
 import (
 	"fmt"
-	"github.com/civo/civogo"
 	"os"
 
+	"github.com/civo/civogo"
+
 	"github.com/civo/cli/common"
 	"github.com/civo/cli/config"
 	"github.com/civo/cli/utility"
 	"github.com/spf13/cobra"
 )
 
 var (
-	cidrV4        string
-	nameserversV4 []string
+	cidrV4                string
+	nameserversV4         []string
+	createDefaultFirewall bool
 )
 
 var networkCreateCmd = &cobra.Command{
 	Use:     "create",
 	Aliases: []string{"new", "add"},
-	Example: "civo network create NAME",
+	Example: "civo network create NAME [flags]",
 	Short:   "Create a new network",
 	Args:    cobra.MinimumNArgs(1),
 	Run: func(cmd *cobra.Command, args []string) {
@@ -57,5 +59,18 @@ var networkCreateCmd = &cobra.Command{
 		default:
 			fmt.Printf("Created a network called %s with ID %s\n", utility.Green(network.Label), utility.Green(network.ID))
 		}
+
+		if createDefaultFirewall {
+			firewall, err := client.NewFirewall(&civogo.FirewallConfig{
+				Name:      fmt.Sprintf("%s - Default", network.Label),
+				NetworkID: network.ID,
+				Region:    client.Region,
+			})
+			if err != nil {
+				utility.Error("%s", err)
+				os.Exit(1)
+			}
+			fmt.Printf("Created a default firewall %s\n", utility.Green(firewall.Name))
+		}
 	},
 }