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"
	"net/http"
	"net/url"
	"time"

	"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
	}

	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
}