#ifndef __PROTO_MSG_H__
#define __PROTO_MSG_H__

#include "wsServer/include/ws.h"
#include <stdint.h>
#include <stdlib.h>
#include "flatcc/flatcc_builder.h"

// Raw FlatBuffers frame copied from the WS thread, queued to the main loop.
typedef struct {
    uint8_t* data;
    size_t   size;
} FbMessage;

typedef struct {
    ws_cli_conn_t clientIdx;
    uint8_t*      msg;
    size_t        msgLen;
} OutgoingMessage;

// Build a ServerMessage{AuthResponse} frame.  Returns a malloc'd buffer;
// ownership is transferred to the caller (dispatchOutgoingMessage will free it).
uint8_t* fb_build_auth_response(uint64_t nonce, uint64_t seat_id, size_t* len_out);

// Build a ServerMessage{CtrlServerMessage{ExecNotifyMessage}} frame.
uint8_t* fb_build_exec_notify(uint64_t nonce, uint64_t tclk, uint8_t state, size_t* len_out);

// Build a ServerMessage{StreamServerMessage{StreamDataPush}} frame.
// B must be a pre-initialized flatcc_builder_t; it is reset before use
// so the caller does not need to reset it manually between calls.
uint8_t* fb_build_stream_data_push(
    flatcc_builder_t *B,
    uint64_t nonce, uint32_t stream_id, uint64_t tclk,
    const uint8_t* data, size_t data_len,
    size_t* len_out);

// Build a ServerMessage{CtrlServerMessage{SetupBuf}} frame (echo back to client).
uint8_t* fb_build_setup_buf(uint64_t nonce, uint32_t lost_buf_size,
    uint64_t client_lifetime_ticks, size_t* len_out);

// Build a ServerMessage{CtrlServerMessage{OrphanedResponse{entries=[]}}} frame.
uint8_t* fb_build_orphaned_response(uint64_t nonce, size_t* len_out);

// Build a ServerMessage{CtrlServerMessage{LostMessagesResponse{seat_id, messages=[]}}} frame.
uint8_t* fb_build_lost_messages_response(uint64_t nonce, uint64_t seat_id, size_t* len_out);

// Build a ServerMessage{MemServerMessage{MemReadResponse}} frame.
uint8_t* fb_build_mem_read_response(
    uint64_t nonce, uint64_t tclk,
    uint32_t device_id, uint32_t seg_id, uint32_t offset,
    const uint8_t* data, size_t data_len,
    size_t* len_out);

// Build a ServerMessage{StreamServerMessage{StreamRegConfirm}} frame.
uint8_t* fb_build_stream_reg_confirm(
    uint64_t nonce, uint32_t stream_id,
    uint32_t device_id, uint32_t seg_id, uint32_t offset, uint32_t length,
    uint8_t mode, size_t* len_out);

#endif // __PROTO_MSG_H__