#pragma once

#include <openssl/ossl_typ.h>
#include <event2/event.h>

#include <tfe_stream.h>
#include <tfe_future.h>
#include <proxy.h>

struct tfe_thread_ctx
{
	pthread_t thr;
	unsigned int thread_id;
	unsigned int load;

	struct event_base * evbase;
	struct evdns_base* dnsbase;
	unsigned char running;
	
	unsigned int nr_modules;
	const struct tfe_plugin * modules;
};

enum tfe_plugin_state
{
	PLUG_STATE_READONLY,
	PLUG_STATE_PREEPTION,
	PLUG_STATE_DETACHED
};

struct plugin_ctx
{
	enum tfe_plugin_state state;
	unsigned int is_plugin_opened;
	void * pme;
};

struct tfe_stream_write_ctx
{
	struct tfe_stream_private * _stream;
	enum tfe_conn_dir dir;
};

struct tfe_conn_private
{
	struct tfe_stream_private * _stream_ref;
	evutil_socket_t fd;
	struct bufferevent * bev;
	uint8_t on_writing;
};

struct tfe_stream_private
{
	struct tfe_stream head;
	char * str_stream_addr;
	void * stream_logger;

	struct tfe_proxy * proxy_ref;
	struct tfe_thread_ctx * thread_ref;

	enum tfe_stream_proto session_type;
	struct tfe_stream_write_ctx * w_ctx_upstream;
	struct tfe_stream_write_ctx * w_ctx_downstream;
	struct tfe_conn_private * conn_upstream;
	struct tfe_conn_private * conn_downstream;

	struct
	{
		struct ssl_mgr * ssl_mgr;
		struct ssl_stream * ssl_downstream;
		struct ssl_stream * ssl_upstream;
	};

	uint8_t is_plugin_opened;
	int calling_idx;

	size_t forward_bytes;
	size_t drop_bytes;

	size_t defer_bytes;
	struct timeval defer_timeval;

	enum tfe_app_proto app_proto;

	/* Plugin Ctxs */
	unsigned int nr_plugin_ctxs;
	struct plugin_ctx * plugin_ctxs;

	/* TCP forward without scan or decode when the passthough is set */
	bool passthough;

	/* For defer connection setup */
	evutil_socket_t defer_fd_downstream;
	evutil_socket_t defer_fd_upstream;

	/* ASYNC UPSTREAM */
	future * future_upstream_create;
	/* ASYNC DOWNSTREAM */
	future * future_downstream_create;

	/* SUSPEND */
	bool is_suspended;
	enum tfe_conn_dir suspended_by;

	/* KEYRING-ID */
	unsigned keyring_id;

	/* ONLY FOR LOG */
	evutil_socket_t log_fd_downstream;
	evutil_socket_t log_fd_upstream;
};

static inline void * __STREAM_LOGGER(struct tfe_stream_private * _stream)
{
	return _stream->proxy_ref->logger;
}