// 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.CodeGeneration;
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)
            {
                var resolved = SymbolKey.ResolveString(key, compilation, cancellationToken: cancellationToken).GetAnySymbol() as IFieldSymbol;
                if (resolved == null)
                    return ImmutableArray<(DocumentId, ImmutableArray<TextChange>)>.Empty;
 
                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);
    }
}