virtigo/main.go

// Copyright 2024 WIT.COM Inc Licensed GPL 3.0

package main

import (
	"embed"
	"os"
	"path/filepath"
	"time"

	"github.com/google/uuid"
	"go.wit.com/dev/alexflint/arg"
	pb "go.wit.com/lib/protobuf/virtbuf"
	"go.wit.com/log"
	"go.wit.com/lib/virtigoxml"
)

var Version string

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

func main() {
	if os.Getenv("VIRTIGO_HOME") == "" {
		homeDir, _ := os.UserHomeDir()
		fullpath := filepath.Join(homeDir, ".config/virtigo")
		os.Setenv("VIRTIGO_HOME", fullpath)
	}
	var pp *arg.Parser
	pp = arg.MustParse(&argv)

	if pp == nil {
		pp.WriteHelp(os.Stdout)
		os.Exit(0)
	}

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

	// set defaults
	me.unstable = time.Now()   // initialize the grid as unstable
	me.delay = 5 * time.Second // how often to poll the hypervisors
	me.changed = false
	me.events = new(pb.Events)
	u := uuid.New()
	me.events.Uuid = u.String()
	me.events.Version = "dirty v1"

	err := cfgfile()
	if err != nil {
		log.Warn("reading config file failed", err)
		os.Exit(-1)
	}

	var newEvents []*pb.Event

	// sanity check the droplets
	checkDroplets(false)
	newe := checkDiskFilenames()
	for _, e := range newe {
		newEvents = append(newEvents, e)
	}
	checkUniqueFilenames()

	for _, filename := range argv.Xml {
		domcfg, err := virtigoxml.ReadXml(filename)
		if err != nil {
			// parsing the libvirt xml file failed
			log.Info("error:", filename, err)
			log.Info("readXml() error", filename)
			log.Info("readXml() error", err)
			log.Info("libvirt XML will have to be fixed by hand")
			os.Exit(-1)
		}
		// this is a new droplet. add it to the cluster
		log.Info("Add XML Droplet here", domcfg.Name)
		_, newe, err := addDomainDroplet(domcfg)
		if err != nil {
			log.Info("addDomainDroplet() error", filename)
			log.Info("addDomainDroplet() error", err)
			log.Info("libvirt XML will have to be fixed by hand")
			os.Exit(-1)
		}
		for _, e := range newe {
			newEvents = append(newEvents, e)
		}
	}
	for i, e := range newEvents {
		log.Info(i, "Event:", e.Droplet, e.FieldName, "orig:", e.OrigVal, "new:", e.NewVal)
		me.changed = true
	}
	if me.changed {
		if argv.Save {
			writeConfigFile()
			writeConfigFileDroplets()
			log.Info("XML changes saved in protobuf config")
			os.Exit(0)
		} else {
			log.Info("Not saving changes (use --save to save)")
			os.Exit(0)
		}
	}
	if len(argv.Xml) != 0 {
		log.Info("No XML changes found")
		os.Exit(0)
	}

	if argv.Start != "" {
		newStart(argv.Start)
		os.Exit(0)
	}

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

	// sit here
	startHTTP()
}