| 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(Key: "FilePath" , Val&: Keys->FilePath); |
| 54 | Io.mapRequired(Key: "Offset" , Val&: Keys->Offset); |
| 55 | Io.mapRequired(Key: "Length" , Val&: Keys->Length); |
| 56 | Io.mapRequired(Key: "ReplacementText" , Val&: 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(Key: "MainSourceFile" , Val&: Doc.MainSourceFile); |
| 66 | Io.mapRequired(Key: "Replacements" , Val&: Doc.Replacements); |
| 67 | } |
| 68 | }; |
| 69 | } // end namespace yaml |
| 70 | } // end namespace llvm |
| 71 | |
| 72 | #endif |
| 73 | |