#pragma once

#include <al/array.h>

#include "util/packet.h"

#include "packet_stream.h"

struct nn_rpc_connection;
struct nn_rpc_callback {
    u32 id;
    void (*callback)(void *, struct nn_rpc_connection *, struct nn_packet *);
    void *userdata;
};

enum {
    NNWT_SKEW_POSITIVE = 0,
    NNWT_SKEW_NEGATIVE
};

struct nn_skew {
    u64 ts;
    bool direction;
};

struct nn_rpc_connection {
    struct nn_packet_stream *stream;
    array(struct nn_rpc_callback) callbacks;
    u32 outgoing;
    bool flushing;
    struct {
        struct nn_skew s;
        s8 query_active;
        u32 repeat;
        array(struct nn_skew) results;
    } skew;
    struct nn_rpc *rpc;
};

struct nn_rpc_command {
    s8 op;
    bool (*callback)(void *, struct nn_rpc_connection *, struct nn_packet *, struct nn_packet *);
    void *userdata;
};

struct nn_rpc {
    struct nn_event_loop *loop;
    u32 increment;
    array(struct nn_rpc_command) commands;
    bool do_query_skew;
    struct nn_rpc_connection *conn; // client
    array(struct nn_rpc_connection *) connections;
    void (*connection_callback)(void *, struct nn_rpc_connection *);
    void (*ready_callback)(void *, struct nn_rpc_connection *);
    void (*connection_closed_callback)(void *, struct nn_rpc_connection *);
    void *userdata;
};

bool nn_rpc_init(struct nn_rpc *rpc, struct nn_event_loop *loop,
    void (*connection_callback)(void *, struct nn_rpc_connection *),
    void (*ready_callback)(void *, struct nn_rpc_connection *),
    void (*connection_closed_callback)(void *, struct nn_rpc_connection *), void *userdata);
void nn_rpc_query_clock_skews(struct nn_rpc *rpc, bool do_query_skew);
void nn_rpc_add_command(struct nn_rpc *rpc, struct nn_rpc_command *command);
void nn_rpc_add_stream(struct nn_rpc *rpc, struct nn_packet_stream *stream);
void nn_rpc_prepare_client(struct nn_rpc *rpc);
bool nn_rpc_connect(struct nn_rpc *rpc, u8 id, u8 type, str *addr, u16 port);
bool nn_rpc_reconnect(struct nn_rpc *rpc, str *addr, u16 port);
void nn_rpc_disconnect(struct nn_rpc *rpc);
struct nn_packet *nn_rpc_get_packet(struct nn_rpc *rpc, s8 op);
void nn_rpc_free(struct nn_rpc *rpc);

// The order of commands per-connection will be respected.
void nn_rpc_connection_command(struct nn_rpc_connection *conn, struct nn_packet *packet,
    void (*callback)(void *, struct nn_rpc_connection *conn, struct nn_packet *), void *userdata);
void nn_rpc_conn_flush(struct nn_rpc_connection *conn);
void nn_rpc_conn_disconnect(struct nn_rpc_connection *conn);