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

#include "frame.h"
#include "class_resolver.h"
#include "string.h"
#include "file.h"

static struct hash_table_entry * load_from_filenames(const char * filenames[], int length, int * hash_table_length)
{
  const uint8_t ** class_names = (const uint8_t **)filenames;
  int class_names_length[length];
  uint8_t * buffers[length];
  size_t file_size[length];

  for (int i = 0; i < length; i++) {
    uint32_t filename_length = string_length(filenames[i]);
    const char * suffix = ".class";
    uint32_t suffix_length = string_length(suffix);
    const char * filename_suffix = &filenames[i][filename_length - suffix_length];
    if (filename_length <= suffix_length || !string_equal(suffix, filename_suffix)) {
      printf("invalid class filename: %s\n", filenames[i]);
      continue;
    }
    class_names_length[i] = filename_length - suffix_length;
    printf("load class: %s\n", filenames[i]);

    buffers[i] = file_read(filenames[i], &file_size[i]);
  }

  struct hash_table_entry * class_hash_table = class_resolver_load_from_buffers(class_names,
                                                                                class_names_length,
                                                                                (const uint8_t **)buffers,
                                                                                length,
                                                                                hash_table_length);

  for (int i = 0; i < length; i++) {
    memset(buffers[i], 0, file_size[i]);
    free(buffers[i]);
  }

  return class_hash_table;
}

int main(int argc, const char * argv[])
{
  assert(argc >= 3);

  const uint8_t * main_class = (const uint8_t *)argv[1];
  int main_class_length = string_length((const char *)main_class);

  const char ** class_filenames = &argv[2];
  int num_class_filenames = argc - 2;

  int class_hash_table_length;
  struct hash_table_entry * class_hash_table = load_from_filenames(class_filenames, num_class_filenames, &class_hash_table_length);

  vm_start(class_hash_table_length,
           class_hash_table,
           main_class,
           main_class_length);
}