#include "wren_inter.h"
#include "../utils/stringBuilder.h"
#include <string.h> 
#include <raylib.h>
#include <stdio.h>
#include <wren.h>

StringBuilder stdout_sb = {0};

void writeFn(WrenVM* vm, const char* text) {
    AppendStringBuilder(&stdout_sb, text, '\n');
}

void errorFn(WrenVM* vm, WrenErrorType errorType,
        const char* module, const int line,
        const char* msg)
{
    switch (errorType)
    {
        case WREN_ERROR_COMPILE:
            {
                // TraceLog(LOG_ERROR, "[%s line %d] [Error] %s", module, line, msg);
                AppendStringBuilder(&stdout_sb, "Comptime error: ", '\n');
                AppendStringBuilder(&stdout_sb, msg, ' ');
            } break;
        case WREN_ERROR_STACK_TRACE:
            {
                AppendStringBuilder(&stdout_sb, "Stack trace: ", '\n');
                AppendStringBuilder(&stdout_sb, msg, ' ');
                // TraceLog(LOG_ERROR, "[%s line %d] in %s", module, line, msg);
            } break;
        case WREN_ERROR_RUNTIME:
            {
                AppendStringBuilder(&stdout_sb, "Runtime error: ", '\n');
                AppendStringBuilder(&stdout_sb, msg, ' ');
                // TraceLog(LOG_ERROR, "[Runtime Error] %s", msg);
            } break;
    }
}

char *EvaluateScript(char *script) {
    TraceLog(LOG_INFO, "Evaluating %s...", script); 
    WrenConfiguration config;
    wrenInitConfiguration(&config);
    config.writeFn = writeFn;
    config.errorFn = errorFn;

    WrenVM* vm = wrenNewVM(&config);
    if (vm == NULL) {
        TraceLog(LOG_ERROR, "Failed to initialize a wren vm");
        return NULL;
    }

    WrenInterpretResult result = wrenInterpret(vm, "main", script);
    switch (result) {
        case WREN_RESULT_SUCCESS: break;
        case WREN_RESULT_COMPILE_ERROR: break;
        case WREN_RESULT_RUNTIME_ERROR: break;
    }
    printf("Wren outp: %s", stdout_sb.val);
    wrenFreeVM(vm);

    if (stdout_sb.val == NULL) return NULL;
    char *outp = strdup(stdout_sb.val);
    FreeStringBuilder(&stdout_sb);
    return outp;
}