/*

  Copyright 2008 Nathan Phillip Brink, Ethan Zonca, Matt 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/>.

*/

/* Ideas for php-side stuff:

 *

 * PHP adds the users... has captcha and email confirmation wth a pseudo-random number that

 * the user must enter on the site. Apache should be running mod_peruser for safety, and I'm

 * thinking zserver2 for web serving for convenience...

 */

#include "options.h"

#include "execio.h"

#include <distren.h>

#include <stdio.h> /* sprintf, printf */

#include <stdlib.h> /* malloc, free */

#include <unistd.h> /* getopt */

#include <libxml/tree.h> /* Happy fun XML time */

#include <libxml/xmlwriter.h>

#include <confuse.h>

#include <string.h> // for strcat

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

{

  char *input;

  char *output;

  char *jobid;

  char curopt;

  distren_t distren;

  distren_job_t distren_job;

  struct options_common *options;

  cfg_t *cfg;

  cfg_opt_t cfg_opts[] =

    {

      CFG_STR_LIST("render_types", NULL, 0),

      CFG_STR("username", NULL, 0),

      CFG_STR("name", NULL, 0),

      CFG_STR("email", NULL, 0),

      CFG_END()

    };

  input = NULL;

  output = NULL;

  while(-1 != (curopt = getopt(argc, argv, "i:o:h")))

    {

      if(curopt == ':')

	{

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

	  return 1;

	}

      else if(curopt == '?')

	{

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

	  return 1;

	}

      else if(curopt == 'h')

	{

	  fprintf(stderr, "Usage: %s -i <inputfile> -o <outputfile>\n", argv[0]);

	  fprintf(stderr, "\t-i\tSpecifies the input file\n\

\t-o\tSpecifies the output file\n\

\t-h\tShows this help message\n");

	  /*

	    don't return here because options_init will have an overall

	    options help page

	  */

	}

      else if(curopt == 'i')

	/* 

	   TODO: is strdup good or bad? what about 

	   the idea that all libs should allow plugging different

	   malloc/free implementations in?

	 */

	input = strdup(optarg);

      else if(curopt == 'o')

	output = strdup(optarg);

    } 

  /*

    parse the config file after the arguments so we can intercept -h for help

  */  

  if(options_init(argc, argv, &cfg, cfg_opts, "client", &options))

    {

      fprintf(stderr, "error getting configuration\n");

      //return 1;

    }

  /* 

     give this error after the general arguments parsing so that

     the general help from options_init can have effect 

  */

  if(!input

     || !output)

    {

      fprintf(stderr, "Input and output files must be specified\n");

      return 1;

    }

  fprintf(stderr, "reading from %s\nwriting to %s\n", input, output);

  distren_init_mf(&distren, &malloc, &free);

  distren_submit_file(distren, &distren_job, input);

  distren_job_getid(distren_job, &jobid);

  fprintf(stdout, "jobid: %s\n", jobid);

  distren_job_free(distren_job);

  distren_free(distren);

  // Please find a better way of doing this :( you can't strcat multiple strings.. meh

  // use something like sprintf

  //char *username = cfg_getstr(cfg, "username");

  //char *hostname = cfg_getstr(cfg, "hostname");

  //strcat(username, "@");

  //strcat(username, hostname);

  // SSH's to client machine with the username speficied.

    char buf[10];

    struct execio *testrem;

    char *execargv[] =

      {

	"ssh",

	"username", //cfg_getstr(cfg, "username") . "@" . cfg_getstr(cfg, "hostname"),

	"-i guest.rsa", //yeaaaah any better ideas?

	"sh",

	"-c",

	"\"echo hello from ${HOSTNAME}\"", // "\"useradd -M -c" strcat( cfg_getstr(cfg, "name") cfg_getstr(cfg, "email")) "-d /home/distren --gid 537" cfg_getstr(cfg, "username") "\"",

	(char *)NULL

      };

    size_t readlen;

    fprintf(stderr, "execio madness is occuring!");

    fprintf(stderr, "execio_open returns %d\n", execio_open(&testrem, "ssh", execargv));

    buf[9] = '\0';

    while(!execio_read(testrem, buf, 9, &readlen))

      {

	if(readlen > 9 )

	  {

	    fprintf(stderr, "execio_read doesn't set readlen correctly or read() is messed up\n");

	    return 1;

	  }

	buf[readlen] = '\0';

	fprintf(stderr, "read \"%s\"\n", buf);

      }

    execio_close(testrem);

  // Can we prune off some of this code up here^? I'm not exactly sure how execio returns output, but it seems like more could be happening in execio itself, and less happening where it is actually called.

  // We need some code to ssh, and if the SSH fails, tell the user to register.

  // Killed the user reg code, as the php interface will take care of this. Although it should check for users...

  options_free(options);

  return 0;

};

/*

  Copyright 2008 Nathan Phillip Brink, Ethan Zonca, Matt 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/>.

*/

#ifndef DISTREN_H

#define DISTREN_H 1

/*

  Public API for DistRen

 */

#include <stddef.h> /* size_t */

typedef struct distren *distren_t;

typedef void* (*distren_malloc_t)(size_t);

typedef void (*distren_free_t)(void*);

typedef struct distren_job *distren_job_t;

/**

 Initialize a libdistren handle. In the spirit of respecting client programs' choices,

 this initialization function allows malloc() and free() compatible function pointers

 to be passed.

*/

int distren_init_mf(distren_t *handle, distren_malloc_t mymalloc, distren_free_t myfree);

/**

 Initializes a libdistren handle using the system malloc() and free() implementation.

*/

int distren_init(distren_t *handle);

/**

 Submits a file to distren.

 @param job must not refer to a valid job.

 @TODO create a stream-based interface

*/

int distren_submit_file(distren_t handle, distren_job_t *job, const char *filename);

/**

   Retrieve a job-ID that can be used to refer to the job after the client's process has recycled. Currently, this will be the only way to retrieve a job handle (a distren_job_t) and, consequently, the only way to retrieve a finished render.

   A job ID is representable using a character string. Hopefully, they will be kept at a reasonably small length. It should be made up of alphanumeric characters so as to be useful in most situations. 

   @param jobid This will be set to a string allocated by the specified malloc implementation

   @param job The distren_job_t from which to retrieve a jobid

*/

int distren_job_getid(distren_job_t job, char **jobid);

/*

 TODO: extraction of a job-ID, retrieving job handle by job-ID, getting information

 for a job handle, retrieving results of a job, callbacks for progress (i.e., job_is_finished callback)

 -- threads?

*/

/**

 Destroys and frees a job handle

*/

int distren_job_free(distren_job_t job);

/**

 Destroys and frees a libdistren handle

*/

int distren_free(distren_t handle);

#endif