// 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.Threading;
using System.Threading.Tasks;
using Microsoft.CodeAnalysis.ExternalAccess.FSharp.Editor.FindUsages;
using Microsoft.CodeAnalysis.FindUsages;
using Microsoft.CodeAnalysis.Shared.Extensions;
 
namespace Microsoft.CodeAnalysis.ExternalAccess.FSharp.Internal.Editor.FindUsages;
 
internal class FSharpFindUsagesContext : IFSharpFindUsagesContext
{
    private readonly IFindUsagesContext _context;
 
    public FSharpFindUsagesContext(IFindUsagesContext context, CancellationToken cancellationToken)
    {
        _context = context;
        CancellationToken = cancellationToken;
    }
 
    public CancellationToken CancellationToken { get; }
 
    public Task OnDefinitionFoundAsync(FSharp.FindUsages.FSharpDefinitionItem definition)
    {
        return _context.OnDefinitionFoundAsync(definition.RoslynDefinitionItem, CancellationToken).AsTask();
    }
 
    public Task OnReferenceFoundAsync(FSharp.FindUsages.FSharpSourceReferenceItem reference)
    {
        return _context.OnReferencesFoundAsync(IAsyncEnumerableExtensions.SingletonAsync(reference.RoslynSourceReferenceItem), CancellationToken).AsTask();
    }
 
    public Task ReportMessageAsync(string message)
    {
        return _context.ReportNoResultsAsync(message, CancellationToken).AsTask();
    }
 
    public Task ReportProgressAsync(int current, int maximum)
    {
        return Task.CompletedTask;
    }
 
    public Task SetSearchTitleAsync(string title)
    {
        return _context.SetSearchTitleAsync(title, CancellationToken).AsTask();
    }
}