// 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.IO;
using System.Threading;
using System.Threading.Tasks;
using Microsoft.CodeAnalysis.CSharp.UnitTests;
using Microsoft.CodeAnalysis.MetadataAsSource;
using Microsoft.CodeAnalysis.PdbSourceDocument;
using Roslyn.Test.Utilities;
using Xunit;

namespace Microsoft.CodeAnalysis.Editor.CSharp.UnitTests.PdbSourceDocument;

public sealed class PdbFileLocatorServiceTests : AbstractPdbSourceDocumentTests
{
    [Fact]
    public Task ReturnsPdbPathFromDebugger()
        => RunTestAsync(async path =>
        {
            MarkupTestFile.GetSpan("""
            public class C
            {
                public event System.EventHandler [|E|] { add { } remove { } }
            }
            """, out var metadataSource, out var expectedSpan);

            var (project, symbol) = await CompileAndFindSymbolAsync(path, Location.OnDisk, Location.OnDisk, metadataSource, c => c.GetMember("C.E"));

            // Move the PDB to a path that only our fake debugger service knows about
            var pdbFilePath = Path.Combine(path, "SourceLink.pdb");
            File.Move(GetPdbPath(path), pdbFilePath);

            var sourceLinkService = new TestSourceLinkService(pdbFilePath: pdbFilePath);
            var service = new PdbFileLocatorService(sourceLinkService, logger: null);

            using var result = await service.GetDocumentDebugInfoReaderAsync(GetDllPath(path), useDefaultSymbolServers: false, new TelemetryMessage(CancellationToken.None), CancellationToken.None);

            Assert.NotNull(result);
        });

    [Fact]
    public Task DoesntReadNonPortablePdbs()
        => RunTestAsync(async path =>
        {
            MarkupTestFile.GetSpan("""
            public class C
            {
                public event System.EventHandler [|E|] { add { } remove { } }
            }
            """, out var metadataSource, out var expectedSpan);

            // Ideally we don't want to pass in true for windowsPdb here, and this is supposed to test that the service ignores non-portable PDBs when the debugger
            // tells us they're not portable, but the debugger has a bug at the moment.
            var (project, symbol) = await CompileAndFindSymbolAsync(path, Location.OnDisk, Location.OnDisk, metadataSource, c => c.GetMember("C.E"), windowsPdb: true);

            // Move the PDB to a path that only our fake debugger service knows about
            var pdbFilePath = Path.Combine(path, "SourceLink.pdb");
            File.Move(GetPdbPath(path), pdbFilePath);

            var sourceLinkService = new TestSourceLinkService(pdbFilePath);
            var service = new PdbFileLocatorService(sourceLinkService, logger: null);

            using var result = await service.GetDocumentDebugInfoReaderAsync(GetDllPath(path), useDefaultSymbolServers: false, new TelemetryMessage(CancellationToken.None), CancellationToken.None);

            Assert.Null(result);
        });

    [Fact]
    public Task NoPdbFoundReturnsNull()
        => RunTestAsync(async path =>
        {
            MarkupTestFile.GetSpan("""
            public class C
            {
                public event System.EventHandler [|E|] { add { } remove { } }
            }
            """, out var metadataSource, out var expectedSpan);

            var (project, symbol) = await CompileAndFindSymbolAsync(path, Location.OnDisk, Location.OnDisk, metadataSource, c => c.GetMember("C.E"));

            // Move the PDB to a path that only our fake debugger service knows about
            var pdbFilePath = Path.Combine(path, "SourceLink.pdb");
            File.Move(GetPdbPath(path), pdbFilePath);

            var sourceLinkService = new TestSourceLinkService(pdbFilePath: null);
            var service = new PdbFileLocatorService(sourceLinkService, logger: null);

            using var result = await service.GetDocumentDebugInfoReaderAsync(GetDllPath(path), useDefaultSymbolServers: false, new TelemetryMessage(CancellationToken.None), CancellationToken.None);

            Assert.Null(result);
        });
}
