// 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.Immutable;
using System.Composition;
using System.Threading;
using System.Threading.Tasks;
using Microsoft.CodeAnalysis.Host.Mef;
using Microsoft.CodeAnalysis.Remote;
using Microsoft.CodeAnalysis.SymbolSearch;
 
namespace Microsoft.CodeAnalysis.AddImport;
 
/// <summary>
/// Used to supply the OOP server a callback that it can use to search for ReferenceAssemblies or
/// nuget packages.  We can't necessarily do that search directly in the OOP server as our 
/// 'SymbolSearchEngine' may actually be running in a *different* process (there is no guarantee
/// that all remote work happens in the same process).  
/// 
/// This does mean, currently, that when we call over to OOP to do a search, it will bounce
/// back to VS, which will then bounce back out to OOP to perform the Nuget/ReferenceAssembly
/// portion of the search.  Ideally we could keep this all OOP.
/// </summary>
[ExportRemoteServiceCallbackDispatcher(typeof(IRemoteMissingImportDiscoveryService)), Shared]
internal sealed class RemoteMissingImportDiscoveryServiceCallbackDispatcher : RemoteServiceCallbackDispatcher, IRemoteMissingImportDiscoveryService.ICallback
{
    [ImportingConstructor]
    [Obsolete(MefConstruction.ImportingConstructorMessage, error: true)]
    public RemoteMissingImportDiscoveryServiceCallbackDispatcher()
    {
    }
 
    private ISymbolSearchService GetService(RemoteServiceCallbackId callbackId)
        => (ISymbolSearchService)GetCallback(callbackId);
 
    public ValueTask<ImmutableArray<PackageWithTypeResult>> FindPackagesWithTypeAsync(RemoteServiceCallbackId callbackId, string source, string name, int arity, CancellationToken cancellationToken)
        => GetService(callbackId).FindPackagesWithTypeAsync(source, name, arity, cancellationToken);
 
    public ValueTask<ImmutableArray<PackageWithAssemblyResult>> FindPackagesWithAssemblyAsync(RemoteServiceCallbackId callbackId, string source, string name, CancellationToken cancellationToken)
        => GetService(callbackId).FindPackagesWithAssemblyAsync(source, name, cancellationToken);
 
    public ValueTask<ImmutableArray<ReferenceAssemblyWithTypeResult>> FindReferenceAssembliesWithTypeAsync(RemoteServiceCallbackId callbackId, string name, int arity, CancellationToken cancellationToken)
        => GetService(callbackId).FindReferenceAssembliesWithTypeAsync(name, arity, cancellationToken);
}