rmad/state_handler/finalize_state_handler.cc

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

#include "rmad/state_handler/finalize_state_handler.h"

#include <base/logging.h>

namespace rmad {

FinalizeStateHandler::FinalizeStateHandler(scoped_refptr<JsonStore> json_store)
    : BaseStateHandler(json_store) {}

RmadErrorCode FinalizeStateHandler::InitializeState() {
  if (!state_.has_finalize()) {
    state_.set_allocated_finalize(new FinalizeState);
  }
  return RMAD_ERROR_OK;
}

BaseStateHandler::GetNextStateCaseReply FinalizeStateHandler::GetNextStateCase(
    const RmadState& state) {
  if (!state.has_finalize()) {
    LOG(ERROR) << "RmadState missing |finalize| state.";
    return {.error = RMAD_ERROR_REQUEST_INVALID, .state_case = GetStateCase()};
  }
  const FinalizeState& finalize = state.finalize();
  if (finalize.shutdown() == FinalizeState::RMAD_FINALIZE_UNKNOWN) {
    LOG(ERROR) << "RmadState missing |shutdown| argument.";
    return {.error = RMAD_ERROR_REQUEST_ARGS_MISSING,
            .state_case = GetStateCase()};
  }

  state_ = state;

  // TODO(chenghan): This is currently fake. We should transition to
  //                 RepairCompleteState after a powerwash.
  return {.error = RMAD_ERROR_OK,
          .state_case = RmadState::StateCase::STATE_NOT_SET};
}

}  // namespace rmad