src/go.chromium.org/chromiumos/test/provision/v2/cros-provision/state-machine/commands/overwrite_install_command.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 commands

import (
	"go.chromium.org/chromiumos/test/provision/v2/cros-provision/service"
	"context"
	"fmt"
	"log"

	conf "go.chromium.org/chromiumos/config/go"
	"go.chromium.org/chromiumos/config/go/test/api"
)

type OverwriteInstalCommand struct {
	ctx context.Context
	cs  *service.CrOSService
}

func NewOverwriteInstalCommand(ctx context.Context, cs *service.CrOSService) *OverwriteInstalCommand {
	return &OverwriteInstalCommand{
		ctx: ctx,
		cs:  cs,
	}
}

func (c *OverwriteInstalCommand) Execute(log *log.Logger) error {
	log.Printf("Start OverwriteInstalCommand Execute")

	if c.cs.OverwritePayload == nil {
		log.Printf("skipping overwrite install, because none was specified.")
		return nil
	}
	log.Printf("OverwriteInstalCommand OverwritePayload Completed")

	if c.cs.OverwritePayload.HostType == conf.StoragePath_LOCAL || c.cs.OverwritePayload.HostType == conf.StoragePath_HOSTTYPE_UNSPECIFIED {
		return fmt.Errorf("only GS copying is implemented")
	}

	err := c.cs.Connection.PipeData(c.ctx, c.cs.OverwritePayload.GetPath(), "tar xf - -C /")
	if err != nil {
		return fmt.Errorf("failed to download and untar file, %s", err)
	}
	log.Printf("OverwriteInstalCommand OverwritePayload.GetPath() Completed")

	if err := c.cs.Connection.Restart(c.ctx); err != nil {
		return fmt.Errorf("failed to restart dut, %s", err)
	}
	log.Printf("OverwriteInstalCommand Restart Completed")
	log.Printf("InstallPartitionsCommand Success")
	return nil
}

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

func (c *OverwriteInstalCommand) GetErrorMessage() string {
	return "failed to overwrite install"
}

func (c *OverwriteInstalCommand) GetStatus() api.InstallResponse_Status {
	return api.InstallResponse_STATUS_PROVISIONING_FAILED
}