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