/*

  Copyright 2009 Nathan Phillip Brink

  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_LIBREMOTEIO_H

#define _DISTREN_LIBREMOTEIO_H

#include "remoteio.h"

/**

  private declarations for remoteio, to be included by options.c and remoteio.c

 */

enum remoteio_method

  {

    REMOTEIO_METHOD_SSH,

    /*    REMOTEIO_METHOD_TCP */ /*< someday, maybe */

    /*    REMOETIO_METHOD_XMLRPC */ /*< again, maybe someday */

  };

struct remoteio_server

{

  struct remoteio_server *next;

  char *name;

  char *username;

  enum remoteio_method method;

  unsigned int types; /*< See ``Server types'' in protocol.h */

};

struct remoteio_opts

{

  char *ssh_command;

};

#endif

/*

  Copyright 2008 Nathan Phillip Brink, Ethan Zonca

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

*/

/* The purpose of this file is to... */

#include "options.h"

/* privat-ish files for components we need to config */

#include "libremoteio.h"

#include <confuse.h>

#include <string.h>

#include <stdio.h>

#include <stdlib.h>

#include <unistd.h>

struct options_common_data

{

  cfg_t *cfg;

};

/**

   Not reentrant because of call to getenv()

 */

int options_init(int argc, char *argv[], cfg_t **mycfg, cfg_opt_t *myopts, char *myname, struct options_common **allopts)

{

  char *configfileprefix;

  char *configfile;

  char *optstring = "hc:";

  char curopt;

  configfileprefix = NULL;

  while((curopt = getopt(argc, argv, optstring)) != -1)

    switch(curopt)

      {

      case 'h':

	fprintf(stderr, "libdistren common options\n\

\n\

\t-h\tShow this help.\n\

\t-c <path>\tBasename for configuration files. For instance, if distrencommon.conf is located at /etc/distren/distrencommon.conf, set this option /etc/distren/distren . Default value: %s\n\

\n", SYSCONFDIR "/distren");

	return 2;

	break;

      case 'c':

	configfileprefix = strdup(optarg);

	if(!configfileprefix)

	  /* return is unnecessary here */

	  fprintf(stderr, "OOM\n");

	break;

      case ':':

	fprintf(stderr, "Option -%c requires an argument\n", optopt);

	return 1;

      }

  /* restore optind for other people who use getopt */

  optind = 1;

  if(!configfileprefix

     && (configfileprefix = getenv("DISTREN_CONFIG_PREFIX")) )

    {

      configfileprefix = strdup(configfileprefix);

      if(!configfileprefix)

	{

	  fprintf(stderr, "OOM\n");

	  return 1;

	}

    }

  /* For those of you who don't know, the following is an example of concatenation of constant strings by the C compiler. Now, that doesn't mean you can do run-time string concatenation ;-)

     strdup is because someday configfile will be customizable via argv[]

  */

  if(!configfileprefix)

      configfileprefix = strdup(SYSCONFDIR "/distren");

  if(!configfileprefix)

    {

      fprintf(stderr, "OOM\n");

      return 1;

    }

  i = strlen(configfileprefix) + strlen(myname) + strlen(".conf") + 1;

  configfile = malloc(i);

  if(!configfile)

    {

      perror("malloc");

      return 1;

    }

  strncpy(configfile, configfileprefix, strlen(configfileprefix) + 1);

  strcat(configfile, myname);

  strcat(configfile, ".conf");

  fprintf(stderr, "using configuration file: %s\n", configfile);

  /* initialize structs */

  *allopts = malloc(sizeof(struct options_common));

  if(!*allopts)

    {

      perror("malloc");

      free(configfile);

      return 1;

    }

  memset(*allopts, '\0', sizeof(struct options_common));

  (*allopts)->data = malloc(sizeof(struct options_common_data));

  if(!(*allopts)->data)

    {

      perror("malloc");

      free(configfile);

      free(*allopts);

      return 1;

    }

  memset((*allopts)->data, '\0', sizeof(struct options_common_data));

  (*allopts)->remoteio = malloc(sizeof(struct remoteio_opts));

  if(!(*allopts)->data)

    {

      perror("malloc");

      free(configfile);

      free((*allopts)->data);

      free(*allopts);

      return 1;

    }

  memset((*allopts)->remoteio, '\0', sizeof(struct remoteio_opts));

  (*allopts)->remoteio->ssh_command = strdup("ssh");

  /** @TODO: check for NULL return above */

  cfg_opt_t common_opts[] =

    {

      CFG_SIMPLE_STR("ssh-command", &(*allopts)->remoteio->ssh_command),

      CFG_END()

    };

  /**

    In these arrays, I should replace NULL with a pointer to a string in the struct which should hold the result of confuse processing an option. This is partially because confuse will not free the values it parses.

    the server sections in the distrencommon.conf describe different servers that the client may connect to

   */

  cfg_opt_t server_opts[] =

    {

      CFG_STR("username", NULL, CFGF_NONE),

      CFG_STR("hostname", NULL, CFGF_NONE),

      CFG_STR("method", "ssh", CFGF_NONE),

      CFG_STR_LIST("types", NULL, CFGF_NONE),

      CFG_END()

    };

  cfg_opt_t opts[] =

    {

      CFG_SEC("common",

	      common_opts,

	      CFGF_NONE),

      CFG_SEC("server",

	      server_opts,

	      CFGF_MULTI | CFGF_TITLE),

      CFG_SEC(myname,

	      myopts,

	      0),

      CFG_FUNC("include",

	       &cfg_include),

      CFG_END()

    };

  (*allopts)->data->cfg = cfg_init(opts, 0);

  switch(cfg_parse((*allopts)->data->cfg, configfile))

    {

    default:

      fprintf(stderr, "cfg_parse returned an unknown error code\n");

    case CFG_FILE_ERROR:

      cfg_error((*allopts)->data->cfg, "Couldn't open file ``%s''\n", configfile);

      /* no break; on purpose */

    case CFG_PARSE_ERROR:

      free((*allopts)->data);

      free(*allopts);

      free(configfile);

      return 1;

    case CFG_SUCCESS:

      break;

    }

  free(configfile);

  *mycfg = cfg_getsec((*allopts)->data->cfg, myname);

  return 0;

}

int options_free(struct options_common *opts)

{

  /* free the other common_options struct's members */

  cfg_free(opts->data->cfg);

  free(opts->remoteio);

  free(opts->data);

  free(opts);

  return 0;

}