djpbessems/Go.NPP-Prepper
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Move routines to subcommands;Add initial logic for datacenter configuration; Move code to package
All checks were successful
continuous-integration/drone/push Build is passing
Details

Browse Source
This commit is contained in:
Danny Bessems
2022-12-28 14:41:03 +01:00
parent 103e355b8e
commit d8c361101e
5 changed files with 101 additions and 33 deletions
Download Patch File Download Diff File Expand all files Collapse all files

65
cmd/vappprop-manager/main.go
View File

@ -10,43 +10,58 @@ import (
"spamasaurus.com/m/pkg/hypervisor"
)
var Global struct {
FQDN string `short:"s" long:"server" description:"FQDN of the vCenter appliance" required:"true"`
Username string `short:"u" long:"username" description:"Username to authenticate with" required:"true"`
Password string `short:"p" long:"password" description:"Password to authenticate with" required:"true"`
}
var Commands struct {
Datacenter struct {
Name string `short:"n" long:"name" description:"Name of datacenter" required:"true"`
Network string `short:"p" long:"portgroup" description:"Name of network portgroup" required:"true"`
} // `command:"datacenter" alias:"dc" description:"Define a Network Protocol Profile within a datacenter"`
VirtualMachine struct {
Datacenter string `short:"d" long:"datacenter" description:"Name of datacenter" required:"true"`
Name string `short:"n" long:"name" description:"Name of virtual machine" required:"true"`
Network string `short:"p" long:"portgroup" description:"Name of network portgroup" required:"true"`
} // `command:"virtualmachine" alias:"vm" description:"Configure a virtual machine for usage of Network Protocol Profiles"`
}
func main() {
var opts struct {
FQDN string `short:"s" long:"server" description:"FQDN of the vCenter appliance" required:"true"`
Username string `short:"u" long:"username" description:"Username to authenticate with" required:"true"`
Password string `short:"p" long:"password" description:"Password to authenticate with" required:"true"`
Datacenter string `short:"d" long:"datacenter" description:"Name of datacenter" required:"true"`
VirtualMachine string `short:"v" long:"virtualmachine" description:"Name of virtual machine" required:"true"`
Network string `short:"n" long:"network" description:"Name of network portgroup" required:"true"`
}
_, err := flags.Parse(&opts)
parser := flags.NewParser(&Global, flags.Default)
parser.AddCommand("datacenter", "Define a Network Protocol Profile within a datacenter", "", &Commands.Datacenter)
parser.AddCommand("dc", "", "", &Commands.Datacenter)
parser.AddCommand("virtualmachine", "Configure a virtual machine for usage of Network Protocol Profiles", "", &Commands.VirtualMachine)
parser.AddCommand("vm", "", "", &Commands.VirtualMachine)
_, err := parser.Parse()
if err != nil {
if e, ok := err.(*flags.Error); ok {
if e.Type == flags.ErrHelp {
os.Exit(0)
}
if flagsErr, ok := err.(*flags.Error); ok && flagsErr.Type == flags.ErrHelp {
os.Exit(0)
} else {
os.Exit(1)
}
os.Exit(1)
}
ctx, cancel := context.WithCancel(context.Background())
defer cancel()
clt, err := hypervisor.NewClient(ctx, opts.FQDN, opts.Username, opts.Password, true)
clt, err := hypervisor.NewClient(ctx, Global.FQDN, Global.Username, Global.Password, true)
if err != nil {
log.Fatalf("[ERROR] Login failed: %s", err)
}
fnd, err := hypervisor.DatacenterFinder(ctx, clt, opts.Datacenter)
if err != nil {
log.Fatalf("[ERROR] Unable to determine datacenter: %s", err)
}
switch parser.Active.Name {
case "datacenter", "dc":
if err := hypervisor.CreateNetworkProtocolProfile(ctx, clt, Commands.Datacenter.Name, Commands.Datacenter.Network); err != nil {
log.Fatalf("[ERROR] Could not create network protocol profile: %s", err)
}
if err := hypervisor.SetVirtualMachineProperties(ctx, fnd, opts.VirtualMachine, opts.Network); err != nil {
log.Fatalf("[ERROR] Could not apply vApp properties: %s", err)
}
log.Printf("[SUCCESS] Hurray! We did it!")
case "virtualmachine", "vm":
if err := hypervisor.SetVirtualMachineProperties(ctx, clt, Commands.VirtualMachine.Datacenter, Commands.VirtualMachine.Name, Commands.VirtualMachine.Network); err != nil {
log.Fatalf("[ERROR] Could not apply vApp properties: %s", err)
}
log.Printf("[SUCCESS] Network protocol profile properties added to virtual machine '%s' and configured for network '%s'", opts.VirtualMachine, opts.Network)
log.Printf("[SUCCESS] Network protocol profile properties added to virtual machine '%s' and configured for network '%s'", Commands.VirtualMachine.Name, Commands.VirtualMachine.Network)
}
}