clang/tools/diagtool/FindDiagnosticID.cpp

   1 //===- FindDiagnosticID.cpp - diagtool tool for finding diagnostic id -----===//
   2 //
   3 // Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
   4 // See https://llvm.org/LICENSE.txt for license information.
   5 // SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
   6 //
   7 //===----------------------------------------------------------------------===//
   8
   9 #include "DiagTool.h"
  10 #include "DiagnosticNames.h"
  11 #include "clang/Basic/AllDiagnostics.h"
  12 #include "llvm/Support/CommandLine.h"
  13 #include <optional>
  14
  15 DEF_DIAGTOOL("find-diagnostic-id", "Print the id of the given diagnostic",
  16              FindDiagnosticID)
  17
  18 using namespace clang;
  19 using namespace diagtool;
  20
  21 static StringRef getNameFromID(StringRef Name) {
  22   int DiagID;
  23   if(!Name.getAsInteger(0, DiagID)) {
  24     const DiagnosticRecord &Diag = getDiagnosticForID(DiagID);
  25     return Diag.getName();
  26   }
  27   return StringRef();
  28 }
  29
  30 static std::optional<DiagnosticRecord>
  31 findDiagnostic(ArrayRef<DiagnosticRecord> Diagnostics, StringRef Name) {
  32   for (const auto &Diag : Diagnostics) {
  33     StringRef DiagName = Diag.getName();
  34     if (DiagName == Name)
  35       return Diag;
  36   }
  37   return std::nullopt;
  38 }
  39
  40 int FindDiagnosticID::run(unsigned int argc, char **argv,
  41                           llvm::raw_ostream &OS) {
  42   static llvm::cl::OptionCategory FindDiagnosticIDOptions(
  43       "diagtool find-diagnostic-id options");
  44
  45   static llvm::cl::opt<std::string> DiagnosticName(
  46       llvm::cl::Positional, llvm::cl::desc("<diagnostic-name>"),
  47       llvm::cl::Required, llvm::cl::cat(FindDiagnosticIDOptions));
  48
  49   std::vector<const char *> Args;
  50   Args.push_back("diagtool find-diagnostic-id");
  51   for (const char *A : llvm::ArrayRef(argv, argc))
  52     Args.push_back(A);
  53
  54   llvm::cl::HideUnrelatedOptions(FindDiagnosticIDOptions);
  55   llvm::cl::ParseCommandLineOptions((int)Args.size(), Args.data(),
  56                                     "Diagnostic ID mapping utility");
  57
  58   ArrayRef<DiagnosticRecord> AllDiagnostics = getBuiltinDiagnosticsByName();
  59   std::optional<DiagnosticRecord> Diag =
  60       findDiagnostic(AllDiagnostics, DiagnosticName);
  61   if (!Diag) {
  62     // Name to id failed, so try id to name.
  63     auto Name = getNameFromID(DiagnosticName);
  64     if (!Name.empty()) {
  65       OS << Name << '\n';
  66       return 0;
  67     }
  68
  69     llvm::errs() << "error: invalid diagnostic '" << DiagnosticName << "'\n";
  70     return 1;
  71   }
  72   OS << Diag->DiagID << "\n";
  73   return 0;
  74 }