/*

  Copyright 2010 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/>.

*/

#include "common/config.h"

#include "slavefuncs.h"

#include "common/asprintf.h"

#include "common/options.h"

#include <stdio.h>

#include <stdlib.h>

#include <string.h>

#include <sys/stat.h>

#include <unistd.h>

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

{

  char *datadir;

  char *server;

  char *username;

  char *password;

  char *hostname;

  multiio_context_t multiio;

  cfg_opt_t myopts[] = {

    CFG_SIMPLE_STR("username", &username),

    CFG_SIMPLE_STR("password", &password),

    CFG_SIMPLE_STR("datadir", &datadir),

    CFG_SIMPLE_STR("server", &server),

    CFG_SIMPLE_STR("hostname", &hostname),

    CFG_END()

  };

  cfg_t * my_cfg;

  struct options_common *commonopts;

  /**

     initializations

  */

  datadir = NULL;

  server = NULL;

  username = NULL;

  password = NULL;

  hostname = NULL;

  char curopt;

  int runBenchmark = 0;

  int updateConf = 0;

  multiio = multiio_context_new();

  while(((char)-1) != (curopt = getopt(argc, argv, "u:p:rh")))

     {

       if(curopt == ':')

         {

           fprintf(stderr, "-%c: is missing an argument\n", optopt);

           return 1;

         }

       else if(curopt == '?')

         {

           fprintf(stderr, "-%c: invalid option specified\n", optopt);

/*

  Copyright 2010 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/>.

*/

#include "common/config.h"

#include "distrenjob.h"

#include "slavefuncs.h"

#include "common/asprintf.h"

#include "common/execio.h"

#include "common/protocol.h"

#include "common/remoteio.h"

#include <curl/curl.h>

#include <curl/easy.h>

#include <curl/types.h>

#include <archive.h>

#include <archive_entry.h>

#include <errno.h>

#include <stdio.h>

#include <string.h>

#include <unistd.h>

#include <stdlib.h>

#include <sys/stat.h>

#include <fcntl.h>

#include <time.h>

#define DEBUG 0

/**

   Grabs the xml DOM node reached by an XPath.

   @param path an XPath that lead to DOM node

   @return the first node associated with the path or NULL if there is no match

 */

xmlNodePtr xml_quickxpath(xmlXPathContextPtr xpathctxt, xmlChar *path)

{

  xmlNodePtr toreturn;

  xmlXPathObjectPtr xmlxpathobjptr;

  xmlxpathobjptr = xmlXPathEval(path, xpathctxt);

  if(!xmlxpathobjptr

     || !xmlxpathobjptr->nodesetval->nodeNr)

    {

      fprintf(stderr, "XPath resolution failed for ``%s'' in ``%s'' (``%s'')\n", path, xpathctxt->doc->name, xpathctxt->doc->URL);

      return (xmlNodePtr)NULL;

    }

  toreturn = *(xmlxpathobjptr->nodesetval->nodeTab);

  xmlXPathFreeObject(xmlxpathobjptr);

  return toreturn;

}

/** Deletes job data from the disk. @TODO: add error checking! */

int delete_jobdata(int jobnum, char *datadir)

{

  char *jobpath;

  char *jobcont;

  _distren_asprintf(&jobpath, "%s/%d", datadir, jobnum);

  _distren_asprintf(&jobcont, "%s/%d/*", datadir, jobnum);

  remove(jobcont);

  rmdir(jobpath);

  fprintf(stderr, "Removed files in %s/%d/ if everything was successful\n", jobpath, jobnum);

  free(jobpath);

  free(jobcont);

  return 0;

}

/** Function referenced by curlget() to write data to disk. */

size_t curl_writetodisk(void *ptr, size_t size, size_t nmemb, FILE *stream)

      if(ret == 0){

        fprintf(stderr, "Job data retrieved successfully\n");

        // free(url); @FIXME: causes doublefree! Curl must free the url?

        return 0;

      }

      else if(ret == 1){

        fprintf(stderr, "Downloading job data from %s failed. Check your network connection.\n",url);

        free(url);

        return 1; // Eventually make a retry loop

      }

    }

  fprintf(stderr, "Tar at %s already exists! Download cancelled.\n", destinationPath);

  return 2;

}

int uploadOutput(char *pathtoOutput, char *urltoOutput, int jobnum, int framenum, int slavekey){

  //fprintf(stderr,"Uploading output %s to url %s for job %d and frame %d for slavekey %d", pathtoOutput, urltoOutput, jobnum, framenum, slavekey);

  if( !curlpost(pathtoOutput, urltoOutput, jobnum, framenum, slavekey)) // Uploads output

    {

      if(DEBUG)

        fprintf(stderr,"Upload successful, removing old output...\n");

      else

        fprintf(stderr,"Upload completed\n");

      remove(pathtoOutput); // Delete the file after its uploaded

      return 0;

    }

  else

    {

      fprintf(stderr,"Upload failed. Check your network connection. Retrying upload...\n");

      int tries=1;

      while(tries<=10 && curlpost(pathtoOutput, urltoOutput, jobnum, framenum, slavekey))

        {

          fprintf(stderr, "Upload failed. Trying again in 10 seconds... (attempt %d of 10)\n", tries);

          tries++;

          sleep(10);

        }

      return 1; // Upload failed after multiple tries

      // @FUTURE: Keep track of files that we were unable to upload, and upload them later

    }

}

void prepareJobPaths(int jobnum, int framenum, char *outputExt, char *datadir, char **urltoTar,char **pathtoTar,char **pathtoTardir,char **pathtoJob, char **pathtoJobfile, char **urltoJobfile, char **urltoOutput,char **pathtoOutput, char **pathtoRenderOutput, char **pathtoOutdir)

{

  // Variable Preparation

  char *jobdatapath;

   _distren_asprintf(&jobdatapath, "job%d", jobnum);

   _distren_asprintf(urltoJobfile, "http://data.distren.org/job%d/job.blend", jobnum); // Prepares URL to download from

   _distren_asprintf(pathtoTardir, "%s/%s/", datadir, jobdatapath); // Prepares destination to save to

   _distren_asprintf(pathtoJob, "%s/%s/", datadir, jobdatapath);

   _distren_asprintf(pathtoJobfile, "%s/%s/job.blend", datadir, jobdatapath ); // Prepares the path to the jobfile

   _distren_asprintf(urltoOutput, "http://distren.org/slaveUpload.php?jobkey=%d&framekey=%d",jobnum,framenum); // Prepares the URL where output is posted

   _distren_asprintf(pathtoRenderOutput, "%s/%s/output/job%d-frame#.%s", datadir, jobdatapath, jobnum, outputExt ); // Note: the # is for blender to put in framenum in output file

   _distren_asprintf(pathtoOutput, "%s/%s/output/job%d-frame%d.%s", datadir, jobdatapath, jobnum, framenum, outputExt ); // Note: the # is for blender to put in framenum in output file

   _distren_asprintf(pathtoOutdir, "%s/%s/output", datadir, jobdatapath);

   free(jobdatapath);

}

int checkUsername(char *username)

{

  if(username == NULL || strcmp(username, "!username") == 0 )

    {

     fprintf(stderr, "\nPlease ensure that your slave ID is present in distrenslave.conf, or use the -u flag to set a slave ID.\n");

     return 1;

   }

   else

     if( username != NULL || strcmp(username, "!username") != 0 )

       {

         // Log in the user

         if(login_user(username) == 1){

           // Logged in OK

           return 0;

         }

         else

           {

             fprintf(stderr, "Login failed, please check your username. If you have not registered, please do so on the DistRen website.\n");

             return 1;

           }

       }

     else

       {

         fprintf(stderr, "Please check your distrenslave.conf, it appears to be incorrectly formatted.\n");

         return 1;

       }

}

void slaveTest(char *datadir)

{

  int command;

  int test = 1;

  int jobnum = 0;