141 lines
2.7 KiB
C
141 lines
2.7 KiB
C
#include "common.h"
|
|
|
|
#include "option.h"
|
|
#include "str.h"
|
|
|
|
int option_str (void *data, int argc, char **argv)
|
|
{
|
|
if (argc<2 || str_empty(argv[1])) {
|
|
gt_print("option `%s' need a string argument\n", argv[0]);
|
|
return -1;
|
|
}
|
|
|
|
memcpy(data, &argv[1], sizeof(argv[1]));
|
|
|
|
return 1;
|
|
}
|
|
|
|
int option_long (void *data, int argc, char **argv)
|
|
{
|
|
if (argc<2 || str_empty(argv[1])) {
|
|
gt_print("option `%s' need an integer argument\n", argv[0]);
|
|
return -1;
|
|
}
|
|
|
|
errno = 0;
|
|
char *end;
|
|
long val = strtol(argv[1], &end, 0);
|
|
|
|
if (errno || argv[1]==end) {
|
|
gt_print("argument `%s' is not a valid integer\n", argv[1]);
|
|
return -1;
|
|
}
|
|
|
|
memcpy(data, &val, sizeof(val));
|
|
|
|
return 1;
|
|
}
|
|
|
|
int option_is_set (struct option *opts, const char *name)
|
|
{
|
|
for (int k=0; opts[k].name; k++) {
|
|
if (!str_cmp(opts[k].name, name))
|
|
return opts[k].set;
|
|
}
|
|
|
|
return 0;
|
|
}
|
|
|
|
int option_option (void *data, int argc, char **argv)
|
|
{
|
|
if (!data)
|
|
return 0;
|
|
|
|
struct option *opts = (struct option *)data;
|
|
|
|
for (int k=0; opts[k].name; k++)
|
|
opts[k].set = 0;
|
|
|
|
for (int i=1; i<argc; i++) {
|
|
int found = 0;
|
|
|
|
for (int k=0; opts[k].name; k++) {
|
|
if (str_cmp(opts[k].name, argv[i]))
|
|
continue;
|
|
|
|
if (opts[k].set) {
|
|
gt_print("option `%s' is already set\n", opts[k].name);
|
|
return -1;
|
|
}
|
|
|
|
int ret = opts[k].call(opts[k].data, argc-i, &argv[i]);
|
|
|
|
if (ret<0)
|
|
return -1;
|
|
|
|
opts[k].set = 1;
|
|
|
|
i += ret;
|
|
found = 1;
|
|
break;
|
|
}
|
|
|
|
if (!found)
|
|
return i-1;
|
|
}
|
|
|
|
return argc;
|
|
}
|
|
|
|
static int option_usage (struct option *opts, int slen)
|
|
{
|
|
if (!opts)
|
|
return 0;
|
|
|
|
int len = 0;
|
|
|
|
for (int k=0; opts[k].name; k++) {
|
|
if (len>40) {
|
|
gt_print("\n%*s", slen, "");
|
|
len = 0;
|
|
}
|
|
|
|
len += gt_print(" [%s", opts[k].name);
|
|
|
|
if (opts[k].call==option_option) {
|
|
len += option_usage((struct option *)opts[k].data, slen+len);
|
|
} else {
|
|
len += gt_print(" ARG");
|
|
}
|
|
|
|
len += gt_print("]");
|
|
}
|
|
|
|
return len;
|
|
}
|
|
|
|
int option (struct option *opts, int argc, char **argv)
|
|
{
|
|
int ret = option_option(opts, argc, argv);
|
|
|
|
if (ret==argc)
|
|
return 0;
|
|
|
|
if (ret<0 || ret+1>=argc)
|
|
return 1;
|
|
|
|
gt_print("option `%s' is unknown\n", argv[ret+1]);
|
|
|
|
int slen = gt_print("usage: %s", argv[0]);
|
|
|
|
if (slen>40) {
|
|
slen = 12;
|
|
gt_print("\n%*s", slen, "");
|
|
}
|
|
|
|
option_usage(opts, slen);
|
|
gt_print("\n");
|
|
|
|
return 1;
|
|
}
|