| // 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. |
| |
| #ifndef RMAD_STATE_HANDLER_BASE_STATE_HANDLER_H_ |
| #define RMAD_STATE_HANDLER_BASE_STATE_HANDLER_H_ |
| |
| #include <base/memory/ref_counted.h> |
| |
| #include "rmad/proto_bindings/rmad.pb.h" |
| #include "rmad/utils/json_store.h" |
| |
| namespace rmad { |
| |
| class BaseStateHandler : public base::RefCounted<BaseStateHandler> { |
| public: |
| explicit BaseStateHandler(JsonStore* json_store); |
| virtual ~BaseStateHandler() = default; |
| |
| // Returns the RmadState that the class handles. This can be declared by the |
| // macro ASSIGN_STATE(state). |
| virtual RmadState GetState() const = 0; |
| |
| // Returns the next RmadState in the RMA flow depending on device status and |
| // user input (e.g. json_store_ content). |
| virtual RmadState GetNextState() const = 0; |
| |
| protected: |
| JsonStore* json_store_; |
| }; |
| |
| #define ASSIGN_STATE(state) \ |
| RmadState GetState() const override { return state; } |
| |
| } // namespace rmad |
| |
| #endif // RMAD_STATE_HANDLER_BASE_STATE_HANDLER_H_ |