path: root/src/lyn.c

#include <errno.h>
#include <assert.h>
#include <stdbool.h>
#include <string.h>
#include <stdio.h>
#include <limits.h>
#include <stdlib.h>

#include <lyn/lyn.h>
#include <lyn/ast.h>
#include <lyn/parser.h>
#include <lyn/debug.h>

static int builtin_let(struct lyn *lyn, struct vec args)
{
	assert(vec_len(&args) == 3);

	struct lyn_value name = lyn_at(args, 1);
	assert(name.kind == LYN_ID);

	struct lyn_value value = lyn_at(args, 2);
	if (lyn_eval(lyn, value))
		return -1;

	value = lyn_res(lyn);
	assert(value.kind == LYN_INT);

	return lyn_create_symbol(lyn, name.s, lyn_var(value));
}

static int builtin_set(struct lyn *lyn, struct vec args)
{
	assert(vec_len(&args) == 3);

	struct lyn_value name = lyn_at(args, 1);
	assert(name.kind == LYN_ID);

	struct lyn_value value = lyn_at(args, 2);
	if (lyn_eval(lyn, value))
		return -1;

	value = lyn_res(lyn);
	assert(value.kind == LYN_INT);

	return lyn_replace_symbol(lyn, name.s, lyn_var(value));
}

static int builtin_for(struct lyn *lyn, struct vec args)
{
	assert(vec_len(&args) == 5);

	struct lyn_value init = lyn_at(args, 1);
	struct lyn_value cond = lyn_at(args, 2);
	struct lyn_value post = lyn_at(args, 3);
	struct lyn_value body = lyn_at(args, 4);

	if (lyn_eval(lyn, init))
		return -1;

top:
	if (lyn_eval(lyn, cond))
		return -1;

	struct lyn_value check = lyn_res(lyn);
	assert(check.kind == LYN_INT);

	if (check.i) {
		if (lyn_eval(lyn, body))
			return -1;

		if (lyn_eval(lyn, post))
			return -1;

		goto top;
	}

	return 0;
}

static int builtin_println(struct lyn *lyn, struct vec args)
{
	assert(vec_len(&args) == 2);

	struct lyn_value value = lyn_at(args, 1);
	if (lyn_eval(lyn, value))
		return -1;

	value = lyn_res(lyn);
	assert(value.kind == LYN_INT);

	printf("%lld\n", value.i);
	return 0;
}

static int builtin_lt(struct lyn *lyn, struct vec args)
{
	assert(vec_len(&args) == 3);
	struct lyn_value left = lyn_at(args, 1);
	struct lyn_value right = lyn_at(args, 2);

	if (lyn_eval(lyn, left))
		return -1;

	left = lyn_res(lyn);
	assert(left.kind == LYN_INT);

	if (lyn_eval(lyn, right))
		return -1;

	right = lyn_res(lyn);
	assert(right.kind == LYN_INT);

	lyn_return(lyn, gen_int(left.i < right.i));
	return 0;
}

static int builtin_plus(struct lyn *lyn, struct vec args)
{
	assert(vec_len(&args) == 3);
	struct lyn_value left = lyn_at(args, 1);
	struct lyn_value right = lyn_at(args, 2);

	if (lyn_eval(lyn, left))
		return -1;

	left = lyn_res(lyn);
	assert(left.kind == LYN_INT);

	if (lyn_eval(lyn, right))
		return -1;

	right = lyn_res(lyn);
	assert(right.kind == LYN_INT);

	lyn_return(lyn, gen_int(left.i + right.i));
	return 0;
}

struct lyn lyn_create()
{
	return (struct lyn){};
}

int lyn_create_scope(struct lyn *lyn)
{
	struct lyn_scope *scope = calloc(1, sizeof(struct lyn_scope));
	if (!scope)
		return -1;

	scope->visible = lookup_create(sizeof(struct lyn_symbol));
	scope->parent = lyn->cur;

	if (!lyn->root)
		lyn->root = scope;

	lyn->cur = scope;
	return 0;
}

static struct lyn_symbol *scope_find(struct lyn_scope *scope, const char *name)
{
	return lookup_at(&scope->visible, name);
}

static struct lyn_symbol *scopes_find(struct lyn_scope *scope, const char *name)
{
	while (scope) {
		struct lyn_symbol *s = scope_find(scope, name);
		if (s)
			return s;

		scope = scope->parent;
	}

	return NULL;
}

static int scope_add(struct lyn_scope *scope, const char *name, struct lyn_symbol symb)
{
	if (!lookup_insert(&scope->visible, name, &symb)) {
		error("%s exists in scope\n", name);
		return -1;
	}

	return 0;
}

int lyn_create_symbol(struct lyn *lyn, const char *name, struct lyn_symbol symb)
{
	return scope_add(lyn->cur, name, symb);
}

struct lyn_symbol *lyn_lookup_symbol(struct lyn *lyn, const char *name)
{
	return scopes_find(lyn->cur, name);
}

int lyn_replace_symbol(struct lyn *lyn, const char *name, struct lyn_symbol symb)
{
	struct lyn_symbol *s = scopes_find(lyn->cur, name);
	if (!s)
		return -1;

	*s = symb;
	return 0;
}

int lyn_init(struct lyn *lyn)
{
	if (lyn_create_scope(lyn))
		return -1;

	if (lyn_create_symbol(lyn, "let", lyn_syntax(builtin_let)))
		return -1;

	if (lyn_create_symbol(lyn, "for", lyn_syntax(builtin_for)))
		return -1;

	if (lyn_create_symbol(lyn, "set", lyn_syntax(builtin_set)))
		return -1;

	if (lyn_create_symbol(lyn, "println", lyn_proc(builtin_println)))
		return -1;

	if (lyn_create_symbol(lyn, "<", lyn_proc(builtin_lt)))
		return -1;

	if (lyn_create_symbol(lyn, "+", lyn_proc(builtin_plus)))
		return -1;

	return 0;
}

static int eval_group(struct lyn *lyn, struct lyn_value group)
{
	assert(group.kind == LYN_GROUP);
	foreach_vec(gi, group.args) {
		struct lyn_value arg = lyn_at(group.args, gi);
		if (lyn_eval(lyn, arg))
			return -1;
	}

	return 0;
}

static int eval_apply(struct lyn *lyn, struct lyn_value apply)
{
	assert(apply.kind == LYN_APPLY);
	foreach_vec(ai, apply.args) {
		struct lyn_value arg = lyn_at(apply.args, ai);
		if (lyn_eval(lyn, arg))
			return -1;
	}

	return 0;
}

int lyn_apply_cmd(struct lyn *lyn, struct lyn_symbol *cmd, struct vec args)
{
	switch (cmd->kind) {
	case LYN_BUILTIN_SYNTAX: return cmd->call(lyn, args);
	case LYN_BUILTIN_PROC: return cmd->call(lyn, args);
	default: abort();
	}

	return 0;
}

static int eval_cmd(struct lyn *lyn, struct lyn_value cmd)
{
	assert(cmd.kind == LYN_CMD);

	struct lyn_value name = lyn_at(cmd.args, 0);

	/* to start with */
	struct lyn_symbol *symb = lyn_lookup_symbol(lyn, name.s);
	assert(symb);

	return lyn_apply_cmd(lyn, symb, cmd.args);
}

static int eval_int(struct lyn *lyn, struct lyn_value i)
{
	assert(i.kind == LYN_INT);
	lyn_return(lyn, i);
	return 0;
}

static int eval_id(struct lyn *lyn, struct lyn_value id)
{
	assert(id.kind == LYN_ID);
	struct lyn_symbol *symb = lyn_lookup_symbol(lyn, id.s);
	assert(symb && symb->kind == LYN_VAR);

	lyn_return(lyn, symb->value);
	return 0;
}

int lyn_eval(struct lyn *lyn, struct lyn_value value)
{
	switch (value.kind) {
	case LYN_GROUP: return eval_group(lyn, value);
	case LYN_APPLY: return eval_apply(lyn, value);
	case LYN_CMD: return eval_cmd(lyn, value);
	case LYN_INT: return eval_int(lyn, value);
	case LYN_ID: return eval_id(lyn, value);
	default: error("unhandled case in interp eval"); break;
	}

	return -1;
}

int lyn_eval_str(struct lyn *lyn, const char *name, const char *str)
{
	struct parser *p = create_parser();
	if (!p)
		return -1;

	parse(p, name, str);
	struct lyn_value ast = p->tree;
	bool failed = p->failed;
	destroy_parser(p);

	if (failed)
		return -1;

	ast_dump(0, ast);
	return lyn_eval(lyn, ast);
}

/**
 * Read whole file into a buffer and return pointer to buffer.
 * Possibly kind of silly to have both \p file and \p f.
 * Apparently there's no standardized way to get the file name of a
 * file pointer.
 *
 * @param file Name of file to read.
 * @param f File pointer.
 * @return Pointer to buffer with file contents.
 */
static char *read_file(const char *file, FILE *f)
{
	fseek(f, 0, SEEK_END);
	/** @todo check how well standardized this actually is */
	long s = ftell(f);
	if (s == LONG_MAX) {
		error("%s might be a directory", file);
		return NULL;
	}

	fseek(f, 0, SEEK_SET);

	char *buf = malloc((size_t)(s + 1));
	if (!buf)
		return NULL;

	fread(buf, (size_t)(s + 1), 1, f);
	/* remember terminating null */
	buf[s] = 0;
	return buf;
}

int lyn_eval_file(struct lyn *lyn, const char *fname)
{
	FILE *f = fopen(fname, "rb");
	if (!f) {
		error("failed opening %s: %s\n", fname, strerror(errno));
		return -1;
	}

	char *buf = read_file(fname, f);
	fclose(f);

	if (!buf)
		return -1;

	int ret = lyn_eval_str(lyn, fname, buf);
	free(buf);

	return ret;
}

void lyn_return(struct lyn *lyn, struct lyn_value ast)
{
	lyn->res = ast;
}

struct lyn_value lyn_res(struct lyn *lyn)
{
	return lyn->res;
}

void lyn_destroy(struct lyn *lyn)
{
}