1 //===- TextDiagnosticBuffer.cpp - Buffer Text Diagnostics -----------------===//
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 // This is a concrete diagnostic client, which buffers the diagnostic messages.
10 //
11 //===----------------------------------------------------------------------===//
12 
13 #include "clang/Frontend/TextDiagnosticBuffer.h"
14 #include "clang/Basic/Diagnostic.h"
15 #include "clang/Basic/LLVM.h"
16 #include "llvm/ADT/SmallString.h"
17 #include "llvm/Support/ErrorHandling.h"
18 
19 using namespace clang;
20 
21 /// HandleDiagnostic - Store the errors, warnings, and notes that are
22 /// reported.
23 void TextDiagnosticBuffer::HandleDiagnostic(DiagnosticsEngine::Level Level,
24                                             const Diagnostic &Info) {
25   // Default implementation (Warnings/errors count).
26   DiagnosticConsumer::HandleDiagnostic(Level, Info);
27 
28   SmallString<100> Buf;
29   Info.FormatDiagnostic(Buf);
30   switch (Level) {
31   default: llvm_unreachable(
32                          "Diagnostic not handled during diagnostic buffering!");
33   case DiagnosticsEngine::Note:
34     All.emplace_back(Level, Notes.size());
35     Notes.emplace_back(Info.getLocation(), std::string(Buf.str()));
36     break;
37   case DiagnosticsEngine::Warning:
38     All.emplace_back(Level, Warnings.size());
39     Warnings.emplace_back(Info.getLocation(), std::string(Buf.str()));
40     break;
41   case DiagnosticsEngine::Remark:
42     All.emplace_back(Level, Remarks.size());
43     Remarks.emplace_back(Info.getLocation(), std::string(Buf.str()));
44     break;
45   case DiagnosticsEngine::Error:
46   case DiagnosticsEngine::Fatal:
47     All.emplace_back(Level, Errors.size());
48     Errors.emplace_back(Info.getLocation(), std::string(Buf.str()));
49     break;
50   }
51 }
52 
53 void TextDiagnosticBuffer::FlushDiagnostics(DiagnosticsEngine &Diags) const {
54   for (const auto &I : All) {
55     auto Diag = Diags.Report(Diags.getCustomDiagID(I.first, "%0"));
56     switch (I.first) {
57     default: llvm_unreachable(
58                            "Diagnostic not handled during diagnostic flushing!");
59     case DiagnosticsEngine::Note:
60       Diag << Notes[I.second].second;
61       break;
62     case DiagnosticsEngine::Warning:
63       Diag << Warnings[I.second].second;
64       break;
65     case DiagnosticsEngine::Remark:
66       Diag << Remarks[I.second].second;
67       break;
68     case DiagnosticsEngine::Error:
69     case DiagnosticsEngine::Fatal:
70       Diag << Errors[I.second].second;
71       break;
72     }
73   }
74 }
75