FlatBufferClient.h

#pragma once

// util
#include <utils/Logger.h>
#include <utils/Image.h>
#include <utils/ColorRgb.h>
#include <utils/Components.h>

// flatbuffer FBS
#include "hyperion_reply_generated.h"
#include "hyperion_request_generated.h"

class QTcpSocket;
class QTimer;
class Hyperion;

namespace flatbuf {
class HyperionRequest;
}

class FlatBufferClient : public QObject
{
    Q_OBJECT
public:
    explicit FlatBufferClient(QTcpSocket* socket, const int &timeout, QObject *parent = nullptr);

signals:
    void registerGlobalInput(const int priority, const hyperion::Components& component, const QString& origin = "System", const QString& owner = "", unsigned smooth_cfg = 0);

    const bool setGlobalInputImage(const int priority, const Image<ColorRgb>& image, const int timeout_ms = -1);

    void clientDisconnected();

public slots:
    void forceClose();

private slots:
    void readyRead();

    void disconnected();

private:
    void handleMessage(const hyperionnet::Request * req);

    void handleRegisterCommand(const hyperionnet::Register *regReq);

    void handleColorCommand(const hyperionnet::Color *colorReq);

    void handleImageCommand(const hyperionnet::Image *image);

    void handleClearCommand(const hyperionnet::Clear *clear);

    void handleNotImplemented();

    void sendMessage();

    void sendSuccessReply();

    void sendErrorReply(const std::string & error);

private:
    Logger *_log;
    QTcpSocket *_socket;
    const QString _clientAddress;
    QTimer *_timeoutTimer;
    int _timeout;
    int _priority;
    Hyperion* _hyperion;

    QByteArray _receiveBuffer;

    // Flatbuffers builder
    flatbuffers::FlatBufferBuilder _builder;
};