#include <stdio.h>
#include <stdlib.h>
#include <dlfcn.h>

#include "parts/symbol.h"
#include "parts/table.h"
size_t total_symbols;
int (*symbol_is_terminal)(symbol s);
int (*symbol_is_input_end)(symbol s);
int (*symbol_is_valid)(symbol s);

#include "parts/grammar.h"

struct production *grammar;
size_t total_productions;

char **semantic_action_str;

#include "util-tables.c"

#define _LAZY_LALR
#include "clr-table.c"

void *xdlsym(void *handle, char *sym)
{
    void *r = dlsym(handle, sym);
    if(!r) {
        fprintf(stderr, "ERROR: Symbol '%s' was not found\n", sym);
        exit(1);
    }
    return r;
}

#define GET_VARIABLE(var, handle)                    \
    var = *(typeof(&var))xdlsym(handle, #var)

int main(int argc, char **argv)
{
    if(argc != 2) return 1;

    void *handle = dlopen(argv[1], RTLD_LAZY);
    if(!handle) { puts(dlerror()); return 1; }

    GET_VARIABLE(total_symbols, handle);
    GET_VARIABLE(symbol_is_terminal, handle);
    GET_VARIABLE(symbol_is_input_end, handle);
    GET_VARIABLE(symbol_is_valid, handle);
    GET_VARIABLE(grammar, handle);
    GET_VARIABLE(total_productions, handle);
    GET_VARIABLE(semantic_action_str, handle);

    util_tables_fill();
    table_fill() && (fprintf(stderr, "ERROR: Table couldn't be generated\n"), exit(1), 1);

    printf("size_t total_symbols = %zu;\n", total_symbols);
    printf("IMPLEMENT_FUNCPTR(int, symbol_is_valid, (symbol s), {return s < total_symbols;})\n");

    printf("struct production _grammar[] = {\n");
    grammar_print_cstyle();
    printf("};\n");

    printf("struct production *grammar = _grammar;");
    printf("size_t total_productions = %zu;\n", total_productions);

    printf("struct action **table = (struct action *([])){\n");
    table_print_cstyle();
    printf("};\n");
    printf("size_t table_states = %zu;\n", table_states);

    for(size_t i = 0; i < total_productions; i++) {
        printf("#define A(n) (*(stack_head-3*%d+3*n-1))\n", grammar[i].nRHS-1);
        printf("int __prod%d_action(int *stack_head)\n", i);
        printf("{ int v;\n");
        printf(semantic_action_str[i]);
        printf("return v; }\n");
        printf("#undef A\n");
    }


    printf("typedef int (*semantic_action_fn)(int *stack_head);\n");

    printf("semantic_action_fn *semantic_actions = (semantic_action_fn[]){\n");
    for(size_t i = 0; i < total_productions; i++)
        printf("__prod%d_action, ", i);
    printf("};");

    table_free();
    util_tables_free();

    dlclose(handle);
    return 0;
}