#include <error.h>
#include <stdio.h>

#include <libcr/coroutine.h>
#include <libnetio/netio.h>
#include <lib9p/srv.h>

#include "static9p.h"
#include "static.h"

/* configuration **************************************************************/

#include "config.h"

#ifndef CONFIG_NETIO_NUM_CONNS
#	error config.h must define CONFIG_NETIO_NUM_CONNS
#endif

/* implementation *************************************************************/

#define UNUSED(name) /* name __attribute__((unused)) */

/* file tree ******************************************************************/

#define FILE_METADATA(NAME) {                                             \
		.header = { .vtable = &static_file_vtable },              \
                                                                          \
		.u_name = "root", .u_num = 0, /* owner user */            \
		.g_name = "root", .g_num = 0, /* owner group */           \
		.m_name = "root", .m_num = 0, /* last-modified-by user */ \
                                                                          \
		.pathnum = __COUNTER__,                                   \
		.name    = NAME,                                          \
		.perm    = 0444,                                          \
		.atime   = 1728337905,                                    \
		.mtime   = 1728337904,                                    \
	}

#define DIR_METADATA(NAME) {                                              \
		.header = { .vtable = &static_dir_vtable },               \
                                                                          \
		.u_name = "root", .u_num = 0, /* owner user */            \
		.g_name = "root", .g_num = 0, /* owner group */           \
		.m_name = "root", .m_num = 0, /* last-modified-by user */ \
                                                                          \
		.pathnum = __COUNTER__,                                   \
		.name    = NAME,                                          \
		.perm    = 0555,                                          \
		.atime   = 1728337905,                                    \
		.mtime   = 1728337904,                                    \
	}

#define STATIC_FILE(STRNAME, SYMNAME)                           \
	&((static struct static_file){                          \
		.metadata   = FILE_METADATA(STRNAME),           \
		.data_start = _binary_static_##SYMNAME##_start, \
		.data_end   = _binary_static_##SYMNAME##_end,   \
	}).metadata.header

static struct static_dir root = {
	.metadata = DIR_METADATA(""),
	.members = {
		&((static struct static_dir){
			.metadata = DIR_METADATA("Documentation"),
			.members = {
				STATIC_FILE("x", Documentation_x),
				NULL
			},
		}).metadata.header,
		STATIC_FILE("README.md", README_md),
		NULL,
	},
};

static struct lib9p_srv_file *get_root(struct lib9p_srv_ctx *UNUSED(ctx), char *UNUSED(treename)) {
	return &root.metadata.header;
}

/* main ***********************************************************************/

int main() {
	int sock = netio_listen(9000);
	if (sock < 0)
		error(1, -sock, "netio_listen");

	struct lib9p_srv srv = {
		.sockfd  = sock,
		.rootdir = get_root,
	};

	for (int i = 0; i < CONFIG_NETIO_NUM_CONNS; i++)
		if (!coroutine_add(lib9p_srv_read_cr, &srv))
			error(1, 0, "coroutine_add(lib9p_srv_read_cr, &srv)");
	for (int i = 0; i < 2*CONFIG_NETIO_NUM_CONNS; i++)
		if (!coroutine_add(lib9p_srv_write_cr, &srv))
			error(1, 0, "coroutine_add(lib9p_srv_write_cr, &srv)");

	coroutine_main();
	return 1;
}