Index: include/clang/Tooling/Refactoring/RefactoringActionRule.h =================================================================== --- include/clang/Tooling/Refactoring/RefactoringActionRule.h +++ include/clang/Tooling/Refactoring/RefactoringActionRule.h @@ -16,6 +16,7 @@ namespace clang { namespace tooling { +class RefactoringOptionConsumer; class RefactoringResultConsumer; class RefactoringRuleContext; @@ -43,6 +44,8 @@ /// Returns true when the rule has a source selection requirement that has /// to be fullfilled before refactoring can be performed. virtual bool hasSelectionRequirement() = 0; + + virtual void visitRefactoringOptions(RefactoringOptionConsumer &Consumer) = 0; }; } // end namespace tooling Index: include/clang/Tooling/Refactoring/RefactoringActionRuleRequirements.h =================================================================== --- include/clang/Tooling/Refactoring/RefactoringActionRuleRequirements.h +++ include/clang/Tooling/Refactoring/RefactoringActionRuleRequirements.h @@ -11,6 +11,7 @@ #define LLVM_CLANG_TOOLING_REFACTOR_REFACTORING_ACTION_RULE_REQUIREMENTS_H #include "clang/Basic/LLVM.h" +#include "clang/Tooling/Refactoring/RefactoringOption.h" #include "clang/Tooling/Refactoring/RefactoringRuleContext.h" #include "llvm/Support/Error.h" #include @@ -53,6 +54,40 @@ } }; +/// A base class for any requirement that requires some refactoring options. +class RefactoringOptionsRequirement : public RefactoringActionRuleRequirement { +public: + virtual ~RefactoringOptionsRequirement() {} + + /// Returns the set of refactoring options that a used when evaluating this + /// requirement. + virtual std::vector> + getRefactoringOptions() const = 0; +}; + +/// A requirement that evaluates to the value of the given \c OptionType when +/// the \c OptionType is a required option. When the \c OptionType is an +/// optional option, the requirement will evaluate to \c None if the option is +/// not specified or to an appropriate value otherwise. +template +class OptionRequirement : public RefactoringOptionsRequirement { +public: + OptionRequirement() : Opt(createRefactoringOption()) {} + + std::vector> + getRefactoringOptions() const final override { + return {Opt}; + } + + Expected().getValue())> + evaluate(RefactoringRuleContext &) const { + return Opt->getValue(); + } + +private: + std::shared_ptr Opt; +}; + } // end namespace tooling } // end namespace clang Index: include/clang/Tooling/Refactoring/RefactoringActionRulesInternal.h =================================================================== --- include/clang/Tooling/Refactoring/RefactoringActionRulesInternal.h +++ include/clang/Tooling/Refactoring/RefactoringActionRulesInternal.h @@ -24,12 +24,23 @@ inline llvm::Error findError() { return llvm::Error::success(); } +inline void ignoreError() {} + +template +void ignoreError(Expected &First, Expected &... Rest) { + if (!First) + llvm::consumeError(First.takeError()); + ignoreError(Rest...); +} + /// Scans the tuple and returns a valid \c Error if any of the values are /// invalid. template llvm::Error findError(Expected &First, Expected &... Rest) { - if (!First) + if (!First) { + ignoreError(Rest...); return First.takeError(); + } return findError(Rest...); } @@ -42,13 +53,45 @@ auto Values = std::make_tuple(std::get(Requirements).evaluate(Context)...); auto Err = findError(std::get(Values)...); - if (Err) + if (Err) { return Consumer.handleError(std::move(Err)); + } // Construct the target action rule by extracting the evaluated // requirements from Expected<> wrappers and then run it. RuleType((*std::get(Values))...).invoke(Consumer, Context); } +inline void gatherRefactoringOptionsImpl( + std::vector> &) {} + +/// Scans the tuple and returns a valid \c Error if any of the values are +/// invalid. +template +void gatherRefactoringOptionsImpl( + std::vector> &Options, + const FirstT &First, const RestT &... Rest) { + struct OptionGatherer { + std::vector> &Options; + + void operator()(const RefactoringOptionsRequirement &Requirement) { + std::vector> Opts = + Requirement.getRefactoringOptions(); + Options.insert(Options.end(), Opts.begin(), Opts.end()); + } + void operator()(const RefactoringActionRuleRequirement &) {} + }; + (OptionGatherer{Options})(First); + return gatherRefactoringOptionsImpl(Options, Rest...); +} + +template +void gatherRefactoringOptions( + std::vector> &Options, + const std::tuple &Requirements, + llvm::index_sequence) { + gatherRefactoringOptionsImpl(Options, std::get(Requirements)...); +} + /// A type trait that returns true when the given type list has at least one /// type whose base is the given base type. template @@ -83,7 +126,11 @@ class Rule final : public RefactoringActionRule { public: Rule(std::tuple Requirements) - : Requirements(Requirements) {} + : Requirements(Requirements) { + internal::gatherRefactoringOptions( + AllRefactoringOptions, this->Requirements, + llvm::index_sequence_for()); + } void invoke(RefactoringResultConsumer &Consumer, RefactoringRuleContext &Context) override { @@ -97,8 +144,14 @@ RequirementTypes...>::value; } + void visitRefactoringOptions(RefactoringOptionConsumer &Consumer) override { + for (const auto &Option : AllRefactoringOptions) + Option->passToConsumer(Consumer); + } + private: std::tuple Requirements; + std::vector> AllRefactoringOptions; }; return llvm::make_unique(std::make_tuple(Requirements...)); Index: include/clang/Tooling/Refactoring/RefactoringOption.h =================================================================== --- /dev/null +++ include/clang/Tooling/Refactoring/RefactoringOption.h @@ -0,0 +1,56 @@ +//===--- RefactoringOption.h - Clang refactoring library ------------------===// +// +// The LLVM Compiler Infrastructure +// +// This file is distributed under the University of Illinois Open Source +// License. See LICENSE.TXT for details. +// +//===----------------------------------------------------------------------===// + +#ifndef LLVM_CLANG_TOOLING_REFACTOR_REFACTORING_OPTION_H +#define LLVM_CLANG_TOOLING_REFACTOR_REFACTORING_OPTION_H + +#include "clang/Basic/LLVM.h" +#include +#include + +namespace clang { +namespace tooling { + +class RefactoringOptionConsumer; + +/// A refactoring option is an interface that describes a value that +/// has an impact on the outcome of a refactoring. +/// +/// Refactoring options can be specified using command-line arguments when +/// the clang-refactor tool is used. +class RefactoringOption { +public: + virtual ~RefactoringOption() {} + + /// Returns the name of the refactoring option. + /// + /// Each refactoring option must have a unique name. + virtual StringRef getName() const = 0; + + virtual StringRef getDescription() const = 0; + + /// True when this option must be specified before invoking the refactoring + /// action. + virtual bool isRequired() const = 0; + + virtual void passToConsumer(RefactoringOptionConsumer &Consumer) = 0; +}; + +/// Constructs a refactoring option of the given type. +template +std::shared_ptr createRefactoringOption() { + static_assert(std::is_base_of::value, + "invalid option type"); + return std::make_shared(); +} + +} // end namespace tooling +} // end namespace clang + +#endif // LLVM_CLANG_TOOLING_REFACTOR_REFACTORING_OPTION_H Index: include/clang/Tooling/Refactoring/RefactoringOptionConsumer.h =================================================================== --- /dev/null +++ include/clang/Tooling/Refactoring/RefactoringOptionConsumer.h @@ -0,0 +1,62 @@ +//===--- RefactoringOptionConsumer.h - Clang refactoring library ----------===// +// +// The LLVM Compiler Infrastructure +// +// This file is distributed under the University of Illinois Open Source +// License. See LICENSE.TXT for details. +// +//===----------------------------------------------------------------------===// + +#ifndef LLVM_CLANG_TOOLING_REFACTOR_REFACTORING_OPTION_CONSUMER_H +#define LLVM_CLANG_TOOLING_REFACTOR_REFACTORING_OPTION_CONSUMER_H + +#include "clang/Basic/LLVM.h" +#include + +namespace clang { +namespace tooling { + +class RefactoringOption; + +/// An interface that declares functions that handle different refactoring +/// option types. +/// +/// A valid refactoring option type must have a corresponding \c handle +/// declaration in this interface. +class RefactoringOptionConsumer { +public: + virtual ~RefactoringOptionConsumer() {} + + virtual void handle(const RefactoringOption &Opt, + Optional &Value) = 0; +}; + +namespace traits { +namespace internal { + +template struct HasHandle { +private: + template + static auto check(ClassT *) -> typename std::is_same< + decltype(std::declval().handle( + std::declval(), *std::declval *>())), + void>::type; + + template static std::false_type check(...); + +public: + using Type = decltype(check(nullptr)); +}; + +} // end namespace internal + +/// A type trait that returns true iff the given type is a type that can be +/// stored in a refactoring option. +template +struct IsValidOptionType : internal::HasHandle::Type {}; + +} // end namespace traits +} // end namespace tooling +} // end namespace clang + +#endif // LLVM_CLANG_TOOLING_REFACTOR_REFACTORING_OPTION_CONSUMER_H Index: include/clang/Tooling/Refactoring/RefactoringOptions.h =================================================================== --- /dev/null +++ include/clang/Tooling/Refactoring/RefactoringOptions.h @@ -0,0 +1,52 @@ +//===--- RefactoringOptions.h - Clang refactoring library -----------------===// +// +// The LLVM Compiler Infrastructure +// +// This file is distributed under the University of Illinois Open Source +// License. See LICENSE.TXT for details. +// +//===----------------------------------------------------------------------===// + +#ifndef LLVM_CLANG_TOOLING_REFACTOR_REFACTORING_OPTIONS_H +#define LLVM_CLANG_TOOLING_REFACTOR_REFACTORING_OPTIONS_H + +#include "clang/Basic/LLVM.h" +#include "clang/Tooling/Refactoring/RefactoringActionRuleRequirements.h" +#include "clang/Tooling/Refactoring/RefactoringOption.h" +#include "clang/Tooling/Refactoring/RefactoringOptionConsumer.h" +#include "llvm/Support/Error.h" +#include + +namespace clang { +namespace tooling { + +/// A refactoring option that stores a value of type \c T. +template ::value>::type> +class OptionalRefactoringOption : public RefactoringOption { +public: + void passToConsumer(RefactoringOptionConsumer &Consumer) final override { + Consumer.handle(*this, Value); + } + + bool isRequired() const override { return false; } + + const Optional &getValue() const { return Value; } + +protected: + Optional Value; +}; + +/// A required refactoring option that stores a value of type \c T. +template ::value>::type> +class RequiredRefactoringOption : public OptionalRefactoringOption { +public: + const T &getValue() const { return *OptionalRefactoringOption::Value; } + bool isRequired() const final override { return true; } +}; + +} // end namespace tooling +} // end namespace clang + +#endif // LLVM_CLANG_TOOLING_REFACTOR_REFACTORING_OPTIONS_H Index: include/clang/Tooling/Refactoring/Rename/RenamingAction.h =================================================================== --- include/clang/Tooling/Refactoring/Rename/RenamingAction.h +++ include/clang/Tooling/Refactoring/Rename/RenamingAction.h @@ -17,6 +17,7 @@ #include "clang/Tooling/Refactoring.h" #include "clang/Tooling/Refactoring/AtomicChange.h" +#include "clang/Tooling/Refactoring/RefactoringOptions.h" #include "clang/Tooling/Refactoring/Rename/SymbolOccurrences.h" #include "llvm/Support/Error.h" @@ -45,12 +46,19 @@ bool PrintLocations; }; +class NewNameOption : public RequiredRefactoringOption { +public: + StringRef getName() const override { return "new-name"; } + StringRef getDescription() const override { + return "The new name to change the symbol to"; + } +}; + /// Returns source replacements that correspond to the rename of the given /// symbol occurrences. llvm::Expected> createRenameReplacements(const SymbolOccurrences &Occurrences, - const SourceManager &SM, - ArrayRef NewNameStrings); + const SourceManager &SM, const SymbolName &NewName); /// Rename all symbols identified by the given USRs. class QualifiedRenamingAction { Index: lib/Tooling/Refactoring/Rename/RenamingAction.cpp =================================================================== --- lib/Tooling/Refactoring/Rename/RenamingAction.cpp +++ lib/Tooling/Refactoring/Rename/RenamingAction.cpp @@ -23,6 +23,8 @@ #include "clang/Tooling/CommonOptionsParser.h" #include "clang/Tooling/Refactoring.h" #include "clang/Tooling/Refactoring/RefactoringAction.h" +#include "clang/Tooling/Refactoring/RefactoringOptions.h" +#include "clang/Tooling/Refactoring/Rename/SymbolName.h" #include "clang/Tooling/Refactoring/Rename/USRFinder.h" #include "clang/Tooling/Refactoring/Rename/USRFindingAction.h" #include "clang/Tooling/Refactoring/Rename/USRLocFinder.h" @@ -75,7 +77,8 @@ class RenameOccurrences final : public SourceChangeRefactoringRule { public: - RenameOccurrences(const NamedDecl *ND) : Finder(ND) {} + RenameOccurrences(const NamedDecl *ND, std::string NewName) + : Finder(ND), NewName(NewName) {} Expected createSourceReplacements(RefactoringRuleContext &Context) { @@ -83,15 +86,15 @@ Finder.findSymbolOccurrences(Context); if (!Occurrences) return Occurrences.takeError(); - // FIXME: This is a temporary workaround that's needed until the refactoring - // options are implemented. - StringRef NewName = "Bar"; + // FIXME: Verify that the new name is valid. + SymbolName Name(NewName); return createRenameReplacements( - *Occurrences, Context.getASTContext().getSourceManager(), NewName); + *Occurrences, Context.getASTContext().getSourceManager(), Name); } private: OccurrenceFinder Finder; + std::string NewName; }; class LocalRename final : public RefactoringAction { @@ -107,7 +110,7 @@ RefactoringActionRules createActionRules() const override { RefactoringActionRules Rules; Rules.push_back(createRefactoringActionRule( - SymbolSelectionRequirement())); + SymbolSelectionRequirement(), OptionRequirement())); return Rules; } }; @@ -120,19 +123,18 @@ Expected> createRenameReplacements(const SymbolOccurrences &Occurrences, - const SourceManager &SM, - ArrayRef NewNameStrings) { + const SourceManager &SM, const SymbolName &NewName) { // FIXME: A true local rename can use just one AtomicChange. std::vector Changes; for (const auto &Occurrence : Occurrences) { ArrayRef Ranges = Occurrence.getNameRanges(); - assert(NewNameStrings.size() == Ranges.size() && + assert(NewName.getNamePieces().size() == Ranges.size() && "Mismatching number of ranges and name pieces"); AtomicChange Change(SM, Ranges[0].getBegin()); for (const auto &Range : llvm::enumerate(Ranges)) { auto Error = Change.replace(SM, CharSourceRange::getCharRange(Range.value()), - NewNameStrings[Range.index()]); + NewName.getNamePieces()[Range.index()]); if (Error) return std::move(Error); } @@ -196,7 +198,7 @@ } // FIXME: Support multi-piece names. // FIXME: better error handling (propagate error out). - StringRef NewNameRef = NewName; + SymbolName NewNameRef(NewName); Expected> Change = createRenameReplacements(Occurrences, SourceMgr, NewNameRef); if (!Change) { Index: test/Refactor/LocalRename/Field.cpp =================================================================== --- test/Refactor/LocalRename/Field.cpp +++ test/Refactor/LocalRename/Field.cpp @@ -1,4 +1,4 @@ -// RUN: clang-refactor local-rename -selection=test:%s %s -- | FileCheck %s +// RUN: clang-refactor local-rename -selection=test:%s -new-name=Bar %s -- | FileCheck %s class Baz { int /*range=*/Foo; // CHECK: int /*range=*/Bar; Index: test/Refactor/tool-test-support.c =================================================================== --- test/Refactor/tool-test-support.c +++ test/Refactor/tool-test-support.c @@ -1,4 +1,4 @@ -// RUN: clang-refactor local-rename -selection=test:%s -v %s -- | FileCheck %s +// RUN: clang-refactor local-rename -selection=test:%s -new-name=test -v %s -- | FileCheck %s /*range=*/int test; @@ -11,12 +11,12 @@ /*range named =+0*/int test5; // CHECK: Test selection group '': -// CHECK-NEXT: 90-90 -// CHECK-NEXT: 143-143 -// CHECK-NEXT: 182-182 +// CHECK-NEXT: 105-105 +// CHECK-NEXT: 158-158 +// CHECK-NEXT: 197-197 // CHECK-NEXT: Test selection group 'named': -// CHECK-NEXT: 117-117 -// CHECK-NEXT: 203-203 +// CHECK-NEXT: 132-132 +// CHECK-NEXT: 218-218 // The following invocations are in the default group: Index: tools/clang-refactor/ClangRefactor.cpp =================================================================== --- tools/clang-refactor/ClangRefactor.cpp +++ tools/clang-refactor/ClangRefactor.cpp @@ -18,6 +18,7 @@ #include "clang/Tooling/CommonOptionsParser.h" #include "clang/Tooling/Refactoring.h" #include "clang/Tooling/Refactoring/RefactoringAction.h" +#include "clang/Tooling/Refactoring/RefactoringOptions.h" #include "clang/Tooling/Refactoring/Rename/RenamingAction.h" #include "clang/Tooling/Tooling.h" #include "llvm/Support/CommandLine.h" @@ -32,10 +33,10 @@ namespace opts { -static cl::OptionCategory CommonRefactorOptions("Common refactoring options"); +static cl::OptionCategory CommonRefactorOptions("Refactoring options"); static cl::opt Verbose("v", cl::desc("Use verbose output"), - cl::cat(CommonRefactorOptions), + cl::cat(cl::GeneralCategory), cl::sub(*cl::AllSubCommands)); } // end namespace opts @@ -116,6 +117,98 @@ return nullptr; } +/// A container that stores the command-line options used by a single +/// refactoring option. +class RefactoringActionCommandLineOptions { +public: + template + void addOption(const RefactoringOption &Option, + std::unique_ptr> CLOption); + + const cl::opt & + getStringOption(const RefactoringOption &Opt) const { + auto It = StringOptions.find(&Opt); + return *It->second; + } + +private: + llvm::DenseMap>> + StringOptions; +}; + +template <> +void RefactoringActionCommandLineOptions::addOption( + const RefactoringOption &Option, + std::unique_ptr> CLOption) { + StringOptions[&Option] = std::move(CLOption); +} + +/// Passes the command-line option values to the options used by a single +/// refactoring action rule. +class CommandLineRefactoringOptionConsumer final + : public RefactoringOptionConsumer { +public: + CommandLineRefactoringOptionConsumer( + const RefactoringActionCommandLineOptions &Options) + : Options(Options) {} + + void handle(const RefactoringOption &Opt, + Optional &Value) override { + const cl::opt &CLOpt = Options.getStringOption(Opt); + if (!CLOpt.getValue().empty()) { + Value = CLOpt.getValue(); + return; + } + Value = None; + if (Opt.isRequired()) + MissingRequiredOptions.push_back(&Opt); + } + + ArrayRef getMissingRequiredOptions() const { + return MissingRequiredOptions; + } + +private: + llvm::SmallVector MissingRequiredOptions; + const RefactoringActionCommandLineOptions &Options; +}; + +/// Creates the refactoring options used by all the rules in a single +/// refactoring action. +class CommandLineRefactoringOptionCreator final + : public RefactoringOptionConsumer { +public: + CommandLineRefactoringOptionCreator( + cl::OptionCategory &Category, cl::SubCommand &Subcommand, + RefactoringActionCommandLineOptions &Options) + : Category(Category), Subcommand(Subcommand), Options(Options) {} + + void handle(const RefactoringOption &Opt, Optional &) override { + if (Visited.insert(&Opt).second) + Options.addOption(Opt, create(Opt)); + } + +private: + template + std::unique_ptr> create(const RefactoringOption &Opt) { + if (!OptionNames.insert(Opt.getName()).second) + llvm::report_fatal_error("Multiple identical refactoring options " + "specified for one refactoring action"); + // FIXME: cl::Required can be specified when this option is present + // in all rules in an action. + return llvm::make_unique>( + Opt.getName(), cl::desc(Opt.getDescription()), cl::Optional, + cl::cat(Category), cl::sub(Subcommand)); + } + + llvm::SmallPtrSet Visited; + llvm::StringSet<> OptionNames; + cl::OptionCategory &Category; + cl::SubCommand &Subcommand; + RefactoringActionCommandLineOptions &Options; +}; + /// A subcommand that corresponds to individual refactoring action. class RefactoringActionSubcommand : public cl::SubCommand { public: @@ -138,6 +231,12 @@ "::)"), cl::cat(Category), cl::sub(*this)); } + // Create the refactoring options. + for (const auto &Rule : this->ActionRules) { + CommandLineRefactoringOptionCreator OptionCreator(Category, *this, + Options); + Rule->visitRefactoringOptions(OptionCreator); + } } ~RefactoringActionSubcommand() { unregisterSubCommand(); } @@ -160,11 +259,17 @@ assert(Selection && "selection not supported!"); return ParsedSelection.get(); } + + const RefactoringActionCommandLineOptions &getOptions() const { + return Options; + } + private: std::unique_ptr Action; RefactoringActionRules ActionRules; std::unique_ptr> Selection; std::unique_ptr ParsedSelection; + RefactoringActionCommandLineOptions Options; }; class ClangRefactorConsumer : public RefactoringResultConsumer { @@ -262,14 +367,22 @@ bool HasSelection = false; for (const auto &Rule : Subcommand.getActionRules()) { + bool SelectionMatches = true; if (Rule->hasSelectionRequirement()) { HasSelection = true; - if (Subcommand.getSelection()) - MatchingRules.push_back(Rule.get()); - else + if (!Subcommand.getSelection()) { MissingOptions.insert("selection"); + SelectionMatches = false; + } + } + CommandLineRefactoringOptionConsumer Consumer(Subcommand.getOptions()); + Rule->visitRefactoringOptions(Consumer); + if (SelectionMatches && Consumer.getMissingRequiredOptions().empty()) { + MatchingRules.push_back(Rule.get()); + continue; } - // FIXME (Alex L): Support custom options. + for (const RefactoringOption *Opt : Consumer.getMissingRequiredOptions()) + MissingOptions.insert(Opt->getName()); } if (MatchingRules.empty()) { llvm::errs() << "error: '" << Subcommand.getName() @@ -326,7 +439,7 @@ ClangRefactorTool Tool; CommonOptionsParser Options( - argc, argv, opts::CommonRefactorOptions, cl::ZeroOrMore, + argc, argv, cl::GeneralCategory, cl::ZeroOrMore, "Clang-based refactoring tool for C, C++ and Objective-C"); // Figure out which action is specified by the user. The user must specify