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

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

package commands

import (
	"context"
	"log"
	"strings"

	"go.chromium.org/chromiumos/test/provision/v2/android-provision/common"
	"go.chromium.org/chromiumos/test/provision/v2/android-provision/common/gsstorage"
	"go.chromium.org/chromiumos/config/go/test/api"
	"go.chromium.org/luci/common/errors"

	"go.chromium.org/chromiumos/test/provision/v2/android-provision/service"
)

type ResolveImagePathCommand struct {
	ctx context.Context
	svc *service.AndroidService
}

func NewResolveImagePathCommand(ctx context.Context, svc *service.AndroidService) *ResolveImagePathCommand {
	return &ResolveImagePathCommand{
		ctx: ctx,
		svc: svc,
	}
}

func (c *ResolveImagePathCommand) Execute(log *log.Logger) error {
	log.Printf("Start ResolveImagePathCommand Execute")
	if c.svc.OS != nil {
		if boardToBuildId := c.svc.OS.ImagePath.BoardToBuildID; boardToBuildId != nil {
			board := c.svc.DUT.Board
			buildId, ok := boardToBuildId[board]
			if !ok {
				err := errors.Reason("Android image not found for %s board", board).Err()
				log.Printf("ResolveImagePathCommand Failure: %v", err)
				return err
			}
			if buildId != c.svc.OS.BuildInfo.Id {
				// DUT has a different Android OS build. Proceeding with provision.
				c.svc.OS.ImagePath.GsPath = gsstorage.GetGsPath("", buildId, board)
			}
		} else if gsPath := c.svc.OS.ImagePath.GsPath; strings.HasPrefix(gsPath, "gs://"+common.GSImageBucketName+"/") {
			s := strings.Split(strings.TrimSuffix(gsPath, "/"), "/")
			if len(s) == 5 {
				buildId, board := s[3], s[4]
				if board != c.svc.DUT.Board {
					err := errors.Reason("Invalid provision request - %s image for %s board", board, c.svc.DUT.Board).Err()
					log.Printf("ResolveImagePathCommand Failure: %v", err)
					return err
				}
				if buildId == c.svc.OS.BuildInfo.Id {
					// DUT has the same Android OS build. Skipping provision.
					c.svc.OS.ImagePath.GsPath = ""
				}
			}
		}
	}
	log.Printf("ResolveImagePathCommand Success")
	return nil
}

func (c *ResolveImagePathCommand) Revert() error {
	return nil
}

func (c *ResolveImagePathCommand) GetErrorMessage() string {
	return "failed to resolve GS image path"
}

func (c *ResolveImagePathCommand) GetStatus() api.InstallResponse_Status {
	return api.InstallResponse_STATUS_PRE_PROVISION_SETUP_FAILED
}