a3c4fe2ce7
Originally done in #166088. This broke 2 of our Flang builds with this error: FAILED: examples/OptSubcommand/CMakeFiles/OptSubcommand.dir/llvm-hello-sub.cpp.o /usr/local/bin/c++ <..> -c /home/tcwg-buildbot/worker/flang-aarch64-rel-assert/llvm-project/llvm/examples/OptSubcommand/llvm-hello-sub.cpp ../llvm-project/llvm/examples/OptSubcommand/llvm-hello-sub.cpp:49:51: error: expected '{' or ',' 49 | OptionSubCommandIDsTable) = default; | ^
138 lines
4.1 KiB
C++
138 lines
4.1 KiB
C++
#include "llvm/ADT/ArrayRef.h"
|
|
#include "llvm/ADT/StringRef.h"
|
|
#include "llvm/Option/ArgList.h"
|
|
#include "llvm/Option/OptTable.h"
|
|
#include "llvm/Support/Error.h"
|
|
#include "llvm/Support/InitLLVM.h"
|
|
#include "llvm/Support/raw_ostream.h"
|
|
|
|
using namespace llvm;
|
|
using namespace llvm::opt;
|
|
|
|
namespace {
|
|
enum ID {
|
|
OPT_INVALID = 0,
|
|
#define OPTION(PREFIXES, NAME, ID, KIND, GROUP, ALIAS, ALIASARGS, FLAGS, \
|
|
VISIBILITY, PARAM, HELPTEXT, HELPTEXTSFORVARIANTS, METAVAR, \
|
|
VALUES, SUBCOMMANDIDS_OFFSET) \
|
|
OPT_##ID,
|
|
#include "Opts.inc"
|
|
#undef OPTION
|
|
};
|
|
#define OPTTABLE_STR_TABLE_CODE
|
|
#include "Opts.inc"
|
|
#undef OPTTABLE_STR_TABLE_CODE
|
|
|
|
#define OPTTABLE_PREFIXES_TABLE_CODE
|
|
#include "Opts.inc"
|
|
#undef OPTTABLE_PREFIXES_TABLE_CODE
|
|
|
|
#define OPTTABLE_SUBCOMMAND_IDS_TABLE_CODE
|
|
#include "Opts.inc"
|
|
#undef OPTTABLE_SUBCOMMAND_IDS_TABLE_CODE
|
|
|
|
#define OPTTABLE_SUBCOMMANDS_CODE
|
|
#include "Opts.inc"
|
|
#undef OPTTABLE_SUBCOMMANDS_CODE
|
|
|
|
static constexpr OptTable::Info InfoTable[] = {
|
|
#define OPTION(...) LLVM_CONSTRUCT_OPT_INFO(__VA_ARGS__),
|
|
#include "Opts.inc"
|
|
#undef OPTION
|
|
};
|
|
|
|
class HelloSubOptTable : public GenericOptTable {
|
|
public:
|
|
HelloSubOptTable()
|
|
: GenericOptTable(OptionStrTable, OptionPrefixesTable, InfoTable,
|
|
/*IgnoreCase=*/false, OptionSubCommands,
|
|
OptionSubCommandIDsTable) {};
|
|
};
|
|
} // namespace
|
|
|
|
int main(int argc, char **argv) {
|
|
InitLLVM X(argc, argv);
|
|
HelloSubOptTable T;
|
|
unsigned MissingArgIndex, MissingArgCount;
|
|
|
|
auto HandleMultipleSubcommands = [](ArrayRef<StringRef> SubCommands) {
|
|
assert(SubCommands.size() > 1);
|
|
llvm::errs() << "error: more than one subcommand passed [\n";
|
|
for (auto SC : SubCommands)
|
|
llvm::errs() << " `" << SC << "`\n";
|
|
llvm::errs() << "]\n";
|
|
llvm::errs() << "See --help.\n";
|
|
exit(1);
|
|
};
|
|
|
|
auto HandleOtherPositionals = [](ArrayRef<StringRef> Positionals) {
|
|
assert(!Positionals.empty());
|
|
llvm::errs() << "error: unknown positional argument(s) [\n";
|
|
for (auto SC : Positionals)
|
|
llvm::errs() << " `" << SC << "`\n";
|
|
llvm::errs() << "]\n";
|
|
llvm::errs() << "See --help.\n";
|
|
exit(1);
|
|
};
|
|
|
|
InputArgList Args = T.ParseArgs(ArrayRef(argv + 1, argc - 1), MissingArgIndex,
|
|
MissingArgCount);
|
|
|
|
StringRef SubCommand = Args.getSubCommand(
|
|
T.getSubCommands(), HandleMultipleSubcommands, HandleOtherPositionals);
|
|
// Handle help. When help options is found, ignore all other options and exit
|
|
// after printing help.
|
|
|
|
if (Args.hasArg(OPT_help)) {
|
|
T.printHelp(llvm::outs(), "llvm-hello-sub [subcommand] [options]",
|
|
"LLVM Hello SubCommand Example", false, false, Visibility(),
|
|
SubCommand);
|
|
return 0;
|
|
}
|
|
|
|
auto HandleSubCommandArg = [&](ID OptionType) {
|
|
if (!Args.hasArg(OptionType))
|
|
return false;
|
|
auto O = T.getOption(OptionType);
|
|
if (!O.isRegisteredSC(SubCommand)) {
|
|
llvm::errs() << "Option [" << O.getName()
|
|
<< "] is not valid for SubCommand [" << SubCommand << "]\n";
|
|
return false;
|
|
}
|
|
return true;
|
|
};
|
|
|
|
bool HasUnknownOptions = false;
|
|
for (const Arg *A : Args.filtered(OPT_UNKNOWN)) {
|
|
HasUnknownOptions = true;
|
|
llvm::errs() << "Unknown option `" << A->getAsString(Args) << "'\n";
|
|
}
|
|
if (HasUnknownOptions) {
|
|
llvm::errs() << "See `OptSubcommand --help`.\n";
|
|
return 1;
|
|
}
|
|
if (SubCommand.empty()) {
|
|
if (Args.hasArg(OPT_version))
|
|
llvm::outs() << "LLVM Hello SubCommand Example 1.0\n";
|
|
} else if (SubCommand == "foo") {
|
|
if (HandleSubCommandArg(OPT_uppercase))
|
|
llvm::outs() << "FOO\n";
|
|
else if (HandleSubCommandArg(OPT_lowercase))
|
|
llvm::outs() << "foo\n";
|
|
|
|
if (HandleSubCommandArg(OPT_version))
|
|
llvm::outs() << "LLVM Hello SubCommand foo Example 1.0\n";
|
|
|
|
} else if (SubCommand == "bar") {
|
|
if (HandleSubCommandArg(OPT_lowercase))
|
|
llvm::outs() << "bar\n";
|
|
else if (HandleSubCommandArg(OPT_uppercase))
|
|
llvm::outs() << "BAR\n";
|
|
|
|
if (HandleSubCommandArg(OPT_version))
|
|
llvm::outs() << "LLVM Hello SubCommand bar Example 1.0\n";
|
|
}
|
|
|
|
return 0;
|
|
}
|