#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include "lake.h"
#include "list.h"

struct context {
    char *s;
    size_t n;
    size_t i;
};
typedef struct context Ctx;

static LakeVal *_parse_expr(Ctx *ctx);

LakeVal *parse_expr(char *s, size_t n)
{
    Ctx ctx = { s, n, 0 };
    LakeVal *result = _parse_expr(&ctx);
    if (ctx.i < ctx.n) {
        printf("ignoring %lu trailing chars: %s\n", ctx.n - ctx.i - 1, ctx.s + ctx.i);
    }
    return result;
}

#define PARSE_EOF -1

static char peek(Ctx *ctx)
{
    if (ctx->i < ctx->n) return ctx->s[ctx->i];
    return PARSE_EOF;
}

static char peek2(Ctx *ctx)
{
    if (ctx->i + 1 < ctx->n) return ctx->s[ctx->i + 1];
    return PARSE_EOF;
}

static void consume(Ctx *ctx, size_t n)
{
    if (ctx->i + n > ctx->n) {
        die("ERROR: cannot consume, no more input");
    }
    ctx->i += n;
}

static char consume1(Ctx *ctx)
{
    char c = peek(ctx);
    consume(ctx, 1);
    return c;
}

static char ch(Ctx *ctx, char expected)
{
    char c = peek(ctx);
    if (c == expected) {
        consume1(ctx);
        return c;
    }
    char *msg = malloc(64);
    sprintf(msg, "parse error, expected '%c' got '%c'", expected, c);
    die(msg);
    return 0; /* placate the compiler */
}

static int maybe_spaces(Ctx *ctx)
{
    char *p;
    while ((p = strchr(" \r\n\t", peek(ctx))) != NULL) {
        consume1(ctx);
    }
    return 1;
}
/*
static int whitespace(Ctx *ctx)
{
    int result = 0;
    char *p;
    while ((p = strchr(" \r\n\t", peek(ctx))) != NULL) {
        consume1(ctx);
        result = 1;
    }
    return result;
}
*/
static int is_letter(char c)
{
    return (c >= 'a' && c <= 'z') || (c >= 'A' && c <= 'Z');
}

static int is_symbol(char c)
{
    return strchr("!$%&|*+-/:<=>?@^_~#", c) != NULL;
}

static int is_digit(char c)
{
    return c >= '0' && c <= '9';
}

static char *parse_while(Ctx *ctx, int (*is_valid)(char))
{
    size_t n = 8;
    size_t i = 0;
    char *s = malloc(n);
    char c;
    while ((c = peek(ctx)) != PARSE_EOF && is_valid(c)) {
        s[i++] = c;
        consume1(ctx);
        /* grow if necessary */
        if (i >= n) {
            n *= 2;
            char *t = realloc(s, n);
            if (!t) oom();
            s = t;
        }
    }
    s[i] = 0;
    return s;
}

static LakeInt *parse_int(Ctx *ctx)
{
    int n = 0;
    char c = peek(ctx);
    char sign = c == '-' ? -1 : 1;
    if (c == '-' || c == '+') {
        consume1(ctx);
    }
    while (is_digit(c = peek(ctx))) {
        n *= 10;
        n += c - '0';
        consume1(ctx);
    }
    return int_from_c(sign * n);
}

static int is_sym_char(char c)
{
    return is_letter(c) || is_symbol(c) || is_digit(c);
}

static LakeSym *parse_sym(Ctx *ctx)
{
    static int size = 1024;
    char s[size];
    char c;
    int i = 0;
    while (is_sym_char(c = peek(ctx)) && i < size - 1) {
        s[i++] = c;
        consume1(ctx);
    }
    s[i] = 0;
    /* TODO: check for #t and #f and return true boolean values (LakeBool *) */
    return sym_intern(s);
}

static char escape_char(char c)
{
    switch (c) {

    case 'n':
        c = '\n';
        break;

    case 'r':
        c = '\r';
        break;
        
    case 't':
        c = '\t';
        break;

    default:
        /* noop */
        break;
        
    }
    return c;
}

static LakeStr *parse_str(Ctx *ctx)
{
    size_t n = 8;
    size_t i = 0;
    char *s = malloc(n);
    char c;
    ch(ctx, '"');
    while ((c = peek(ctx)) != PARSE_EOF && c != '"') {
        /* handle backslash escapes */
        if (c == '\\') {
            consume1(ctx);
            c = escape_char(peek(ctx));
            if (c == PARSE_EOF) break;
        }
        s[i++] = c;
        consume1(ctx);
        /* grow if necessary */
        if (i >= n) {
            n *= 2;
            char *t = realloc(s, n);
            if (!t) oom();
            s = t;
        }
    }
    s[i] = 0;
    ch(ctx, '"');
    LakeStr *str = str_from_c(s);
    free(s);
    return str;
}

static LakeList* parse_list(Ctx *ctx)
{
    LakeList *list = list_make();
    ch(ctx, '(');
    char c;
    while ((c = peek(ctx)) != ')') {
        if (c == PARSE_EOF) {
            printf("error: end of input while parsing list");
            return NIL;
        }
        list_append(list, _parse_expr(ctx));
    }
    ch(ctx, ')');
    return list;
}

static int is_not_newline(char c)
{
    return !(c == '\n' || c == '\r');
}

static void parse_comment(Ctx *ctx)
{
    free(parse_while(ctx, is_not_newline));
}

static LakeVal *_parse_expr(Ctx *ctx)
{
    LakeVal *result = NIL;
    char c = peek(ctx);
    /*char d =*/ peek2(ctx);
    if (c >= '0' && c <= '9') {
        result = (LakeVal *)parse_int(ctx);
    }
    /* TODO: chars
    else if (c == '#' && d == '\\') {
        result = (LakeVal *)parse_char(ctx);
    }
    */
    else if (is_letter(c) || is_symbol(c)) {
        result = (LakeVal *)parse_sym(ctx);
    }
    else if (c == '"') {
        result = (LakeVal *)parse_str(ctx);
    }
    /* TODO: quote
    else if (c == '\'') {
        result = (LakeVal *)parse_quoted(ctx);
    }
    */
    else if (c == '(') {
        result = (LakeVal *)parse_list(ctx);
    }
    else if (c == ';') {
        parse_comment(ctx);
    }
    else if (c == PARSE_EOF) {
        printf("error: end of input, expected an expression");
    }
    else {
        char msg[32];
        sprintf(msg, "unexpected char '%c'", c);
        die(msg);
    }
    maybe_spaces(ctx);
    return result;
}