//
// Created by MightyPork on 2017/12/23.
//

#include "platform.h"

#include <TinyFrame.h>

#include "messages.h"
#include "utils/payload_parser.h"
#include "utils/payload_builder.h"

/**
 * TF listener for the bulk write transaction
 */
static TF_Result bulkwrite_lst(TinyFrame *tf, TF_Msg *msg)
{
    BulkWrite *bulk = msg->userdata;

    // this is a final call before timeout, to clean up
    if (msg->data == NULL) {
        if (bulk->offset != bulk->len) {
            TF_Error("Bulk write ended prematurely with %d written out of %d", (int)bulk->offset, (int)bulk->len);
        }
        goto close;
    }

    assert_param(NULL != bulk);

    if (msg->type == MSG_BULK_ABORT) {
        goto close;
    }
    else if (msg->type == MSG_BULK_DATA || msg->type == MSG_BULK_END) {
        // if past len, speak up
        if (bulk->offset >= bulk->len) {
            com_respond_error(bulk->frame_id, E_PROTOCOL_BREACH);
            goto close;
        }

        bulk->write(bulk, msg->data, msg->len);

        // Return SUCCESS
        com_respond_ok(msg->frame_id);

        // advance the position pointer (can be used by the write callback)
        bulk->offset += msg->len;

        if (msg->type == MSG_BULK_END) {
            // this was the last chunk
            goto close;
        }
    }

    return TF_RENEW;

close:
    if (bulk) {
        // Ask user to free the bulk and userdata
        bulk->write(bulk, NULL, 0);
        msg->userdata = NULL;
    }
    return TF_CLOSE;
}


void bulkwrite_start(TinyFrame *tf, BulkWrite *bulk)
{
    assert_param(bulk);
    assert_param(bulk->len);
    assert_param(bulk->write);

    bulk->offset = 0;

    {
        uint8_t buf[8];
        PayloadBuilder pb = pb_start(buf, 8, NULL);
        pb_u32(&pb, bulk->len); // total expected len
        pb_u32(&pb, TF_MAX_PAYLOAD_RX); // max chunk size

        // We use userdata1 to hold a reference to the bulk transfer
        TF_Msg msg = {
                .type = MSG_BULK_WRITE_OFFER,
                .frame_id = bulk->frame_id,
                .is_response = true, // this ensures the frame_id is not re-generated
                .data = buf,
                .len = (TF_LEN) pb_length(&pb),
                .userdata = bulk,
        };

        TF_Query(tf, &msg, bulkwrite_lst, BULK_LST_TIMEOUT_MS);
    }
}