1//===--- TextDiagnosticBuffer.cpp - Buffer Text Diagnostics ---------------===//
2//
3//                     The LLVM Compiler Infrastructure
4//
5// This file is distributed under the University of Illinois Open Source
6// License. See LICENSE.TXT for details.
7//
8//===----------------------------------------------------------------------===//
9//
10// This is a concrete diagnostic client, which buffers the diagnostic messages.
11//
12//===----------------------------------------------------------------------===//
13
14#include "clang/Frontend/TextDiagnosticBuffer.h"
15#include "llvm/ADT/SmallString.h"
16#include "llvm/Support/ErrorHandling.h"
17using namespace clang;
18
19/// HandleDiagnostic - Store the errors, warnings, and notes that are
20/// reported.
21///
22void TextDiagnosticBuffer::HandleDiagnostic(DiagnosticsEngine::Level Level,
23                                            const Diagnostic &Info) {
24  // Default implementation (Warnings/errors count).
25  DiagnosticConsumer::HandleDiagnostic(Level, Info);
26
27  SmallString<100> Buf;
28  Info.FormatDiagnostic(Buf);
29  switch (Level) {
30  default: llvm_unreachable(
31                         "Diagnostic not handled during diagnostic buffering!");
32  case DiagnosticsEngine::Note:
33    Notes.push_back(std::make_pair(Info.getLocation(), Buf.str()));
34    break;
35  case DiagnosticsEngine::Warning:
36    Warnings.push_back(std::make_pair(Info.getLocation(), Buf.str()));
37    break;
38  case DiagnosticsEngine::Error:
39  case DiagnosticsEngine::Fatal:
40    Errors.push_back(std::make_pair(Info.getLocation(), Buf.str()));
41    break;
42  }
43}
44
45/// \brief Escape diagnostic texts to avoid problems when they are fed into the
46/// diagnostic formatter a second time.
47static StringRef escapeDiag(StringRef Str, SmallVectorImpl<char> &Buf) {
48  size_t Pos = Str.find('%');
49  if (Pos == StringRef::npos)
50    return Str;
51
52  // We found a '%'. Replace this and all following '%' with '%%'.
53  Buf.clear();
54  Buf.append(Str.data(), Str.data() + Pos);
55  for (size_t I = Pos, E = Str.size(); I != E; ++I) {
56    if (Str[I] == '%')
57      Buf.push_back('%');
58    Buf.push_back(Str[I]);
59  }
60
61  return StringRef(Buf.data(), Buf.size());
62}
63
64void TextDiagnosticBuffer::FlushDiagnostics(DiagnosticsEngine &Diags) const {
65  SmallVector<char, 64> Buf;
66  // FIXME: Flush the diagnostics in order.
67  for (const_iterator it = err_begin(), ie = err_end(); it != ie; ++it)
68    Diags.Report(Diags.getCustomDiagID(DiagnosticsEngine::Error,
69                                       escapeDiag(it->second, Buf)));
70  for (const_iterator it = warn_begin(), ie = warn_end(); it != ie; ++it)
71    Diags.Report(Diags.getCustomDiagID(DiagnosticsEngine::Warning,
72                                       escapeDiag(it->second, Buf)));
73  for (const_iterator it = note_begin(), ie = note_end(); it != ie; ++it)
74    Diags.Report(Diags.getCustomDiagID(DiagnosticsEngine::Note,
75                                       escapeDiag(it->second, Buf)));
76}
77
78