package main import "net/http" import "encoding/json" import "io/ioutil" import "log" import "github.com/davecgh/go-spew/spew" // import "github.com/golang/protobuf/proto" import pb "git.wit.com/wit/witProtobuf" const ( API = "http://stackapi:4000/" MAX_NETS = 10 MAX_VMS = 10 ) type network struct { Subdomain string `json:"subdomain"` } func fetchNetworks(token string) []network { var networks []network req, err := http.NewRequest("GET", API + "clusters", nil) if err != nil { log.Fatal("can't make new req") } req.Header.Set("X-Wit-Auth", token) client := &http.Client{} resp, err := client.Do(req) if err != nil { log.Printf("Can't exec the req to list networks: %s", err) } body, err := ioutil.ReadAll(resp.Body) if err != nil { log.Fatal("can't read resp") } log.Printf("NETWORKS =", body) if body == nil { log.Println("networks is nil") } else { len := len(body) err = json.Unmarshal(body, &networks) if err != nil { log.Println("body = ", body) log.Println("body len = ", len) log.Println("can't unmarshal resp for list networks: %s", err) } } spew.Dump(networks) return networks } /* { "name": "vm10", "cpu": 4, "ram": 2048, "created_on": "2019-03-18T19:22:10.755428Z", "scope": "private", "role": "k8s-node", "stage": "Creating...", "seconds_left": 300, "base_image": "debcore_guest_latest", "address_list": [ "192.168.0.1/32" ] }, */ type address struct { Address string } type vm struct { Name string `json:"name"` Role string `json:"role"` BaseImage string `json:"base_image"` } func fetchVMs(token string, network string) []vm { var vms []vm req, err := http.NewRequest("GET", API + "vms/" + network, nil) if err != nil { log.Fatal("can't make new req") } req.Header.Set("X-Wit-Auth", token) client := &http.Client{} resp, err := client.Do(req) if err != nil { log.Printf("Can't exec the req to fetch vms for %s: %s", network, err) } body, err := ioutil.ReadAll(resp.Body) if err != nil { log.Fatal("can't read resp") } err = json.Unmarshal(body, &vms) if err != nil { log.Fatal("can't unmarshal resp for fetch vms: ", err, "\n\n", network, string(body)) } spew.Dump(vms) return vms } func processGetEvent(ge *pb.Event) *pb.Event { log.Println("processGetEvent() START") log.Println("processGetEvent() ge.Token =", ge.Token) e := pb.Event{} e.Name = "stablesid.org" networks := fetchNetworks(ge.Token) for i, network := range(networks) { if i >= MAX_NETS { break } log.Println("network.Subdomain =",network.Subdomain) log.Println("network=",network) enetwork := pb.Event_Network{} enetwork.Name = network.Subdomain e.Networks = append(e.Networks, &enetwork) vms := fetchVMs(ge.Token, network.Subdomain) for j, vm := range(vms) { if j >= MAX_VMS { break } evm := pb.Event_VM{} evm.Name = vm.Name evm.Cpus = 1 evm.Memory = 512 evm.Disk = 20 evm.Hostname = vm.Name + "." + enetwork.Name + ".lab.wit.com" e.Vms = append(e.Vms, &evm) log.Println("\tvm.Name =",vm.Name) log.Println("\tvm =",vm) } } return &e }