t/helper/test-env-helper.c

   1 #include "test-tool.h"
   2 #include "parse.h"
   3 #include "parse-options.h"
   4
   5 static char const * const env__helper_usage[] = {
   6         "test-tool env-helper --type=[bool|ulong] <options> <env-var>",
   7         NULL
   8 };
   9
  10 enum cmdmode {
  11         ENV_HELPER_TYPE_BOOL = 1,
  12         ENV_HELPER_TYPE_ULONG
  13 };
  14
  15 static int option_parse_type(const struct option *opt, const char *arg,
  16                              int unset)
  17 {
  18         enum cmdmode *cmdmode = opt->value;
  19
  20         BUG_ON_OPT_NEG(unset);
  21
  22         if (!strcmp(arg, "bool"))
  23                 *cmdmode = ENV_HELPER_TYPE_BOOL;
  24         else if (!strcmp(arg, "ulong"))
  25                 *cmdmode = ENV_HELPER_TYPE_ULONG;
  26         else
  27                 die("unrecognized --type argument, %s", arg);
  28
  29         return 0;
  30 }
  31
  32 int cmd__env_helper(int argc, const char **argv)
  33 {
  34         int exit_code = 0;
  35         const char *env_variable = NULL;
  36         const char *env_default = NULL;
  37         int ret;
  38         int ret_int, default_int;
  39         unsigned long ret_ulong, default_ulong;
  40         enum cmdmode cmdmode = 0;
  41         struct option opts[] = {
  42                 OPT_CALLBACK_F(0, "type", &cmdmode, "type",
  43                                "value is given this type", PARSE_OPT_NONEG,
  44                                option_parse_type),
  45                 OPT_STRING(0, "default", &env_default, "value",
  46                            "default for git_env_*(...) to fall back on"),
  47                 OPT_BOOL(0, "exit-code", &exit_code,
  48                          "be quiet only use git_env_*() value as exit code"),
  49                 OPT_END(),
  50         };
  51
  52         argc = parse_options(argc, argv, NULL, opts, env__helper_usage,
  53                              PARSE_OPT_KEEP_UNKNOWN_OPT);
  54         if (env_default && !*env_default)
  55                 usage_with_options(env__helper_usage, opts);
  56         if (!cmdmode)
  57                 usage_with_options(env__helper_usage, opts);
  58         if (argc != 1)
  59                 usage_with_options(env__helper_usage, opts);
  60         env_variable = argv[0];
  61
  62         switch (cmdmode) {
  63         case ENV_HELPER_TYPE_BOOL:
  64                 if (env_default) {
  65                         default_int = git_parse_maybe_bool(env_default);
  66                         if (default_int == -1) {
  67                                 error("option `--default' expects a boolean value with `--type=bool`, not `%s`",
  68                                       env_default);
  69                                 usage_with_options(env__helper_usage, opts);
  70                         }
  71                 } else {
  72                         default_int = 0;
  73                 }
  74                 ret_int = git_env_bool(env_variable, default_int);
  75                 if (!exit_code)
  76                         puts(ret_int ? "true" : "false");
  77                 ret = ret_int;
  78                 break;
  79         case ENV_HELPER_TYPE_ULONG:
  80                 if (env_default) {
  81                         if (!git_parse_ulong(env_default, &default_ulong)) {
  82                                 error("option `--default' expects an unsigned long value with `--type=ulong`, not `%s`",
  83                                       env_default);
  84                                 usage_with_options(env__helper_usage, opts);
  85                         }
  86                 } else {
  87                         default_ulong = 0;
  88                 }
  89                 ret_ulong = git_env_ulong(env_variable, default_ulong);
  90                 if (!exit_code)
  91                         printf("%lu\n", ret_ulong);
  92                 ret = ret_ulong;
  93                 break;
  94         default:
  95                 BUG("unknown <type> value");
  96                 break;
  97         }
  98
  99         return !ret;
 100 }