kygui/include/client.hpp

#ifndef CLIENT_HPP
#define CLIENT_HPP

#include <QDialog>
#include <QComboBox>
#include <QPushButton>
#include <QMessageBox>
#include <QLineEdit>
#include <QUuid>
#include <QLabel>
#include <QString>
#include <QVector>
#include <QQueue>
#include <QThread>
#include <QMetaType>
#include <thread>
#include <string>
#include <utility>
#include <headers/util.hpp>

static constexpr int MESSAGE_UPDATE_TYPE = 1;
static constexpr int COMMANDS_UPDATE_TYPE = 2;
static constexpr int EVENT_UPDATE_TYPE = 3;
static constexpr int PROCESS_REQUEST_TYPE = 4;

enum TaskType {
    INSTAGRAM = 1,
    OTHER = 2
};

namespace TaskCode {
static constexpr int IGTASKBYTE = 0xFF;
static constexpr int GENMSGBYTE = 0xFE;
}

typedef std::map<int, std::string> CommandMap;
typedef std::map<int, std::vector<std::string>> CommandArgMap;
typedef QVector<QString> StringVec;

struct SentFile {
    int timestamp;
    QString name;
    FileType type;
};

Q_DECLARE_METATYPE(StringVec)
Q_DECLARE_METATYPE(QVector<QByteArray>);

class Client : public QDialog
{
    Q_OBJECT
    QThread workerThread;

public:

    class MessageHandler {
    public:
        MessageHandler(std::function<void()> cb) : m_cb(cb) {}

        void operator()() { m_cb(); }

    private:
        std::function<void()> m_cb;
    };

    Client(QWidget *parent = nullptr);
    Client(QWidget *parent, int count, char** arguments);
    ~Client();
    void start();
    void closeConnection();
    void execute();
    QString getAppName(int mask);
    int getSelectedApp();
    // Move this to private after moving responsibilities to Client
    void scheduleTask(std::vector<std::string> task_args, bool file_pending);
    MessageHandler createMessageHandler(std::function<void()> cb);

public slots:
    void sendMessage(const QString& s);
    void setSelectedApp(std::vector<QString> app_names);
    void sendFiles(QVector<KFileData> files);

signals:
    void messageReceived(int t, QString s, QVector<QString> args);
    void eventReceived(int t, std::string event, StringVec args);

private:
    void sendEncoded(std::string message);
    void sendFileEncoded(QByteArray bytes);
    void sendTaskEncoded(TaskType type, std::vector<std::string> args);
    void processFileQueue();
    void handleMessages();
    void sendPackets(uint8_t* data, int size);
    int argc;
    char** argv;
    int m_client_socket_fd;
    std::vector<std::string> m_task;
    bool executing;
    CommandMap m_commands;
    CommandArgMap m_command_arg_map;
    std::vector<int> selected_commands;
    QQueue<KFileData> outgoing_files;
    std::vector<SentFile> sent_files;
};
#endif // CLIENT_HPP