NetworkReader.hpp

#pragma once
 
#include "Array.hpp"
#include "ArraySegment.hpp"
#include "Platform.hpp"
#include <string>
#include <memory>
 
namespace KapMirror {
    class NetworkReader {
      private:
        byte* buffer = nullptr;
        int position;
 
      public:
        explicit NetworkReader(byte* _buffer) : buffer(_buffer) { position = 0; }
 
        explicit NetworkReader(const ArraySegment<byte>& segment) {
            buffer = segment.toArray();
            position = 0;
        }
 
        explicit NetworkReader(const std::shared_ptr<ArraySegment<byte>>& segment) {
            buffer = segment->toArray();
            position = 0;
        }
 
        ~NetworkReader() = default;
 
        inline void reset() { position = 0; }
 
        template <typename T>
        inline T read() {
            T value = *(T*)(buffer + position);
            position += sizeof(T);
            return value;
        }
 
        byte* readBytes(int count) {
            byte* array = new byte[count];
            std::copy(buffer + position, buffer + position + count, array);
            position += count;
            return array;
        }
 
        std::string readString() {
            auto length = read<short>();
            byte* array = readBytes(length);
            std::string str((char*)array, length);
            delete[] array;
            return str;
        }
    };
} // namespace KapMirror