#include <stdio.h>
#include <assert.h>

#include "code/binaryFileParser.hpp"
#include "runtime/universe.hpp"
#include "object/hiList.hpp"
#include "object/hiDouble.hpp"

BinaryFileParser::BinaryFileParser(BufferedInputStream* buf_file_stream) {
    file_stream = buf_file_stream;
}

CodeObject* BinaryFileParser::parse() {
    int magic_number = file_stream->read_int();
    printf("magic number is 0x%x\n", magic_number);
    int moddate = file_stream->read_int();
    printf("moddate is 0x%x\n", moddate);

    char object_type = file_stream->read();

    if (object_type == 'c') {
        CodeObject* result = get_code_object();
        printf("parse OK!\n");
        return result;
    }

    return NULL;
}

CodeObject* BinaryFileParser::get_code_object() {
    int argcount  = file_stream->read_int();
    int nlocals   = file_stream->read_int();
    int stacksize = file_stream->read_int();
    int flags     = file_stream->read_int();
    printf("flags is 0x%x\n", flags);

    HiString* byte_codes = get_byte_codes();
    ArrayList<HiObject*>* consts     = get_consts();
    ArrayList<HiObject*>* names      = get_names();
    ArrayList<HiObject*>* var_names  = get_var_names();
    ArrayList<HiObject*>* free_vars  = get_free_vars();
    ArrayList<HiObject*>* cell_vars  = get_cell_vars();

    HiString* file_name   = get_file_name();
    HiString* module_name = get_name();
    int begin_line_no     = file_stream->read_int();
    HiString* lnotab      = get_no_table();

    return new CodeObject(argcount, nlocals, stacksize, flags, byte_codes,
        consts, names, var_names, free_vars, cell_vars, file_name, module_name,
        begin_line_no, lnotab);
}

HiString* BinaryFileParser::get_string() {
    int length = file_stream->read_int();
    char* str_value = new char[length];

    for (int i = 0; i < length; i++) {
        str_value[i] = file_stream->read();
    }

    HiString* s = new HiString(str_value, length);
    delete[] str_value;

    return s;
}

HiString* BinaryFileParser::get_name() {
    char ch = file_stream->read();

    if (ch == 's') {
        return get_string();
    }
    else if (ch == 't') {
        HiString* str = get_string();
        _string_table.add(str);
        return str;
    }
    else if (ch == 'R') {
        return _string_table.get(file_stream->read_int());
    }
    
    return NULL;
}

HiString* BinaryFileParser::get_file_name() {
    return get_name();
}

HiString* BinaryFileParser::get_byte_codes() {
    assert(file_stream->read() == 's');

    return get_string();
}

HiString* BinaryFileParser::get_no_table() {
    char ch = file_stream->read();
    
    if (ch != 's' && ch != 't') {
        file_stream->unread();
        return NULL;
    }

    return get_string();
}

ArrayList<HiObject*>* BinaryFileParser::get_consts() {
    if (file_stream->read() == '(') {
        return get_tuple();
    }

    file_stream->unread();
    return NULL;
}

ArrayList<HiObject*>* BinaryFileParser::get_names() {
    if (file_stream->read() == '(') {
        return get_tuple();
    }

    file_stream->unread();
    return NULL;
}

ArrayList<HiObject*>* BinaryFileParser::get_var_names() {
    if (file_stream->read() == '(') {
        return get_tuple();
    }

    file_stream->unread();
    return NULL;
}

ArrayList<HiObject*>* BinaryFileParser::get_free_vars() {
    if (file_stream->read() == '(') {
        return get_tuple();
    }

    file_stream->unread();
    return NULL;
}

ArrayList<HiObject*>* BinaryFileParser::get_cell_vars() {
    if (file_stream->read() == '(') {
        return get_tuple();
    }

    file_stream->unread();
    return NULL;
}

ArrayList<HiObject*>* BinaryFileParser::get_tuple() {
    int length = file_stream->read_int();
    HiString * str;

    ArrayList<HiObject*>* list = new ArrayList<HiObject*>(length);
    for (int i = 0; i < length; i++) {
        char obj_type = file_stream->read();

        switch (obj_type) {
        case 'c':
            printf("got a code object\n");
            list->add(get_code_object());
            break;
        case 'i':
            list->add(new HiInteger(file_stream->read_int()));
            break;
        case 'g':
            list->add(new HiDouble(file_stream->read_double()));
            break;
        case 'N':
            list->add(Universe::HiNone);
            break;
        case 't':
            str = get_string();
            list->add(str);
            _string_table.add(str);
            break;
        case 's':
            list->add(get_string());
            break;
        case 'R':
            list->add(_string_table.get(file_stream->read_int()));
            break;
        case '(':
            list->add(new HiList(get_tuple()));
            break;
        default:
            printf("parser, unrecognized type : %c\n", obj_type);
        }
    }

    return list;
}