122 lines
2.6 KiB
Go
122 lines
2.6 KiB
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"
|
|
)
|
|
|
|
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 := 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 != "" {
|
|
startDropletXml(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()
|
|
}
|