// 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;
using System.Composition;
using System.Windows;
using System.Windows.Controls;
using Microsoft.CodeAnalysis.Editor.Shared.Extensions;
using Microsoft.CodeAnalysis.Editor.Shared.Preview;
using Microsoft.CodeAnalysis.Editor.Shared.Utilities;
using Microsoft.CodeAnalysis.Host.Mef;
using Microsoft.CodeAnalysis.Options;
using Microsoft.CodeAnalysis.Text;
using Microsoft.CodeAnalysis.Text.Shared.Extensions;
using Microsoft.VisualStudio.Text;
using Microsoft.VisualStudio.Text.Editor;
using Microsoft.VisualStudio.Text.Projection;
using Microsoft.VisualStudio.Utilities;
 
namespace Microsoft.CodeAnalysis.Editor.QuickInfo;
 
[ExportWorkspaceService(typeof(IContentControlService), layer: ServiceLayer.Editor), Shared]
internal sealed partial class ContentControlService : IContentControlService
{
    private readonly IThreadingContext _threadingContext;
    private readonly ITextEditorFactoryService _textEditorFactoryService;
    private readonly IContentTypeRegistryService _contentTypeRegistryService;
    private readonly IProjectionBufferFactoryService _projectionBufferFactoryService;
    private readonly EditorOptionsService _editorOptionsService;
 
    [ImportingConstructor]
    [Obsolete(MefConstruction.ImportingConstructorMessage, error: true)]
    public ContentControlService(
        IThreadingContext threadingContext,
        ITextEditorFactoryService textEditorFactoryService,
        IContentTypeRegistryService contentTypeRegistryService,
        IProjectionBufferFactoryService projectionBufferFactoryService,
        EditorOptionsService editorOptionsService)
    {
        _threadingContext = threadingContext;
        _textEditorFactoryService = textEditorFactoryService;
        _contentTypeRegistryService = contentTypeRegistryService;
        _projectionBufferFactoryService = projectionBufferFactoryService;
        _editorOptionsService = editorOptionsService;
    }
 
    public void AttachToolTipToControl(FrameworkElement element, Func<DisposableToolTip> createToolTip)
        => LazyToolTip.AttachTo(element, _threadingContext, createToolTip);
 
    public DisposableToolTip CreateDisposableToolTip(Document document, ITextBuffer textBuffer, Span contentSpan, object backgroundResourceKey)
    {
        var control = CreateViewHostingControl(textBuffer, contentSpan);
 
        // Create the actual tooltip around the region of that text buffer we want to show.
        var toolTip = new ToolTip
        {
            Content = control
        };
 
        toolTip.SetResourceReference(Control.BackgroundProperty, backgroundResourceKey);
 
        // Create a preview workspace for this text buffer and open it's corresponding
        // document.
        //
        // our underlying preview tagger and mechanism to attach tagger to associated buffer of
        // opened document will light up automatically
        var workspace = new PreviewWorkspace(document.Project.Solution);
        workspace.OpenDocument(document.Id, textBuffer.AsTextContainer());
 
        return new DisposableToolTip(toolTip, workspace);
    }
 
    public DisposableToolTip CreateDisposableToolTip(ITextBuffer textBuffer, object backgroundResourceKey)
    {
        var control = CreateViewHostingControl(textBuffer, textBuffer.CurrentSnapshot.GetFullSpan().Span);
 
        // Create the actual tooltip around the region of that text buffer we want to show.
        var toolTip = new ToolTip
        {
            Content = control
        };
 
        toolTip.SetResourceReference(Control.BackgroundProperty, backgroundResourceKey);
 
        // we have stand alone view that is not associated with roslyn solution
        return new DisposableToolTip(toolTip, workspaceOpt: null);
    }
 
    public ViewHostingControl CreateViewHostingControl(ITextBuffer textBuffer, Span contentSpan)
    {
        var snapshotSpan = textBuffer.CurrentSnapshot.GetSpan(contentSpan);
 
        var contentType = _contentTypeRegistryService.GetContentType(
            IProjectionBufferFactoryServiceExtensions.RoslynPreviewContentType);
 
        var roleSet = _textEditorFactoryService.CreateTextViewRoleSet(
            TextViewRoles.PreviewRole,
            PredefinedTextViewRoles.Analyzable,
            PredefinedTextViewRoles.Document,
            PredefinedTextViewRoles.Editable);
 
        var contentControl = ProjectionBufferContent.Create(
            _threadingContext,
            [snapshotSpan],
            _projectionBufferFactoryService,
            _editorOptionsService,
            _textEditorFactoryService,
            contentType,
            roleSet);
 
        return contentControl;
    }
}