/*
 * $Id: awk.c,v 1.7 2006-01-09 16:03:55 bacon Exp $
 */

#include <xp/awk/awk.h>
#include <xp/bas/memory.h>
#include <xp/bas/assert.h>

xp_awk_t* xp_awk_open (xp_awk_t* awk)
{
	if (awk == XP_NULL) {
		awk = (xp_awk_t*) xp_malloc (xp_sizeof(awk));
		if (awk == XP_NULL) return XP_NULL;
		awk->__dynamic = xp_true;
	}
	else awk->__dynamic = xp_false;

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

	awk->tree = XP_NULL;
	awk->errnum = XP_AWK_ENOERR;

	awk->src_func = XP_NULL;
	awk->inp_func = XP_NULL;
	awk->outp_func = XP_NULL;

	awk->src_arg = XP_NULL;
	awk->inp_arg = XP_NULL;
	awk->outp_arg = XP_NULL;

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

	return awk;
}

static void __collapse_tree (xp_awk_t* awk)
{
	/* TODO: collapse the tree */
	/* TODO */
	awk->tree = XP_NULL;
}

int xp_awk_close (xp_awk_t* awk)
{

	if (awk->tree != XP_NULL) __collapse_tree (awk);

	if (xp_awk_detsrc(awk) == -1) return -1;
	xp_str_close (&awk->token.name);
	if (awk->__dynamic) xp_free (awk);
	return 0;
}

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;
}