#ifndef PL_H
#define PL_H

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

#include "common.h"

int PlString(term_t t, std::wstring* pWs, int iFlags = CVT_WRITE);

struct Frame
{
	Frame();
	~Frame();
	void Close();
	void Discard();
	void Rewind();
private:
	fid_t m_f;
};

struct Mark
{
	Mark();
	~Mark();
private:
	buf_mark_t m_m;
};

struct Query
{
	Query(module_t ctx, predicate_t p, term_t t0);
	~Query();
	int Cut();
	int Close();
	int NextSolution();
private:
	qid_t m_q;
};

/* Polymorphic aliases for PL_put_*, PL_get_*. */
inline int PlPut(term_t t, int x) { return PL_put_integer(t, x); }
inline int PlPut(term_t t, long x) { return PL_put_integer(t, x); }
inline int PlPut(term_t t, long long x) { return PL_put_int64(t, x); }
inline int PlPut(term_t t, atom_t x) { return PL_put_atom(t, x); }
inline int PlPut(term_t t, char* x) { return PL_put_atom_chars(t, x); }
inline int PlPut(term_t t, const char* x) { return PL_put_atom_chars(t, x); }
inline int PlPut(term_t, int*) { return -1; }
inline int PlPut(term_t, long*) { return -1; }
inline int PlPut(term_t, long long*) { return -1; }
inline int PlPut(term_t, atom_t*) { return -1; }
inline int PlPut(term_t, char**) { return -1; }
inline int PlPut(term_t, std::string*) { return -1; }
inline int PlPut(term_t, std::wstring*) { return -1; }

inline int PlGet(term_t, int) { return -1; }
inline int PlGet(term_t, long) { return -1; }
inline int PlGet(term_t, long long) { return -1; }
inline int PlGet(term_t, atom_t) { return -1; }
inline int PlGet(term_t, char*) { return -1; }
inline int PlGet(term_t, const char*) { return -1; }
inline int PlGet(term_t t, int* x) { return PL_get_integer(t, x); }
inline int PlGet(term_t t, long* x) { return PL_get_long(t, x); }
inline int PlGet(term_t t, long long* x) { return PL_get_int64(t, x); }
inline int PlGet(term_t t, atom_t* x) { return PL_get_atom(t, x); }
inline int PlGet(term_t t, char** x) { return PL_get_atom_chars(t, x); }
inline int PlGet(term_t t, std::string* x) { 
	Mark m;
	char* sz;
	if (!PlGet(t, &sz)) return 0;
	*x = sz;
	return 1;
}
inline int PlGet(term_t t, std::wstring* x) { 
	Mark m;
	char* sz;
	if (!PlGet(t, &sz)) return 0;
	*x = WsFromSz(sz);
	return 1; /* or catch potential exception from BstrFromSz? */
}

/* Put in or get from a term reference an arbitrary number of values,
 * returning false if any value could not be put/got. */
template <typename = void>
inline bool PlPutN(term_t) { return true; }
template <typename T, typename... U>
inline bool PlPutN(term_t t, T x, U... xs) { return PlPut(t, x) && PlPutN(t+1, xs...); }

template <typename = void>
inline bool PlGetN(term_t) { return true; }
template <typename T, typename... U>
inline bool PlGetN(term_t t, T x, U... xs) { return PlGet(t, x) && PlGetN(t+1, xs...); }

/* Call Prolog predicate, propagating Prolog exceptions. */
template <typename... T>
int Plx(const char* const szMod, const char* const szPred, T... xs)
{
	Frame f;
	const term_t t = PL_new_term_refs(sizeof...(T));
	if (!PlPutN(t, xs...)) return 0;
	Query q(NULL, PL_predicate(szPred, sizeof...(T), szMod), t);
	if (!q.NextSolution()) return 0;
	if (!PlGetN(t, xs...)) return 0;
	return 1;
}

/* Call Prolog predicate, ignoring Prolog exceptions. */
template <typename... T>
int Pl(const char* const szMod, const char* const szPred, T... xs)
{
	try {
		return Plx(szMod, szPred, xs...);
	} catch (const term_t& t) {
		return 0;
	}
}

#endif