Initial commit

main.go

@@ -0,0 +1,84 @@
+package main
+
+import (
+	"fmt"
+	"math/rand"
+	"sync"
+
+	"github.com/sirupsen/logrus"
+	"github.com/spf13/cobra"
+
+	"mkvm/libvirtx"
+)
+
+var (
+	wg      sync.WaitGroup
+	rootCmd = cobra.Command{
+		Use:   "mkvm name [name [name]]",
+		Short: "create virtual machine(s) via libvirt",
+		Args:  cobra.MinimumNArgs(1),
+		Run: func(cmd *cobra.Command, args []string) {
+			logrus.SetLevel(logrus.DebugLevel)
+
+			conn, err := libvirtx.New()
+			if err != nil {
+				logrus.WithError(err).Fatal("error connecting to libvirt")
+				return
+			}
+			defer conn.Close()
+
+			network, err := getNetwork(conn)
+			if err != nil {
+				logrus.WithError(err).Fatal("error finding requested network")
+			}
+
+			serverBindIP := network.IPs[0].Address
+			port := rand.Intn(65535-1025) + 1025
+			serverBind := fmt.Sprintf("%s:%d", serverBindIP, port)
+
+			go runHTTPServer(serverBind)
+
+			for i, name := range args {
+				metadataURL := fmt.Sprintf("http://%s/%d", serverBind, i)
+				if err := buildCloudConfig(i, name, metadataURL); err != nil {
+					logrus.WithError(err).WithField("vm", name).Error("error building cloudconfig for vm")
+					continue
+				}
+
+				wg.Add(1)
+				if err := mkvm(conn, metadataURL, name); err != nil {
+					logrus.WithError(err).Error("unexpected error building VM")
+					wg.Done()
+				}
+			}
+
+			logrus.Info("waiting for VM(s) to finish provisioning")
+			wg.Wait()
+		},
+	}
+
+	argMemoryMB   int
+	argCPUs       int
+	argImage      string
+	argDiskSizeGB int
+
+	// cloudinit args
+	argSSHKeys  []string
+	argPackages []string
+)
+
+func main() {
+	if err := rootCmd.Execute(); err != nil {
+		panic(err)
+	}
+}
+
+func init() {
+	rootCmd.Flags().IntVarP(&argMemoryMB, "memory", "m", 1024, "amount of memory (in MB) to assign to the VM")
+	rootCmd.Flags().IntVarP(&argCPUs, "cpu", "c", 2, "the number of vCPU cores to assign to the VM")
+	rootCmd.Flags().StringVarP(&argImage, "image", "", "https://cloud.debian.org/images/cloud/bookworm/latest/debian-12-generic-amd64.qcow2", "URL of the image to download")
+	rootCmd.Flags().IntVarP(&argDiskSizeGB, "disk", "d", 25, "disk size (in GB)")
+
+	rootCmd.Flags().StringArrayVarP(&argSSHKeys, "ssh-keys", "s", nil, "SSH key(s) authorzed to access the VM")
+	rootCmd.Flags().StringArrayVarP(&argPackages, "packages", "p", nil, "packages to install on the VM")
+}