/*
 * Copyright (c) 1997 Kungliga Tekniska Högskolan
 * (Royal Institute of Technology, Stockholm, Sweden). 
 * All rights reserved. 
 *
 * Redistribution and use in source and binary forms, with or without 
 * modification, are permitted provided that the following conditions 
 * are met: 
 *
 * 1. Redistributions of source code must retain the above copyright 
 *    notice, this list of conditions and the following disclaimer. 
 *
 * 2. Redistributions in binary form must reproduce the above copyright 
 *    notice, this list of conditions and the following disclaimer in the 
 *    documentation and/or other materials provided with the distribution. 
 *
 * 3. All advertising materials mentioning features or use of this software 
 *    must display the following acknowledgement: 
 *      This product includes software developed by Kungliga Tekniska 
 *      Högskolan and its contributors. 
 *
 * 4. Neither the name of the Institute nor the names of its contributors 
 *    may be used to endorse or promote products derived from this software 
 *    without specific prior written permission. 
 *
 * THIS SOFTWARE IS PROVIDED BY THE INSTITUTE AND CONTRIBUTORS ``AS IS'' AND 
 * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE 
 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE 
 * ARE DISCLAIMED.  IN NO EVENT SHALL THE INSTITUTE OR CONTRIBUTORS BE LIABLE 
 * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL 
 * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS 
 * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) 
 * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT 
 * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY 
 * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF 
 * SUCH DAMAGE. 
 */

#ifdef HAVE_CONFIG_H
#include <config.h>
RCSID("$Id$");
#endif

#include "getarg.h"

static int
arg_match(struct getargs *arg, char *argv)
{
    char *optarg;
    int negate = 0;
    if(arg->long_name){
	int len = strlen(arg->long_name);
	if(strncmp(arg->long_name, argv, len))
	    if(arg->type == arg_flag && strlen(argv) > 3 && 
	       strncmp(arg->long_name, argv + 3, len) == 0){
		optarg = argv + 3 + len;
		negate = 1;
	    } else
		return ARG_ERR_NO_MATCH;
	else
	    optarg = argv + len;
	
	if(*optarg != '=' && (arg->type != arg_flag && *optarg == 0))
	    return ARG_ERR_NO_MATCH;
	switch(arg->type){
	case arg_integer:
	    {
		int tmp;
		if(sscanf(optarg + 1, "%d", &tmp) != 1)
		    return ARG_ERR_BAD_ARG;
		*(int*)arg->value = tmp;
		return 0;
	    }
	case arg_string:
	    {
		*(char**)arg->value = optarg + 1;
		return 0;
	    }
	case arg_flag:
	    {
		int *flag = arg->value;
		if(*optarg == 0 ||
		   strcmp(optarg + 1, "yes") == 0 || 
		   strcmp(optarg + 1, "true") == 0){
		    *flag = !negate;
		    return 0;
		} else {
		    *flag = negate;
		    return 0;
		}
		return ARG_ERR_BAD_ARG;
	    }
	}
    }
    return ARG_ERR_NO_MATCH;
}

int
getarg(struct getargs *args, size_t num_args, 
       int argc, char **argv, int *optind)
{
    int i, j, k;
    int ret = 0;
    (*optind)++;
    for(i = *optind; i < argc; i++){
	if(argv[i][0] != '-')
	    break;
	if(argv[i][1] == '-'){
	    if(argv[i][2] == 0){
		i++;
		break;
	    }
	    for(j = 0; j < num_args; j++){
		ret = arg_match(&args[j], argv[i] + 2);
		if(ret == ARG_ERR_NO_MATCH)
		    continue;
		break;
	    }
	    if(ret)
		break;
	}else{
	    for(j = 1; argv[i][j]; j++){
		for(k = 0; k < num_args; k++){
		    char *optarg;
		    if(args[k].short_name == 0)
			continue;
		    if(argv[i][j] == args[k].short_name){
			if(args[k].type == arg_flag){
			    *(int*)args[k].value = 1;
			    break;
			}
			if(args[k].type == arg_negative_flag){
			    *(int*)args[k].value = 0;
			    break;
			}
			if(argv[i][j + 1])
			    optarg = &argv[i][j + 1];
			else{
			    i++;
			    optarg = argv[i];
			}
			if(optarg == NULL)
			    return ARG_ERR_NO_ARG;
			if(args[k].type == arg_integer){
			    int tmp;
			    if(sscanf(optarg, "%d", &tmp) != 1)
				return ARG_ERR_BAD_ARG;
			    *(int*)args[k].value = tmp;
			    break;
			}else if(args[k].type == arg_string){
			    *(char**)args[k].value = optarg;
			    break;
			}
			return ARG_ERR_BAD_ARG;
		    }
			
		}
	    }
	}
    }
    *optind = i;
    return 0;
}

#if TEST
int foo_flag = 2;
int flag1 = 0;
int flag2 = 0;
int bar_int;
char *baz_string;

struct getargs args[] = {
    { NULL, '1', arg_flag, &flag1, NULL },
    { NULL, '2', arg_flag, &flag2, NULL },
    { "foo", 'f', arg_flag, &foo_flag, NULL },
    { "bar", 'b', arg_integer, &bar_int, NULL },
    { "baz", 'x', arg_string, &baz_string, NULL },
};

int main(int argc, char **argv)
{
    int optind = 0;
    while(getarg(args, 5, argc, argv, &optind))
	printf("Bad arg: %s\n", argv[optind]);
    printf("flag1 = %d\n", flag1);  
    printf("flag2 = %d\n", flag2);  
    printf("foo_flag = %d\n", foo_flag);  
    printf("bar_int = %d\n", bar_int);
    printf("baz_flag = %s\n", baz_string);
}
#endif