// Copyright 2021 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.

// Second step of CrOSInstall State Machine. Responsible for stateful provisioning
package crosservice

import (
	"chromiumos/test/provision/cmd/provisionserver/bootstrap/services"
	"context"
	"fmt"
)

type CrOSPostInstallState struct {
	service CrOSService
}

func (s CrOSPostInstallState) Execute(ctx context.Context) error {
	if !s.service.preserverStateful {
		if err := s.service.WipeStateful(ctx); err != nil {
			return fmt.Errorf("failed to wipe stateful, %s", err)
		}

		if err := s.service.connection.Restart(ctx); err != nil {
			return fmt.Errorf("failed to restart dut, %s", err)
		}
	}

	if err := s.service.ProvisionStateful(ctx); err != nil {
		return fmt.Errorf("failed to provision stateful, %s", err)
	}

	if err := s.service.connection.Restart(ctx); err != nil {
		return fmt.Errorf("failed to restart dut, %s", err)
	}

	return nil
}

func (s CrOSPostInstallState) Next() services.ServiceState {
	return CrOSVerifyState{
		service: s.service,
	}
}

func (s CrOSPostInstallState) Name() string {
	return "CrOS Post-Install"
}