src/Workspaces/CoreTest/CodeCleanup/Extensions.cs - main - githubmirror/roslyn

src/Workspaces/CoreTest/CodeCleanup/Extensions.cs
71 строка3 KB
Cyrus Najmabadi
Use file scoped namespaces. (#77854)
27 мар 2025, 09:55
27 мар 2025, 09:5598d41b8
// Licensed to the .NET Foundation under one or more agreements.
// The .NET Foundation licenses this file to you under the MIT license.
// See the LICENSE file in the project root for more information.

#nullable disable

using System.Collections.Generic;
using System.Linq;
using Microsoft.CodeAnalysis.Text;

namespace Microsoft.CodeAnalysis.UnitTests.CodeCleanup;

using Microsoft.CodeAnalysis;
using CSharp = Microsoft.CodeAnalysis.CSharp;

public static class Extensions
{
    public static TextSpan GetCodeCleanupSpan(this SyntaxNode node)
    {
        var previousToken = node.GetFirstToken(includeZeroWidth: true, includeSkipped: true, includeDirectives: true, includeDocumentationComments: true).GetPreviousToken(includeZeroWidth: true, includeSkipped: true, includeDirectives: true, includeDocumentationComments: true);
        var endToken = node.GetLastToken(includeZeroWidth: true, includeSkipped: true, includeDirectives: true, includeDocumentationComments: true).GetNextToken(includeZeroWidth: true, includeSkipped: true, includeDirectives: true, includeDocumentationComments: true);

        return TextSpan.FromBounds(previousToken.SpanStart, endToken.Span.End);
    }

    public static T GetMember<T>(this Document document, int index) where T : SyntaxNode
        => (T)document.GetSyntaxRootAsync().Result.GetMember(index);

    public static T GetMember<T>(this T node, int index) where T : SyntaxNode
    {
        dynamic d = node;
        return (T)d.Members[index];
    }

    public static T RemoveCSharpMember<T>(this T node, int index) where T : SyntaxNode
    {
        var newMembers = CSharp.SyntaxFactory.List(node.RemoveMember<CSharp.Syntax.MemberDeclarationSyntax>(index));

        dynamic d = node;
        return (T)d.WithMembers(newMembers);
    }

    public static T AddCSharpMember<T>(this T node, CSharp.Syntax.MemberDeclarationSyntax member, int index) where T : SyntaxNode
    {
        var newMembers = CSharp.SyntaxFactory.List(node.AddMember<CSharp.Syntax.MemberDeclarationSyntax>(member, index));

        dynamic d = node;
        return (T)d.WithMembers(newMembers);
    }

    public static IEnumerable<M> RemoveMember<M>(this SyntaxNode node, int index) where M : SyntaxNode
    {
        dynamic d = node;

        var members = ((IEnumerable<M>)d.Members).ToList();
        members.RemoveAt(index);

        return members;
    }

    public static IEnumerable<M> AddMember<M>(this SyntaxNode node, M member, int index)
        where M : SyntaxNode
    {
        dynamic d = node;

        var members = ((IEnumerable<M>)d.Members).ToList();
        members.Insert(index, member);

        return members;
    }
}