2 * Copyright 2015 Facebook, Inc.
4 * Licensed under the Apache License, Version 2.0 (the "License");
5 * you may not use this file except in compliance with the License.
6 * You may obtain a copy of the License at
8 * http://www.apache.org/licenses/LICENSE-2.0
10 * Unless required by applicable law or agreed to in writing, software
11 * distributed under the License is distributed on an "AS IS" BASIS,
12 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13 * See the License for the specific language governing permissions and
14 * limitations under the License.
18 #include <folly/Conv.h>
19 #include <folly/experimental/ProgramOptions.h>
20 #include <glog/logging.h>
22 DEFINE_bool(flag_bool_true, true, "Bool with true default value");
23 DEFINE_bool(flag_bool_false, false, "Bool with false default value");
24 DEFINE_int32(flag_int, 42, "Integer flag");
25 DEFINE_string(flag_string, "foo", "String flag");
27 namespace po = ::boost::program_options;
31 void print(const po::variables_map& vm, const std::string& name) {
35 folly::to<std::string>(v.as<T>()).c_str());
39 int main(int argc, char *argv[]) {
40 po::options_description desc;
41 auto styleEnv = getenv("PROGRAM_OPTIONS_TEST_STYLE");
43 CHECK(styleEnv) << "PROGRAM_OPTIONS_TEST_STYLE is required";
44 bool gnuStyle = !strcmp(styleEnv, "GNU");
45 CHECK(gnuStyle || !strcmp(styleEnv, "GFLAGS"))
46 << "Invalid value for PROGRAM_OPTIONS_TEST_STYLE";
49 gnuStyle ? folly::ProgramOptionsStyle::GNU :
50 folly::ProgramOptionsStyle::GFLAGS));
55 auto result = folly::parseNestedCommandLine(argc, argv, desc);
56 po::store(result.options, vm);
59 if (vm.count("help")) {
64 print<bool>(vm, gnuStyle ? "flag-bool-true" : "flag_bool_true");
65 print<bool>(vm, gnuStyle ? "flag-bool-false" : "flag_bool_false");
66 print<int32_t>(vm, gnuStyle ? "flag-int" : "flag_int");
67 print<std::string>(vm, gnuStyle ? "flag-string" : "flag_string");
70 printf("command %s\n", result.command->c_str());
73 for (auto& arg : result.rest) {
74 printf("arg %s\n", arg.c_str());