1e5dd7070Spatrick //===-- DiagnosticsYaml.h -- Serialiazation for Diagnosticss ---*- C++ -*-===//
2e5dd7070Spatrick //
3e5dd7070Spatrick // Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
4e5dd7070Spatrick // See https://llvm.org/LICENSE.txt for license information.
5e5dd7070Spatrick // SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
6e5dd7070Spatrick //
7e5dd7070Spatrick //===----------------------------------------------------------------------===//
8e5dd7070Spatrick ///
9e5dd7070Spatrick /// \file
10e5dd7070Spatrick /// This file defines the structure of a YAML document for serializing
11e5dd7070Spatrick /// diagnostics.
12e5dd7070Spatrick ///
13e5dd7070Spatrick //===----------------------------------------------------------------------===//
14e5dd7070Spatrick
15e5dd7070Spatrick #ifndef LLVM_CLANG_TOOLING_DIAGNOSTICSYAML_H
16e5dd7070Spatrick #define LLVM_CLANG_TOOLING_DIAGNOSTICSYAML_H
17e5dd7070Spatrick
18e5dd7070Spatrick #include "clang/Tooling/Core/Diagnostic.h"
19e5dd7070Spatrick #include "clang/Tooling/ReplacementsYaml.h"
20e5dd7070Spatrick #include "llvm/Support/YAMLTraits.h"
21e5dd7070Spatrick #include <string>
22e5dd7070Spatrick
23e5dd7070Spatrick LLVM_YAML_IS_SEQUENCE_VECTOR(clang::tooling::Diagnostic)
LLVM_YAML_IS_SEQUENCE_VECTOR(clang::tooling::DiagnosticMessage)24e5dd7070Spatrick LLVM_YAML_IS_SEQUENCE_VECTOR(clang::tooling::DiagnosticMessage)
25ec727ea7Spatrick LLVM_YAML_IS_SEQUENCE_VECTOR(clang::tooling::FileByteRange)
26e5dd7070Spatrick
27e5dd7070Spatrick namespace llvm {
28e5dd7070Spatrick namespace yaml {
29e5dd7070Spatrick
30ec727ea7Spatrick template <> struct MappingTraits<clang::tooling::FileByteRange> {
31ec727ea7Spatrick static void mapping(IO &Io, clang::tooling::FileByteRange &R) {
32ec727ea7Spatrick Io.mapRequired("FilePath", R.FilePath);
33ec727ea7Spatrick Io.mapRequired("FileOffset", R.FileOffset);
34ec727ea7Spatrick Io.mapRequired("Length", R.Length);
35ec727ea7Spatrick }
36ec727ea7Spatrick };
37ec727ea7Spatrick
38e5dd7070Spatrick template <> struct MappingTraits<clang::tooling::DiagnosticMessage> {
39e5dd7070Spatrick static void mapping(IO &Io, clang::tooling::DiagnosticMessage &M) {
40e5dd7070Spatrick Io.mapRequired("Message", M.Message);
41e5dd7070Spatrick Io.mapOptional("FilePath", M.FilePath);
42e5dd7070Spatrick Io.mapOptional("FileOffset", M.FileOffset);
43e5dd7070Spatrick std::vector<clang::tooling::Replacement> Fixes;
44e5dd7070Spatrick for (auto &Replacements : M.Fix) {
45*12c85518Srobert llvm::append_range(Fixes, Replacements.second);
46e5dd7070Spatrick }
47e5dd7070Spatrick Io.mapRequired("Replacements", Fixes);
48e5dd7070Spatrick for (auto &Fix : Fixes) {
49e5dd7070Spatrick llvm::Error Err = M.Fix[Fix.getFilePath()].add(Fix);
50e5dd7070Spatrick if (Err) {
51e5dd7070Spatrick // FIXME: Implement better conflict handling.
52e5dd7070Spatrick llvm::errs() << "Fix conflicts with existing fix: "
53e5dd7070Spatrick << llvm::toString(std::move(Err)) << "\n";
54e5dd7070Spatrick }
55e5dd7070Spatrick }
56a9ac8606Spatrick Io.mapOptional("Ranges", M.Ranges);
57e5dd7070Spatrick }
58e5dd7070Spatrick };
59e5dd7070Spatrick
60e5dd7070Spatrick template <> struct MappingTraits<clang::tooling::Diagnostic> {
61e5dd7070Spatrick /// Helper to (de)serialize a Diagnostic since we don't have direct
62e5dd7070Spatrick /// access to its data members.
63e5dd7070Spatrick class NormalizedDiagnostic {
64e5dd7070Spatrick public:
65e5dd7070Spatrick NormalizedDiagnostic(const IO &)
66e5dd7070Spatrick : DiagLevel(clang::tooling::Diagnostic::Level::Warning) {}
67e5dd7070Spatrick
68e5dd7070Spatrick NormalizedDiagnostic(const IO &, const clang::tooling::Diagnostic &D)
69e5dd7070Spatrick : DiagnosticName(D.DiagnosticName), Message(D.Message), Notes(D.Notes),
70a9ac8606Spatrick DiagLevel(D.DiagLevel), BuildDirectory(D.BuildDirectory) {}
71e5dd7070Spatrick
72e5dd7070Spatrick clang::tooling::Diagnostic denormalize(const IO &) {
73e5dd7070Spatrick return clang::tooling::Diagnostic(DiagnosticName, Message, Notes,
74a9ac8606Spatrick DiagLevel, BuildDirectory);
75e5dd7070Spatrick }
76e5dd7070Spatrick
77e5dd7070Spatrick std::string DiagnosticName;
78e5dd7070Spatrick clang::tooling::DiagnosticMessage Message;
79e5dd7070Spatrick SmallVector<clang::tooling::DiagnosticMessage, 1> Notes;
80e5dd7070Spatrick clang::tooling::Diagnostic::Level DiagLevel;
81e5dd7070Spatrick std::string BuildDirectory;
82e5dd7070Spatrick };
83e5dd7070Spatrick
84e5dd7070Spatrick static void mapping(IO &Io, clang::tooling::Diagnostic &D) {
85e5dd7070Spatrick MappingNormalization<NormalizedDiagnostic, clang::tooling::Diagnostic> Keys(
86e5dd7070Spatrick Io, D);
87e5dd7070Spatrick Io.mapRequired("DiagnosticName", Keys->DiagnosticName);
88e5dd7070Spatrick Io.mapRequired("DiagnosticMessage", Keys->Message);
89e5dd7070Spatrick Io.mapOptional("Notes", Keys->Notes);
90ec727ea7Spatrick Io.mapOptional("Level", Keys->DiagLevel);
91ec727ea7Spatrick Io.mapOptional("BuildDirectory", Keys->BuildDirectory);
92e5dd7070Spatrick }
93e5dd7070Spatrick };
94e5dd7070Spatrick
95e5dd7070Spatrick /// Specialized MappingTraits to describe how a
96e5dd7070Spatrick /// TranslationUnitDiagnostics is (de)serialized.
97e5dd7070Spatrick template <> struct MappingTraits<clang::tooling::TranslationUnitDiagnostics> {
98e5dd7070Spatrick static void mapping(IO &Io, clang::tooling::TranslationUnitDiagnostics &Doc) {
99e5dd7070Spatrick Io.mapRequired("MainSourceFile", Doc.MainSourceFile);
100e5dd7070Spatrick Io.mapRequired("Diagnostics", Doc.Diagnostics);
101e5dd7070Spatrick }
102e5dd7070Spatrick };
103ec727ea7Spatrick
104ec727ea7Spatrick template <> struct ScalarEnumerationTraits<clang::tooling::Diagnostic::Level> {
105ec727ea7Spatrick static void enumeration(IO &IO, clang::tooling::Diagnostic::Level &Value) {
106ec727ea7Spatrick IO.enumCase(Value, "Warning", clang::tooling::Diagnostic::Warning);
107ec727ea7Spatrick IO.enumCase(Value, "Error", clang::tooling::Diagnostic::Error);
108a9ac8606Spatrick IO.enumCase(Value, "Remark", clang::tooling::Diagnostic::Remark);
109ec727ea7Spatrick }
110ec727ea7Spatrick };
111ec727ea7Spatrick
112e5dd7070Spatrick } // end namespace yaml
113e5dd7070Spatrick } // end namespace llvm
114e5dd7070Spatrick
115e5dd7070Spatrick #endif // LLVM_CLANG_TOOLING_DIAGNOSTICSYAML_H
116