// 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.Linq;
using System.Threading;
using System.Threading.Tasks;
using Microsoft.CodeAnalysis.BracePairs;
using Microsoft.CodeAnalysis.PooledObjects;
using Microsoft.CodeAnalysis.Shared.Extensions;
using Microsoft.CodeAnalysis.Text;
using Roslyn.Test.Utilities;
using Roslyn.Utilities;
using Xunit;

namespace Microsoft.CodeAnalysis.Test.Utilities.BracePairs;

[UseExportProvider]
public abstract class AbstractBracePairsTests
{
    protected abstract EditorTestWorkspace CreateWorkspace(string input);

    public async Task Test(string test)
    {
        using var workspace = CreateWorkspace(test);

        var document = workspace.CurrentSolution.Projects.Single().Documents.Single();
        var service = document.GetRequiredLanguageService<IBracePairsService>();

        var text = await document.GetTextAsync();

        using var _ = ArrayBuilder<BracePairData>.GetInstance(out var bracePairs);
        await service.AddBracePairsAsync(document, new TextSpan(0, text.Length), bracePairs, CancellationToken.None);

        var expected = workspace.Documents.Single().AnnotatedSpans;

        Assert.Equal(expected.Count, bracePairs.Count);

        foreach (var bracePair in bracePairs)
        {
            if (!FindMatch(expected, bracePair))
                Assert.Fail($"Unexpected brace pair: {bracePair}");
        }
    }

    private static bool FindMatch(IDictionary<string, ImmutableArray<TextSpan>> expected, BracePairData bracePair)
    {
        foreach (var (_, expectedSpans) in expected)
        {
            Assert.Equal(2, expectedSpans.Length);
            var sortedSpans = expectedSpans.Sort((s1, s2) => s1.Start - s2.Start);

            if (sortedSpans.SequenceEqual(ImmutableArray.Create(bracePair.Start, bracePair.End)))
                return true;
        }

        return false;
    }
}
