github.com/cloudfoundry-incubator/stembuild@v0.0.0-20211223202937-5b61d62226c6/commandparser/construct.go

package commandparser

import (
	"context"
	"flag"
	"fmt"
	"os"
	"path/filepath"

	vcenter_client_factory "github.com/cloudfoundry-incubator/stembuild/iaas_cli/iaas_clients/factory"
	"github.com/cloudfoundry-incubator/stembuild/iaas_cli/iaas_clients/guest_manager"
	"github.com/cloudfoundry-incubator/stembuild/iaas_cli/iaas_clients/vcenter_manager"
	"github.com/vmware/govmomi/guest"
	"github.com/vmware/govmomi/object"

	"github.com/cloudfoundry-incubator/stembuild/construct/config"
	"github.com/google/subcommands"
)

//go:generate go run github.com/maxbrunsfeld/counterfeiter/v6 . VmConstruct
type VmConstruct interface {
	PrepareVM() error
}

//go:generate go run github.com/maxbrunsfeld/counterfeiter/v6 . VCenterManager
type VCenterManager interface {
	OperationsManager(ctx context.Context, vm *object.VirtualMachine) *guest.OperationsManager
	GuestManager(ctx context.Context, opsManager vcenter_manager.OpsManager, username, password string) (*guest_manager.GuestManager, error)
	FindVM(ctx context.Context, inventoryPath string) (*object.VirtualMachine, error)
	Login(ctx context.Context) error
}

//go:generate go run github.com/maxbrunsfeld/counterfeiter/v6 . VMPreparerFactory
type VMPreparerFactory interface {
	VMPreparer(config config.SourceConfig, vCenterManager VCenterManager) (VmConstruct, error)
}

//go:generate go run github.com/maxbrunsfeld/counterfeiter/v6 . ManagerFactory
type ManagerFactory interface {
	VCenterManager(ctx context.Context) (*vcenter_manager.VCenterManager, error)
	SetConfig(config vcenter_client_factory.FactoryConfig)
}

//go:generate go run github.com/maxbrunsfeld/counterfeiter/v6 . ConstructCmdValidator
type ConstructCmdValidator interface {
	PopulatedArgs(...string) bool
	LGPOInDirectory() bool
}

//go:generate go run github.com/maxbrunsfeld/counterfeiter/v6 . ConstructMessenger
type ConstructMessenger interface {
	ArgumentsNotProvided()
	LGPONotFound()
	CannotConnectToVM(err error)
	CannotPrepareVM(err error)
}

type ConstructCmd struct {
	ctx            context.Context
	sourceConfig   config.SourceConfig
	prepFactory    VMPreparerFactory
	managerFactory ManagerFactory
	validator      ConstructCmdValidator
	messenger      ConstructMessenger
	GlobalFlags    *GlobalFlags
}

func NewConstructCmd(ctx context.Context, prepFactory VMPreparerFactory, managerFactory ManagerFactory, validator ConstructCmdValidator, messenger ConstructMessenger) *ConstructCmd {
	return &ConstructCmd{ctx: ctx, prepFactory: prepFactory, managerFactory: managerFactory, validator: validator, messenger: messenger}
}

func (*ConstructCmd) Name() string { return "construct" }
func (*ConstructCmd) Synopsis() string {
	return "Provisions and syspreps an existing VM on vCenter, ready to be packaged into a stemcell"
}

func (*ConstructCmd) Usage() string {
	return fmt.Sprintf(`%[1]s construct -vm-ip <IP of VM> -vm-username <vm username> -vm-password <vm password>  -vcenter-url <vCenter URL> -vcenter-username <vCenter username> -vcenter-password <vCenter password> -vm-inventory-path <vCenter VM inventory path>

Prepares a VM to be used by stembuild package. It leverages stemcell automation scripts to provision a VM to be used as a stemcell.

Requirements:
	LGPO.zip in current working directory
	Running Windows VM with:
		- Up to date Operating System
		- Reachable by IP
		- Username and password with Administrator privileges
		- vCenter URL, username and password
		- vCenter Inventory Path
	The [vm-ip], [vm-username], [vm-password], [vcenter-url], [vcenter-username], [vcenter-password], [vm-inventory-path] must be specified

Example:
	%[1]s construct -vm-ip '10.0.0.5' -vm-username Admin -vm-password 'password' -vcenter-url vcenter.example.com -vcenter-username root -vcenter-password 'password' -vm-inventory-path '/datacenter/vm/folder/vm-name'

Flags:
`, filepath.Base(os.Args[0]))
}

func (p *ConstructCmd) SetFlags(f *flag.FlagSet) {
	f.StringVar(&p.sourceConfig.GuestVmIp, "vm-ip", "", "IP of target machine")
	f.StringVar(&p.sourceConfig.GuestVMUsername, "vm-username", "", "Username of target machine")
	f.StringVar(&p.sourceConfig.GuestVMPassword, "vm-password", "", "Password of target machine. Needs to be wrapped in single quotations.")
	f.StringVar(&p.sourceConfig.VCenterUrl, "vcenter-url", "", "vCenter url")
	f.StringVar(&p.sourceConfig.VCenterUsername, "vcenter-username", "", "vCenter username")
	f.StringVar(&p.sourceConfig.VCenterPassword, "vcenter-password", "", "vCenter password")
	f.StringVar(&p.sourceConfig.VmInventoryPath, "vm-inventory-path", "", "vCenter VM inventory path. (e.g: <datacenter>/vm/<vm-folder>/<vm-name>)")
	f.StringVar(&p.sourceConfig.CaCertFile, "vcenter-ca-certs", "", "filepath for custom ca certs")
}

func (p *ConstructCmd) Execute(_ context.Context, f *flag.FlagSet, _ ...interface{}) subcommands.ExitStatus {
	c := p.sourceConfig
	if !p.validator.PopulatedArgs(c.GuestVmIp, c.GuestVMUsername, c.GuestVMPassword, c.VCenterUrl, c.VCenterUsername, c.VCenterPassword, c.VmInventoryPath) {
		p.messenger.ArgumentsNotProvided()
		return subcommands.ExitFailure
	}
	if !p.validator.LGPOInDirectory() {
		p.messenger.LGPONotFound()
		return subcommands.ExitFailure
	}

	p.managerFactory.SetConfig(vcenter_client_factory.FactoryConfig{
		p.sourceConfig.VCenterUrl,
		p.sourceConfig.VCenterUsername,
		p.sourceConfig.VCenterPassword,
		&vcenter_client_factory.ClientCreator{},
		&vcenter_client_factory.GovmomiFinderCreator{},
		p.sourceConfig.CaCertFile,
	})

	vCenterManager, err := p.managerFactory.VCenterManager(p.ctx)
	if err != nil {
		p.messenger.CannotPrepareVM(err)
		return subcommands.ExitFailure
	}

	vmConstruct, err := p.prepFactory.VMPreparer(p.sourceConfig, vCenterManager)
	if err != nil {
		p.messenger.CannotPrepareVM(err)
		return subcommands.ExitFailure
	}

	err = vmConstruct.PrepareVM()
	if err != nil {
		p.messenger.CannotPrepareVM(err)
		return subcommands.ExitFailure
	}

	return subcommands.ExitSuccess
}