如何使用 Roslyn 将 SyntaxTree 或 UnitCompilationRoot 包装在区域或注释中？

Question

我正在使用CSharpSyntaxTree.ParseText从任意代码块创建语法树； 我希望能够将该SyntaxTree或其UnitCompilationRoot包装或封装在一个区域中。 我知道 Quoter 网站允许您查看如何使用 Roslyn API 编写任意代码，但是它将代码分解为其原始组件的方式无助于添加区域的用例到SyntaxTree或UnitCompilationRoot 。

使用任意代码的示例

// Create syntax tree from template
var str = @"public class Foo { }"
var syntaxTree = CSharpSyntaxTree.ParseText(str);

// Add Region
// ????

Answer 1

区域由 Roslyn 表示为特殊类型的SyntaxTrivia 。 因此，为了将代码块包装到区域中，您可以获取目标SyntaxNode并将其前导和尾随琐事替换为区域琐事。 以下代码演示了如何做到这一点：

public static T AddRegion<T>(this T node, string regionName) where T : SyntaxNode
{
    return node
        // inserts #region RegionName before node
        .WithLeadingTrivia(node.GetLeadingTrivia().Insert(0, GetRegionLeadingTrivia(regionName)))
        // inserts #endregion after node
        .WithTrailingTrivia(node.GetTrailingTrivia().Add(GetRegionTrailingTrivia()));
}

public static SyntaxTrivia GetRegionLeadingTrivia(string regionName)
{
    return SyntaxFactory.Trivia(
        SyntaxFactory
            .RegionDirectiveTrivia(true)
            .WithEndOfDirectiveToken(
                SyntaxFactory.Token(
                    SyntaxFactory.TriviaList(SyntaxFactory.PreprocessingMessage(regionName)),
                    SyntaxKind.EndOfDirectiveToken,
                    SyntaxFactory.TriviaList())));
}

public static SyntaxTrivia GetRegionTrailingTrivia()
{
    return SyntaxFactory.Trivia(SyntaxFactory.EndRegionDirectiveTrivia(true));
}

要执行更复杂的操作，您可以使用CSharpSyntaxRewriter class 。 可以在此处找到使用 class 的一些示例。