// 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.ComponentModel.Composition;
using Microsoft.CodeAnalysis;
using Microsoft.CodeAnalysis.Editor;
using Microsoft.CodeAnalysis.Editor.Host;
using Microsoft.CodeAnalysis.Editor.Shared.Utilities;
using Microsoft.CodeAnalysis.Host.Mef;
using Microsoft.CodeAnalysis.Options;
using Microsoft.CodeAnalysis.Shared.TestHooks;
using Microsoft.CodeAnalysis.Text;
using Microsoft.VisualStudio.Text.Classification;
using Microsoft.VisualStudio.Text.Editor;
using Microsoft.VisualStudio.Text.Tagging;
using Microsoft.VisualStudio.Utilities;
 
namespace Microsoft.VisualStudio.LanguageServices.Implementation.InheritanceMargin;
 
[Export(typeof(IWpfTextViewMarginProvider))]
[ContentType(ContentTypeNames.CSharpContentType)]
[ContentType(ContentTypeNames.VisualBasicContentType)]
[Name(nameof(InheritanceMarginViewMarginProvider))]
// Place our margin inside Left Selection Margin Container. And keep it to the left-most location.
[MarginContainer(PredefinedMarginNames.LeftSelection)]
[Order(After = DefaultOrderings.Lowest)]
[TextViewRole(PredefinedTextViewRoles.Document)]
internal class InheritanceMarginViewMarginProvider : IWpfTextViewMarginProvider
{
    private readonly Workspace _workspace;
    private readonly IViewTagAggregatorFactoryService _tagAggregatorFactoryService;
    private readonly IThreadingContext _threadingContext;
    private readonly IStreamingFindUsagesPresenter _streamingFindUsagesPresenter;
    private readonly IClassificationFormatMapService _classificationFormatMapService;
    private readonly ClassificationTypeMap _classificationTypeMap;
    private readonly IUIThreadOperationExecutor _operationExecutor;
    private readonly IEditorFormatMapService _editorFormatMapService;
    private readonly IAsynchronousOperationListenerProvider _listenerProvider;
    private readonly IGlobalOptionService _globalOptions;
 
    [ImportingConstructor]
    [Obsolete(MefConstruction.ImportingConstructorMessage, error: true)]
    public InheritanceMarginViewMarginProvider(
        VisualStudioWorkspace workspace,
        IThreadingContext threadingContext,
        IStreamingFindUsagesPresenter streamingFindUsagesPresenter,
        ClassificationTypeMap classificationTypeMap,
        IClassificationFormatMapService classificationFormatMapService,
        IUIThreadOperationExecutor operationExecutor,
        IViewTagAggregatorFactoryService tagAggregatorFactoryService,
        IEditorFormatMapService editorFormatMapService,
        IGlobalOptionService globalOptions,
        IAsynchronousOperationListenerProvider listenerProvider)
    {
        _workspace = workspace;
        _threadingContext = threadingContext;
        _streamingFindUsagesPresenter = streamingFindUsagesPresenter;
        _classificationTypeMap = classificationTypeMap;
        _classificationFormatMapService = classificationFormatMapService;
        _operationExecutor = operationExecutor;
        _tagAggregatorFactoryService = tagAggregatorFactoryService;
        _editorFormatMapService = editorFormatMapService;
        _globalOptions = globalOptions;
        _listenerProvider = listenerProvider;
    }
 
    public IWpfTextViewMargin? CreateMargin(IWpfTextViewHost wpfTextViewHost, IWpfTextViewMargin marginContainer)
    {
        var textView = wpfTextViewHost.TextView;
        var tagAggregator = _tagAggregatorFactoryService.CreateTagAggregator<InheritanceMarginTag>(textView);
        var editorFormatMap = _editorFormatMapService.GetEditorFormatMap(textView);
 
        var document = wpfTextViewHost.TextView.TextBuffer.CurrentSnapshot.GetOpenDocumentInCurrentContextWithChanges();
        if (document == null)
        {
            return null;
        }
 
        var listener = _listenerProvider.GetListener(FeatureAttribute.InheritanceMargin);
        return new InheritanceMarginViewMargin(
            _workspace,
            textView,
            _threadingContext,
            _streamingFindUsagesPresenter,
            _operationExecutor,
            _classificationFormatMapService.GetClassificationFormatMap("tooltip"),
            _classificationTypeMap,
            tagAggregator,
            editorFormatMap,
            _globalOptions,
            listener,
            document.Project.Language);
    }
}