#include <sys/stat.h>

#include <fcntl.h>
#include <stdio.h>
#include <stdlib.h>
#include <unistd.h>

#include <gmp.h>

#include "alloc.h"
#include "analyzer.h"
#include "codegen.h"
#include "common.h"
#include "errors.h"
#include "lexer.h"
#include "parser.h"

/* Read the contents of FILE into a dynamically allocated buffer and
   return it, storing the buffer size in BUFSZ. */
static char *readfile(const char *file, size_t *bufsz)
	__attribute__((returns_nonnull, nonnull));

int
main(int argc, char **argv)
{
	if (argc != 2) {
		fputs("Usage: oryx file\n", stderr);
		exit(EXIT_FAILURE);
	}

	size_t srclen;
	char *src = readfile(argv[1], &srclen);

	aux_t aux;
	mpq_t *folds;
	type_t *types;
	scope_t *scps;
	arena_t a = NULL;

	lexemes_t toks = lexstring(src, srclen);
	ast_t ast = parsetoks(toks, &aux);
	analyzeprog(ast, aux, toks, &a, &types, &scps, &folds);
	codegen(argv[1], folds, scps, types, ast, aux, toks);

#if DEBUG
	for (size_t i = 0; i < ast.len; i++) {
		if ((*folds[i])._mp_den._mp_d != NULL)
			mpq_clear(folds[i]);
	}

	free(folds);
	free(scps);
	free(src);
	free(types);
	lexemes_free(toks);
	ast_free(ast);
	aux_free(aux);
	arena_free(&a);
#endif
	return EXIT_SUCCESS;
}

char *
readfile(const char *filename, size_t *n)
{
	int fd = open(filename, O_RDONLY);
	if (fd == -1)
		err("open: %s:", filename);

	struct stat sb;
	if (fstat(fd, &sb) == -1)
		err("fstat: %s:", filename);

	char *p = bufalloc(NULL, sb.st_size + 4, 1);

	ssize_t nr;
	for (size_t off = 0; (nr = read(fd, p + off, sb.st_blksize)) > 0; off += nr)
		;
	if (nr == -1)
		err("read: %s:", filename);

	p[sb.st_size + 0] =
	p[sb.st_size + 1] =
	p[sb.st_size + 2] =
	p[sb.st_size + 3] = 0;

	*n = sb.st_size;
	close(fd);
	return p;
}