MessageRecorder.hpp

Go to the documentation of this file.

#ifndef HOPS_MESSAGERECORDER_HPP
#define HOPS_MESSAGERECORDER_HPP
 
#include "../../FileWriter/FileWriter.hpp"
#include "IsClearRecordsAvailable.hpp"
#include "IsInstallDataObjectAvailable.hpp"
#include "IsStoreRecordAvailable.hpp"
#include "IsWriteRecordsToFileAvailable.hpp"
#include "../../Utility/Data.hpp"
 
#include <string>
#include <vector>
 
namespace hops {
    template<typename MarkovChainImpl>
    class MessageRecorder : public MarkovChainImpl {
    public:
        explicit MessageRecorder(const MarkovChainImpl &markovChainImpl) : MarkovChainImpl(markovChainImpl) {}
 
        void installDataObject(ChainData& chainData) {
            if constexpr(IsInstallDataObjectAvailable<MarkovChainImpl>::value) {
                MarkovChainImpl::installDataObject(chainData);
            }
        }
 
        void writeRecordsToFile(const FileWriter *const fileWriter) const {
            fileWriter->write("messages", records);
            if constexpr(IsWriteRecordsToFileAvailable<MarkovChainImpl>::value) {
                MarkovChainImpl::writeRecordsToFile(fileWriter);
            }
        };
 
        void storeRecord() {
            records.emplace_back(latestMessage);
            latestMessage = "";
            if constexpr(IsStoreRecordAvailable<MarkovChainImpl>::value) {
                MarkovChainImpl::storeRecord();
            }
        }
 
        void addMessage(const std::string &message) {
            latestMessage += message;
        }
 
        void clearRecords() {
            records.clear();
        }
 
    private:
        std::vector<std::string> records;
        std::string latestMessage;
    };
}
 
#endif //HOPS_MESSAGERECORDER_HPP