#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include "./ubixfs.h"

#define typeFile      1
#define typeContainer 2
#define typeDirectory 4


//Michelle My Bell

struct blockAllocationTableEntry *BAT = 0x0;

int findFreeBlock(int cBlock,int size) {
  int i = 0x0;
  for (i=1;i<size;i++) {
    if (BAT[i].attributes == 0x0) {
      if (cBlock != -1) {
        BAT[cBlock].nextBlock = i;
        }
      BAT[i].attributes = 1;
      return(i);
      }
    }
  return(0x0);
  }

int main(int argc,char **argv) {
  FILE *driveFd,*fileFd;
  int startSector = 0x0,size = 0x0,structSize = 0x0,i = 0x0,x = 0x4,block = 0x0,batSize = 0x0;
  int blockCount = 0x0,file = 0x0;
  unsigned long counter = 0x0;
  struct directoryEntry *dirEntry = 0x0;
  struct partitionInformation *partInfo = 0x0;
  if (argc <= 4) {
    printf("Error:\nformat start size files....\n");
    exit(1);
    }
  driveFd = fopen(argv[3],"wb");
  startSector = atoi(argv[1])-1;
  size = atoi(argv[2])*512;
  structSize = sizeof(struct blockAllocationTableEntry);
  structSize *= (size/4096);
  batSize = (((structSize+511)/512)*512);
  BAT = (struct blockAllocationTableEntry *)malloc(structSize);
  dirEntry = (struct directoryEntry *)malloc(0x4000);
  memset(dirEntry,0x0,0x4000);
  partInfo = (struct partitionInformation *)malloc(512);
  partInfo->size = (size/512);
  partInfo->startSector = (startSector+1);
  partInfo->blockAllocationTable = (startSector+1);
  partInfo->rootDirectory = ((startSector+1) + (batSize/512));
  partInfo->rootDirectorySize = 0x4000/512;
  fseek(driveFd,(startSector * 512),0);
  fwrite(partInfo,512,1,driveFd);
  startSector++;
  dirEntry[0].attributes   = typeDirectory;
  dirEntry[0].permissions  = 0xEAA;  
  dirEntry[0].gid          = 0x0;
  dirEntry[0].uid          = 0x0;
  dirEntry[0].startCluster = 0x0;
  dirEntry[0].size         = 0x4000;
  sprintf(dirEntry[0].fileName,"/");  
  dirEntry[1].attributes   = typeDirectory;
  dirEntry[1].permissions  = 0xEAA;
  dirEntry[1].gid          = 0x0;
  dirEntry[1].uid          = 0x0;
  dirEntry[1].startCluster = 0x0;
  dirEntry[1].size         = 0x4000;
  sprintf(dirEntry[1].fileName,"..");
  BAT[0].nextBlock  = 0x1;
  BAT[0].attributes = 1;
  BAT[0].realSector = (batSize/512);
  BAT[1].nextBlock  = 0x2;
  BAT[1].attributes = 1;
  BAT[1].realSector = ((batSize/512) + 1 * 8);
  BAT[2].nextBlock  = 0x3;
  BAT[2].attributes = 1;
  BAT[2].realSector = ((batSize/512) + 2 * 8);
  BAT[3].nextBlock  = -1;
  BAT[3].attributes = 1;
  BAT[3].realSector = ((batSize/512) + 3 * 8);      
  
  for (i=4;i<(size/4096);i++) {
    BAT[i].nextBlock  =  -1;
    BAT[i].attributes = 0x0;
    BAT[i].realSector = ((batSize/512) + (i*8));
    BAT[i].reserved   = 0x0;
    }
  file = 0x2;
  while (x < argc) {
    counter = 0x0;
    blockCount = 0x0;
    fileFd = fopen(argv[x],"rb");
    block = findFreeBlock(-1,(size/4096));
    dirEntry[file].startCluster = block;
    dirEntry[file].attributes   = typeFile;
    dirEntry[file].permissions  = atoi(argv[x+1]);
    rewind(driveFd);
    /* fseek(driveFd,((BAT[startSector].realSector * 512) + ((batSize/512)+(BAT[block].realSector*4096))),0); */
    fseek(driveFd,((startSector + BAT[block].realSector) * 512),0);
    while (!feof(fileFd)) {
      if (4096 == (counter - (blockCount * 4096))) {
        block = findFreeBlock(block,(size/4096));
        printf("Block: [%i][%s]\n",block,argv[x]);
        rewind(driveFd);
        /* fseek(driveFd,((startSector * 512) + ((batSize/512)+(BAT[block].realSector*4096))),0); */
        fseek(driveFd,((startSector + BAT[block].realSector) * 512),0);
        blockCount++;
        }
      fputc(fgetc(fileFd),driveFd);
      counter++;
      }
    i = 0;
    while((counter + i)%512) {
      fputc(0x0,driveFd);
      i++;
      }
    fclose(fileFd);
    dirEntry[file].size = (counter - 1);
    dirEntry[file].uid  = 0x0;
    dirEntry[file].gid  = 0x0;
    sprintf(dirEntry[file].fileName,"%s",argv[x]);
    x += 2;
    file++;
    }
  dirEntry[1].attributes   = typeDirectory;
  dirEntry[1].permissions  = 0xEAA;
  dirEntry[1].gid          = 0x0;
  dirEntry[1].uid          = 0x0;
  dirEntry[1].startCluster = 0x0;
  dirEntry[1].size         = 0x4000;
  sprintf(dirEntry[1].fileName,"fakeDir");    
  rewind(driveFd);  
  fseek(driveFd,(long)(((startSector) * 512) + batSize),0);
  fwrite(dirEntry,0x4000,1,driveFd);
  rewind(driveFd);
  fseek(driveFd,((startSector) * 512),0);
  if (fwrite(BAT,batSize,1,driveFd) >= 1) {
/*
 *    printf("size         [%i]\n",partInfo->size);
 *    printf("startSector: [%i]\n",partInfo->startSector);
 *    printf("bat:         [%i]\n",partInfo->blockAllocationTable);
 *    printf("rootDir:     [%i]\n",partInfo->rootDirectory);
 *    printf("sizeof:      [%i]\n",sizeof(struct blockAllocationTableEntry));
 *    printf("size:        [%i]\n",(size/4096));
 */
    printf("Formatted!\n");
    }
  else {
    printf("Error Formatting!!\n");
    }
  fclose(driveFd);
  return(0);
  }