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
|
package network
import (
"context"
"fmt"
"net"
"github.com/spf13/cobra"
"github.com/hetznercloud/cli/internal/cmd/base"
"github.com/hetznercloud/cli/internal/cmd/cmpl"
"github.com/hetznercloud/cli/internal/hcapi2"
"github.com/hetznercloud/cli/internal/state"
"github.com/hetznercloud/hcloud-go/v2/hcloud"
)
var AddSubnetCommand = base.Cmd{
BaseCobraCommand: func(client hcapi2.Client) *cobra.Command {
cmd := &cobra.Command{
Use: "add-subnet NETWORK FLAGS",
Short: "Add a subnet to a network",
Args: cobra.ExactArgs(1),
ValidArgsFunction: cmpl.SuggestArgs(cmpl.SuggestCandidatesF(client.Network().Names)),
TraverseChildren: true,
DisableFlagsInUseLine: true,
}
cmd.Flags().String("type", "", "Type of subnet (required)")
cmd.RegisterFlagCompletionFunc("type", cmpl.SuggestCandidates("cloud", "server", "vswitch"))
cmd.MarkFlagRequired("type")
cmd.Flags().String("network-zone", "", "Name of network zone (required)")
cmd.RegisterFlagCompletionFunc("network-zone", cmpl.SuggestCandidatesF(client.Location().NetworkZones))
cmd.MarkFlagRequired("network-zone")
cmd.Flags().IPNet("ip-range", net.IPNet{}, "Range to allocate IPs from")
cmd.Flags().Int64("vswitch-id", 0, "ID of the vSwitch")
return cmd
},
Run: func(ctx context.Context, client hcapi2.Client, waiter state.ActionWaiter, cmd *cobra.Command, args []string) error {
subnetType, _ := cmd.Flags().GetString("type")
networkZone, _ := cmd.Flags().GetString("network-zone")
ipRange, _ := cmd.Flags().GetIPNet("ip-range")
vSwitchID, _ := cmd.Flags().GetInt64("vswitch-id")
idOrName := args[0]
network, _, err := client.Network().Get(ctx, idOrName)
if err != nil {
return err
}
if network == nil {
return fmt.Errorf("network not found: %s", idOrName)
}
subnet := hcloud.NetworkSubnet{
Type: hcloud.NetworkSubnetType(subnetType),
NetworkZone: hcloud.NetworkZone(networkZone),
}
if ipRange.IP != nil && ipRange.Mask != nil {
subnet.IPRange = &ipRange
}
if subnetType == "vswitch" {
subnet.VSwitchID = vSwitchID
}
opts := hcloud.NetworkAddSubnetOpts{
Subnet: subnet,
}
action, _, err := client.Network().AddSubnet(ctx, network, opts)
if err != nil {
return err
}
if err := waiter.ActionProgress(ctx, action); err != nil {
return err
}
fmt.Printf("Subnet added to network %d\n", network.ID)
return nil
},
}
|