/*
  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 "protocol.h" /*< for DISTREN_SERVERTYPE_* */

#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()
   @todo replace abort()s with something useful
 */
int options_init(int argc, char *argv[], cfg_t **mycfg, cfg_opt_t *myopts, char *myname, struct options_common **allopts)
{
  char *configfileprefix;
  char *configfile;

  size_t i, j;

  char *optstring = "hc:";
  char curopt;

  struct remoteio_server *aserver;

  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);


  /*
   * libdistrencommon's config options:
   */

  /* remoteio -- iterate through servers */
  aserver = malloc(sizeof(struct remoteio_server));
  if(!aserver)
    {
      fprintf(stderr, "@todo cleanup!\n");
      abort();
    }
  (*allopts)->remoteio->servers = aserver;

  j = cfg_size((*allopts)->data->cfg, "server"); 
  for(i = 0; i < j; i ++)
    {
      cfg_t *cfg_aserver;
      
      cfg_aserver = cfg_getnsec((*allopts)->data->cfg, "server", i);
      
      if(!aserver) /*< if the malloc in the previous loop failed */
	abort();
      
      aserver->name = strdup(cfg_title(cfg_aserver));
      aserver->hostname = strdup(cfg_getstr(cfg_aserver, "hostname"));
      aserver->username = strdup(cfg_getstr(cfg_aserver, "username"));
      if(strcmp(cfg_getstr(cfg_aserver, "method"), "ssh") == 0)
	aserver->method = REMOTEIO_METHOD_SSH;
      else
	abort();
      
      if(i < j - 1)
	{
	  aserver->next = malloc(sizeof(struct remoteio_server));
	  aserver = aserver->next;
	}
    }
  aserver->next = NULL;

  return 0;
}

/**
   @todo free remoteio_servers
*/
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;
}