#include <windows.h>
#include <SWI-Prolog.h>

#include "pl.h"

/* Frame: Wrapper for opening and closing Prolog foreign frames. */

Frame::Frame()
{
	m_f = PL_open_foreign_frame();
}

Frame::~Frame()
{
	PL_close_foreign_frame(m_f);
}

void Frame::Close()
{
	PL_close_foreign_frame(m_f);
}

/* Close foreign frame and invalidate associated data. */
void Frame::Discard()
{
	PL_discard_foreign_frame(m_f);
}

void Frame::Rewind()
{
	PL_rewind_foreign_frame(m_f);
}

/* Mark: Wrapper for marking and releasing strings on the Prolog stack. */

Mark::Mark()
{
	PL_mark_string_buffers(&m_m);
}

Mark::~Mark()
{
	PL_release_string_buffers_from_mark(m_m);
}

/* Query: Wrapper for opening, solving and cutting Prolog query. */

Query::Query(const module_t ctx, const predicate_t p, const term_t t0)
{
	m_q = PL_open_query(ctx, PL_Q_CATCH_EXCEPTION, p, t0);
}

Query::~Query()
{
	PL_cut_query(m_q);
}

int Query::Cut()
{
	if (PL_cut_query(m_q)) return 1;
	if (const term_t t = PL_exception(m_q)) throw t;
	return 0;
}

/* Cut query and invalidate associated data. */
int Query::Close()
{
	if (PL_close_query(m_q)) return 1;
	if (const term_t t = PL_exception(m_q)) throw t;
	return 0;
}

int Query::NextSolution()
{
	if (PL_next_solution(m_q)) return 1;
	if (const term_t t = PL_exception(m_q)) throw t;
	return 0;
}

/* Convert Prolog term to wide characters. */
int PlString(const term_t t, std::wstring* const pWstr, const int iFlags)
{
	char* sz;
	int r = PL_get_chars(t, &sz, iFlags);
	if (r) *pWstr = WstrFromSz(sz);
	return r;
}