virtigo/doDroplet.go

// Copyright 2017-2025 WIT.COM Inc. All rights reserved.
// Use of this source code is governed by the GPL 3.0

package main

// An app to submit patches for the 30 GO GUI repos

import (
	"fmt"
	"math/rand"
	"net/http"
	"net/url"
	"path/filepath"
	"strings"
	"time"

	"github.com/google/uuid"
	"go.wit.com/lib/gui/shell"
	"go.wit.com/lib/protobuf/virtpb"
	"go.wit.com/log"
)

func doDroplet() (string, error) {
	err := me.clusters.ConfigLoad()
	if err != nil {
		return "", err
	}

	msg := []byte(`{"message": "Hello"}`)

	// Initialize a persistent client with a custom Transport
	client = &http.Client{
		Transport: &http.Transport{
			DisableKeepAlives: false, // Ensure Keep-Alive is enabled
		},
		Timeout: 10 * time.Second, // Set a reasonable timeout
	}

	me.cmap = make(map[*virtpb.Cluster]*adminT)
	for c := range me.clusters.IterAll() {
		var err error
		admin := new(adminT)
		if admin.cluster == nil {
			admin.cluster = new(virtpb.Cluster)
		}
		me.cmap[c] = admin
		log.Info("found in the config file", c.URL[0])
		// a.makeClusterGroup(c)
		admin.url, err = url.Parse(c.URL[0])
		if err != nil {
			return "", err
		}

		// update the droplet list
		if data, err := postData(admin.url.String()+"/DropletsPB", msg); err != nil {
			log.Info("/DropletsPB Error:", err)
			continue
		} else {
			admin.cluster.Droplets = new(virtpb.Droplets)
			if err := admin.cluster.Droplets.Unmarshal(data); err != nil {
				log.Printf("DropletsPB Response len:%d\n", len(data))
				log.Println("droplets marshal failed", err)
				continue
			}
		}
		log.Printf("Cluster Name: %s\n", c.Name)
		log.Printf("Number of Droplets: %d\n", admin.cluster.Droplets.Len())

		if argv.Droplet.Name == "" {
			return "", fmt.Errorf("--name droplet name was empty")
		}

		var found *virtpb.Droplets
		found = virtpb.NewDroplets()
		all := admin.cluster.Droplets.All()
		for all.Scan() {
			vm := all.Next()
			if argv.Droplet.Name == vm.Hostname {
				if argv.Droplet.Show != nil {
					log.Info(vm.SprintHeader())
					txt := vm.FormatTEXT()
					log.Info(txt)
					return "droplet status", nil
				}
				if argv.Droplet.Start != nil {
					log.Info("should start droplet here")
					log.Info(vm.SprintHeader())
					e := new(virtpb.Event)
					e.Etype = virtpb.EventType_POWERON
					e.DropletUuid = vm.Uuid

					if err := admin.postEvent(e); err != nil {
						return "droplet start err", err
					}
					return "droplet start", nil
				}
				return "droplet found", fmt.Errorf("do what to the droplet?")
			}
			found.Append(vm)
		}
		log.Println("On Droplet count=", found.Len())
	}
	return "", fmt.Errorf("droplet %s not found", argv.Droplet.Name)
}

func doEvent(e *virtpb.Event) *virtpb.Event {
	result := new(virtpb.Event)
	if e.Etype == virtpb.EventType_POWERON {
		log.Println("power on droplet on local cluster here", e.DropletUuid)
		result.State = virtpb.Event_DONE
		rs, err := Start(e.DropletUuid)
		log.Println("Start() returned", rs)
		log.Println("Start() returned err", err)
		if err != nil {
			result.Error = fmt.Sprintf("%v", err)
		}
		return result
	}

	if e.Etype == virtpb.EventType_EDIT {
		log.Println("edit event", e.DropletUuid)
		result.State = virtpb.Event_DONE
		if e.Droplet != nil {
			return updateDroplet(e.Droplet)
		}
		log.Println("unknown edit event")
		result.State = virtpb.Event_FAIL
		return result
	}

	if e.Etype == virtpb.EventType_ADD {
		log.Println("START ADD droplet event", e.Droplet.FormatTEXT())
		if e.Droplet == nil {
			result.State = virtpb.Event_FAIL
			return result
		}
		result.DropletName = e.Droplet.Hostname
		result.Error = e.Droplet.FormatTEXT() // feedback to the other side for debugging

		// attempt to create the new droplet
		if err := createDroplet(e.Droplet, result); err != nil {
			result.Error += fmt.Sprintf("createDroplet() err: %v", err)
			result.State = virtpb.Event_FAIL
			return result
		}
		log.Println("create droplet worked", e.Droplet.FormatTEXT())
		result.State = virtpb.Event_DONE
		return result
	}

	log.Println("unknown event", e)
	result.Etype = e.Etype
	result.State = virtpb.Event_FAIL
	return result
}

func updateDroplet(newd *virtpb.Droplet) *virtpb.Event {
	var changed bool = false
	result := new(virtpb.Event)

	if newd == nil {
		result.Error = "updateDroplet() d == nil"
		result.State = virtpb.Event_FAIL
		return result
	}

	d := me.cluster.FindDropletByUuid(newd.Uuid)
	if d == nil {
		result.Error = "updateDroplet() could not find uuid"
		result.State = virtpb.Event_FAIL
		return result
	}
	log.Println("found droplet to update:", newd.Uuid, newd.Hostname, newd.Cpus, newd.Memory)

	if d.Hostname != newd.Hostname && newd.Hostname != "" {
		d.Hostname = newd.Hostname
		changed = true
	}

	if d.Cpus != newd.Cpus && newd.Cpus > 0 {
		d.Cpus = newd.Cpus
		changed = true
	}

	// arbitrary check. don't make vm's with less than 64 MB of RAM
	// big enough most things will load with some stdout
	if d.Memory != newd.Memory && newd.Memory > (64*1024*1024) {
		d.Memory = newd.Memory
		changed = true
	}

	if changed {
		if err := me.cluster.ConfigSave(); err != nil {
			log.Info("configsave error", err)
			result.Error = fmt.Sprintf("%v", err)
			result.State = virtpb.Event_FAIL
			return result
		}
	} else {
		log.Println("nothing changed in", newd.Uuid, newd.Hostname)
	}

	result.State = virtpb.Event_DONE
	return result
}

func createDroplet(newd *virtpb.Droplet, result *virtpb.Event) error {
	if newd == nil {
		return fmt.Errorf("droplet protobuf == nil")
	}

	if newd.Uuid == "" {
		newd.Uuid = uuid.New().String()
	}

	d := me.cluster.FindDropletByUuid(newd.Uuid)
	if d != nil {
		return fmt.Errorf("droplet uuid already used")
	}

	log.Println("found droplet to update:", newd.Uuid, newd.Hostname, newd.Cpus, newd.Memory)

	if newd.Hostname == "" {
		return fmt.Errorf("Hostname can not be blank")
	}
	d = me.cluster.FindDropletByName(newd.Hostname)
	if d != nil {
		return fmt.Errorf("hostname already defined")
	}

	// by default, on locally imported domains, set the preferred hypervisor!
	newd.LocalOnly = "yes on: " + "farm03"

	newd.PreferredHypervisor = "farm03"
	newd.StartState = virtpb.DropletState_OFF

	newd.Current = new(virtpb.Current)
	newd.Current.State = virtpb.DropletState_OFF

	// create the network
	if err := createNetwork(newd); err != nil {
		return err
	}

	// create the disks
	if err := createDisks(newd); err != nil {
		return err
	}

	// append the protobuf and save it
	me.cluster.AddDroplet(newd)
	if err := me.cluster.ConfigSave(); err != nil {
		log.Info("configsave error", err)
		return fmt.Errorf("ConfigSave() error: %v", err)
	}

	return nil
}

func findDisks(d *virtpb.Droplet) error {
	log.Info("need to do this")
	return nil
}

func createDisks(d *virtpb.Droplet) error {
	if d.Disks != nil {
		return findDisks(d)
	}

	newdisk := new(virtpb.Disk)
	newdisk.Filename = d.Hostname + ".qcow2"
	newdisk.Filepath = "/home/nfs2"
	d.Disks = append(d.Disks, newdisk)

	basefile := "/home/nfs2/base2025.wit-5.qcow2"
	newfile := filepath.Join(newdisk.Filepath, newdisk.Filename)

	if !shell.Exists(newdisk.Filepath) {
		return fmt.Errorf("disk image path missing: %s", newdisk.Filepath)
	}

	if !shell.Exists(basefile) {
		return fmt.Errorf("basefile %s missing", basefile)
	}

	if shell.Exists(newfile) {
		return fmt.Errorf("disk image already exists: %s", newfile)
	}

	cmd := []string{"dd", "bs=100M", "status=progress", "oflag=dsync", "if=" + basefile, "of=" + newfile}
	result := shell.RunRealtime(cmd)
	if result.Exit != 0 {
		return fmt.Errorf("dd to %s failed %d\n%s\n%s", newfile, result.Exit, strings.Join(result.Stdout, "\n"), strings.Join(result.Stderr, "\n"))
	}

	return nil
}

func createNetwork(d *virtpb.Droplet) error {
	if d.Networks != nil {
		// network already done
		return nil
	}
	if len(d.Networks) > 0 {
		// network already done
		return nil
	}

	n := new(virtpb.Network)
	n.Mac = getNewMac()
	n.Name = "worldbr"
	d.Networks = append(d.Networks, n)

	return nil
}

func getNewMac() string {
	// mac address map to check for duplicates
	var macs map[string]string
	macs = make(map[string]string)

	loop := me.cluster.DropletsAll() // get the list of droplets
	for loop.Scan() {
		d := loop.Next()
		for _, n := range d.Networks {
			// log.Println("network:", n.Mac, d.Uuid, d.Hostname)
			if _, ok := macs[n.Mac]; ok {
				// UUID already exists
				log.Info("duplicate MAC", n.Mac, macs[n.Mac])
				log.Info("duplicate MAC", n.Mac, d.Hostname)
				return ""
			}
			macs[n.Mac] = d.Hostname
		}
	}

	return generateMAC(macs)
}

func generateMAC(macs map[string]string) string {
	prefix := []byte{0x22, 0x22, 0x22}
	for {
		// Generate last 3 bytes randomly
		suffix := make([]byte, 3)
		if _, err := rand.Read(suffix); err != nil {
			log.Fatalf("Failed to generate random bytes: %v", err)
		}

		// Format full MAC address
		mac := fmt.Sprintf("%02x:%02x:%02x:%02x:%02x:%02x",
			prefix[0], prefix[1], prefix[2],
			suffix[0], suffix[1], suffix[2])

		// Check if MAC is already used
		if _, exists := macs[mac]; !exists {
			log.Println("Using new MAC:", mac)
			return mac
		}
		log.Println("MAC already defined:", mac)
	}
}