/*
 * Copyright (c) Meta Platforms, Inc. and affiliates.
 *
 * This source code is licensed under the MIT license found in the
 * LICENSE file in the root directory of this source tree.
 */

#pragma once

#include <map>
#include <memory>
#include <mutex>
#include <sstream>
#include <string>
#include <vector>

class FlipperStep;
class FlipperStateUpdateListener;

namespace facebook {
namespace flipper {

enum State { success, in_progress, failed };

class StateElement {
 public:
  StateElement(std::string name, State state) : name_(name), state_(state){};
  std::string name_;
  State state_;
};

} // namespace flipper
} // namespace facebook

class FlipperState {
  friend FlipperStep;

 public:
  FlipperState();
  // Update listeners are responsible for their own
  // synchronization. There is no guarantee about which thread they
  // may be called on.
  void setUpdateListener(std::shared_ptr<FlipperStateUpdateListener>);
  std::string getState();
  std::vector<facebook::flipper::StateElement> getStateElements();

  /* To record a state update, call start() with the name of the step to get a
   FlipperStep object. Call complete on this to register it successful,
   the absense of the completion call when it is destructed will register as a
   step failure. */
  std::shared_ptr<FlipperStep> start(std::string step);

 private:
  void success(std::string);
  void failed(std::string, std::string);
  void started(std::string);
  void ensureLogsCapacity();

  std::mutex mutex; // Protects all our member variables.
  std::shared_ptr<FlipperStateUpdateListener> mListener = nullptr;
  std::stringstream logs;
  std::vector<std::string> insertOrder;
  std::map<std::string, facebook::flipper::State> stateMap;
};