clang  10.0.0svn
FixItRewriter.h
Go to the documentation of this file.
1 //===- FixItRewriter.h - Fix-It Rewriter Diagnostic Client ------*- 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 is a diagnostic client adaptor that performs rewrites as
10 // suggested by code modification hints attached to diagnostics. It
11 // then forwards any diagnostics to the adapted diagnostic client.
12 //
13 //===----------------------------------------------------------------------===//
14 
15 #ifndef LLVM_CLANG_REWRITE_FRONTEND_FIXITREWRITER_H
16 #define LLVM_CLANG_REWRITE_FRONTEND_FIXITREWRITER_H
17 
18 #include "clang/Basic/Diagnostic.h"
19 #include "clang/Basic/LLVM.h"
23 #include <memory>
24 #include <string>
25 #include <utility>
26 #include <vector>
27 
28 namespace clang {
29 
30 class LangOptions;
31 class SourceManager;
32 
33 class FixItOptions {
34 public:
35  FixItOptions() = default;
36  virtual ~FixItOptions();
37 
38  /// This file is about to be rewritten. Return the name of the file
39  /// that is okay to write to.
40  ///
41  /// \param fd out parameter for file descriptor. After the call it may be set
42  /// to an open file descriptor for the returned filename, or it will be -1
43  /// otherwise.
44  virtual std::string RewriteFilename(const std::string &Filename, int &fd) = 0;
45 
46  /// True if files should be updated in place. RewriteFilename is only called
47  /// if this is false.
48  bool InPlace = false;
49 
50  /// Whether to abort fixing a file when not all errors could be fixed.
51  bool FixWhatYouCan = false;
52 
53  /// Whether to only fix warnings and not errors.
54  bool FixOnlyWarnings = false;
55 
56  /// If true, only pass the diagnostic to the actual diagnostic consumer
57  /// if it is an error or a fixit was applied as part of the diagnostic.
58  /// It basically silences warnings without accompanying fixits.
59  bool Silent = false;
60 };
61 
63  /// The diagnostics machinery.
64  DiagnosticsEngine &Diags;
65 
66  edit::EditedSource Editor;
67 
68  /// The rewriter used to perform the various code
69  /// modifications.
70  Rewriter Rewrite;
71 
72  /// The diagnostic client that performs the actual formatting
73  /// of error messages.
74  DiagnosticConsumer *Client;
75  std::unique_ptr<DiagnosticConsumer> Owner;
76 
77  /// Turn an input path into an output path. NULL implies overwriting
78  /// the original.
79  FixItOptions *FixItOpts;
80 
81  /// The number of rewriter failures.
82  unsigned NumFailures = 0;
83 
84  /// Whether the previous diagnostic was not passed to the consumer.
85  bool PrevDiagSilenced = false;
86 
87 public:
88  /// Initialize a new fix-it rewriter.
90  const LangOptions &LangOpts, FixItOptions *FixItOpts);
91 
92  /// Destroy the fix-it rewriter.
93  ~FixItRewriter() override;
94 
95  /// Check whether there are modifications for a given file.
96  bool IsModified(FileID ID) const {
97  return Rewrite.getRewriteBufferFor(ID) != nullptr;
98  }
99 
101 
102  // Iteration over files with changes.
103  iterator buffer_begin() { return Rewrite.buffer_begin(); }
104  iterator buffer_end() { return Rewrite.buffer_end(); }
105 
106  /// Write a single modified source file.
107  ///
108  /// \returns true if there was an error, false otherwise.
109  bool WriteFixedFile(FileID ID, raw_ostream &OS);
110 
111  /// Write the modified source files.
112  ///
113  /// \returns true if there was an error, false otherwise.
114  bool WriteFixedFiles(
115  std::vector<std::pair<std::string, std::string>> *RewrittenFiles = nullptr);
116 
117  /// IncludeInDiagnosticCounts - This method (whose default implementation
118  /// returns true) indicates whether the diagnostics handled by this
119  /// DiagnosticConsumer should be included in the number of diagnostics
120  /// reported by DiagnosticsEngine.
121  bool IncludeInDiagnosticCounts() const override;
122 
123  /// HandleDiagnostic - Handle this diagnostic, reporting it to the user or
124  /// capturing it to a log as needed.
125  void HandleDiagnostic(DiagnosticsEngine::Level DiagLevel,
126  const Diagnostic &Info) override;
127 
128  /// Emit a diagnostic via the adapted diagnostic client.
129  void Diag(SourceLocation Loc, unsigned DiagID);
130 };
131 
132 } // namespace clang
133 
134 #endif // LLVM_CLANG_REWRITE_FRONTEND_FIXITREWRITER_H
static DiagnosticBuilder Diag(DiagnosticsEngine *Diags, const LangOptions &Features, FullSourceLoc TokLoc, const char *TokBegin, const char *TokRangeBegin, const char *TokRangeEnd, unsigned DiagID)
Produce a diagnostic highlighting some portion of a literal.
bool FixOnlyWarnings
Whether to only fix warnings and not errors.
Definition: FixItRewriter.h:54
Abstract interface, implemented by clients of the front-end, which formats and prints fully processed...
Definition: Diagnostic.h:1496
Keeps track of the various options that can be enabled, which controls the dialect of C or C++ that i...
Definition: LangOptions.h:49
bool Silent
If true, only pass the diagnostic to the actual diagnostic consumer if it is an error or a fixit was ...
Definition: FixItRewriter.h:59
Forward-declares and imports various common LLVM datatypes that clang wants to use unqualified...
Concrete class used by the front-end to report problems and issues.
Definition: Diagnostic.h:149
Defines the Diagnostic-related interfaces.
buffer_iterator buffer_end()
Definition: Rewriter.h:195
StringRef Filename
Definition: Format.cpp:1711
FixItOptions()=default
Encodes a location in the source.
bool FixWhatYouCan
Whether to abort fixing a file when not all errors could be fixed.
Definition: FixItRewriter.h:51
const RewriteBuffer * getRewriteBufferFor(FileID FID) const
getRewriteBufferFor - Return the rewrite buffer for the specified FileID.
Definition: Rewriter.h:187
std::map< FileID, RewriteBuffer >::iterator buffer_iterator
Definition: Rewriter.h:54
An opaque identifier used by SourceManager which refers to a source file (MemoryBuffer) along with it...
Dataflow Directional Tag Classes.
Rewriter::buffer_iterator iterator
virtual ~FixItOptions()
buffer_iterator buffer_begin()
Definition: Rewriter.h:194
Rewriter - This is the main interface to the rewrite buffers.
Definition: Rewriter.h:32
Defines the clang::SourceLocation class and associated facilities.
Level
The level of the diagnostic, after it has been through mapping.
Definition: Diagnostic.h:152
bool IsModified(FileID ID) const
Check whether there are modifications for a given file.
Definition: FixItRewriter.h:96
virtual std::string RewriteFilename(const std::string &Filename, int &fd)=0
This file is about to be rewritten.
A little helper class (which is basically a smart pointer that forwards info from DiagnosticsEngine) ...
Definition: Diagnostic.h:1324
bool InPlace
True if files should be updated in place.
Definition: FixItRewriter.h:48
This class handles loading and caching of source files into memory.