shithub: pprolog

ref: 28e7dd47d568908702264977d70860c25467fb6e
dir: /repl.c/

#include <u.h>
#include <libc.h>
#include <bio.h>

#include "dat.h"
#include "fns.h"

Rune parsefindmore(int);

void
repl(void)
{
	int fd = 0; /* Standard input */
	while(1){
		print("?- ");
		Term *query = parse(fd, nil, 1);
		Binding *bindings = nil;
		choicestack = nil;
		goalstack = nil;
		int success;
		int firsttime = 1;
FindMore:
		success = evalquery(query, &bindings);
		if(firsttime){
			print(" ");
			firsttime = 0;
		}
		if(success == 0)
			print("  false.\n");
		else{
			if(bindings == nil)
				print("  true");
			else{
				while(bindings){
					print("  %S = %S%s", 
						bindings->name, 
						prettyprint(bindings->value, 0, 0, 0), 
						bindings->next ? ",\n " : "");
					bindings = bindings->next;
				}
			}
			if(choicestack != nil){
				print("\n");
				if(parsefindmore(fd) == L';'){
					print(";");
					goto FindMore;
				}else
					print(".\n");
			}else{
				print(".\n");
			}
		}

		vlong amount = collectgarbage();
		if(amount != 0)
			print("Collected %lld bytes of garbage\n", amount);
	}
}

Rune
parsefindmore(int fd)
{
	int consctl = open("/dev/consctl", OWRITE);
	if(consctl > 0)
		write(consctl, "rawon", 5);
	else{
		print("Could not open /dev/consctl\n");
		exits("open");
	}

	fd = dup(fd, -1);
	Biobuf *input = Bfdopen(fd, OREAD);
	Rune peek = Bgetrune(input);
	Bterm(input);

	if(consctl > 0){
		write(consctl, "rawoff", 6);
		close(consctl);
	}
	return peek;
}