vm.go

package compute

import (
	"github.com/DataDog/test-infra-definitions/common/config"
	"github.com/DataDog/test-infra-definitions/common/utils"
	"github.com/DataDog/test-infra-definitions/resources/gcp"
	"github.com/pulumi/pulumi-gcp/sdk/v7/go/gcp/compute"
	"github.com/pulumi/pulumi/sdk/v3/go/pulumi"
)

func NewLinuxInstance(e gcp.Environment, name string, imageName string, instanceType string, opts ...pulumi.ResourceOption) (*compute.Instance, error) {

	sshPublicKey, err := utils.GetSSHPublicKey(e.DefaultPublicKeyPath())
	if err != nil {
		return nil, err
	}
	instance, err := compute.NewInstance(e.Ctx(), e.Namer.ResourceName(name), &compute.InstanceArgs{
		NetworkInterfaces: compute.InstanceNetworkInterfaceArray{
			&compute.InstanceNetworkInterfaceArgs{
				AccessConfigs: compute.InstanceNetworkInterfaceAccessConfigArray{
					nil,
				},
				Network:    pulumi.String(e.DefaultNetworkName()),
				Subnetwork: pulumi.String(e.DefaultSubnet()),
			},
		},
		Name:                   e.Namer.DisplayName(63, pulumi.String(name)),
		MachineType:            pulumi.String(instanceType),
		AllowStoppingForUpdate: pulumi.Bool(true),
		AdvancedMachineFeatures: &compute.InstanceAdvancedMachineFeaturesArgs{
			EnableNestedVirtualization: pulumi.Bool(true),
		},
		Tags: pulumi.StringArray{
			pulumi.String("appgate-gateway"),
			pulumi.String("nat-us-central1"),
		},
		BootDisk: &compute.InstanceBootDiskArgs{
			InitializeParams: &compute.InstanceBootDiskInitializeParamsArgs{
				Image: pulumi.String(imageName),
				Labels: pulumi.StringMap{
					"my_label": pulumi.String("value"),
				},
				Size: pulumi.Int(100),
			},
		},
		Metadata: pulumi.StringMap{
			"enable-oslogin": pulumi.String("false"),
			"ssh-keys":       pulumi.Sprintf("gce:%s", sshPublicKey),
		},
		ServiceAccount: &compute.InstanceServiceAccountArgs{
			Email: pulumi.String(e.DefaultVMServiceAccount()),
			Scopes: pulumi.StringArray{
				pulumi.String("cloud-platform"),
			},
		},
	}, append(opts, e.WithProviders(config.ProviderGCP))...)
	if err != nil {
		return nil, err
	}

	return instance, nil
}