130 lines
3.8 KiB
Go
130 lines
3.8 KiB
Go
package driver
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
"net/url"
|
|
"time"
|
|
|
|
packersdk "github.com/hashicorp/packer/packer-plugin-sdk/packer"
|
|
"github.com/vmware/govmomi"
|
|
"github.com/vmware/govmomi/find"
|
|
"github.com/vmware/govmomi/object"
|
|
"github.com/vmware/govmomi/session"
|
|
"github.com/vmware/govmomi/vapi/library"
|
|
"github.com/vmware/govmomi/vapi/rest"
|
|
"github.com/vmware/govmomi/vim25"
|
|
"github.com/vmware/govmomi/vim25/soap"
|
|
"github.com/vmware/govmomi/vim25/types"
|
|
)
|
|
|
|
type Driver interface {
|
|
NewVM(ref *types.ManagedObjectReference) VirtualMachine
|
|
FindVM(name string) (VirtualMachine, error)
|
|
FindCluster(name string) (*Cluster, error)
|
|
PreCleanVM(ui packersdk.Ui, vmPath string, force bool) error
|
|
CreateVM(config *CreateConfig) (VirtualMachine, error)
|
|
|
|
NewDatastore(ref *types.ManagedObjectReference) Datastore
|
|
FindDatastore(name string, host string) (Datastore, error)
|
|
GetDatastoreName(id string) (string, error)
|
|
GetDatastoreFilePath(datastoreID, dir, filename string) (string, error)
|
|
|
|
NewFolder(ref *types.ManagedObjectReference) *Folder
|
|
FindFolder(name string) (*Folder, error)
|
|
NewHost(ref *types.ManagedObjectReference) *Host
|
|
FindHost(name string) (*Host, error)
|
|
NewNetwork(ref *types.ManagedObjectReference) *Network
|
|
FindNetwork(name string) (*Network, error)
|
|
FindNetworks(name string) ([]*Network, error)
|
|
NewResourcePool(ref *types.ManagedObjectReference) *ResourcePool
|
|
FindResourcePool(cluster string, host string, name string) (*ResourcePool, error)
|
|
|
|
FindContentLibraryByName(name string) (*Library, error)
|
|
FindContentLibraryItem(libraryId string, name string) (*library.Item, error)
|
|
FindContentLibraryFileDatastorePath(isoPath string) (string, error)
|
|
}
|
|
|
|
type VCenterDriver struct {
|
|
// context that controls the authenticated sessions used to run the VM commands
|
|
ctx context.Context
|
|
client *govmomi.Client
|
|
vimClient *vim25.Client
|
|
restClient *RestClient
|
|
finder *find.Finder
|
|
datacenter *object.Datacenter
|
|
}
|
|
|
|
type ConnectConfig struct {
|
|
VCenterServer string
|
|
Username string
|
|
Password string
|
|
InsecureConnection bool
|
|
Datacenter string
|
|
}
|
|
|
|
func NewDriver(config *ConnectConfig) (Driver, error) {
|
|
ctx := context.TODO()
|
|
|
|
vcenterUrl, err := url.Parse(fmt.Sprintf("https://%v/sdk", config.VCenterServer))
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
credentials := url.UserPassword(config.Username, config.Password)
|
|
vcenterUrl.User = credentials
|
|
|
|
soapClient := soap.NewClient(vcenterUrl, config.InsecureConnection)
|
|
vimClient, err := vim25.NewClient(ctx, soapClient)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
vimClient.RoundTripper = session.KeepAlive(vimClient.RoundTripper, 10*time.Minute)
|
|
client := &govmomi.Client{
|
|
Client: vimClient,
|
|
SessionManager: session.NewManager(vimClient),
|
|
}
|
|
|
|
err = client.SessionManager.Login(ctx, credentials)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
finder := find.NewFinder(client.Client, false)
|
|
datacenter, err := finder.DatacenterOrDefault(ctx, config.Datacenter)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
finder.SetDatacenter(datacenter)
|
|
|
|
d := &VCenterDriver{
|
|
ctx: ctx,
|
|
client: client,
|
|
vimClient: vimClient,
|
|
restClient: &RestClient{
|
|
client: rest.NewClient(vimClient),
|
|
credentials: credentials,
|
|
},
|
|
datacenter: datacenter,
|
|
finder: finder,
|
|
}
|
|
return d, nil
|
|
}
|
|
|
|
// The rest.Client requires vCenter.
|
|
// RestClient is to modularize the rest.Client session and use it only when is necessary.
|
|
// This will allow users without vCenter to use the other features that doesn't use the rest.Client.
|
|
// To use the client login/logout must be done to create an authenticated session.
|
|
type RestClient struct {
|
|
client *rest.Client
|
|
credentials *url.Userinfo
|
|
}
|
|
|
|
func (r *RestClient) Login(ctx context.Context) error {
|
|
return r.client.Login(ctx, r.credentials)
|
|
}
|
|
|
|
func (r *RestClient) Logout(ctx context.Context) error {
|
|
return r.client.Logout(ctx)
|
|
}
|