// 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 Microsoft.VisualStudio.Text;
using Microsoft.VisualStudio.Text.Editor;
using System;
using System.Collections.Immutable;
 
namespace Microsoft.CodeAnalysis.Editor.UnitTests;
 
internal static class TextEditorFactoryExtensions
{
    public static DisposableTextView CreateDisposableTextView(this ITextEditorFactoryService textEditorFactory)
        => new DisposableTextView(textEditorFactory.CreateTextView());
 
    public static DisposableTextView CreateDisposableTextView(this ITextEditorFactoryService textEditorFactory, ITextBuffer buffer, ImmutableArray<string> roles = default)
    {
        // Every default role but outlining. Starting in 15.2, the editor
        // OutliningManager imports JoinableTaskContext in a way that's 
        // difficult to satisfy in our unit tests. Since we don't directly
        // depend on it, just disable it
        if (roles.IsDefault)
        {
            roles =
            [
                PredefinedTextViewRoles.Analyzable,
                PredefinedTextViewRoles.Document,
                PredefinedTextViewRoles.Editable,
                PredefinedTextViewRoles.Interactive,
                PredefinedTextViewRoles.Zoomable,
            ];
        }
 
        var roleSet = textEditorFactory.CreateTextViewRoleSet(roles);
        return new DisposableTextView(textEditorFactory.CreateTextView(buffer, roleSet));
    }
}
 
public class DisposableTextView : IDisposable
{
    public DisposableTextView(IWpfTextView textView)
        => this.TextView = textView;
 
    public IWpfTextView TextView { get; }
 
    public void Dispose()
        => TextView.Close();
}