blob: b922fd7c0e5ed6d21f15ce05a6ea1afa1540229f [file] [log] [blame]
// 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/state_handler_manager.h"
#include <utility>
#include <base/check.h>
#include <base/memory/scoped_refptr.h>
#include "rmad/state_handler/welcome_screen_state_handler.h"
namespace rmad {
StateHandlerManager::StateHandlerManager(scoped_refptr<JsonStore> json_store)
: json_store_(json_store) {}
void StateHandlerManager::RegisterStateHandler(
scoped_refptr<BaseStateHandler> handler) {
RmadState state = handler->GetState();
auto res = state_handler_map_.insert(std::make_pair(state, handler));
// Check if there are StateId collision.
DCHECK(res.second) << "Registered handlers should have unique RmadStates.";
void StateHandlerManager::InitializeStateHandlers() {
scoped_refptr<BaseStateHandler> StateHandlerManager::GetStateHandler(
RmadState state) const {
auto it = state_handler_map_.find(state);
if (it == state_handler_map_.end()) {
// Unregistered RmadState, return an empty pointer.
return scoped_refptr<BaseStateHandler>(nullptr);
return it->second;
} // namespace rmad