#include <tasking.h>
#include <stdlib.h>
#include <mailboxes.h>
#include <ipc/vfs.h>
#include <memory.h>
#include <grub/text_fb_info.h>
#include <vfs.h>
#include <ipc/devfs.h>
#include <string.h>
#include <dbg.h>
#include <initrd.h>

#define VFS_PID 2
#define DEVFS_PID 3

uint32_t box;
char* initrd;
long size;

void initrd_puts(vfs_message* vfs_msg) {
  char* data=malloc(sizeof(char)*vfs_msg->data);
  Message msg;
  msg.msg=data;
  mailbox_get_msg(box,&msg,vfs_msg->data);
  while (msg.from==0 && msg.size==0) {
    yield();
    mailbox_get_msg(box,&msg,sizeof(vfs_message));
  }
  if (msg.from==0) {
    serial_print("Could not recieve fputs data from the VFS\n");
    vfs_msg->flags=2;
    return;
  }
  free(data);
  vfs_msg->flags=2;
  return;
}

char* initrd_gets(vfs_message* vfs_msg) {
  char str[256];
  serial_print("File pos initrd:");
  int_to_ascii(vfs_msg->pos,str);
  serial_print(str);
  serial_print("\n");
  long max_data=size-vfs_msg->pos;
  serial_print("max_data=");
  int_to_ascii(max_data,str);
  serial_print(str);
  serial_print("\n");
  serial_print("Amount requested:");
  int_to_ascii(vfs_msg->data,str);
  serial_print(str);
  serial_print("\n");
  if (vfs_msg->data>max_data) {
    serial_print("OVER\n");
    vfs_msg->flags=2;
    return NULL;
  }
  char* data=malloc(sizeof(char)*vfs_msg->data);
  for (long i=0;i<vfs_msg->data;i++) {
    data[i]=(uint8_t)initrd[i+vfs_msg->pos];
    // if (i<sizeof(vfs_message)) {
    //   serial_print("data[");
    //   char str[256];
    //   int_to_ascii(i,str);
    //   serial_print(str);
    //   serial_print("]=");
    //   hex_to_ascii((uint8_t)data[i],str);
    //   serial_print(str);
    //   serial_print("\n");
    // }
  }
  vfs_msg->flags=0;
  return data;
}

void process_vfs_msg(vfs_message* vfs_msg, uint32_t from) {
  char* gets_data;
  switch (vfs_msg->type) {
    case VFS_PUTS:
    initrd_puts(vfs_msg);
    break;
    case VFS_GETS:
    gets_data=initrd_gets(vfs_msg);
    break;
    break;
    default:
    vfs_msg->flags=1;
  }
  Message msg;
  msg.to=from;
  msg.from=box;
  msg.size=sizeof(vfs_message);
  msg.msg=vfs_msg;
  serial_print("Sending message with flags of ");
  char str[256];
  int_to_ascii(vfs_msg->flags,str);
  serial_print(str);
  serial_print("\n");
  mailbox_send_msg(&msg);
  if (gets_data) {
    serial_print("Amount sending:");
    char str[256];
    int_to_ascii(vfs_msg->data,str);
    serial_print(str);
    serial_print("\n");
    msg.size=vfs_msg->data;
    msg.msg=gets_data;
    mailbox_send_msg(&msg);
    free(gets_data);
  }
}


int main() {
  size=initrd_sz();
  initrd=malloc(size);
  initrd_get(initrd);
  box=mailbox_new(16,"initrd");
  devfs_message* msg_data=malloc(sizeof(devfs_message));
  msg_data->msg_type=DEVFS_REGISTER;
  msg_data->dev_type=DEV_GLOBAL;
  msg_data->mbox=box;
  strcpy(&msg_data->name[0],"initrd");
  Message msg;
  msg.from=box;
  msg.to=mailbox_find_by_name("devfs_register");
  msg.size=sizeof(devfs_message);
  msg.msg=msg_data;
  mailbox_send_msg(&msg);
  free(msg_data);
  yieldToPID(DEVFS_PID);
  for (;;) {
    Message msg;
    msg.msg=malloc(sizeof(vfs_message));
    mailbox_get_msg(box,&msg,sizeof(vfs_message));
    if (msg.from==0) {
      free(msg.msg);
    } else {
      vfs_message* vfs_msg=(vfs_message*)msg.msg;
      process_vfs_msg(vfs_msg,msg.from);
      free(vfs_msg);
      yieldToPID(VFS_PID);
    }
    yield();
  }
}