cannam@127: /*
cannam@127:  * Copyright (c) 2003, 2007-14 Matteo Frigo
cannam@127:  * Copyright (c) 2003, 2007-14 Massachusetts Institute of Technology
cannam@127:  *
cannam@127:  * This program is free software; you can redistribute it and/or modify
cannam@127:  * it under the terms of the GNU General Public License as published by
cannam@127:  * the Free Software Foundation; either version 2 of the License, or
cannam@127:  * (at your option) any later version.
cannam@127:  *
cannam@127:  * This program is distributed in the hope that it will be useful,
cannam@127:  * but WITHOUT ANY WARRANTY; without even the implied warranty of
cannam@127:  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
cannam@127:  * GNU General Public License for more details.
cannam@127:  *
cannam@127:  * You should have received a copy of the GNU General Public License
cannam@127:  * along with this program; if not, write to the Free Software
cannam@127:  * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA  02110-1301  USA
cannam@127:  *
cannam@127:  */
cannam@127: 
cannam@127: #include <string.h>
cannam@127: #include <stdio.h>
cannam@127: 
cannam@127: #include "config.h"
cannam@127: #include "my-getopt.h"
cannam@127: 
cannam@127: int my_optind = 1;
cannam@127: const char *my_optarg = 0;
cannam@127: static const char *scan_pointer = 0;
cannam@127: 
cannam@127: void my_usage(const char *progname, const struct my_option *opt)
cannam@127: {
cannam@127:     int i;
cannam@127:     size_t col = 0;
cannam@127: 
cannam@127:     fprintf(stdout, "Usage: %s", progname);
cannam@127:     col += (strlen(progname) + 7);
cannam@127:     for (i = 0; opt[i].long_name; i++) {
cannam@127: 	size_t option_len;
cannam@127: 
cannam@127: 	option_len = strlen(opt[i].long_name);
cannam@127: 	if (col >= 80 - (option_len + 16)) {
cannam@127: 	    fputs("\n\t", stdout);
cannam@127: 	    col = 8;
cannam@127: 	}
cannam@127: 	fprintf(stdout, " [--%s", opt[i].long_name);
cannam@127: 	col += (option_len + 4);
cannam@127: 	if (opt[i].short_name < 128) {
cannam@127: 	    fprintf(stdout, " | -%c", opt[i].short_name);
cannam@127: 	    col += 5;
cannam@127: 	}
cannam@127: 	switch (opt[i].argtype) {
cannam@127: 	    case REQARG:
cannam@127: 		 fputs(" arg]", stdout);
cannam@127: 		 col += 5;
cannam@127: 		 break;
cannam@127: 	    case OPTARG:
cannam@127: 		 fputs(" [arg]]", stdout);
cannam@127: 		 col += 10;
cannam@127: 		 break;
cannam@127: 	    default:
cannam@127: 		 fputs("]", stdout);
cannam@127: 		 col++;
cannam@127: 	}
cannam@127:     }
cannam@127: 
cannam@127:     fputs ("\n", stdout);
cannam@127: }
cannam@127: 
cannam@127: int my_getopt(int argc, char *argv[], const struct my_option *optarray)
cannam@127: {
cannam@127:      const char *p;
cannam@127:      const struct my_option *l;
cannam@127: 
cannam@127:      if (scan_pointer && *scan_pointer) {
cannam@127: 	  /* continue a previously scanned argv[] element */
cannam@127: 	  p = scan_pointer;
cannam@127: 	  goto short_option;
cannam@127:      } else {
cannam@127: 	  /* new argv[] element */
cannam@127: 	  if (my_optind >= argc)
cannam@127: 	       return -1; /* no more options */
cannam@127: 
cannam@127: 	  p = argv[my_optind];
cannam@127:      
cannam@127: 	  if (*p++ != '-')  
cannam@127: 	       return (-1); /* not an option */
cannam@127: 
cannam@127: 	  if (!*p) 
cannam@127: 	       return (-1); /* string is exactly '-' */
cannam@127: 	       
cannam@127: 	  ++my_optind;
cannam@127:      }
cannam@127: 
cannam@127:      if (*p == '-') {
cannam@127: 	  /* long option */
cannam@127: 	  scan_pointer = 0;
cannam@127: 	  my_optarg = 0;
cannam@127: 
cannam@127: 	  ++p;
cannam@127: 	  
cannam@127: 	  for (l = optarray; l->short_name; ++l) {
cannam@127: 	       size_t len = strlen(l->long_name);
cannam@127: 	       if (!strncmp(l->long_name, p, len) && 
cannam@127: 		   (!p[len] || p[len] == '=')) {
cannam@127: 		    switch (l->argtype) {
cannam@127: 			case NOARG: 
cannam@127: 			     goto ok;
cannam@127: 			case OPTARG: 
cannam@127: 			     if (p[len] == '=')
cannam@127: 				  my_optarg = p + len + 1;
cannam@127: 			     goto ok;
cannam@127: 			case REQARG: 
cannam@127: 			     if (p[len] == '=') {
cannam@127: 				  my_optarg = p + len + 1;
cannam@127: 				  goto ok;
cannam@127: 			     }
cannam@127: 			     if (my_optind >= argc) {
cannam@127: 				  fprintf(stderr, 
cannam@127: 					  "option --%s requires an argument\n",
cannam@127: 					  l->long_name);
cannam@127: 				  return '?';
cannam@127: 			     }
cannam@127: 			     my_optarg = argv[my_optind];
cannam@127: 			     ++my_optind;
cannam@127: 			     goto ok;
cannam@127: 		    }
cannam@127: 	       }
cannam@127: 	  }
cannam@127:      } else {
cannam@127:      short_option:
cannam@127: 	  scan_pointer = 0;
cannam@127: 	  my_optarg = 0;
cannam@127: 
cannam@127: 	  for (l = optarray; l->short_name; ++l) {
cannam@127: 	       if (l->short_name == (char)l->short_name &&
cannam@127: 		   *p == l->short_name) {
cannam@127: 		    ++p;
cannam@127: 		    switch (l->argtype) {
cannam@127: 			case NOARG: 
cannam@127: 			     scan_pointer = p;
cannam@127: 			     goto ok;
cannam@127: 			case OPTARG: 
cannam@127: 			     if (*p)
cannam@127: 				  my_optarg = p;
cannam@127: 			     goto ok;
cannam@127: 			case REQARG: 
cannam@127: 			     if (*p) {
cannam@127: 				  my_optarg = p;
cannam@127: 			     } else {
cannam@127: 				  if (my_optind >= argc) {
cannam@127: 				       fprintf(stderr, 
cannam@127: 					  "option -%c requires an argument\n",
cannam@127: 					  l->short_name);
cannam@127: 				       return '?';
cannam@127: 				  }
cannam@127: 				  my_optarg = argv[my_optind];
cannam@127: 				  ++my_optind;
cannam@127: 			     }
cannam@127: 			     goto ok;
cannam@127: 		    }
cannam@127: 	       }
cannam@127: 	  }
cannam@127:      }
cannam@127: 
cannam@127:      fprintf(stderr, "unrecognized option %s\n", argv[my_optind - 1]);
cannam@127:      return '?';
cannam@127: 
cannam@127:  ok:
cannam@127:      return l->short_name;
cannam@127: }
cannam@127: