virtigo/main.go

// Copyright 2024 WIT.COM Inc Licensed GPL 3.0

package main

import (
	"embed"
	"fmt"
	"os"
	"time"

	"go.wit.com/dev/alexflint/arg"
	"go.wit.com/log"
)

var Version string

//go:embed resources/*
var resources embed.FS

func main() {
	pp := arg.MustParse(&argv)

	if !argv.Uptime {
		pp.WriteHelp(os.Stdout)
		os.Exit(0)
	}

	if argv.Daemon {
		log.DaemonMode(true)
	}

	readConfigFile()
	readDropletFile("droplets")
	readHypervisorFile("hypervisor")
	writeConfigFile()

	// initialize the grid as unstable
	me.unstable = time.Now()

	log.Info("command line hypervisors:", argv.Hosts)
	for _, name := range argv.Hosts {
		h := findHypervisor(name)
		if h != nil {
			log.Info("command line hypervisor", name, "already in config file")
			continue
		}
		h = addHypervisor(name)
		h.Active = true
	}

	if argv.Start != "" {
		d := findDroplet(argv.Start)
		if d == nil {
			log.Info("droplet is unknown:", argv.Start)
			os.Exit(0)
		}
		log.Info("start droplet here:", d.Hostname)
		domcfg := makeStandardXml(d)

		fmt.Printf("Virt type %s\n", domcfg.Type)
		fmt.Printf("Virt name %s\n", domcfg.Name)
		fmt.Printf("Virt UUID %s\n", domcfg.UUID)
		fmt.Printf("Virt Memory %d %s\n", domcfg.Memory.Value, domcfg.Memory.Unit)

		// test add some ethernet devices
		macs := getMacs(domcfg)
		fmt.Printf("Virt mac addr:%s\n", macs)

		// clearEthernet(domcfg)

		addEthernet(domcfg, "04:44:33:11:22:11", "worldbr")
		addEthernet(domcfg, "04:44:33:33:44:55", "greenbr")

		// add a check here to make these unique
		// setRandomMacs(domcfg)

		// print out the final mac addresses
		macs = getMacs(domcfg)
		fmt.Printf("Virt mac addr:%s\n", macs)

		qcow := "/home/nfs/" + d.Hostname + ".qcow2"
		setSimpleDisk(domcfg, qcow)

		writeoutXml(domcfg, "blahcarr")
		os.Exit(0)
	}

	// start the watchdog polling for each hypervisor
	for _, h := range me.hypers {
		log.Info("starting polling on", h.Hostname)
		go h.NewWatchdog()
	}

	// sit here
	startHTTP()
}