src/go.chromium.org/chromiumos/test/provision/v2/android-provision/state-machine/package_fetch_state.go

// Copyright 2022 The ChromiumOS Authors
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.

package state_machine

import (
	"context"
	"fmt"
	"log"

	"go.chromium.org/chromiumos/config/go/test/api"
	"go.chromium.org/luci/common/errors"
	"google.golang.org/protobuf/types/known/anypb"

	"go.chromium.org/chromiumos/test/provision/v2/android-provision/common"
	"go.chromium.org/chromiumos/test/provision/v2/android-provision/service"
	"go.chromium.org/chromiumos/test/provision/v2/android-provision/state-machine/commands"
	common_utils "go.chromium.org/chromiumos/test/provision/v2/common-utils"
)

type PackageFetchState struct {
	svc *service.AndroidService
}

func (s PackageFetchState) Execute(ctx context.Context, log *log.Logger) (*anypb.Any, api.InstallResponse_Status, error) {
	log.Println("State: Execute AndroidPackageFetchState")
	ctx = context.WithValue(ctx, "stage", common.PackageFetch)
	cmds := []common_utils.CommandInterface{
		commands.NewResolveCIPDPackageCommand(ctx, s.svc),
		commands.NewFetchCIPDPackageCommand(ctx, s.svc),
		commands.NewExtractZipCommand(ctx, s.svc),
		commands.NewUploadAPKToGSCommand(ctx, s.svc),
		commands.NewCopyDataCommand(ctx, s.svc),
	}
	for i, c := range cmds {
		if err := c.Execute(log); err != nil {
			log.Printf("State: Execute AndroidPackageFetchState failure %s\n", err)
			log.Println("State: Revert AndroidPackageFetchState")
			for ; i >= 0; i-- {
				if e := cmds[i].Revert(); e != nil {
					err = errors.Annotate(err, "failure while reverting %s", e).Err()
					break
				}
			}
			return nil, c.GetStatus(), fmt.Errorf("%s: %s", c.GetErrorMessage(), err)
		}
	}
	log.Println("State: AndroidPackageFetchState Completed")
	return nil, api.InstallResponse_STATUS_SUCCESS, nil
}

func (s PackageFetchState) Next() common_utils.ServiceState {
	return PackageInstallState{
		svc: s.svc,
	}
}

func (s PackageFetchState) Name() string {
	return "Android Package Fetch State"
}