Index: cfe/trunk/include/clang/Tooling/Refactoring/Stencil.h =================================================================== --- cfe/trunk/include/clang/Tooling/Refactoring/Stencil.h +++ cfe/trunk/include/clang/Tooling/Refactoring/Stencil.h @@ -23,6 +23,7 @@ #include "clang/AST/ASTContext.h" #include "clang/AST/ASTTypeTraits.h" #include "clang/ASTMatchers/ASTMatchFinder.h" +#include "clang/Tooling/Refactoring/MatchConsumer.h" #include "clang/Tooling/Refactoring/RangeSelector.h" #include "llvm/ADT/StringRef.h" #include "llvm/Support/Error.h" @@ -175,6 +176,10 @@ return ifBound(Id, text(TrueText), text(FalseText)); } +/// Wraps a MatchConsumer in a StencilPart, so that it can be used in a Stencil. +/// This supports user-defined extensions to the Stencil language. +StencilPart run(MatchConsumer C); + /// For debug use only; semantics are not guaranteed. /// /// \returns the string resulting from calling the node's print() method. Index: cfe/trunk/lib/Tooling/Refactoring/Stencil.cpp =================================================================== --- cfe/trunk/lib/Tooling/Refactoring/Stencil.cpp +++ cfe/trunk/lib/Tooling/Refactoring/Stencil.cpp @@ -26,6 +26,7 @@ using ast_matchers::MatchFinder; using llvm::errc; using llvm::Error; +using llvm::Expected; using llvm::StringError; // A down_cast function to safely down cast a StencilPartInterface to a subclass @@ -102,6 +103,12 @@ return A.Id == B.Id && A.TruePart == B.TruePart && A.FalsePart == B.FalsePart; } +// Equality is not defined over MatchConsumers, which are opaque. +bool isEqualData(const MatchConsumer &A, + const MatchConsumer &B) { + return false; +} + // The `evalData()` overloads evaluate the given stencil data to a string, given // the match result, and append it to `Result`. We define an overload for each // type of stencil data. @@ -159,6 +166,15 @@ .eval(Match, Result); } +Error evalData(const MatchConsumer &Fn, + const MatchFinder::MatchResult &Match, std::string *Result) { + Expected Value = Fn(Match); + if (!Value) + return Value.takeError(); + *Result += *Value; + return Error::success(); +} + template class StencilPartImpl : public StencilPartInterface { T Data; @@ -233,3 +249,9 @@ return StencilPart(std::make_shared>( Id, std::move(TruePart), std::move(FalsePart))); } + +StencilPart stencil::run(MatchConsumer Fn) { + return StencilPart( + std::make_shared>>( + std::move(Fn))); +} Index: cfe/trunk/unittests/Tooling/StencilTest.cpp =================================================================== --- cfe/trunk/unittests/Tooling/StencilTest.cpp +++ cfe/trunk/unittests/Tooling/StencilTest.cpp @@ -29,6 +29,7 @@ using stencil::cat; using stencil::dPrint; using stencil::ifBound; +using stencil::run; using stencil::text; // Create a valid translation-unit from a statement. @@ -283,6 +284,15 @@ EXPECT_THAT_EXPECTED(Stencil.eval(StmtMatch->Result), HasValue("field")); } +TEST_F(StencilTest, RunOp) { + StringRef Id = "id"; + auto SimpleFn = [Id](const MatchResult &R) { + return std::string(R.Nodes.getNodeAs(Id) != nullptr ? "Bound" + : "Unbound"); + }; + testExpr(Id, "3;", cat(run(SimpleFn)), "Bound"); +} + TEST(StencilEqualityTest, Equality) { auto Lhs = cat("foo", dPrint("dprint_id")); auto Rhs = cat("foo", dPrint("dprint_id")); @@ -307,4 +317,12 @@ auto S2 = cat(node("node")); EXPECT_NE(S1, S2); } + +// `run` is opaque. +TEST(StencilEqualityTest, InEqualityRun) { + auto F = [](const MatchResult &R) { return "foo"; }; + auto S1 = cat(run(F)); + auto S2 = cat(run(F)); + EXPECT_NE(S1, S2); +} } // namespace