/* 
 * $Id: awk.c,v 1.41 2006-04-16 04:31:38 bacon Exp $ 
 */

#include <xp/awk/awk_i.h>

#ifndef XP_AWK_STAND_ALONE
#include <xp/bas/memory.h>
#include <xp/bas/assert.h>
#endif

static void __free_func (xp_awk_t* awk, void* func);
static void __free_namedval (xp_awk_t* awk, void* val);

xp_awk_t* xp_awk_open (void)
{	
	xp_awk_t* awk;

	awk = (xp_awk_t*) xp_malloc (xp_sizeof(xp_awk_t));
	if (awk == XP_NULL) return XP_NULL;

	if (xp_str_open(&awk->token.name, 128) == XP_NULL) {
		xp_free (awk);
		return XP_NULL;	
	}

	if (xp_awk_map_open (&awk->tree.funcs, awk, 256, __free_func) == XP_NULL) {
		xp_str_close (&awk->token.name);
		xp_free (awk);
		return XP_NULL;	
	}

	if (xp_awk_tab_open(&awk->parse.globals) == XP_NULL) {
		xp_str_close (&awk->token.name);
		xp_awk_map_close (&awk->tree.funcs);
		xp_free (awk);
		return XP_NULL;	
	}

	if (xp_awk_tab_open(&awk->parse.locals) == XP_NULL) {
		xp_str_close (&awk->token.name);
		xp_awk_map_close (&awk->tree.funcs);
		xp_awk_tab_close (&awk->parse.globals);
		xp_free (awk);
		return XP_NULL;	
	}

	if (xp_awk_tab_open(&awk->parse.params) == XP_NULL) {
		xp_str_close (&awk->token.name);
		xp_awk_map_close (&awk->tree.funcs);
		xp_awk_tab_close (&awk->parse.globals);
		xp_awk_tab_close (&awk->parse.locals);
		xp_free (awk);
		return XP_NULL;	
	}

/* TODO: initial map size... */
	if (xp_awk_map_open(&awk->run.named,awk,256,__free_namedval) == XP_NULL) {
		xp_str_close (&awk->token.name);
		xp_awk_map_close (&awk->tree.funcs);
		xp_awk_tab_close (&awk->parse.globals);
		xp_awk_tab_close (&awk->parse.locals);
		xp_awk_tab_close (&awk->parse.params);
		xp_free (awk);
		return XP_NULL;	
	}

	awk->opt.parse = 0;
	awk->opt.run = 0;
	awk->errnum = XP_AWK_ENOERR;
	awk->src_func = XP_NULL;
	awk->in_func = XP_NULL;
	awk->out_func = XP_NULL;
	awk->src_arg = XP_NULL;
	awk->in_arg = XP_NULL;
	awk->out_arg = XP_NULL;

	awk->parse.nlocals_max = 0;

	awk->tree.nglobals = 0;
	awk->tree.begin = XP_NULL;
	awk->tree.end = XP_NULL;
	awk->tree.chain = XP_NULL;
	awk->tree.chain_tail = XP_NULL;

	awk->run.stack = XP_NULL;
	awk->run.stack_top = 0;
	awk->run.stack_base = 0;
	awk->run.stack_limit = 0;
	awk->run.exit_level = 0;
	awk->run.icache_count = 0;
	awk->run.rcache_count = 0;

	awk->lex.curc = XP_CHAR_EOF;
	awk->lex.ungotc_count = 0;

	return awk;
}

int xp_awk_close (xp_awk_t* awk)
{
	xp_awk_clear (awk);

	if (xp_awk_detsrc(awk) == -1) return -1;

	xp_awk_map_close (&awk->run.named);
	xp_awk_map_close (&awk->tree.funcs);
	xp_awk_tab_close (&awk->parse.globals);
	xp_awk_tab_close (&awk->parse.locals);
	xp_awk_tab_close (&awk->parse.params);
	xp_str_close (&awk->token.name);

	xp_free (awk);
	return 0;
}

/* TODO: write a function to clear awk->parse data structure.
         this would be need either as a separate function or as a part of xp_awk_clear...
         do i have to pass an option to xp_awk_clear to do this??? */

void xp_awk_clear (xp_awk_t* awk)
{
	xp_awk_map_clear (&awk->run.named);

	xp_awk_tab_clear (&awk->parse.globals);
	xp_awk_tab_clear (&awk->parse.locals);
	xp_awk_tab_clear (&awk->parse.params);
	awk->parse.nlocals_max = 0; 

	/* clear parse trees */	
	awk->tree.nglobals = 0;	
	xp_awk_map_clear (&awk->tree.funcs);

	if (awk->tree.begin != XP_NULL) {
		xp_assert (awk->tree.begin->next == XP_NULL);
		xp_awk_clrpt (awk->tree.begin);
		awk->tree.begin = XP_NULL;
	}
	if (awk->tree.end != XP_NULL) {
		xp_assert (awk->tree.end->next == XP_NULL);
		xp_awk_clrpt (awk->tree.end);
		awk->tree.end = XP_NULL;
	}
	while (awk->tree.chain != XP_NULL) {
		xp_awk_chain_t* next = awk->tree.chain->next;
		if (awk->tree.chain->pattern != XP_NULL)
			xp_awk_clrpt (awk->tree.chain->pattern);
		if (awk->tree.chain->action != XP_NULL)
			xp_awk_clrpt (awk->tree.chain->action);
		xp_free (awk->tree.chain);
		awk->tree.chain = next;
	}
	awk->tree.chain_tail = XP_NULL;	

	/* TODO: destroy function list */
}

void xp_awk_setparseopt (xp_awk_t* awk, int opt)
{
	awk->opt.parse = opt;
}

void xp_awk_setrunopt (xp_awk_t* awk, int opt)
{
	awk->opt.run = opt;
}

int xp_awk_attsrc (xp_awk_t* awk, xp_awk_io_t src, void* arg)
{
	if (xp_awk_detsrc(awk) == -1) return -1;

	xp_assert (awk->src_func == XP_NULL);
	if (src(XP_AWK_IO_OPEN, arg, XP_NULL, 0) == -1) {
		awk->errnum = XP_AWK_ESRCOP;
		return -1;
	}

	awk->src_func = src;
	awk->src_arg = arg;
	awk->lex.curc = XP_CHAR_EOF;
	awk->lex.ungotc_count = 0;
	return 0;
}

int xp_awk_detsrc (xp_awk_t* awk)
{
	if (awk->src_func != XP_NULL) {
		if (awk->src_func(XP_AWK_IO_CLOSE, awk->src_arg, XP_NULL, 0) == -1) {
			awk->errnum = XP_AWK_ESRCCL;
			return -1;
		}
		awk->src_func = XP_NULL;
		awk->src_arg = XP_NULL;
		awk->lex.curc = XP_CHAR_EOF;
		awk->lex.ungotc_count = 0;
	}

	return 0;
}

static void __free_func (xp_awk_t* awk, void* func)
{
	xp_awk_func_t* f = (xp_awk_func_t*) func;

	/* f->name doesn't have to be freed */
	/*xp_free (f->name);*/
	xp_awk_clrpt (f->body);
	xp_free (f);
}

static void __free_namedval (xp_awk_t* awk, void* val)
{
	xp_awk_refdownval (awk, val);
}