105 lines
2.9 KiB
Go
105 lines
2.9 KiB
Go
package create
|
|
|
|
import (
|
|
"errors"
|
|
"fmt"
|
|
"github.com/cirruslabs/orchard/pkg/client"
|
|
v1 "github.com/cirruslabs/orchard/pkg/resource/v1"
|
|
"github.com/spf13/cobra"
|
|
"os"
|
|
"strings"
|
|
)
|
|
|
|
var ErrVMFailed = errors.New("failed to create VM")
|
|
|
|
var image string
|
|
var cpu uint64
|
|
var memory uint64
|
|
var netSoftnet bool
|
|
var netBridged string
|
|
var headless bool
|
|
var resources map[string]string
|
|
var restartPolicy string
|
|
var startupScript string
|
|
|
|
func newCreateVMCommand() *cobra.Command {
|
|
command := &cobra.Command{
|
|
Use: "vm NAME",
|
|
Short: "Create a VM",
|
|
RunE: runCreateVM,
|
|
Args: cobra.ExactArgs(1),
|
|
}
|
|
|
|
command.PersistentFlags().StringVar(&image, "image", "ghcr.io/cirruslabs/macos-ventura-base:latest", "image to use")
|
|
command.PersistentFlags().Uint64Var(&cpu, "cpu", 4, "number of CPUs to use")
|
|
command.PersistentFlags().Uint64Var(&memory, "memory", 8*1024, "megabytes of memory to use")
|
|
command.PersistentFlags().BoolVar(&netSoftnet, "net-softnet", false, "whether to use Softnet network isolation")
|
|
command.PersistentFlags().StringVar(&netBridged, "net-bridged", "", "whether to use Bridged network mode")
|
|
command.PersistentFlags().BoolVar(&headless, "headless", true, "whether to run without graphics")
|
|
command.PersistentFlags().StringToStringVar(&resources, "resources", map[string]string{},
|
|
"resources to request for this VM")
|
|
command.PersistentFlags().StringVar(&restartPolicy, "restart-policy", "Never",
|
|
"restart policy for this VM: specify \"Never\" to never restart or \"OnFailure\" "+
|
|
"to only restart when the VM fails")
|
|
command.PersistentFlags().StringVar(&startupScript, "startup-script", "",
|
|
"startup script (e.g. --startup-script=\"sync\") or a path to a script file prefixed with \"@\" "+
|
|
"(e.g. \"--startup-script=@script.sh\")")
|
|
|
|
return command
|
|
}
|
|
|
|
func runCreateVM(cmd *cobra.Command, args []string) error {
|
|
name := args[0]
|
|
|
|
vm := &v1.VM{
|
|
Meta: v1.Meta{
|
|
Name: name,
|
|
},
|
|
Image: image,
|
|
CPU: cpu,
|
|
Memory: memory,
|
|
NetSoftnet: netSoftnet,
|
|
NetBridged: netBridged,
|
|
Headless: headless,
|
|
}
|
|
|
|
// Convert resources
|
|
var err error
|
|
|
|
vm.Resources, err = v1.NewResourcesFromStringToString(resources)
|
|
if err != nil {
|
|
return fmt.Errorf("%w: %v", ErrVMFailed, err)
|
|
}
|
|
|
|
// Convert restart policy
|
|
vm.RestartPolicy, err = v1.NewRestartPolicyFromString(restartPolicy)
|
|
if err != nil {
|
|
return fmt.Errorf("%w: %v", ErrVMFailed, err)
|
|
}
|
|
|
|
// Convert startup script, optionally reading it from the file system
|
|
const scriptFilePrefix = "@"
|
|
|
|
if strings.HasPrefix(startupScript, scriptFilePrefix) {
|
|
startupScriptBytes, err := os.ReadFile(strings.TrimPrefix(startupScript, scriptFilePrefix))
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
vm.StartupScript = &v1.VMScript{
|
|
ScriptContent: string(startupScriptBytes),
|
|
}
|
|
} else if startupScript != "" {
|
|
vm.StartupScript = &v1.VMScript{
|
|
ScriptContent: startupScript,
|
|
}
|
|
}
|
|
|
|
client, err := client.New()
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
return client.VMs().Create(cmd.Context(), vm)
|
|
}
|