/********************************************************************************
 *    Copyright (C) 2014 GSI Helmholtzzentrum fuer Schwerionenforschung GmbH    *
 *                                                                              *
 *              This software is distributed under the terms of the             *
 *              GNU Lesser General Public Licence (LGPL) version 3,             *
 *                  copied verbatim in the file "LICENSE"                       *
 ********************************************************************************/

#include "Header.h"

#include <fairmq/Device.h>
#include <fairmq/runDevice.h>

namespace bpo = boost::program_options;

struct Sink : fair::mq::Device
{
    Sink()
    {
        OnData("data", &Sink::HandleData);
    }

    bool HandleData(fair::mq::Parts& parts, int)
    {
        LOG(info) << "Received message with " << parts.Size() << " parts";

        example_multipart::Header header;
        header.stopFlag = (static_cast<example_multipart::Header*>(parts.At(0)->GetData()))->stopFlag;

        LOG(info) << "Received part 1 (header) with stopFlag: " << header.stopFlag;
        LOG(info) << "Received part 2 of size: " << parts.At(1)->GetSize();
        assert(parts.At(1)->GetSize() == 1000);
        LOG(info) << "Received part 3 of size: " << parts.At(2)->GetSize();
        assert(parts.At(2)->GetSize() == 500);
        LOG(info) << "Received part 4 of size: " << parts.At(3)->GetSize();
        assert(parts.At(3)->GetSize() == 600);
        LOG(info) << "Received part 5 of size: " << parts.At(4)->GetSize();
        assert(parts.At(4)->GetSize() == 700);
        LOG(info) << "Received part 6 of size: " << parts.At(5)->GetSize();
        assert(parts.At(5)->GetSize() == 0);
        LOG(info) << "Received part 7 of size: " << parts.At(6)->GetSize();
        assert(parts.At(6)->GetSize() == 100);

        if (header.stopFlag == 1) {
            LOG(info) << "stopFlag is 1, going IDLE";
            return false;
        }

        return true;
    }

};

void addCustomOptions(bpo::options_description& /*options*/)
{
}

std::unique_ptr<fair::mq::Device> getDevice(fair::mq::ProgOptions& /*config*/)
{
    return std::make_unique<Sink>();
}