summaryrefslogtreecommitdiff
path: root/doDroplet.go
blob: 38d99033bdc11ca32504bcdf9ecb540455231d4f (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
// 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)
	}

	return result
}