// 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;
using System.Collections.Generic;
using System.Collections.Immutable;
using System.Threading;
using System.Threading.Tasks;
using Microsoft.CodeAnalysis.CodeActions;
using Microsoft.CodeAnalysis.CodeRefactorings.PullMemberUp.Dialog;
using Microsoft.CodeAnalysis.PullMemberUp;

namespace Microsoft.CodeAnalysis.CodeRefactorings.PullMemberUp;

internal abstract partial class AbstractPullMemberUpRefactoringProvider
{
    private sealed class PullMemberUpWithDialogCodeAction(
        Document document,
        ImmutableArray<ISymbol> selectedMembers,
        IPullMemberUpOptionsService service) : CodeActionWithOptions
    {
        /// <summary>
        /// Member which user initially selects. It will be selected initially when the dialog pops up.
        /// </summary>
        private readonly ImmutableArray<ISymbol> _selectedMembers = selectedMembers;
        private readonly Document _document = document;
        private readonly IPullMemberUpOptionsService _service = service;

        public override string Title => FeaturesResources.Pull_members_up_to_base_type;

        internal override bool IsOptionServiceAvailable()
            => _service is not null;

        public override object GetOptions(CancellationToken cancellationToken)
        {
            return _service.GetPullMemberUpOptions(_document, _selectedMembers);
        }

        protected override async Task<IEnumerable<CodeActionOperation>> ComputeOperationsAsync(
            object options, IProgress<CodeAnalysisProgress> progressTracker, CancellationToken cancellationToken)
        {
            if (options is PullMembersUpOptions pullMemberUpOptions)
            {
                var changedSolution = await MembersPuller.PullMembersUpAsync(_document, pullMemberUpOptions, cancellationToken).ConfigureAwait(false);
                return [new ApplyChangesOperation(changedSolution)];
            }
            else
            {
                // If user click cancel button, options will be null and hit this branch
                return [];
            }
        }
    }
}