//
// Created by Kosho on 2020年8月14日.
//

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

#include "runtime/universe.hpp"
#include "util/bufferedInputStream.hpp"
#include "object/hiInteger.hpp"
#include "binaryFileParser.hpp"
#include "object/hiList.hpp"

BinaryFileParser::BinaryFileParser(BufferedInputStream *stream) {
 file_stream = 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();
 printf("argcount is 0x%x\n", argcount);
 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();

 HiString *file_path = new HiString(file_stream->get_file_path());

 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, file_path);
}


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

 return get_string();
}

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 *str = new HiString(str_value, length);
 delete[] str_value;

 return str;
}

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

 if (ch != 's' && ch != 't') {
 file_stream->unread();
 return NULL;
 }

 return get_string();
}

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();
}

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':
 // CodeObject
 printf("got a code object\n");
 list->add(get_code_object());
 break;
 case 'i':
 // Integer
 list->add(new HiInteger(file_stream->read_int()));
 break;
 case 'N':
 list->add(Universe::HiNone);
 break;
 case 't':
 // String
 str = get_string();
 list->add(str);
 _string_table.add(str);
 break;
 case 's':
 // String
 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;
}