#include "stdafx.h"
#include "defs.h"

extern int scan(char *);
extern void setup(void);
extern void init(void);
extern void selftest(void);
extern void cmdisplay(U *);
extern int symbol_level;
extern int test_flag;
extern U *varlist;
static void check_stack(void);
static void print_mem_info(void);
static int dash_dash_command(char *);
jmp_buf stop_return;
static char *errstr;
static char buf[100];

void
stop(char *str)
{
	errstr = str; // don't print str now, jmp_buf might be redirected
	longjmp(stop_return, 1);
}

void
run(char *s)
{
	int n;

	esc_flag = 0;

	if (setjmp(stop_return)) {
		restore_symbols(0);
		if (errstr) {
			printstr("Stop: ");
			printstr(errstr);
			printstr("\n");
		}
		return;
	}

	init();

	tos = 0;

	frame = stack + TOS;

	symbol_level = 0;

	varlist = nil;

	if (dash_dash_command(s))
		return;

	while (1) {

		n = scan(s);

		if (n == 0)
			break;

		s += n;

		setup();

		p1 = pop();
		check_stack();

		push(p1);
		eval();

		p2 = pop();
		check_stack();

		symbol(LAST)->u.sym.binding = last;
		symbol(LAST)->u.sym.binding2 = nil;

		// print string w/o quotes

		if (p2->k == STR) {
			printstr(p2->u.str);
			printstr("\n");
			continue;
		}

		// don't print nil unless it was due to eval of a symbol

		if (p2 == nil && !issymbol(p1))
			continue;

		if (equal(symbol(TTY)->u.sym.binding, one) || test_flag) // tty mode?
			printline(p2);
		else {
#ifdef LINUX
			display(p2);
#else
			if (p1->k == SYM
			&& p1 != symbol(LAST)
			&& p1 != symbol(TTY)
			&& p1 != p2) {
				push_symbol(SETQ);
				push(p1);
				push(p2);
				list(3);
				p2 = pop();
			}
			cmdisplay(p2);
#endif
		}
	}
}

static int
dash_dash_command(char *s)
{
	if (strncmp(s, "--mem", 5) == 0) {
		print_mem_info();
		return 1;
	}
	if (strncmp(s, "--gc", 4) == 0) {
		gc();
		return 1;
	}
	if (strncmp(s, "--test", 6) == 0) {
		selftest();
		return 1;
	}
	return 0;
}

static void
check_stack(void)
{
	if (tos != 0)
		stop("stack error");
	if (frame != stack + TOS)
		stop("frame error");
}

void
echo_input(char *s)
{
	if (nil && equal(symbol(TTY)->u.sym.binding, one)) // tty mode?
		printstr("\n");
	printstr(s);
	printstr("\n");
}

extern int total_count;
extern int free_count;

static void
print_mem_info(void)
{
	sprintf(buf, "total atoms %d (%dM)   free atoms %d   used atoms %d   sizeof atom %d\n",
		total_count,
		total_count / 1000000,
		free_count,
		total_count - free_count,
		(int) sizeof (U));
	printstr(buf);
}