ProgramOptionsTestHelper.cpp File Reference

#include <iostream>
#include <glog/logging.h>
#include <folly/Conv.h>
#include <folly/experimental/ProgramOptions.h>

Go to the source code of this file.

Functions
	DEFINE_bool (flag_bool_true, true,"Bool with true default value")
	DEFINE_bool (flag_bool_false, false,"Bool with false default value")
	DEFINE_int32 (flag_int, 42,"Integer flag")
	DEFINE_string (flag_string,"foo","String flag")
int	main (int argc, char *argv[])

Function Documentation

DEFINE_bool	(	flag_bool_true	,
		true	,
		"Bool with true default value"
	)

DEFINE_bool	(	flag_bool_false	,
		false	,
		"Bool with false default value"
	)

DEFINE_int32	(	flag_int	,
		42	,
		"Integer flag"
	)

DEFINE_string	(	flag_string	,
		"foo"	,
		"String flag"
	)

int main	(	int	argc,
		char *	argv[]
	)

Definition at line 39 of file ProgramOptionsTestHelper.cpp.

References folly::getGFlags(), folly::GFLAGS, folly::GNU, and folly::parseNestedCommandLine().

                                 {
  po::options_description desc;
  auto styleEnv = getenv("PROGRAM_OPTIONS_TEST_STYLE");

  CHECK(styleEnv) << "PROGRAM_OPTIONS_TEST_STYLE is required";
  bool gnuStyle = !strcmp(styleEnv, "GNU");
  CHECK(gnuStyle || !strcmp(styleEnv, "GFLAGS"))
      << "Invalid value for PROGRAM_OPTIONS_TEST_STYLE";

  // clang-format off
  desc.add(getGFlags(
      gnuStyle ? folly::ProgramOptionsStyle::GNU :
      folly::ProgramOptionsStyle::GFLAGS));
  desc.add_options()
    ("help,h", "help");
  // clang-format on

  po::variables_map vm;
  auto result = folly::parseNestedCommandLine(argc, argv, desc);
  po::store(result.options, vm);
  po::notify(vm);

  if (vm.count("help")) {
    std::cout << desc;
    return 1;
  }

  print<bool>(vm, gnuStyle ? "flag-bool-true" : "flag_bool_true");
  print<bool>(vm, gnuStyle ? "flag-bool-false" : "flag_bool_false");
  print<int32_t>(vm, gnuStyle ? "flag-int" : "flag_int");
  print<std::string>(vm, gnuStyle ? "flag-string" : "flag_string");

  if (result.command) {
    printf("command %s\n", result.command->c_str());
  }

  for (auto& arg : result.rest) {
    printf("arg %s\n", arg.c_str());
  }

  return 0;
}