#pragma once

#include <algorithm>
#include <memory>
#include <set>

#include "common/prefix.h"
#include "tools/cabana/streams/abstractstream.h"
#include "tools/replay/replay.h"

class ReplayStream : public AbstractStream {
public:
  ReplayStream();
  ~ReplayStream();
  void start() override { replay->start(); }
  bool loadRoute(const std::string &route, const std::string &data_dir, uint32_t replay_flags = REPLAY_FLAG_NONE, bool auto_source = false);
  bool eventFilter(const Event *event);
  void seekTo(double ts) override { replay->seekTo(std::max(double(0), ts), false); }
  bool liveStreaming() const override { return false; }
  inline std::string routeName() const override { return replay->route().name(); }
  inline std::string carFingerprint() const override { return replay->carFingerprint(); }
  double minSeconds() const override { return replay->minSeconds(); }
  double maxSeconds() const { return replay->maxSeconds(); }
  inline std::chrono::system_clock::time_point beginDateTime() const override {
    return std::chrono::system_clock::from_time_t(replay->routeDateTime());
  }
  inline uint64_t beginMonoTime() const override { return replay->routeStartNanos(); }
  inline void setSpeed(float speed) override { replay->setSpeed(speed); }
  inline float getSpeed() const { return replay->getSpeed(); }
  inline Replay *getReplay() const { return replay.get(); }
  inline bool isPaused() const override { return replay->isPaused(); }
  void pause(bool pause) override;


  Observable<std::shared_ptr<LogReader>> qLogLoaded;

private:
  void mergeSegments();
  std::unique_ptr<Replay> replay = nullptr;
  Connection settings_connection_;
  std::set<int> processed_segments;
  std::unique_ptr<OpenpilotPrefix> op_prefix;
};