// 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.
 
#nullable disable
 
using System.Linq;
using System.Threading.Tasks;
using Microsoft.CodeAnalysis.BraceMatching;
using Microsoft.CodeAnalysis.Editor.Tagging;
using Microsoft.CodeAnalysis.Test.Utilities;
using Microsoft.CodeAnalysis.Text;
using Microsoft.CodeAnalysis.Text.Shared.Extensions;
using Microsoft.VisualStudio.Text;
using Roslyn.Test.Utilities;
using Xunit;
 
namespace Microsoft.CodeAnalysis.Editor.UnitTests.BraceHighlighting
{
    [UseExportProvider]
    public abstract class AbstractBraceHighlightingTests
    {
        protected async Task TestBraceHighlightingAsync(
            string markup, ParseOptions options = null, bool swapAnglesWithBrackets = false)
        {
            MarkupTestFile.GetPositionAndSpans(markup,
                out var text, out var cursorPosition, out var expectedSpans);
 
            // needed because markup test file can't support [|[|] to indicate selecting
            // just an open bracket.
            if (swapAnglesWithBrackets)
            {
                text = text.Replace("<", "[").Replace(">", "]");
            }
 
            using var workspace = CreateWorkspace(text, options);
            WpfTestRunner.RequireWpfFact($"{nameof(AbstractBraceHighlightingTests)}.{nameof(TestBraceHighlightingAsync)} creates asynchronous taggers");
 
            var provider = new BraceHighlightingViewTaggerProvider(
                workspace.GetService<TaggerHost>(),
                GetBraceMatchingService(workspace));
 
            var testDocument = workspace.Documents.First();
            var buffer = testDocument.GetTextBuffer();
            var document = buffer.CurrentSnapshot.GetRelatedDocumentsWithChanges().FirstOrDefault();
            var context = new TaggerContext<BraceHighlightTag>(
                document, buffer.CurrentSnapshot, frozenPartialSemantics: false,
                new SnapshotPoint(buffer.CurrentSnapshot, cursorPosition));
            await provider.GetTestAccessor().ProduceTagsAsync(context);
 
            var expectedHighlights = expectedSpans.Select(ts => ts.ToSpan()).OrderBy(s => s.Start).ToList();
            var actualHighlights = context.TagSpans.Select(ts => ts.Span.Span).OrderBy(s => s.Start).ToList();
 
            Assert.Equal(expectedHighlights, actualHighlights);
        }
 
        internal virtual IBraceMatchingService GetBraceMatchingService(EditorTestWorkspace workspace)
            => workspace.GetService<IBraceMatchingService>();
 
        protected abstract EditorTestWorkspace CreateWorkspace(string markup, ParseOptions options);
    }
}