spec.go

package platform

import (
	"strings"

	"github.com/otiai10/hotsub/params"

	"github.com/otiai10/dkmachine"
	"github.com/otiai10/hotsub/core"
)

// DefineMachineSpec is a factory layer to connect cli.Context to CreateOptions.
func DefineMachineSpec(ctx params.Context) (*dkmachine.CreateOptions, error) {
	opt := &dkmachine.CreateOptions{
		// AWS
		AmazonEC2Region:             ctx.String("aws-region"),
		AmazonEC2VpcID:              ctx.String("aws-vpc-id"),
		AmazonEC2SubnetID:           ctx.String("aws-subnet-id"),
		AmazonEC2RootSize:           ctx.Int("disk-size"),
		AmazonEC2InstanceType:       ctx.String("aws-ec2-instance-type"),
		AmazonEC2IAMInstanceProfile: DefaultAWSInstanceProfileNameForCompute,
		AmazonEC2SecurityGroup:      DefaultAWSSecurityGroupName,
		// GCP
		GoogleProject:  ctx.String("google-project"),
		GoogleZone:     ctx.String("google-zone"),
		GoogleDiskSize: ctx.Int("disk-size"),
		GoogleTags:     []string{DefaultGoogleInstanceTag},
		GoogleScopes: strings.Join([]string{
			"https://www.googleapis.com/auth/devstorage.read_write",
			"https://www.googleapis.com/auth/logging.write",
			"https://www.googleapis.com/auth/monitoring.write",
		}, ","),
	}
	switch Provider(ctx.String("provider")) {
	case AWS:
		opt.Driver = string(AmazonEC2)
	case GCP:
		opt.Driver = string(Google)
	case VBOX:
		opt.Driver = string(Vbox)
	case HYPERV:
		opt.Driver = string(Hyperv)
	default:
		opt.Driver = string(AmazonEC2)
	}
	return opt, nil
}

// DefineSharedDataInstanceSpec ...
// It defines ALL the specifications for docker-machine,
// though it's a bit verbose ;)
func DefineSharedDataInstanceSpec(shared core.Inputs, ctx params.Context) (*dkmachine.CreateOptions, error) {
	opt := &dkmachine.CreateOptions{
		Name:                        "Shared-Data-Instance",
		AmazonEC2VpcID:              ctx.String("aws-vpc-id"),
		AmazonEC2SubnetID:           ctx.String("aws-subnet-id"),
		AmazonEC2Region:             ctx.String("aws-region"),
		AmazonEC2IAMInstanceProfile: DefaultAWSInstanceProfileNameForCompute,
		AmazonEC2SecurityGroup:      DefaultAWSSecurityGroupName,

		AmazonEC2InstanceType: ctx.String("aws-shared-instance-type"),
		AmazonEC2RootSize:     ctx.Int("shareddata-disksize"),
		// GCP
		GoogleProject:  ctx.String("google-project"),
		GoogleZone:     ctx.String("google-zone"),
		GoogleDiskSize: ctx.Int("shareddata-disksize"),
		GoogleTags:     []string{DefaultGoogleInstanceTag},
		GoogleScopes: strings.Join([]string{
			"https://www.googleapis.com/auth/devstorage.read_write",
			"https://www.googleapis.com/auth/logging.write",
			"https://www.googleapis.com/auth/monitoring.write",
		}, ","),
	}
	switch Provider(ctx.String("provider")) {
	case AWS:
		opt.Driver = string(AmazonEC2)
	case GCP:
		opt.Driver = string(Google)
	default:
		opt.Driver = string(AmazonEC2)
	}
	return opt, nil
}