#include "aare.hpp"
#include "aare/examples/defs.hpp"
#include "zmq.h"
#include <cassert>
#include <fmt/core.h>
#include <string>
using namespace aare;
using namespace std;

string setup(int argc, char **argv) {
    logger::set_verbosity(logger::DEBUG);

    ArgParser parser("Zmq task ventilator");
    parser.add_option("port", "p", true, false, "5555", "port number");
    auto args = parser.parse(argc, argv);
    int port = std::stoi(args["port"]);

    return "tcp://127.0.0.1:" + to_string(port);
}

int process(const std::string &endpoint) {
    // 0. connect to slsReceiver
    ZmqSocketReceiver receiver(endpoint, ZMQ_SUB);
    receiver.connect();

    // 1. create ventilator
    ZmqVentilator ventilator("tcp://*:4321");

    while (true) {
        // 2. receive frame from slsReceiver
        ZmqFrame zframe = receiver.receive_zmqframe();
        if (zframe.header.data == 0)
            continue;
        logger::info("Received frame, frame_number=", zframe.header.frameNumber);
        logger::info(zframe.header.to_string());

        // 3. create task
        Task *task = Task::init(zframe.frame.data(), zframe.frame.bytes());
        task->opcode = (size_t)Task::Operation::PEDESTAL;
        task->id = zframe.header.frameNumber;

        // 4. push task to ventilator
        ventilator.push(task);
        Task::destroy(task);
    }
}
int main(int argc, char **argv) {
    // read the command line arguments
    string endpoint = setup(argc, argv);
    int ret = process(endpoint);
    return ret;
}