go/src/infra/libs/skylab/autotest/hostinfo/convert.go

// Copyright 2019 The Chromium OS Authors. All rights reserved.
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.

package hostinfo

import (
	"infra/libs/skylab/inventory"
	"infra/libs/skylab/inventory/autotest/labels"
)

// ConvertDut converts the inventory DUT struct to Autotest hostinfo.
func ConvertDut(d *inventory.DeviceUnderTest) *HostInfo {
	var hi HostInfo
	convertDutAttributes(&hi, d)
	convertDutLabels(&hi, d)
	return &hi
}

func convertDutAttributes(hi *HostInfo, d *inventory.DeviceUnderTest) {
	if hi.Attributes == nil {
		hi.Attributes = make(map[string]string)
	}
	for _, a := range d.GetCommon().GetAttributes() {
		hi.Attributes[a.GetKey()] = a.GetValue()
	}
}

func convertDutLabels(hi *HostInfo, d *inventory.DeviceUnderTest) {
	sl := d.GetCommon().GetLabels()
	hi.Labels = labels.Convert(sl)
}

// RevertDut reverts the Autotest hostinfo back to an inventory DUT struct.
func RevertDut(d *inventory.DeviceUnderTest, hi *HostInfo) {
	revertDutAttributes(d, hi)
	revertDutLabels(d, hi)
}

func revertDutAttributes(d *inventory.DeviceUnderTest, hi *HostInfo) {
	c := d.GetCommon()
	c.Attributes = nil
	for k, v := range hi.Attributes {
		k, v := k, v
		c.Attributes = append(c.Attributes, &inventory.KeyValue{Key: &k, Value: &v})
	}
}

func revertDutLabels(d *inventory.DeviceUnderTest, hi *HostInfo) {
	c := d.GetCommon()
	c.Labels = labels.Revert(hi.Labels)
}