#include "postgres.h"
#include "fmgr.h"

#include "query_util.h"

PG_FUNCTION_INFO_V1(tsquery_numnode);
Datum           tsquery_numnode(PG_FUNCTION_ARGS);

Datum
tsquery_numnode(PG_FUNCTION_ARGS) {
	QUERYTYPE  *query = (QUERYTYPE *) DatumGetPointer(PG_DETOAST_DATUM_COPY(PG_GETARG_DATUM(0)));
	int nnode = query->size;
	PG_FREE_IF_COPY(query,0);
	PG_RETURN_INT32(nnode);
}

static int
CompareTSQ( QUERYTYPE *a, QUERYTYPE *b ) {
	if ( a->size != b->size ) {
		return ( a->size < b->size ) ? -1 : 1;
	} else if ( a->len != b->len ) {
		return ( a->len < b->len ) ? -1 : 1;
	} else {
		QTNode	*an = QT2QTN( GETQUERY(a), GETOPERAND(a) ); 
		QTNode	*bn = QT2QTN( GETQUERY(b), GETOPERAND(b) );
		int res = QTNodeCompare(an, bn);

		QTNFree(an);
		QTNFree(bn);

		return res; 
	}

	return 0;
}

PG_FUNCTION_INFO_V1(tsquery_cmp);                                                                              \
Datum   tsquery_cmp(PG_FUNCTION_ARGS);         

Datum
tsquery_cmp(PG_FUNCTION_ARGS) {
	QUERYTYPE  *a = (QUERYTYPE *) DatumGetPointer(PG_DETOAST_DATUM_COPY(PG_GETARG_DATUM(0)));
	QUERYTYPE  *b = (QUERYTYPE *) DatumGetPointer(PG_DETOAST_DATUM_COPY(PG_GETARG_DATUM(1)));
	int res = CompareTSQ(a,b);

	PG_FREE_IF_COPY(a,0);
	PG_FREE_IF_COPY(b,1);

	PG_RETURN_INT32(res);
}

#define CMPFUNC( NAME, ACTION ) 									\
PG_FUNCTION_INFO_V1(NAME);										\
Datum	NAME(PG_FUNCTION_ARGS);										\
													\
Datum													\
NAME(PG_FUNCTION_ARGS) { 										\
	QUERYTYPE  *a = (QUERYTYPE *) DatumGetPointer(PG_DETOAST_DATUM_COPY(PG_GETARG_DATUM(0)));	\
	QUERYTYPE  *b = (QUERYTYPE *) DatumGetPointer(PG_DETOAST_DATUM_COPY(PG_GETARG_DATUM(1)));	\
	int res = CompareTSQ(a,b);									\
													\
	PG_FREE_IF_COPY(a,0);										\
	PG_FREE_IF_COPY(b,1);										\
													\
	PG_RETURN_BOOL( ACTION );									\
}

CMPFUNC( tsquery_lt, res <0 ); 
CMPFUNC( tsquery_le, res<=0 ); 
CMPFUNC( tsquery_eq, res==0 ); 
CMPFUNC( tsquery_ge, res>=0 ); 
CMPFUNC( tsquery_gt, res >0 );
CMPFUNC( tsquery_ne, res!=0 );