src/Workspaces/Remote/ServiceHub/Services/EncapsulateField/RemoteEncapsulateFieldService.cs - main - githubmirror/roslyn

src/Workspaces/Remote/ServiceHub/Services/EncapsulateField/RemoteEncapsulateFieldService.cs

56 строк2 KB

Cyrus Najmabadi

Use pattern matching

10 май 2025, 05:44

10 май 2025, 05:44d0e9b44

// 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.

using System.Collections.Immutable;
using System.Threading;
using System.Threading.Tasks;
using Microsoft.CodeAnalysis.EncapsulateField;
using Microsoft.CodeAnalysis.Host;
using Microsoft.CodeAnalysis.PooledObjects;
using Microsoft.CodeAnalysis.Shared.Extensions;
using Microsoft.CodeAnalysis.Text;

namespace Microsoft.CodeAnalysis.Remote;

internal sealed class RemoteEncapsulateFieldService(in BrokeredServiceBase.ServiceConstructionArguments arguments)
    : BrokeredServiceBase(arguments), IRemoteEncapsulateFieldService
{
    internal sealed class Factory : FactoryBase<IRemoteEncapsulateFieldService>
    {
        protected override IRemoteEncapsulateFieldService CreateService(in ServiceConstructionArguments arguments)
            => new RemoteEncapsulateFieldService(arguments);
    }

    public ValueTask<ImmutableArray<(DocumentId, ImmutableArray<TextChange>)>> EncapsulateFieldsAsync(
        Checksum solutionChecksum,
        DocumentId documentId,
        ImmutableArray<string> fieldSymbolKeys,
        bool updateReferences,
        CancellationToken cancellationToken)
    {
        return RunServiceAsync(solutionChecksum, async solution =>
        {
            var document = solution.GetRequiredDocument(documentId);

            using var _ = ArrayBuilder<IFieldSymbol>.GetInstance(out var fields);
            var compilation = await document.Project.GetRequiredCompilationAsync(cancellationToken).ConfigureAwait(false);

            foreach (var key in fieldSymbolKeys)
            {
                if (SymbolKey.ResolveString(key, compilation, cancellationToken: cancellationToken).GetAnySymbol() is not IFieldSymbol resolved)
                    return [];

                fields.Add(resolved);
            }

            var service = document.GetRequiredLanguageService<IEncapsulateFieldService>();

            var newSolution = await service.EncapsulateFieldsAsync(
                document, fields.ToImmutable(), updateReferences, cancellationToken).ConfigureAwait(false);

            return await RemoteUtilities.GetDocumentTextChangesAsync(
                solution, newSolution, cancellationToken).ConfigureAwait(false);
        }, cancellationToken);
    }
}