#include "priv.h"

#include <stdlib.h>
#include <string.h>

#include "lua.h"
#include "lauxlib.h"
#include "lualib.h"

#include "compiler/compiler.lua.h"

TYC_RESULT code_assemble(const char* code, uint8_t** bytecode, size_t* bytecode_sz)
{
    lua_State* L = luaL_newstate();
    luaL_openlibs(L);

    int r = luaL_loadbufferx(L, (const char *) lib_compiler_compiler_out, lib_compiler_compiler_out_len, "compiler", "b");
    if (r == LUA_ERRSYNTAX)
        abort();
    else if (r == LUA_ERRMEM)
        out_of_memory();

    lua_call(L, 0, 1);

    lua_pushstring(L, code);
    r = lua_pcall(L, 1, 1, 0);
    if (r == LUA_ERRMEM) {
        out_of_memory();
    } else if (r == LUA_ERRERR) {
        abort();
    } else if (r == LUA_ERRRUN) {
        fprintf(stderr, "%s\n", lua_tostring(L, -1));
        return T_ERR_ASSEMBLER_SYNTAX_ERROR;
    }

    if (!lua_istable(L, -1))
        abort();
    *bytecode_sz = luaL_len(L, -1);
    *bytecode = malloc(*bytecode_sz);
    for (size_t i = 0; i < *bytecode_sz; ++i) {
        lua_geti(L, -1, i + 1);
        (*bytecode)[i] = (uint8_t) lua_tointeger(L, -1);
        lua_pop(L, 1);
    }

    lua_close(L);
    return T_OK;
}