llvm-project

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));
36
    break;
37
  case DiagnosticsEngine::Warning:
38
    All.emplace_back(Level, Warnings.size());
39
    Warnings.emplace_back(Info.getLocation(), std::string(Buf));
40
    break;
41
  case DiagnosticsEngine::Remark:
42
    All.emplace_back(Level, Remarks.size());
43
    Remarks.emplace_back(Info.getLocation(), std::string(Buf));
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));
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