/* 
 * $Id$
 *
 * (c) 2004 by Kalle Wallin <kaw@linux.se>
 *
 * This program is free software; you can redistribute it and/or modify
 * it under the terms of the GNU General Public License as published by
 * the Free Software Foundation; either version 2 of the License, or
 * (at your option) any later version.
 *
 * This program 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 General Public License for more details.
 * You should have received a copy of the GNU General Public License
 * along with this program; if not, write to the Free Software
 * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
 *
 */

#include "options.h"
#include "config.h"
#include "ncmpc.h"
#include "support.h"
#include "command.h"
#include "conf.h"

#include <stdlib.h>
#include <string.h>

#define MAX_LONGOPT_LENGTH 32

#define ERROR_UNKNOWN_OPTION    0x01
#define ERROR_BAD_ARGUMENT      0x02
#define ERROR_GOT_ARGUMENT      0x03
#define ERROR_MISSING_ARGUMENT  0x04

typedef struct {
	int shortopt;
	const char *longopt;
	const char *argument;
	const char *descrition;
} arg_opt_t;


typedef void (*option_callback_fn_t)(int c, const char *arg);


options_t options;

static arg_opt_t option_table[] = {
  { '?', "help",     NULL,   "Show this help message" },
  { 'V', "version",  NULL,   "Display version information" },
  { 'c', "colors",   NULL,   "Enable colors" },
  { 'C', "no-colors", NULL,  "Disable colors" },
#ifdef HAVE_GETMOUSE
  { 'm', "mouse",    NULL,   "Enable mouse" },
  { 'M', "no-mouse", NULL,   "Disable mouse" },
#endif
  { 'e', "exit",     NULL,   "Exit on connection errors" },
  { 'p', "port",  "PORT", "Connect to server on port [" DEFAULT_PORT_STR "]" },
  { 'h', "host",  "HOST", "Connect to server on host [" DEFAULT_HOST "]" },
  { 'P', "password","PASSWORD", "Connect with password" },
  { 'f', "config",  "FILE",     "Read configuration from file" },
  { 'k', "key-file","FILE",     "Read configuration from file" },
  { 'S', "no-splash", NULL, "Don't show the splash screen" },
#ifndef NDEBUG
  { 'K', "dump-keys", NULL,     "Dump key bindings to stdout" },
  { 'D', "debug",   NULL,   "Enable debug output on stderr" },
#endif
  { 0, NULL, NULL, NULL },
};

static arg_opt_t *
lookup_option(int s, char *l)
{
  int i;

  i=0;
  while( option_table[i].descrition )
    {
      if( l && strcmp(l, option_table[i].longopt) == 0 )
	return &option_table[i];;
      if( s && s==option_table[i].shortopt )
	return &option_table[i];;
      i++;
    }
  return NULL;
}

static void
option_error(int error, const char *option, const char *arg)
{
  switch(error)
    {
    case ERROR_UNKNOWN_OPTION:
      fprintf(stderr, PACKAGE ": invalid option %s\n", option);
      break;
    case ERROR_BAD_ARGUMENT:
      fprintf(stderr, PACKAGE ": bad argument: %s\n", option);
      break;
    case ERROR_GOT_ARGUMENT:
      fprintf(stderr, PACKAGE ": invalid option %s=%s\n", option, arg);
      break;
    case ERROR_MISSING_ARGUMENT:
      fprintf(stderr, PACKAGE ": missing value for %s option\n", option);
      break;
    default:
      fprintf(stderr, PACKAGE ": internal error %d\n", error);
      break;
    }
  exit(EXIT_FAILURE);
}

static void 
display_help(void)
{
  int i = 0;

  printf("Usage: %s [OPTION]...\n", PACKAGE);
  while( option_table[i].descrition )
    {
      char tmp[MAX_LONGOPT_LENGTH];

      if( option_table[i].argument )
	g_snprintf(tmp, MAX_LONGOPT_LENGTH, "%s=%s", 
		   option_table[i].longopt, 
		   option_table[i].argument);
      else
	g_strlcpy(tmp, option_table[i].longopt, 64);

      printf("  -%c, --%-20s %s\n", 
	     option_table[i].shortopt, 
	     tmp,
	     option_table[i].descrition);
      i++;
    }
}

static void 
handle_option(int c, const char *arg)
{
  D("option callback -%c %s\n", c, arg);
  switch(c)
    {
    case '?': /* --help */
      display_help();
      exit(EXIT_SUCCESS);
    case 'V': /* --version */
      printf("%s version: %s\n", PACKAGE, VERSION);
      printf("build options:");
#ifndef NDEBUG
      printf(" debug");
#endif
#ifdef ENABLE_NLS
      printf(" nls");
#endif
#ifdef HAVE_GETMOUSE
      printf(" getmouse");
#endif
#ifdef ENABLE_ARTIST_SCREEN
      printf(" artist-screen");
#endif
#ifdef ENABLE_SEARCH_SCREEN
      printf(" search-screen");
#endif
#ifdef ENABLE_KEYDEF_SCREEN
      printf(" key-screen");
#endif
#ifdef ENABLE_CLOCK_SCREEN
      printf(" clock-screen");
#endif
      printf("\n");
      exit(EXIT_SUCCESS);
    case 'c': /* --colors */
      options.enable_colors = TRUE;
      break;
    case 'C': /* --no-colors */
      options.enable_colors = FALSE;
      break;
    case 'm': /* --mouse */
     options.enable_mouse = TRUE;
      break;
    case 'M': /* --no-mouse */
      options.enable_mouse = FALSE;
      break;
    case 'e': /* --exit */
      options.reconnect = FALSE;
      break;
    case 'p': /* --port */
      options.port = atoi(arg);
      break;
    case 'h': /* --host */
      if( options.host )
	g_free(options.host);
      options.host = g_strdup(arg);
      break;
    case 'P': /* --password */
      if( options.password )
	g_free(options.password);
      options.password = locale_to_utf8(arg);
      break;
    case 'f': /* --config */
      if( options.config_file )
	g_free(options.config_file);
      options.config_file = g_strdup(arg);
      break;
    case 'k': /* --key-file */
      if( options.key_file )
	g_free(options.key_file);
      options.key_file = g_strdup(arg);
      break;
    case 'S': /* --key-file */
      /* the splash screen was removed */
      break;
#ifndef NDEBUG
    case 'K': /* --dump-keys */
      read_configuration(&options);
      write_key_bindings(stdout, KEYDEF_WRITE_ALL | KEYDEF_COMMENT_ALL);
      exit(EXIT_SUCCESS);
      break;
    case 'D': /* --debug */
      options.debug = TRUE;
      break;
#endif
    default:
      fprintf(stderr,"Unknown Option %c = %s\n", c, arg);
      break;
    }
}

options_t *
options_parse(int argc, const char *argv[])
{
	int i;
	arg_opt_t *opt = NULL;
	option_callback_fn_t option_cb = handle_option;

	i=1;
	while (i < argc) {
		const char *arg = argv[i];
		size_t len = strlen(arg);

		/* check for a long option */
		if (g_str_has_prefix(arg, "--")) {
			char *name, *value;

			/* make shure we got an argument for the previous option */
			if( opt && opt->argument )
				option_error(ERROR_MISSING_ARGUMENT, opt->longopt, opt->argument);

			/* retreive a option argument */
			if ((value=g_strrstr(arg+2, "="))) {
				*value = '\0';
				name = g_strdup(arg);
				*value = '=';
				value++;
			} else
				name = g_strdup(arg);

			/* check if the option exists */
			if( (opt=lookup_option(0, name+2)) == NULL )
				option_error(ERROR_UNKNOWN_OPTION, name, NULL);
			g_free(name);

			/* abort if we got an argument to the option and dont want one */
			if( value && opt->argument==NULL )
				option_error(ERROR_GOT_ARGUMENT, arg, value);

			/* execute option callback */
			if (value || opt->argument==NULL) {
				option_cb (opt->shortopt, value);
				opt = NULL;
			}
		}
		/* check for short options */
		else if (len>=2 && g_str_has_prefix(arg, "-")) {
			int j;

			for(j=1; j<len; j++) {
				/* make shure we got an argument for the previous option */
				if( opt && opt->argument )
					option_error(ERROR_MISSING_ARGUMENT,
						     opt->longopt, opt->argument);

				/* check if the option exists */
				if( (opt=lookup_option(arg[j], NULL))==NULL )
					option_error(ERROR_UNKNOWN_OPTION, arg, NULL);

				/* if no option argument is needed execute callback */
				if( opt->argument==NULL ) {
					option_cb (opt->shortopt, NULL);
					opt = NULL;
				}
			}
		} else {
			/* is this a option argument? */
			if( opt && opt->argument) {
				option_cb (opt->shortopt, arg);
				opt = NULL;
			} else
				option_error(ERROR_BAD_ARGUMENT, arg, NULL);
		}
		i++;
	}

	if( opt && opt->argument==NULL)
		option_cb (opt->shortopt, NULL);
	else if( opt && opt->argument )
		option_error(ERROR_MISSING_ARGUMENT, opt->longopt, opt->argument);

	return  &options;
}

options_t *
options_init( void )
{
	const char *value;
	char *tmp;

	memset(&options, 0, sizeof(options_t));

	/* get initial values for host and password from MPD_HOST (enviroment) */
	if ((value=g_getenv(MPD_HOST_ENV)))
		options.host = g_strdup(value);
	else
		options.host = g_strdup(DEFAULT_HOST);

	if ((tmp=g_strstr_len(options.host, strlen(options.host), "@"))) {
		char *oldhost = options.host;
		*tmp  = '\0';
		options.password = locale_to_utf8(oldhost);
		options.host = g_strdup(tmp+1);
		g_free(oldhost);
	}

	/* get initial values for port from MPD_PORT (enviroment) */
	if ((value=g_getenv(MPD_PORT_ENV)))
		options.port = atoi(value);
	else
		options.port = DEFAULT_PORT;

	/* default option values */
	options.reconnect = TRUE;
	options.find_wrap = TRUE;
	options.wide_cursor = TRUE;
	options.audible_bell = TRUE;
	options.crossfade_time = DEFAULT_CROSSFADE_TIME;
	options.seek_time = 1;
	options.screen_list = g_strsplit_set(DEFAULT_SCREEN_LIST, " ", 0);
	options.timedisplay_type = g_strdup(DEFAULT_TIMEDISPLAY_TYPE);
	options.lyrics_timeout = DEFAULT_LYRICS_TIMEOUT;
	options.scroll = DEFAULT_SCROLL;
	options.scroll_sep = g_strdup(DEFAULT_SCROLL_SEP);

	return &options;
}