#ifndef _TR_TEST_SESSION_CONTEXT_HPP_
#define _TR_TEST_SESSION_CONTEXT_HPP_

#include <sstream>
#include <vector>
#include <TorrentContext.hpp>

class TestSessionSettings {
public:
  int download_rate_limit = 1;
  int upload_rate_limit = 2;
  int listen_interfaces = 3;
  int enable_dht = 4;
  bool enable_dht_ = true;
  int download_rate_limit_ = 0;
  int upload_rate_limit_ = 0;
  std::string listen_interfaces_ = "0.0.0.0:0";
  void set_bool(int type, int value);
  void set_int(int type, int value);
  void set_str(int type, std::string value);
  int get_int(int type) const;
  bool get_bool(int type);
  std::string get_str(int type);
 
};

class TestSession {
public:
  bool valid = false;
  bool paused = false;
  TestSessionSettings settings_;
  bool is_valid();
  TestSessionSettings get_settings();
  bool is_paused() const;
  void pause();
  void resume();
  int listen_port();
  bool is_dht_running() const;
  std::vector<TestTorrent> get_torrents() const;
  void apply_settings(TestSessionSettings settings);
};

#endif