// 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.Linq;
using System.Threading;
using System.Threading.Tasks;
using Roslyn.LanguageServer.Protocol;
using Xunit;
 
namespace Microsoft.CodeAnalysis.LanguageServer.UnitTests.DocumentChanges;
 
public sealed partial class DocumentChangesTests
{
    [Theory, CombinatorialData]
    public async Task LinkedDocuments_AllTracked(bool mutatingLspWorkspace)
    {
        var documentText = "class C { }";
        var workspaceXml =
@$"<Workspace>
    <Project Language=""C#"" CommonReferences=""true"" AssemblyName=""CSProj1"">
        <Document FilePath=""C:\C.cs"">{documentText}{{|caret:|}}</Document>
    </Project>
    <Project Language=""C#"" CommonReferences=""true"" AssemblyName=""CSProj2"">
        <Document IsLinkFile=""true"" LinkFilePath=""C:\C.cs"" LinkAssemblyName=""CSProj1""></Document>
    </Project>
</Workspace>";
 
        await using var testLspServer = await CreateXmlTestLspServerAsync(workspaceXml, mutatingLspWorkspace);
        var caretLocation = testLspServer.GetLocations("caret").Single();
 
        await DidOpen(testLspServer, caretLocation.DocumentUri);
 
        var trackedDocuments = testLspServer.GetTrackedTexts();
        Assert.Equal(1, trackedDocuments.Length);
 
        var solution = await GetLSPSolutionAsync(testLspServer, caretLocation.DocumentUri).ConfigureAwait(false);
 
        foreach (var document in solution.Projects.First().Documents)
        {
            Assert.Equal(documentText, document.GetTextSynchronously(CancellationToken.None).ToString());
        }
 
        await DidClose(testLspServer, caretLocation.DocumentUri);
 
        Assert.Empty(testLspServer.GetTrackedTexts());
    }
 
    [Theory, CombinatorialData]
    public async Task LinkedDocuments_AllTextChanged(bool mutatingLspWorkspace)
    {
        var initialText =
@"class A
{
    void M()
    {
        {|caret:|}
    }
}";
        var workspaceXml =
@$"<Workspace>
    <Project Language=""C#"" CommonReferences=""true"" AssemblyName=""CSProj1"">
        <Document FilePath=""C:\C.cs"">{initialText}</Document>
    </Project>
    <Project Language=""C#"" CommonReferences=""true"" AssemblyName=""CSProj2"">
        <Document IsLinkFile=""true"" LinkFilePath=""C:\C.cs"" LinkAssemblyName=""CSProj1""></Document>
    </Project>
</Workspace>";
 
        await using var testLspServer = await CreateXmlTestLspServerAsync(workspaceXml, mutatingLspWorkspace);
        var caretLocation = testLspServer.GetLocations("caret").Single();
 
        var updatedText =
@"class A
{
    void M()
    {
        // hi there
    }
}";
 
        await DidOpen(testLspServer, caretLocation.DocumentUri);
 
        Assert.Equal(1, testLspServer.GetTrackedTexts().Length);
 
        await DidChange(testLspServer, caretLocation.DocumentUri, (4, 8, "// hi there"));
 
        var solution = await GetLSPSolutionAsync(testLspServer, caretLocation.DocumentUri).ConfigureAwait(false);
 
        foreach (var document in solution.Projects.First().Documents)
        {
            Assert.Equal(updatedText, document.GetTextSynchronously(CancellationToken.None).ToString());
        }
 
        await DidClose(testLspServer, caretLocation.DocumentUri);
 
        Assert.Empty(testLspServer.GetTrackedTexts());
    }
 
    private static async Task<Solution> GetLSPSolutionAsync(TestLspServer testLspServer, DocumentUri uri)
    {
        var (_, _, lspDocument) = await testLspServer.GetManager().GetLspDocumentInfoAsync(new TextDocumentIdentifier { DocumentUri = uri }, CancellationToken.None).ConfigureAwait(false);
        Contract.ThrowIfNull(lspDocument);
        return lspDocument.Project.Solution;
    }
}