clang  10.0.0svn
RefactoringRuleContext.h
Go to the documentation of this file.
1 //===--- RefactoringRuleContext.h - Clang refactoring library -------------===//
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 #ifndef LLVM_CLANG_TOOLING_REFACTOR_REFACTORING_RULE_CONTEXT_H
10 #define LLVM_CLANG_TOOLING_REFACTOR_REFACTORING_RULE_CONTEXT_H
11 
15 
16 namespace clang {
17 
18 class ASTContext;
19 
20 namespace tooling {
21 
22 /// The refactoring rule context stores all of the inputs that might be needed
23 /// by a refactoring action rule. It can create the specialized
24 /// \c ASTRefactoringOperation or \c PreprocessorRefactoringOperation values
25 /// that can be used by the refactoring action rules.
26 ///
27 /// The following inputs are stored by the operation:
28 ///
29 /// - SourceManager: a reference to a valid source manager.
30 ///
31 /// - SelectionRange: an optional source selection ranges that can be used
32 /// to represent a selection in an editor.
34 public:
35  RefactoringRuleContext(const SourceManager &SM) : SM(SM) {}
36 
37  const SourceManager &getSources() const { return SM; }
38 
39  /// Returns the current source selection range as set by the
40  /// refactoring engine. Can be invalid.
41  SourceRange getSelectionRange() const { return SelectionRange; }
42 
43  void setSelectionRange(SourceRange R) { SelectionRange = R; }
44 
45  bool hasASTContext() const { return AST; }
46 
48  assert(AST && "no AST!");
49  return *AST;
50  }
51 
52  void setASTContext(ASTContext &Context) { AST = &Context; }
53 
54  /// Creates an llvm::Error value that contains a diagnostic.
55  ///
56  /// The errors should not outlive the context.
58  return DiagnosticError::create(Loc, PartialDiagnostic(DiagID, DiagStorage));
59  }
60 
62  return createDiagnosticError(SourceLocation(), DiagID);
63  }
64 
65  void setASTSelection(std::unique_ptr<SelectedASTNode> Node) {
66  ASTNodeSelection = std::move(Node);
67  }
68 
69 private:
70  /// The source manager for the translation unit / file on which a refactoring
71  /// action might operate on.
72  const SourceManager &SM;
73  /// An optional source selection range that's commonly used to represent
74  /// a selection in an editor.
75  SourceRange SelectionRange;
76  /// An optional AST for the translation unit on which a refactoring action
77  /// might operate on.
78  ASTContext *AST = nullptr;
79  /// The allocator for diagnostics.
81 
82  // FIXME: Remove when memoized.
83  std::unique_ptr<SelectedASTNode> ASTNodeSelection;
84 };
85 
86 } // end namespace tooling
87 } // end namespace clang
88 
89 #endif // LLVM_CLANG_TOOLING_REFACTOR_REFACTORING_RULE_CONTEXT_H
Defines the SourceManager interface.
SourceRange getSelectionRange() const
Returns the current source selection range as set by the refactoring engine.
Holds long-lived AST nodes (such as types and decls) that can be referred to throughout the semantic ...
Definition: ASTContext.h:154
llvm::Error createDiagnosticError(SourceLocation Loc, unsigned DiagID)
Creates an llvm::Error value that contains a diagnostic.
void setASTSelection(std::unique_ptr< SelectedASTNode > Node)
The refactoring rule context stores all of the inputs that might be needed by a refactoring action ru...
An allocator for Storage objects, which uses a small cache to objects, used to reduce malloc()/free()...
Encodes a location in the source.
const SourceManager & getSources() const
llvm::Error createDiagnosticError(unsigned DiagID)
ast_type_traits::DynTypedNode Node
Dataflow Directional Tag Classes.
static llvm::Error create(SourceLocation Loc, PartialDiagnostic Diag)
Creates a new DiagnosticError that contains the given diagnostic at the given location.
A trivial tuple used to represent a source range.
This class handles loading and caching of source files into memory.