bencode_value.hpp

Go to the documentation of this file.

#pragma once
 
#include <map>
#include <memory>
#include <string>
#include <vector>
 
namespace fur::bencode {
 
enum class BencodeType { Integer, String, List, Dict };
 
class BencodeValue {
 public:
  [[nodiscard]] virtual std::string to_string() const = 0;
  [[nodiscard]] virtual BencodeType get_type() const = 0;
  virtual ~BencodeValue() = default;
};
 
class BencodeInt : public BencodeValue {
 private:
  int64_t _val;
 
 public:
  explicit BencodeInt(int64_t data);
  [[nodiscard]] std::string to_string() const override;
  [[nodiscard]] BencodeType get_type() const override;
  [[nodiscard]] int64_t value() const;
};
 
class BencodeString : public BencodeValue {
 private:
  std::string _val;
 
 public:
  explicit BencodeString(std::string data);
  [[nodiscard]] std::string to_string() const override;
  [[nodiscard]] BencodeType get_type() const override;
  [[nodiscard]] std::string& value();
};
 
class BencodeList : public BencodeValue {
 private:
  std::vector<std::unique_ptr<BencodeValue>> _val;
 
 public:
  explicit BencodeList(std::vector<std::unique_ptr<BencodeValue>> data);
  [[nodiscard]] std::string to_string() const override;
  [[nodiscard]] BencodeType get_type() const override;
  [[nodiscard]] std::vector<std::unique_ptr<BencodeValue>>& value();
};
 
class BencodeDict : public BencodeValue {
 private:
  std::map<std::string, std::unique_ptr<BencodeValue>> _val;
 
 public:
  explicit BencodeDict(
      std::map<std::string, std::unique_ptr<BencodeValue>> data);
  [[nodiscard]] std::string to_string() const override;
  [[nodiscard]] BencodeType get_type() const override;
  [[nodiscard]] std::map<std::string, std::unique_ptr<BencodeValue>>& value();
};
 
}  // namespace fur::bencode