#include "common/execio.h"

#include "common/options.h"

#include "common/protocol.h"

#include <confuse.h>

#include <malloc.h>

#include <stdio.h>

#include <stdlib.h>

#include <string.h>

#include <sys/stat.h>

#include <sys/types.h>

#include <time.h>

#include <unistd.h>

#include <libxml/encoding.h>

#include <libxml/parser.h>

#include <libxml/tree.h>

#include <libxml/xmlmemory.h>

#include <libxml/xmlreader.h>

#include <libxml/xmlwriter.h>

/* ******************* Structs ************************ */

struct general_info

{

  struct distrenjob head;

  struct distrend_config *config;

  struct

  {

    /** general_info.xml */

    char *geninfo;

  } files;

  int jobs_in_queue;

  unsigned int free_clients;

  unsigned int rendering_clients;

  unsigned int total_finished_jobs;

  unsigned int total_frames_rendered;

  unsigned int highest_jobnum;

  int hibernate;

  time_t timestamp;

  unsigned long total_render_power;

  unsigned long total_priority_pieces;

};

/* ************General Functions************* */

int xml_dump();

int distrend_do();

int start_data(struct general_info *general_info);

void finish_frame(struct general_info *geninfo, struct distrenjob *distrenjob, int frame);

int mortition(struct general_info *geninfo, struct distrenjob *job);

int restoreJobState(struct distrenjob *job);

int prepare_distrenjob(struct general_info *geninfo, int type, char *name, char *submitter, int priority, int start_frame, int end_frame, int width, int height);

int distrenjob_enqueue(struct general_info *geninfo, struct distrenjob *job);

int change_job_priority(struct general_info *geninfo, struct distrenjob *job, int new_priority);

int find_jobframe(struct general_info *geninfo, struct distrenjob **job, struct frameset **frame);

int find_jobframe_from_job(struct distrenjob *distrenjob_ptr, struct distrenjob **job, struct frameset **frame);

int find_jobframe_new(struct general_info *geninfo, int rend_pwr, struct distrenjob **job, struct frameset **frame);

int find_jobframe_again(struct general_info *geninfo, int jobnum, int rend_pwr, struct distrenjob **job, struct frameset **frame);

void frame_watchdog(struct distrenjob *distrenjob_head);

struct distrenjob *distrenjob_get(struct distrenjob *head, jobnum_t jobnum);

void distrenjob_remove(struct general_info *geninfo, struct distrenjob *bj);

int distrend_do_config(int argc, char *argv[], struct distrend_config **config);

int distrend_config_free(struct distrend_config *config);

/* **************XML Functions**************** */

void update_general_info(struct general_info *geninfo);

int import_general_info(struct general_info *general_info);

int restore_distrenjob(struct general_info *geninfo, struct distrenjob **distrenjob, jobnum_t jobnum);

int updateJobStatsXML(struct distrenjob *job);

int update_xml_joblist(struct general_info *geninfo);

int createQueueFromXML(struct general_info *geninfo);

int reCreateQueueFromXML(struct general_info *geninfo, xmlDocPtr doc, xmlNodePtr current);

/* **************Test Functions**************** */

int printFrameInfo(struct frameset *frame);

int printJob(struct distrenjob *job);

int printJobInfo(struct distrenjob *job);

int printAllJobnums(struct distrenjob *head);

int interactiveTest(int test, struct general_info general_info);

int job_getserialfilename(char **filename, struct general_info *geninfo, unsigned int jobnum, int createdir);

int distren_mkdir_recurse(char *dirname);

int main(int argc, char *argv[])

{

  /* Parse arguments */

  int counter;

  int test = 0; /*< Interactive mode if 1 */

  struct general_info general_info;

  struct distrend_clientset *clients;

  enum clientstatus

  {

    CLIENTSTATUS_UNINITIALIZED = 0,

    CLIENTSTATUS_BUSY = 1,

    CLIENTSTATUS_IDLE = 2

  } clientstatus;

  xmlinit();

  for(counter = 0; counter < argc; counter ++)

    {

      if(strcmp(argv[counter], "-h") == 0)

      {

    	  fprintf(stderr, "Usage: distrend [option] \nStarts the distrend server\n\t-h\tshow this help\n\t-t\tlaunches queue testing interface \n");

      }

      else if(strcmp(argv[counter], "-t") == 0)

      {

    	  fprintf(stderr, "Entering into test mode...\n\n");

      }

    }

  if(distrend_do_config(argc, argv, &general_info.config))

    return 1;

  /** preset paths */

  _distren_asprintf(&general_info.files.geninfo, "%s/general_info.xml",

		    general_info.config->datadir);

  if(start_data(&general_info))

    {

      fprintf(stderr, "%s:%d: start_data() failed\n", __FILE__, __LINE__);

      return 1;

    }

  /** pre-loaded jobs for testing */

  prepare_distrenjob(&general_info, 1, "awesome", "LordOfWar", 8, 1, 100, 640, 480);

  prepare_distrenjob(&general_info, 1, "hamburger", "ohnobinki", 3, 1, 50, 1280, 720);

  distrend_listen(general_info.config, &clients);

  while(!general_info.config->die)

    {

      int clientsays = 0; /*< temporary example variable, will be replaced when we can handle messages */

      distrend_accept(general_info.config, clients);

      /* Make the following code more event-driven */

      frame_watchdog(&general_info.head);

      struct frameset *frame;

      struct distrenjob *job;

      /* If the client is idle, must be modified for climbing through linked list of clients (client->clientnum) */

      if(clientstatus == CLIENTSTATUS_IDLE)

	{

	  int returnnum = find_jobframe(&general_info, &job, &frame); // Finds a frame to render

	  if(returnnum)

	    {

	      fprintf(stderr,"No frames are available to render at this time. Idling...\n");

	      sleep(10);

	    }

	  else

	    remotio_send_to_client(frame->num, job->jobnum); // Pseudo-sends data to client

	}

      /* If the client states that they finished the frame */

      	if(clientsays == DISTREN_REQUEST_DONEFRAME){

      	  clientstatus = CLIENTSTATUS_IDLE; // Sets the client back to idle

      	  finish_frame(&general_info, job, frame->num); // @TODO: Make sure this actually works.

      	}

    } /* while(!general_info.config->die) */

  distrend_unlisten(general_info.config->listens, clients);

  distrend_config_free(general_info.config);

  xmlcleanup();

  /** free() paths */

  free(general_info.files.geninfo);

  return 0;

}

/* ********************** Functions ************************* */

/** Dumps all data in RAM to an xml file (such as current jobs, etc) which is parsed by start_data. Remember to invoke this before shutting down! @TODO: Fill this stub*/

int xml_dump()

{

  return 0;

}

/**

   Performs command stored in a client's request. @TODO: Fill stub

*/

int distrend_do()

{

  return 0;

}

}

/**

   inserts jobs at front of queue, starting

   with the last job in the job_list xml file

   to preserve the order of the queue

*/

int reCreateQueueFromXML(struct general_info *geninfo, xmlDocPtr doc, xmlNodePtr current)

{

  struct distrenjob *holder; // holds the job that "was" after head, so that the new struct can be inserted after head

  struct distrenjob *job; // job to be added

  xmlChar *tmp;

  jobnum_t job_num;

  if(current == NULL) // base case, if element doesn't exist then don't do anything

    return 0;

  /**

     recursively call itself so that the next job in

     the queue is added to the front before the current one

  */

  if(reCreateQueueFromXML(geninfo, doc, current->next))

    /**

       bail if failure recursevely

        ^-- is this the best thing to do in this case?

        @TODO bail-out cleanup

     */

    return 1;

  /** actual work is done now: add the job to the front of the queue */

  holder = geninfo->head.next;

  tmp = xmlNodeListGetString(doc, current->xmlChildrenNode, 1); // get job number

  job_num = atoi((char*)tmp);

  xmlFree(tmp);

  if(restore_distrenjob(geninfo, &job, job_num))

    return 1;

  /**

     insert job at front of the queue

  */

  geninfo->head.next = job;

  job->next = holder;

  return 0;

}

/* ************************** Test Functions ************************* */

int printFrameInfo(struct frameset *frame)

{

  char *status;

  status = NULL;

  switch(frame->status)

  {

    case FRAMESETSTATUS_UNASSIGNED:

      _distren_asprintf(&status, "%s", "unassigned");

      break;

    case FRAMESETSTATUS_ASSIGNED:

      _distren_asprintf(&status, "%s", "assigned");

      break;

    case FRAMESETSTATUS_DONE:

      _distren_asprintf(&status, "%s", "completed");

      break;

    case FRAMESETSTATUS_CANCELED:

      _distren_asprintf(&status, "%s", "canceled");

  }

  printf("frame #: %d --> %s\n", frame->num, status);

  free(status);

  return 0;

}

int printJob(struct distrenjob *job)

{

  int counter;

  fprintf(stderr, "frame_num: status\n");

  for(counter = 0; counter < job->total_frames; counter++)

    {

      printFrameInfo(&job->frameset[counter]);

    }

  return 1;

}

int printJobInfo(struct distrenjob *job)

{

  fprintf(stderr, "type: %d\n", job->type);

  fprintf(stderr, "name: %s\n", job->name);

  fprintf(stderr, "submitter: %s\n", job->submitter);

  fprintf(stderr, "priority: %d\n", job->priority);

  fprintf(stderr, "completed: %d\n", job->completed_frames);

  fprintf(stderr, "assigned: %d\n", job->assigned_frames);

  fprintf(stderr, "total: %d\n", job->total_frames);

  fprintf(stderr, "watchdog: %d\n", job->watchdog_forgiveness);

  fprintf(stderr, "hibernate: %d\n", job->hibernate);

  fprintf(stderr, "prev_frame: %d\n", job->prev_frame_index);

  fprintf(stderr, "render power: %ld\n", job->assigned_render_power);

  return 1;

}

int printAllJobnums(struct distrenjob *head)

{

  struct distrenjob *current_job;

  int total_jobs;

  fprintf(stderr, "job numbers in the order they will be processed:\n");

  total_jobs = 0;

  for(current_job = head->next; current_job; current_job = current_job->next)

    {

      fprintf(stderr, "%d: %s\n", current_job->jobnum, current_job->name);

      total_jobs++;

    }

  fprintf(stderr, "\n%d jobs in queue\n\n", total_jobs);

  return 1;

}

int interactiveTest(int test, struct general_info general_info){

  int command;

  jobnum_t jobnum;

  struct distrenjob *tmp_job;

  struct frameset *tmp_frame;

  int type;

  char *name;

  char *submitter;

  char *email;

  int priority;

  int width;

  int height;

  int start_frame;

  int end_frame;

  size_t read_buf;

  while(test == 1)

   {

     fprintf(stderr, "Welcome to DistRen Alpha Interactive Test Mode\n\n");

     fprintf(stderr, "\t1 \tPrint all frames in a job\n");

     fprintf(stderr, "\t2 \tExamine certain job\n");

     fprintf(stderr, "\t3 \tGet a frame to render\n");

     fprintf(stderr, "\t4 \tAdd a job\n");

     fprintf(stderr, "\t5 \tDelete a job\n");

     fprintf(stderr, "\t6 \tPrint jobnums in queue\n");

     fprintf(stderr, "\t7 \tPrint general info\n");

     fprintf(stderr, "\t8 \tQuit\n");

     scanf("%d", &command);

     switch(command)

     {

     case 1:

       fprintf(stderr, "Job number: ");

       scanf("%d", &jobnum);

       printJob(distrenjob_get(&general_info.head, jobnum));

       break;

     case 2:

       fprintf(stderr, "Job number: ");

       scanf("%d", &jobnum);

       printJobInfo(distrenjob_get(&general_info.head, jobnum));

       break;

     case 3:

       general_info.total_render_power ++;

       if(!find_jobframe_again(&general_info, -1, 1, &tmp_job, &tmp_frame))

       {

           fprintf(stderr, "frame was found, details below\n");

           fprintf(stderr, "Job#:%d\n", tmp_job->jobnum);

     case 5:

       fprintf(stderr, "\nJob number: ");

       scanf("%d", &jobnum);

       distrenjob_remove(&general_info, distrenjob_get(&general_info.head, jobnum));

       break;

     case 6:

       printAllJobnums(&general_info.head);

       break;

     case 7:

       fprintf(stderr, "\nHighest job number: %d", general_info.highest_jobnum);

       fprintf(stderr, "\nJobs in queue: %d", general_info.jobs_in_queue);

       fprintf(stderr, "\nTotal frames rendered: %d", general_info.total_frames_rendered);

       fprintf(stderr, "\nTimestamp: %lu", (long)general_info.timestamp);

       fprintf(stderr, "\nTotal priority pieces: %ld", general_info.total_priority_pieces);

       fprintf(stderr, "\nTotal render power: %ld\n", general_info.total_render_power);

       break;

     case 8:

       fprintf(stderr,"Goodbye.\n");

       test = 0;

     default:

       fprintf(stderr, "Invalid input, please try again.\n");

     }

   }

  return 0;

}

/**

   constructs the filename for a distrenjob's serialized XML

   to be stored/retrieved from/in.

   @arg filename pointer to filename allocated using something

           malloc() and free() compatible. Must be free()ed by the

	   caller

   @return 0 on success

 */

int job_getserialfilename(char **filename, struct general_info *geninfo, unsigned int jobnum, int createdir)

{

  _distren_asprintf(filename, "%s/stor/job/%d/distrenjob.xml",

		    geninfo->config->datadir,

		    jobnum);

  if(!*filename)

    return 1;

  if(createdir)

    return distren_mkdir_recurse(*filename);

  return 0;

}

int distren_mkdir_recurse(char *dirname)

{

  size_t counter;

  char *nextdir;

  nextdir = strdup(dirname);

  for(counter = 0; nextdir[counter]; counter ++)

    {

      /** @TODO OS-portabalize the path-separators */

      if(nextdir[counter] == '/')

	{

	  nextdir[counter] = '\0';

	  mkdir(nextdir, S_IRWXU | S_IRGRP | S_IROTH);

	  nextdir[counter] = '/';

	}

    }

  return 0;

}