#pragma once
#include <sys/socket.h>
#include <stdlib.h>
#include <stdio.h>
#include <errno.h>
#include <string.h>
#include "evt_core.h"
#include "algo_utils.h"
#include "capture_traffic.h"
#include "url.h"
#include "utils.h"
#include "packet.h"

struct algo_params {
  uint8_t is_waiting_bootstrap;
  uint8_t is_healing;
  char *algo_name, *capture_file;
  int links, fresh_data, redundant_data;
};

struct algo_ctx;
typedef void (*algo_init)(struct evt_core_ctx* ctx, struct algo_ctx* app_ctx, struct algo_params* ap);
typedef int  (*algo_ctx_on_buffer)(struct evt_core_ctx* ctx, struct evt_core_fdinfo* fdinfo, struct buffer_packet* bp);
typedef int  (*algo_ctx_on_event)(struct evt_core_ctx* ctx, struct evt_core_fdinfo* fdinfo);

typedef void (*algo_ctx_free_misc)(void*);

struct algo_desc {
  char* name;
  algo_init init;
  algo_ctx_on_buffer on_stream;
  algo_ctx_on_buffer on_datagram;
  algo_ctx_on_event on_err;
};

struct algo_ctx {
  struct algo_desc* desc;
  uint8_t link_count;
  uint8_t is_rdy;
  struct algo_params ap;
  int ref_count;
  uint64_t udp_rcv, udp_sent, cell_rcv, cell_sent;
  struct capture_ctx cap;
  struct buffer_resources br;
  void* misc;                       // Additional structures
  algo_ctx_free_misc free_misc;     // Fx ptr to free misc
};

void algo_naive_init(struct evt_core_ctx* ctx, struct algo_ctx* app_ctx, struct algo_params* ap);
int  algo_naive_on_stream(struct evt_core_ctx* ctx, struct evt_core_fdinfo* fdinfo, struct buffer_packet* bp);
int  algo_naive_on_datagram(struct evt_core_ctx* ctx, struct evt_core_fdinfo* fdinfo, struct buffer_packet* bp);
int  algo_naive_on_err(struct evt_core_ctx *ctx, struct evt_core_fdinfo *fdinfo);

void algo_dup2_init(struct evt_core_ctx* ctx, struct algo_ctx* app_ctx, struct algo_params* ap);
int algo_dup2_on_stream(struct evt_core_ctx* ctx, struct evt_core_fdinfo* fdinfo, struct buffer_packet* bp);
int algo_dup2_on_datagram(struct evt_core_ctx* ctx, struct evt_core_fdinfo* fdinfo, struct buffer_packet* bp);
int algo_dup2_on_err(struct evt_core_ctx *ctx, struct evt_core_fdinfo *fdinfo);

void algo_thunder_init(struct evt_core_ctx* ctx, struct algo_ctx* app_ctx, struct algo_params* ap);
int algo_thunder_on_stream(struct evt_core_ctx* ctx, struct evt_core_fdinfo* fdinfo, struct buffer_packet* bp);
int algo_thunder_on_datagram(struct evt_core_ctx* ctx, struct evt_core_fdinfo* fdinfo, struct buffer_packet* bp);
int algo_thunder_on_err(struct evt_core_ctx *ctx, struct evt_core_fdinfo *fdinfo);

static struct algo_desc available_algo[] = {
  {
    .name = "naive",
    .init = algo_naive_init,
    .on_stream = algo_naive_on_stream,
    .on_datagram = algo_naive_on_datagram,
    .on_err = algo_naive_on_err
  },
  {
    .name = "dup2",
    .init = algo_dup2_init,
    .on_stream = algo_dup2_on_stream,
    .on_datagram = algo_dup2_on_datagram,
    .on_err = algo_dup2_on_err
  },
  {
    .name = "thunder",
    .init = algo_thunder_init,
    .on_stream = algo_thunder_on_stream,
    .on_datagram = algo_thunder_on_datagram,
    .on_err = algo_thunder_on_err
  }
};

void algo_main_init(struct evt_core_ctx* evt, struct algo_params* ap);

int main_on_tcp_co(struct evt_core_ctx* ctx, struct evt_core_fdinfo* fdinfo);
int main_on_tcp_read(struct evt_core_ctx* ctx, struct evt_core_fdinfo* fdinfo);
int main_on_udp_read(struct evt_core_ctx* ctx, struct evt_core_fdinfo* fdinfo);
int main_on_tcp_write(struct evt_core_ctx* ctx, struct evt_core_fdinfo* fdinfo);
int main_on_udp_write (struct evt_core_ctx* ctx, struct evt_core_fdinfo* fdinfo);
int main_on_timer(struct evt_core_ctx* ctx, struct evt_core_fdinfo* fdinfo);
int main_on_err(struct evt_core_ctx* ctx, struct evt_core_fdinfo* fdinfo);