#include <assert.h>
#include <stddef.h>

#include "uclisp.h"
#include "internal.h"
#include "utility.h"
#include "state.h"

// TODO: remove string.h
#include <string.h>

struct ucl_object *ucl_evaluate_builtin_form(struct ucl_state *state, struct ucl_object *list) {
    // TODO: Reasonably split builtin and non-builtin evaluation
    struct ucl_object *evaluated_list = ucl_nil_create();

    FOREACH_LIST(list, iter, item) {
        struct ucl_object *obj = ucl_evaluate(state, item);
        UCL_RET_IF_ERROR(obj);
        ucl_list_append(evaluated_list, obj);
    };

    struct ucl_object *fun = ucl_car(evaluated_list);
    struct ucl_object *result = NULL;

    if (fun->type == UCL_TYPE_BUILTIN) {
        struct ucl_object *args = ucl_cdr(evaluated_list);
        result = fun->builtin(state, args);
    } else if (fun->type == UCL_TYPE_CELL) {
        struct ucl_state *fun_state = ucl_state_create_child(state);
        struct ucl_object *fun_arg_syms = ucl_car(fun);
        struct ucl_object *fun_forms = ucl_cdr(fun);
        int i = 0;
        FOREACH_LIST(fun_arg_syms, iter, sym) {
            ucl_state_put(fun_state, sym->symbol, ucl_list_nth(evaluated_list, i + 1));
            i++;
        }
        result = ucl_progn(fun_state, fun_forms);
    } else {
        assert(0);
    }

    // TODO: cleanup

    return (result == NULL) ? ucl_nil_create() : result;
}

struct ucl_object *ucl_evaluate_special_form(struct ucl_state *state, struct ucl_object *list) {
    // TODO: Recursively eval args
    const char *fun_sym = ucl_car(list)->symbol;

    struct ucl_object *fun = ucl_state_get(state, fun_sym);
    struct ucl_object *args = ucl_cdr(list);
    struct ucl_object *result = NULL;

    result = fun->special(state, args);

    return result;
}

struct ucl_object *ucl_evaluate_list(struct ucl_state *state, struct ucl_object *list) {
    if (list->cell.car == NULL) {
        return list;
    }

    struct ucl_object *fun = ucl_evaluate(state, ucl_car(list));
    UCL_RET_IF_ERROR(fun);

    if (fun->type == UCL_TYPE_SPECIAL) {
        return ucl_evaluate_special_form(state, list);
    } else if (fun->type == UCL_TYPE_BUILTIN || fun->type == UCL_TYPE_CELL) {
        return ucl_evaluate_builtin_form(state, list);
    } else {
        assert(0);
        // TODO: Lisp functions and other errors
    }
}

struct ucl_object *ucl_evaluate(struct ucl_state *state, struct ucl_object *obj) {
    assert(obj != NULL);

    switch (obj->type) {
        case UCL_TYPE_CELL:
            return ucl_evaluate_list(state, obj);
        case UCL_TYPE_SYMBOL:
            return ucl_state_get(state, obj->symbol);
        case UCL_TYPE_INT:
        case UCL_TYPE_STRING:
        case UCL_TYPE_ERROR:
            return obj;
        case UCL_TYPE_BUILTIN:
        case UCL_TYPE_SPECIAL:
        case UCL_TYPE_COUNT:
            assert(0);
            return NULL;
    }
}