| 1 | /** |
| 2 | * Copyright (C) 2012-2015 Yecheng Fu <cofyc.jackson at gmail dot com> |
| 3 | * All rights reserved. |
| 4 | * |
| 5 | * Use of this source code is governed by a MIT-style license that can be found |
| 6 | * in the LICENSE file. |
| 7 | */ |
| 8 | #ifndef ARGPARSE_H |
| 9 | #define ARGPARSE_H |
| 10 | |
| 11 | /* For c++ compatibility */ |
| 12 | #ifdef __cplusplus |
| 13 | extern "C" { |
| 14 | #endif |
| 15 | |
| 16 | #include <stdint.h> |
| 17 | |
| 18 | struct argparse; |
| 19 | struct argparse_option; |
| 20 | |
| 21 | typedef int argparse_callback (struct argparse *self, |
| 22 | const struct argparse_option *option); |
| 23 | |
| 24 | enum argparse_flag { |
| 25 | ARGPARSE_STOP_AT_NON_OPTION = 1, |
| 26 | }; |
| 27 | |
| 28 | enum argparse_option_type { |
| 29 | /* special */ |
| 30 | ARGPARSE_OPT_END, |
| 31 | ARGPARSE_OPT_GROUP, |
| 32 | /* options with no arguments */ |
| 33 | ARGPARSE_OPT_BOOLEAN, |
| 34 | ARGPARSE_OPT_BIT, |
| 35 | /* options with arguments (optional or required) */ |
| 36 | ARGPARSE_OPT_INTEGER, |
| 37 | ARGPARSE_OPT_FLOAT, |
| 38 | ARGPARSE_OPT_STRING, |
| 39 | }; |
| 40 | |
| 41 | enum argparse_option_flags { |
| 42 | OPT_NONEG = 1, /* disable negation */ |
| 43 | }; |
| 44 | |
| 45 | /** |
| 46 | * argparse option |
| 47 | * |
| 48 | * `type`: |
| 49 | * holds the type of the option, you must have an ARGPARSE_OPT_END last in your |
| 50 | * array. |
| 51 | * |
| 52 | * `short_name`: |
| 53 | * the character to use as a short option name, '\0' if none. |
| 54 | * |
| 55 | * `long_name`: |
| 56 | * the long option name, without the leading dash, NULL if none. |
| 57 | * |
| 58 | * `value`: |
| 59 | * stores pointer to the value to be filled. |
| 60 | * |
| 61 | * `help`: |
| 62 | * the short help message associated to what the option does. |
| 63 | * Must never be NULL (except for ARGPARSE_OPT_END). |
| 64 | * |
| 65 | * `callback`: |
| 66 | * function is called when corresponding argument is parsed. |
| 67 | * |
| 68 | * `data`: |
| 69 | * associated data. Callbacks can use it like they want. |
| 70 | * |
| 71 | * `flags`: |
| 72 | * option flags. |
| 73 | */ |
| 74 | struct argparse_option { |
| 75 | enum argparse_option_type type; |
| 76 | const char short_name; |
| 77 | const char *long_name; |
| 78 | void *value; |
| 79 | const char *help; |
| 80 | argparse_callback *callback; |
| 81 | intptr_t data; |
| 82 | int flags; |
| 83 | }; |
| 84 | |
| 85 | /** |
| 86 | * argpparse |
| 87 | */ |
| 88 | struct argparse { |
| 89 | // user supplied |
| 90 | const struct argparse_option *options; |
| 91 | const char *const *usages; |
| 92 | int flags; |
| 93 | const char *description; // a description after usage |
| 94 | const char *epilog; // a description at the end |
| 95 | // internal context |
| 96 | int argc; |
| 97 | const char **argv; |
| 98 | const char **out; |
| 99 | int cpidx; |
| 100 | const char *optvalue; // current option value |
| 101 | }; |
| 102 | |
| 103 | // built-in callbacks |
| 104 | int argparse_help_cb(struct argparse *self, |
| 105 | const struct argparse_option *option); |
| 106 | |
| 107 | // built-in option macros |
| 108 | #define OPT_END() { ARGPARSE_OPT_END, 0, NULL, NULL, 0, NULL, 0, 0 } |
| 109 | #define OPT_BOOLEAN(...) { ARGPARSE_OPT_BOOLEAN, __VA_ARGS__ } |
| 110 | #define OPT_BIT(...) { ARGPARSE_OPT_BIT, __VA_ARGS__ } |
| 111 | #define OPT_INTEGER(...) { ARGPARSE_OPT_INTEGER, __VA_ARGS__ } |
| 112 | #define OPT_FLOAT(...) { ARGPARSE_OPT_FLOAT, __VA_ARGS__ } |
| 113 | #define OPT_STRING(...) { ARGPARSE_OPT_STRING, __VA_ARGS__ } |
| 114 | #define OPT_GROUP(h) { ARGPARSE_OPT_GROUP, 0, NULL, NULL, h, NULL, 0, 0 } |
| 115 | #define OPT_HELP() OPT_BOOLEAN('h', "help", NULL, \ |
| 116 | "show this help message and exit", \ |
| 117 | argparse_help_cb, 0, OPT_NONEG) |
| 118 | |
| 119 | int argparse_init(struct argparse *self, struct argparse_option *options, |
| 120 | const char *const *usages, int flags); |
| 121 | void argparse_describe(struct argparse *self, const char *description, |
| 122 | const char *epilog); |
| 123 | int argparse_parse(struct argparse *self, int argc, const char **argv); |
| 124 | void argparse_usage(struct argparse *self); |
| 125 | |
| 126 | #ifdef __cplusplus |
| 127 | } |
| 128 | #endif |
| 129 | |
| 130 | #endif |
| 131 | |