%% @file yapi.yap
%% @brief support yap shell
%%

% :- yap_flag(verbose, verbose).

:- use_module( library(python) ).

:- module(yapi, [
		 python_ouput/0,
		 show_answer/2,
		 show_answer/3,
		 yap_query/4,
		 python_query/2,
		 yapi_query/2
		 ]).

:- use_module( library(lists) ).
:- use_module( library(maplist) ).
:- use_module( library(rbtrees) ).
:- use_module( library(terms) ).
:- use_module( library(python) ).

:- python_import(yap4py.yapi).

	%% @pred yapi_query( + VarList, - Dictionary)
	%%
	%% dictionary, Examples
	%%
	%%
	yapi_query( VarNames, Self ) :-
		show_answer(VarNames, Dict),
		Self.bindings := Dict.

:- initialization set_preds.

set_preds :-
	current_predicate(P, Q),
	functor(Q,P,A),
	atom_string(P,S),
	:= yap4py.yapi.named( S, A),
	fail.
set_preds :-
	system_predicate(P/A),
	atom_string(P,S),
	:= yap4py.yapi.named( S, A),
	fail.
set_preds.

argi(N,I,I1) :-
    atomic_concat(`A`,I,N),
	I1 is I+1.

python_query( Self, String ) :-
	yap_flag(typein_module, Mod),
 	atomic_to_term( String, Goal, VarNames ),
	query_to_answer( Mod:Goal, VarNames, Status, Bindings),
	maplist(in_dict(Self.bindings), Bindings),
	write_query_answer( Bindings ),
	nl( user_error ),
	Self.port := Status.

in_dict(Dict, var([V0,V|Vs])) :- !,
	Dict[V] := V0,
	in_dict( Dict, var([V0|Vs])).
in_dict(Dict, nonvar([V0|Vs],G)) :- !,
	Dict[V0] := G,
	in_dict( Dict, var([V0|Vs])).
in_dict(_, _).