autoroll/vm.go

package main

import (
	"path"
	"runtime"

	"go.skia.org/infra/go/androidbuildinternal/v2beta1"
	"go.skia.org/infra/go/auth"
	"go.skia.org/infra/go/gce"
	"go.skia.org/infra/go/gce/server"
)

func AutoRollBase(name, ipAddress string) *gce.Instance {
	vm := server.AddGitConfigs(server.Server20170613(name), name)
	vm.DataDisk.SizeGb = 64
	vm.DataDisk.Type = gce.DISK_TYPE_PERSISTENT_STANDARD
	if ipAddress != "" {
		vm.ExternalIpAddress = ipAddress
	}
	vm.MachineType = gce.MACHINE_TYPE_STANDARD_2
	vm.Metadata["owner_primary"] = "borenet"
	vm.Metadata["owner_secondary"] = "rmistry"
	vm.Scopes = append(vm.Scopes,
		auth.SCOPE_GERRIT,
	)
	return vm
}

func Skia() *gce.Instance {
	return AutoRollBase("skia-autoroll", "" /* Use ephemeral IP */)
}

func SkiaInternal() *gce.Instance {
	return AutoRollBase("skia-internal-autoroll", "" /* Use ephemeral IP */)
}

func Catapult() *gce.Instance {
	return AutoRollBase("catapult-autoroll", "" /* Use ephemeral IP */)
}

func NaCl() *gce.Instance {
	return AutoRollBase("nacl-autoroll", "" /* Use ephemeral IP */)
}

func PDFium() *gce.Instance {
	return AutoRollBase("pdfium-autoroll", "" /* Use ephemeral IP */)
}

func Fuchsia() *gce.Instance {
	return AutoRollBase("fuchsia-autoroll", "" /* Use ephemeral IP */)
}

func AddAndroidConfigs(vm *gce.Instance) *gce.Instance {
	vm.DataDisk.SizeGb = 512
	vm.MachineType = gce.MACHINE_TYPE_HIGHMEM_16
	vm.Scopes = append(vm.Scopes, androidbuildinternal.AndroidbuildInternalScope)

	_, filename, _, _ := runtime.Caller(0)
	dir := path.Dir(filename)
	vm.SetupScript = path.Join(dir, "setup-script-android.sh")
	return vm
}

func AndroidMaster() *gce.Instance {
	return AddAndroidConfigs(AutoRollBase("android-master-autoroll", "130.211.199.63" /* Needs whitelisted static IP */))
}

func AndroidO() *gce.Instance {
	return AddAndroidConfigs(AutoRollBase("android-o-autoroll", "104.198.73.244" /* Needs whitelisted static IP */))
}

func main() {
	server.Main(gce.ZONE_DEFAULT, map[string]*gce.Instance{
		"skia":           Skia(),
		"skia-internal":  SkiaInternal(),
		"catapult":       Catapult(),
		"nacl":           NaCl(),
		"pdfium":         PDFium(),
		"fuchsia":        Fuchsia(),
		"android-master": AndroidMaster(),
		"android-o":      AndroidO(),
	})
}