skolo/go/powercycle-cli/autofix.go

package main

import (
	"encoding/json"
	"io/ioutil"
	"os"

	"go.skia.org/infra/go/httputils"
	"go.skia.org/infra/go/skerr"
	"go.skia.org/infra/go/util"
	"go.skia.org/infra/power/go/gatherer"
)

// getAutoFixCandidates polls the given url (e.g. power.skia.org) for a
// list of down bots and devices. The list will be filtered to only
// those that match this hostname and are not silenced. An error will
// be returned for any of the various steps that could go wrong.
func getAutoFixCandidates(url string) ([]string, error) {
	hostname, err := os.Hostname()
	if err != nil {
		return nil, skerr.Wrapf(err, "getting hostname")
	}

	client := httputils.NewTimeoutClient()
	resp, err := client.Get(url)
	if err != nil {
		return nil, skerr.Wrapf(err, "problem GET against %s", url)
	}
	defer util.Close(resp.Body)

	body, err := ioutil.ReadAll(resp.Body)
	if err != nil {
		return nil, skerr.Wrapf(err, "reading http response from %s", url)
	}

	return getMatchingCandidates(body, hostname)
}

type downBotsResponse struct {
	List []gatherer.DownBot `json:"list"`
}

// getMatchingCandidates parses the json returned by power-controller/main.go
// It then filters the list to only those that match this hostname and are
// not silenced.
func getMatchingCandidates(response []byte, hostname string) ([]string, error) {
	r := downBotsResponse{}
	if err := json.Unmarshal(response, &r); err != nil {
		return nil, skerr.Wrapf(err, "parsing JSON")
	}
	var rv []string
	for _, b := range r.List {
		if b.HostID == hostname && !b.Silenced {
			if b.Status == gatherer.STATUS_DEVICE_MISSING {
				rv = append(rv, b.BotID+"-device")
			} else {
				rv = append(rv, b.BotID)
			}
		}
	}

	return rv, nil
}