go/src/infra/cmd/skylab_swarming_worker/side_effects.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 main

import (
	"fmt"
	"io"
	"strings"

	"github.com/golang/protobuf/jsonpb"

	"go.chromium.org/chromiumos/infra/proto/go/test_platform/side_effects"
	"go.chromium.org/luci/common/errors"

	"infra/cmd/skylab_swarming_worker/internal/annotations"
	"infra/libs/skylab/sideeffects"
)

// parseSideEffectsConfig parses and validates a side_effects.Config JSONpb
// string.
func parseSideEffectsConfig(content string, logdogOutput io.Writer) (sec *side_effects.Config, err error) {
	annotations.SeedStep(logdogOutput, "Parse side_effects.Config")
	annotations.StepCursor(logdogOutput, "Parse side_effects.Config")
	annotations.StepStarted(logdogOutput)
	defer annotations.StepClosed(logdogOutput)

	defer func() {
		if err != nil {
			annotations.StepException(logdogOutput)
			fmt.Fprint(logdogOutput, err.Error())
		}
	}()

	fmt.Fprintf(logdogOutput, "Validating side_effects.Config JSONpb:\n%s\n", content)

	var c side_effects.Config
	u := jsonpb.Unmarshaler{AllowUnknownFields: true}
	if err := u.Unmarshal(strings.NewReader(content), &c); err != nil {
		return nil, errors.Annotate(err, "parse side_effects.Config").Err()
	}
	if err := sideeffects.ValidateConfig(&c); err != nil {
		return nil, errors.Annotate(err, "parse side_effects.Config").Err()
	}
	return &c, nil
}

// dropSideEffectsConfig write a side_effects.Config to
// <dir>/side_effects_config.json as a JSONpb.
func dropSideEffectsConfig(sec *side_effects.Config, dir string, logdogOutput io.Writer) (err error) {
	annotations.SeedStep(logdogOutput, "Write side_effects_config.json")
	annotations.StepCursor(logdogOutput, "Write side_effects_config.json")
	annotations.StepStarted(logdogOutput)
	defer annotations.StepClosed(logdogOutput)

	defer func() {
		if err != nil {
			annotations.StepException(logdogOutput)
			fmt.Fprint(logdogOutput, err.Error())
		}
	}()

	fmt.Fprintf(logdogOutput, "Writing side_effects.Config to %s", dir)
	if err := sideeffects.WriteConfigToDisk(dir, sec); err != nil {
		return errors.Annotate(err, "drop side_effects.Config").Err()
	}
	return nil
}