Details | Last modification | View Log | RSS feed
Rev | Author | Line No. | Line |
---|---|---|---|
14 | pmbaty | 1 | //===--- AtomicChange.h - AtomicChange class --------------------*- 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 | // This file defines AtomicChange which is used to create a set of source |
||
10 | // changes, e.g. replacements and header insertions. |
||
11 | // |
||
12 | //===----------------------------------------------------------------------===// |
||
13 | |||
14 | #ifndef LLVM_CLANG_TOOLING_REFACTORING_ATOMICCHANGE_H |
||
15 | #define LLVM_CLANG_TOOLING_REFACTORING_ATOMICCHANGE_H |
||
16 | |||
17 | #include "clang/Basic/SourceManager.h" |
||
18 | #include "clang/Format/Format.h" |
||
19 | #include "clang/Tooling/Core/Replacement.h" |
||
20 | #include "llvm/ADT/Any.h" |
||
21 | #include "llvm/ADT/StringRef.h" |
||
22 | #include "llvm/Support/Error.h" |
||
23 | |||
24 | namespace clang { |
||
25 | namespace tooling { |
||
26 | |||
27 | /// An atomic change is used to create and group a set of source edits, |
||
28 | /// e.g. replacements or header insertions. Edits in an AtomicChange should be |
||
29 | /// related, e.g. replacements for the same type reference and the corresponding |
||
30 | /// header insertion/deletion. |
||
31 | /// |
||
32 | /// An AtomicChange is uniquely identified by a key and will either be fully |
||
33 | /// applied or not applied at all. |
||
34 | /// |
||
35 | /// Calling setError on an AtomicChange stores the error message and marks it as |
||
36 | /// bad, i.e. none of its source edits will be applied. |
||
37 | class AtomicChange { |
||
38 | public: |
||
39 | /// Creates an atomic change around \p KeyPosition with the key being a |
||
40 | /// concatenation of the file name and the offset of \p KeyPosition. |
||
41 | /// \p KeyPosition should be the location of the key syntactical element that |
||
42 | /// is being changed, e.g. the call to a refactored method. |
||
43 | AtomicChange(const SourceManager &SM, SourceLocation KeyPosition); |
||
44 | |||
45 | AtomicChange(const SourceManager &SM, SourceLocation KeyPosition, |
||
46 | llvm::Any Metadata); |
||
47 | |||
48 | /// Creates an atomic change for \p FilePath with a customized key. |
||
49 | AtomicChange(llvm::StringRef FilePath, llvm::StringRef Key) |
||
50 | : Key(Key), FilePath(FilePath) {} |
||
51 | |||
52 | AtomicChange(AtomicChange &&) = default; |
||
53 | AtomicChange(const AtomicChange &) = default; |
||
54 | |||
55 | AtomicChange &operator=(AtomicChange &&) = default; |
||
56 | AtomicChange &operator=(const AtomicChange &) = default; |
||
57 | |||
58 | bool operator==(const AtomicChange &Other) const; |
||
59 | |||
60 | /// Returns the atomic change as a YAML string. |
||
61 | std::string toYAMLString(); |
||
62 | |||
63 | /// Converts a YAML-encoded automic change to AtomicChange. |
||
64 | static AtomicChange convertFromYAML(llvm::StringRef YAMLContent); |
||
65 | |||
66 | /// Returns the key of this change, which is a concatenation of the |
||
67 | /// file name and offset of the key position. |
||
68 | const std::string &getKey() const { return Key; } |
||
69 | |||
70 | /// Returns the path of the file containing this atomic change. |
||
71 | const std::string &getFilePath() const { return FilePath; } |
||
72 | |||
73 | /// If this change could not be created successfully, e.g. because of |
||
74 | /// conflicts among replacements, use this to set an error description. |
||
75 | /// Thereby, places that cannot be fixed automatically can be gathered when |
||
76 | /// applying changes. |
||
77 | void setError(llvm::StringRef Error) { this->Error = std::string(Error); } |
||
78 | |||
79 | /// Returns whether an error has been set on this list. |
||
80 | bool hasError() const { return !Error.empty(); } |
||
81 | |||
82 | /// Returns the error message or an empty string if it does not exist. |
||
83 | const std::string &getError() const { return Error; } |
||
84 | |||
85 | /// Adds a replacement that replaces the given Range with |
||
86 | /// ReplacementText. |
||
87 | /// \returns An llvm::Error carrying ReplacementError on error. |
||
88 | llvm::Error replace(const SourceManager &SM, const CharSourceRange &Range, |
||
89 | llvm::StringRef ReplacementText); |
||
90 | |||
91 | /// Adds a replacement that replaces range [Loc, Loc+Length) with |
||
92 | /// \p Text. |
||
93 | /// \returns An llvm::Error carrying ReplacementError on error. |
||
94 | llvm::Error replace(const SourceManager &SM, SourceLocation Loc, |
||
95 | unsigned Length, llvm::StringRef Text); |
||
96 | |||
97 | /// Adds a replacement that inserts \p Text at \p Loc. If this |
||
98 | /// insertion conflicts with an existing insertion (at the same position), |
||
99 | /// this will be inserted before/after the existing insertion depending on |
||
100 | /// \p InsertAfter. Users should use `replace` with `Length=0` instead if they |
||
101 | /// do not want conflict resolving by default. If the conflicting replacement |
||
102 | /// is not an insertion, an error is returned. |
||
103 | /// |
||
104 | /// \returns An llvm::Error carrying ReplacementError on error. |
||
105 | llvm::Error insert(const SourceManager &SM, SourceLocation Loc, |
||
106 | llvm::StringRef Text, bool InsertAfter = true); |
||
107 | |||
108 | /// Adds a header into the file that contains the key position. |
||
109 | /// Header can be in angle brackets or double quotation marks. By default |
||
110 | /// (header is not quoted), header will be surrounded with double quotes. |
||
111 | void addHeader(llvm::StringRef Header); |
||
112 | |||
113 | /// Removes a header from the file that contains the key position. |
||
114 | void removeHeader(llvm::StringRef Header); |
||
115 | |||
116 | /// Returns a const reference to existing replacements. |
||
117 | const Replacements &getReplacements() const { return Replaces; } |
||
118 | |||
119 | Replacements &getReplacements() { return Replaces; } |
||
120 | |||
121 | llvm::ArrayRef<std::string> getInsertedHeaders() const { |
||
122 | return InsertedHeaders; |
||
123 | } |
||
124 | |||
125 | llvm::ArrayRef<std::string> getRemovedHeaders() const { |
||
126 | return RemovedHeaders; |
||
127 | } |
||
128 | |||
129 | const llvm::Any &getMetadata() const { return Metadata; } |
||
130 | |||
131 | private: |
||
132 | AtomicChange() {} |
||
133 | |||
134 | AtomicChange(std::string Key, std::string FilePath, std::string Error, |
||
135 | std::vector<std::string> InsertedHeaders, |
||
136 | std::vector<std::string> RemovedHeaders, |
||
137 | clang::tooling::Replacements Replaces); |
||
138 | |||
139 | // This uniquely identifies an AtomicChange. |
||
140 | std::string Key; |
||
141 | std::string FilePath; |
||
142 | std::string Error; |
||
143 | std::vector<std::string> InsertedHeaders; |
||
144 | std::vector<std::string> RemovedHeaders; |
||
145 | tooling::Replacements Replaces; |
||
146 | |||
147 | // This field stores metadata which is ignored for the purposes of applying |
||
148 | // edits to source, but may be useful for other consumers of AtomicChanges. In |
||
149 | // particular, consumers can use this to direct how they want to consume each |
||
150 | // edit. |
||
151 | llvm::Any Metadata; |
||
152 | }; |
||
153 | |||
154 | using AtomicChanges = std::vector<AtomicChange>; |
||
155 | |||
156 | // Defines specs for applying changes. |
||
157 | struct ApplyChangesSpec { |
||
158 | // If true, cleans up redundant/erroneous code around changed code with |
||
159 | // clang-format's cleanup functionality, e.g. redundant commas around deleted |
||
160 | // parameter or empty namespaces introduced by deletions. |
||
161 | bool Cleanup = true; |
||
162 | |||
163 | format::FormatStyle Style = format::getNoStyle(); |
||
164 | |||
165 | // Options for selectively formatting changes with clang-format: |
||
166 | // kAll: Format all changed lines. |
||
167 | // kNone: Don't format anything. |
||
168 | // kViolations: Format lines exceeding the `ColumnLimit` in `Style`. |
||
169 | enum FormatOption { kAll, kNone, kViolations }; |
||
170 | |||
171 | FormatOption Format = kNone; |
||
172 | }; |
||
173 | |||
174 | /// Applies all AtomicChanges in \p Changes to the \p Code. |
||
175 | /// |
||
176 | /// This completely ignores the file path in each change and replaces them with |
||
177 | /// \p FilePath, i.e. callers are responsible for ensuring all changes are for |
||
178 | /// the same file. |
||
179 | /// |
||
180 | /// \returns The changed code if all changes are applied successfully; |
||
181 | /// otherwise, an llvm::Error carrying llvm::StringError is returned (the Error |
||
182 | /// message can be converted to string with `llvm::toString()` and the |
||
183 | /// error_code should be ignored). |
||
184 | llvm::Expected<std::string> |
||
185 | applyAtomicChanges(llvm::StringRef FilePath, llvm::StringRef Code, |
||
186 | llvm::ArrayRef<AtomicChange> Changes, |
||
187 | const ApplyChangesSpec &Spec); |
||
188 | |||
189 | } // end namespace tooling |
||
190 | } // end namespace clang |
||
191 | |||
192 | #endif // LLVM_CLANG_TOOLING_REFACTORING_ATOMICCHANGE_H |