/**

   This file stores the distrenjob and frameset structs and prototypes for some functions to manipulate/use these.

 */

#include <time.h> /* clock_t, time_t */

typedef unsigned int jobnum_t;

/**

   Stores Blender Job Info

*/

struct distrenjob {

  struct distrenjob *next; /*< next will be NULL unless if there is another distrenjob */

  char *name;

  char *submitter;

  char *email; /*< This should be looked up based on the value of submitter, not stored in this struct */

  jobnum_t jobnum;

  int priority;  // 1 is lowest, 10 is highest, 0 means the job is done

  float percent_done;

  int completed_frames; // number of completed frames for stats/etc

  int assigned_frames; // number of assigned frames (that are not yet completed) for stats/etc

  int total_frames; // how many frames are in the animation for stats/etc (unassigned frames)

  time_t avg_render_time; // average seconds it took to render a frame

  time_t time_remaining; /*< estimated seconds remaining till render is complete (up to 49, 710 days) */

  struct frameset *frameset;

};

/**

   Frameset Structure

*/

struct frameset {

  int num; /*< frame number to render */

  char slave_name; /*< user that frame is assigned to */

  int status; /*< status of frame, 0= unassigned, 1= taken, 2= done */

  clock_t start_time; /*< time the frame was started */

  int time_to_render; /*< the total seconds it took to render the frame */

}; /* Frameset array is generated by status_report_generator() */

/*

related functions 

*/

/**

   @param distrenjob the address where we will store the pointer of a malloc()ed

  Copyright 2009 Nathan Phillip Brink, Ethan Zonca, Matthew Orlando

  This file is a part of DistRen.

  DistRen is free software: you can redistribute it and/or modify

  it under the terms of the GNU Affero General Public License as published by

  the Free Software Foundation, either version 3 of the License, or

  (at your option) any later version.

  DistRen is distributed in the hope that it will be useful,

  but WITHOUT ANY WARRANTY; without even the implied warranty of

  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the

  GNU Affero General Public License for more details.

  You should have received a copy of the GNU Affero General Public License

  along with DistRen.  If not, see <http://www.gnu.org/licenses/>.

 */

// Provides DISTREN_REQUEST* which is in reality, DISTREN_SEND(signal) in the minds of your average joe

#include "protocol.h"

#include "options.h" // Confuse, etc.

#include "slavefuncs.h" // Slave functions

#include "asprintf.h"

#include <string.h>

#include <stdio.h>

#include <unistd.h>

#include <sys/stat.h> /*< mkdir() */

#include <stdlib.h>

int slavestatus = 0; // Ugh global vars

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

{

int counter;

/* Parses arguments, skips if there are no args */

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

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

		fprintf(stderr, "Usage: distrenslave [option] <username> <emailaddr>\nStarts a distren slave\n\t-h\tshow this help\n\t-c\tregisters a user with [username] and [emailaddr] \n");

		return 2;

	}

}

	if(argc>1

		&& (strcmp("-c", argv[1]) == 0))

	{

      }

      else{

        fprintf(stderr, "Using cached job file...\n");

      }

    _distren_asprintf(&outdir, "/tmp/distren/job%d", jobnum); /*< @todo free() */

    mkdir("/tmp/distren", 0750); /* @TODO: Make this less *nix-specific */

    mkdir(outdir, 0750);

    _distren_asprintf(&tarcmd, "tar -xvf \"%s\" -C \"%s\"", pathtoTar, outdir); /* @TODO: Make this portable. Libtar or something? */

    system(tarcmd);

    free(tarcmd);

    // Parses a job's XML file, puts data in the myjob struct

    if(xml2distrenjob(&myjob, pathtoXml) == 0){

      fprintf(stderr,"Well, the XML craziness may have worked. Maybe. \n");

      distrenjob_free(&myjob); // Frees things up if it was successful. xml2distrenjob() really (usually) only fails if malloc'ing inside it fails

    }

    else{

      fprintf(stderr,"I think the XML craziness may have failed, so I'll terminate just for fun.\n");

      return 1;

    }

    /* Variable-fillers which require XML */

    _distren_asprintf(&pathtoOutput, "%s/job%d/output/frame%d.%s", datadir, jobnum, framenum, outputExt ); // Prepares the path to the jobfile

    // Consider placing the following in the exec_blender() function

    while(busy == 1){

      tell_the_server(DISTREN_REQUEST_PROGRESS);

        fprintf(stderr, "Rendering frame %d in job %d...",framenum,jobnum);

        sleep(5); // or not... this should be more event-driven, but should still give a heartbeat to the server

      }

    busy = 2;

    // busy = 1; // @TODO: When done rendering, set this.

     }

    // When blender is finished, run this...

      if(busy==2){

        fprintf(stderr, "Finished frame %d in job %d, uploading...",framenum,jobnum);

        char *outputurl;

        _distren_asprintf(&outputurl, "http://protofusion.org/distren/stor/job%d/",jobnum); // Aggregates the output url

        curlpost(pathtoOutput, urltoOutput); // uploads (HTML POST) the output at outputpath to the server at outputurl

        tell_the_server(DISTREN_REQUEST_DONEFRAME); // Tells the server that it's finished rendering this frame

        busy=0;

        // Slave now becomes idle, doesn't need to tell the server anything, ssh handles this.

      }

  return 0;

}