#include <xp/lsp/lsp.h>
#include <xp/bas/stdio.h>
#include <xp/bas/ctype.h>
#include <xp/bas/stdcli.h>
#include <xp/bas/locale.h>
#include <xp/bas/sio.h>

#ifdef __linux
#include <mcheck.h>
#endif

static xp_ssize_t get_input (int cmd, void* arg, xp_char_t* data, xp_size_t size)
{
	xp_ssize_t n;

	switch (cmd) {
	case XP_LSP_IO_OPEN:
	case XP_LSP_IO_CLOSE:
		return 0;

	case XP_LSP_IO_DATA:
		if (size < 0) return -1;
		n = xp_sio_getc (xp_sio_in, data);
		if (n == 0) return 0;
		if (n != 1) return -1;
		return n;
	}

	return -1;
}

static xp_ssize_t put_output (int cmd, void* arg, xp_char_t* data, xp_size_t size)
{

	switch (cmd) {
	case XP_LSP_IO_OPEN:
	case XP_LSP_IO_CLOSE:
		return 0;

	case XP_LSP_IO_DATA:
		return xp_sio_putsx (xp_sio_out, data, size);
	}

	return -1;
}


int to_int (const xp_char_t* str)
{
	int r = 0;

	while (*str != XP_CHAR('\0')) {
		if (!xp_isdigit(*str))	break;
		r = r * 10 + (*str - XP_CHAR('0'));
		str++;
	}

	return r;
}

#include <locale.h>

int handle_cli_error (
	const xp_cli_t* cli, int code, 
	const xp_char_t* name, const xp_char_t* value)
{
	xp_printf (XP_TEXT("usage: %s /memory=nnn /increment=nnn\n"), cli->verb);

	if (code == XP_CLI_ERROR_INVALID_OPTNAME) {
		xp_printf (XP_TEXT("unknown option - %s\n"), name);
	}
	else if (code == XP_CLI_ERROR_MISSING_OPTNAME) {
		xp_printf (XP_TEXT("missing option - %s\n"), name);
	}
	else if (code == XP_CLI_ERROR_REDUNDANT_OPTVAL) {
		xp_printf (XP_TEXT("redundant value %s for %s\n"), value, name);
	}
	else if (code == XP_CLI_ERROR_MISSING_OPTVAL) {
		xp_printf (XP_TEXT("missing value for %s\n"), name);
	}
	else if (code == XP_CLI_ERROR_MEMORY) {
		xp_printf (XP_TEXT("memory error in processing %s\n"), name);
	}
	else {
		xp_printf (XP_TEXT("error code: %d\n"), code);
	}

	return -1;
}

xp_cli_t* parse_cli (int argc, xp_char_t* argv[])
{
	static const xp_char_t* optsta[] =
	{
		XP_TEXT("/"), XP_TEXT("--"), XP_NULL
	};

	static xp_cliopt_t opts[] =
	{
		{ XP_TEXT("memory"), XP_CLI_OPTNAME | XP_CLI_OPTVAL },
        { XP_TEXT("increment"), XP_CLI_OPTNAME | XP_CLI_OPTVAL },
        { XP_NULL, 0 }
    };

	static xp_cli_t cli =
	{
		handle_cli_error,
		optsta,
		XP_TEXT("="),
		opts
	};

	if (xp_parsecli (argc, argv, &cli) == -1) return XP_NULL;
	return &cli;
}

int xp_main (int argc, xp_char_t* argv[])
{
	xp_lsp_t* lsp;
	xp_lsp_obj_t* obj;
	xp_cli_t* cli;
	int mem, inc;

#ifdef __linux
	mtrace ();
#endif

	if (xp_setlocale () == -1) {
		xp_fprintf (xp_stderr,
			XP_TEXT("error: cannot set locale\n"));
		return -1;
	}

	if ((cli = parse_cli (argc, argv)) == XP_NULL) return -1;
	mem = to_int(xp_getclioptval(cli, XP_TEXT("memory")));
	inc = to_int(xp_getclioptval(cli, XP_TEXT("increment")));
	xp_clearcli (cli);

	if (mem <= 0) {
		xp_fprintf (xp_stderr,
			XP_TEXT("error: invalid memory size given\n"));
		return -1;
	}

	lsp = xp_lsp_open (XP_NULL, mem, inc);
	if (lsp == XP_NULL) {
		xp_fprintf (xp_stderr, 
			XP_TEXT("error: cannot create a lsp instance\n"));
		return -1;
	}

	xp_printf (XP_TEXT("LSP 0.0001\n"));

	xp_lsp_attach_input (lsp, get_input, XP_NULL);
	xp_lsp_attach_output (lsp, put_output, XP_NULL);

	for (;;) {
		xp_sio_puts (xp_sio_out, argv[0]);
		xp_sio_puts (xp_sio_out, XP_TEXT("> "));
		xp_sio_flush (xp_sio_out);

		obj = xp_lsp_read (lsp);
		if (obj == XP_NULL) {
			if (lsp->errnum != XP_LSP_ERR_END && 
			    lsp->errnum != XP_LSP_ERR_ABORT) {
				xp_fprintf (xp_stderr, 
					XP_TEXT("error while reading: %d\n"), lsp->errnum);
			}

			if (lsp->errnum < XP_LSP_ERR_SYNTAX) break;
			continue;
		}

		if ((obj = xp_lsp_eval (lsp, obj)) != XP_NULL) {
			xp_lsp_print (lsp, obj);
			xp_sio_puts (xp_sio_out, XP_TEXT("\n"));
		}
		else {
			int errnum;
			xp_char_t errstr[256];

			errnum = xp_lsp_error (lsp, errstr, xp_countof(errstr));
			if (errnum == XP_LSP_ERR_ABORT) break;
			xp_fprintf (xp_stderr, 
				XP_TEXT("error: [%d] %s\n"), 
				errnum, errstr);
		}
	}

	xp_lsp_close (lsp);

#ifdef __linux
	muntrace ();
#endif
	return 0;
}