117 lines
4.4 KiB
Go
117 lines
4.4 KiB
Go
package googlecompute
|
|
|
|
import (
|
|
"crypto/rsa"
|
|
"time"
|
|
|
|
compute "google.golang.org/api/compute/v1"
|
|
)
|
|
|
|
// Driver is the interface that has to be implemented to communicate
|
|
// with GCE. The Driver interface exists mostly to allow a mock implementation
|
|
// to be used to test the steps.
|
|
type Driver interface {
|
|
// CreateImage creates an image from the given disk in Google Compute
|
|
// Engine.
|
|
CreateImage(name, description, family, zone, disk string, image_labels map[string]string, image_licenses []string, image_encryption_key *compute.CustomerEncryptionKey, imageStorageLocation []string) (<-chan *Image, <-chan error)
|
|
|
|
// DeleteImage deletes the image with the given name.
|
|
DeleteImage(name string) <-chan error
|
|
|
|
// DeleteInstance deletes the given instance, keeping the boot disk.
|
|
DeleteInstance(zone, name string) (<-chan error, error)
|
|
|
|
// DeleteDisk deletes the disk with the given name.
|
|
DeleteDisk(zone, name string) (<-chan error, error)
|
|
|
|
// GetImage gets an image; tries the default and public projects. If
|
|
// fromFamily is true, name designates an image family instead of a
|
|
// particular image.
|
|
GetImage(name string, fromFamily bool) (*Image, error)
|
|
|
|
// GetImageFromProject gets an image from a specific projects.
|
|
// Returns the image from the first project in slice it can find one
|
|
// If fromFamily is true, name designates an image family instead of a particular image.
|
|
GetImageFromProjects(project []string, name string, fromFamily bool) (*Image, error)
|
|
|
|
// GetImageFromProject gets an image from a specific project. If fromFamily
|
|
// is true, name designates an image family instead of a particular image.
|
|
GetImageFromProject(project, name string, fromFamily bool) (*Image, error)
|
|
|
|
// GetInstanceMetadata gets a metadata variable for the instance, name.
|
|
GetInstanceMetadata(zone, name, key string) (string, error)
|
|
|
|
// GetInternalIP gets the GCE-internal IP address for the instance.
|
|
GetInternalIP(zone, name string) (string, error)
|
|
|
|
// GetNatIP gets the NAT IP address for the instance.
|
|
GetNatIP(zone, name string) (string, error)
|
|
|
|
// GetSerialPortOutput gets the Serial Port contents for the instance.
|
|
GetSerialPortOutput(zone, name string) (string, error)
|
|
|
|
// ImageExists returns true if the specified image exists. If an error
|
|
// occurs calling the API, this method returns false.
|
|
ImageExists(name string) bool
|
|
|
|
// RunInstance takes the given config and launches an instance.
|
|
RunInstance(*InstanceConfig) (<-chan error, error)
|
|
|
|
// WaitForInstance waits for an instance to reach the given state.
|
|
WaitForInstance(state, zone, name string) <-chan error
|
|
|
|
// CreateOrResetWindowsPassword creates or resets the password for a user on an Windows instance.
|
|
CreateOrResetWindowsPassword(zone, name string, config *WindowsPasswordConfig) (<-chan error, error)
|
|
}
|
|
|
|
type InstanceConfig struct {
|
|
AcceleratorType string
|
|
AcceleratorCount int64
|
|
Address string
|
|
Description string
|
|
DisableDefaultServiceAccount bool
|
|
DiskSizeGb int64
|
|
DiskType string
|
|
EnableSecureBoot bool
|
|
EnableVtpm bool
|
|
EnableIntegrityMonitoring bool
|
|
Image *Image
|
|
Labels map[string]string
|
|
MachineType string
|
|
Metadata map[string]string
|
|
MinCpuPlatform string
|
|
Name string
|
|
Network string
|
|
NetworkProjectId string
|
|
OmitExternalIP bool
|
|
OnHostMaintenance string
|
|
Preemptible bool
|
|
Region string
|
|
ServiceAccountEmail string
|
|
Scopes []string
|
|
Subnetwork string
|
|
Tags []string
|
|
Zone string
|
|
}
|
|
|
|
// WindowsPasswordConfig is the data structure that GCE needs to encrypt the created
|
|
// windows password.
|
|
type WindowsPasswordConfig struct {
|
|
key *rsa.PrivateKey
|
|
password string
|
|
UserName string `json:"userName"`
|
|
Modulus string `json:"modulus"`
|
|
Exponent string `json:"exponent"`
|
|
Email string `json:"email"`
|
|
ExpireOn time.Time `json:"expireOn"`
|
|
}
|
|
|
|
type windowsPasswordResponse struct {
|
|
UserName string `json:"userName"`
|
|
PasswordFound bool `json:"passwordFound"`
|
|
EncryptedPassword string `json:"encryptedPassword"`
|
|
Modulus string `json:"modulus"`
|
|
Exponent string `json:"exponent"`
|
|
ErrorMessage string `json:"errorMessage"`
|
|
}
|