mirror of https://github.com/janet-lang/janet
119 lines
2.8 KiB
C
119 lines
2.8 KiB
C
#include <stdlib.h>
|
|
#include <stdio.h>
|
|
#include "datatypes.h"
|
|
#include "vm.h"
|
|
#include "parse.h"
|
|
#include "compile.h"
|
|
#include "value.h"
|
|
#include "disasm.h"
|
|
|
|
void StringPut(uint8_t * string) {
|
|
uint32_t i;
|
|
uint32_t len = VStringSize(string);
|
|
for (i = 0; i < len; ++i)
|
|
fputc(string[i], stdout);
|
|
}
|
|
|
|
/* Test c function */
|
|
Value print(VM * vm) {
|
|
uint32_t j, count;
|
|
Value nil;
|
|
count = VMCountArgs(vm);
|
|
for (j = 0; j < count; ++j) {
|
|
uint8_t * string = ValueToString(vm, VMGetArg(vm, j));
|
|
StringPut(string);
|
|
fputc('\n', stdout);
|
|
}
|
|
nil.type = TYPE_NIL;
|
|
return nil;
|
|
}
|
|
|
|
/* A simple repl for debugging */
|
|
void debugRepl() {
|
|
char buffer[1024] = {0};
|
|
const char * reader = buffer;
|
|
Value func;
|
|
VM vm;
|
|
Parser p;
|
|
Compiler c;
|
|
|
|
VMInit(&vm);
|
|
|
|
for (;;) {
|
|
|
|
/* Run garbage collection */
|
|
VMMaybeCollect(&vm);
|
|
|
|
/* Reset state */
|
|
ParserInit(&p, &vm);
|
|
|
|
/* Get and parse input until we have a full form */
|
|
while (p.status == PARSER_PENDING) {
|
|
/* Get some input if we are done */
|
|
if (*reader == '\0') {
|
|
printf(">> ");
|
|
if (!fgets(buffer, sizeof(buffer), stdin)) {
|
|
return;
|
|
}
|
|
p.index = 0;
|
|
reader = buffer;
|
|
}
|
|
reader += ParserParseCString(&p, reader);
|
|
}
|
|
|
|
/* Check for parsing errors */
|
|
if (p.error) {
|
|
unsigned i;
|
|
printf("\n");
|
|
printf("%s\n", buffer);
|
|
for (i = 0; i < p.index; ++i) {
|
|
printf(" ");
|
|
}
|
|
printf("^\n");
|
|
printf("\nParse error: %s\n", p.error);
|
|
reader = buffer; /* Flush the input buffer */
|
|
buffer[0] = '\0';
|
|
continue;
|
|
}
|
|
|
|
/* Try to compile generated AST */
|
|
CompilerInit(&c, &vm);
|
|
CompilerAddGlobalCFunc(&c, "print", print);
|
|
func.type = TYPE_FUNCTION;
|
|
func.data.func = CompilerCompile(&c, p.value);
|
|
|
|
/* Check for compilation errors */
|
|
if (c.error) {
|
|
printf("Compiler error: %s\n", c.error);
|
|
reader = buffer;
|
|
buffer[0] = 0;
|
|
continue;
|
|
}
|
|
|
|
/* Print asm */
|
|
//printf("\n");
|
|
//dasmFunc(stdout, func.data.func);
|
|
//printf("\n");
|
|
|
|
/* Execute function */
|
|
VMLoad(&vm, func);
|
|
if (VMStart(&vm)) {
|
|
printf("VM error: %s\n", vm.error);
|
|
reader = buffer;
|
|
buffer[0] = 0;
|
|
continue;
|
|
} else {
|
|
uint8_t * string = ValueToString(&vm, vm.ret);
|
|
StringPut(string);
|
|
printf("\n");
|
|
}
|
|
}
|
|
|
|
}
|
|
|
|
int main() {
|
|
printf("Super cool interpreter v0.0\n");
|
|
debugRepl();
|
|
return 0;
|
|
}
|