1 //===-- ReplacementsYaml.h -- Serialiazation for Replacements ---*- C++ -*-===//
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 /// \file
11 /// \brief This file defines the structure of a YAML document for serializing
12 /// replacements.
13 ///
14 //===----------------------------------------------------------------------===//
15
16 #ifndef LLVM_CLANG_TOOLING_REPLACEMENTSYAML_H
17 #define LLVM_CLANG_TOOLING_REPLACEMENTSYAML_H
18
19 #include "clang/Tooling/Refactoring.h"
20 #include "llvm/Support/YAMLTraits.h"
21 #include <string>
22 #include <vector>
23
LLVM_YAML_IS_SEQUENCE_VECTOR(clang::tooling::Replacement)24 LLVM_YAML_IS_SEQUENCE_VECTOR(clang::tooling::Replacement)
25
26 namespace llvm {
27 namespace yaml {
28
29 /// \brief Specialized MappingTraits to describe how a Replacement is
30 /// (de)serialized.
31 template <> struct MappingTraits<clang::tooling::Replacement> {
32 /// \brief Helper to (de)serialize a Replacement since we don't have direct
33 /// access to its data members.
34 struct NormalizedReplacement {
35 NormalizedReplacement(const IO &)
36 : FilePath(""), Offset(0), Length(0), ReplacementText("") {}
37
38 NormalizedReplacement(const IO &, const clang::tooling::Replacement &R)
39 : FilePath(R.getFilePath()), Offset(R.getOffset()),
40 Length(R.getLength()), ReplacementText(R.getReplacementText()) {}
41
42 clang::tooling::Replacement denormalize(const IO &) {
43 return clang::tooling::Replacement(FilePath, Offset, Length,
44 ReplacementText);
45 }
46
47 std::string FilePath;
48 unsigned int Offset;
49 unsigned int Length;
50 std::string ReplacementText;
51 };
52
53 static void mapping(IO &Io, clang::tooling::Replacement &R) {
54 MappingNormalization<NormalizedReplacement, clang::tooling::Replacement>
55 Keys(Io, R);
56 Io.mapRequired("FilePath", Keys->FilePath);
57 Io.mapRequired("Offset", Keys->Offset);
58 Io.mapRequired("Length", Keys->Length);
59 Io.mapRequired("ReplacementText", Keys->ReplacementText);
60 }
61 };
62
63 /// \brief Specialized MappingTraits to describe how a
64 /// TranslationUnitReplacements is (de)serialized.
65 template <> struct MappingTraits<clang::tooling::TranslationUnitReplacements> {
66 static void mapping(IO &Io,
67 clang::tooling::TranslationUnitReplacements &Doc) {
68 Io.mapRequired("MainSourceFile", Doc.MainSourceFile);
69 Io.mapOptional("Context", Doc.Context, std::string());
70 Io.mapRequired("Replacements", Doc.Replacements);
71 }
72 };
73 } // end namespace yaml
74 } // end namespace llvm
75
76 #endif
77