blob: 11e79ed593410d5fda5adaf3ceac339fca76e5d1 [file] [log] [blame]
//===-- llvm-driver.cpp ---------------------------------------------------===//
//
// Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
// See https://llvm.org/LICENSE.txt for license information.
// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
//
//===----------------------------------------------------------------------===//
#include "llvm/ADT/StringExtras.h"
#include "llvm/ADT/StringRef.h"
#include "llvm/ADT/StringSwitch.h"
#include "llvm/Support/CommandLine.h"
#include "llvm/Support/ErrorHandling.h"
#include "llvm/Support/LLVMDriver.h"
#include "llvm/Support/Path.h"
#include "llvm/Support/WithColor.h"
using namespace llvm;
#define LLVM_DRIVER_TOOL(tool, entry) \
int entry##_main(int argc, char **argv, const llvm::ToolContext &);
#include "LLVMDriverTools.def"
constexpr char subcommands[] =
#define LLVM_DRIVER_TOOL(tool, entry) " " tool "\n"
#include "LLVMDriverTools.def"
;
static void printHelpMessage() {
llvm::outs() << "OVERVIEW: llvm toolchain driver\n\n"
<< "USAGE: llvm [subcommand] [options]\n\n"
<< "SUBCOMMANDS:\n\n"
<< subcommands
<< "\n Type \"llvm <subcommand> --help\" to get more help on a "
"specific subcommand\n\n"
<< "OPTIONS:\n\n --help - Display this message";
}
static int findTool(int Argc, char **Argv, const char *Argv0) {
if (!Argc) {
printHelpMessage();
return 1;
}
StringRef ToolName = Argv[0];
if (ToolName == "--help") {
printHelpMessage();
return 0;
}
StringRef Stem = sys::path::stem(ToolName);
auto Is = [=](StringRef Tool) {
auto IsImpl = [=](StringRef Stem) {
auto I = Stem.rfind_insensitive(Tool);
return I != StringRef::npos && (I + Tool.size() == Stem.size() ||
!llvm::isAlnum(Stem[I + Tool.size()]));
};
for (StringRef S : {Stem, sys::path::filename(ToolName)})
if (IsImpl(S))
return true;
return false;
};
auto MakeDriverArgs = [=]() -> llvm::ToolContext {
if (ToolName != Argv0)
return {Argv0, ToolName.data(), true};
return {Argv0, sys::path::filename(Argv0).data(), false};
};
#define LLVM_DRIVER_TOOL(tool, entry) \
if (Is(tool)) \
return entry##_main(Argc, Argv, MakeDriverArgs());
#include "LLVMDriverTools.def"
if (Is("llvm") || Argv0 == Argv[0])
return findTool(Argc - 1, Argv + 1, Argv0);
printHelpMessage();
return 1;
}
int main(int Argc, char **Argv) { return findTool(Argc, Argv, Argv[0]); }