Add usings on paste

src/EditorFeatures/Core.Wpf/InlineHints/InlineHintsTaggerProvider.cs

@@ -5,6 +5,7 @@
 using System;
 using System.ComponentModel.Composition;
 using Microsoft.CodeAnalysis.Editor.Host;
+using Microsoft.CodeAnalysis.Editor.Shared.Extensions;
 using Microsoft.CodeAnalysis.Editor.Shared.Utilities;
 using Microsoft.CodeAnalysis.Host.Mef;
 using Microsoft.VisualStudio.Text;
@@ -56,9 +57,7 @@ public InlineHintsTaggerProvider(
 
         public ITagger<T>? CreateTagger<T>(ITextView textView, ITextBuffer buffer) where T : ITag
         {
-            // Determining of the textView's buffer does not match the buffer in order to skip showing the hints for
-            // the interactive window
-            if (buffer != textView.TextBuffer)
+            if (textView.IsBufferInInteractiveWindow(buffer))
             {
                 return null;
             }

src/EditorFeatures/Core/EditorFeaturesResources.resx

@@ -945,6 +945,13 @@ Do you want to proceed?</value>
   <data name="Error_creating_instance_of_CodeFixProvider" xml:space="preserve">
     <value>Error creating instance of CodeFixProvider</value>
   </data>
+  <data name="Adding_missing_import_directives" xml:space="preserve">
+    <value>Adding missing import directives...</value>
+    <comment>Used as a thread awaited dialog letting the user know that work is being done on paste. "import" is a language specific term</comment>
+  </data>
+  <data name="Add_Missing_Imports_On_Paste" xml:space="preserve">
+    <value>Add Missing Imports On Paste</value>
+  </data>
   <data name="User_Types_Records" xml:space="preserve">
     <value>User Types - Records</value>
   </data>

src/EditorFeatures/Core/Extensibility/Commands/PredefinedCommandHandlerNames.cs

@@ -169,6 +169,11 @@ internal static class PredefinedCommandHandlerNames
         /// </summary>
         public const string PasteTrackingPaste = "Paste Tracking Paste Command Handler";
 
+        /// <summary>
+        /// Command handler name for Paste in Add Imports.
+        /// </summary>
+        public const string AddImportsPaste = "Add Imports Paste Command Handler";
+
         /// <summary>
         /// Command handler name for Edit and Continue file save handler.
         /// </summary>

src/EditorFeatures/Core/Implementation/AddImports/AddImportsPasteCommandHandler.cs

@@ -0,0 +1,120 @@
+// 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 System.Threading;
+using System.Threading.Tasks;
+using Microsoft.CodeAnalysis.AddMissingImports;
+using Microsoft.CodeAnalysis.Editor.Shared.Extensions;
+using Microsoft.CodeAnalysis.Editor.Shared.Options;
+using Microsoft.CodeAnalysis.Editor.Shared.Utilities;
+using Microsoft.CodeAnalysis.Host.Mef;
+using Microsoft.CodeAnalysis.Shared.Extensions;
+using Microsoft.CodeAnalysis.Text;
+using Microsoft.VisualStudio.Commanding;
+using Microsoft.VisualStudio.Text;
+using Microsoft.VisualStudio.Text.Editor.Commanding.Commands;
+using Microsoft.VisualStudio.Utilities;
+
+namespace Microsoft.CodeAnalysis.Editor.Implementation.AddImports
+{
+    [Export]
+    [Export(typeof(ICommandHandler))]
+    [ContentType(ContentTypeNames.RoslynContentType)]
+    [Name(PredefinedCommandHandlerNames.AddImportsPaste)]
+    // Order is important here, this command needs to execute before PasteTracking
+    // since it may modify the pasted span. Paste tracking dismisses if 
+    // the span is modified. It doesn't need to be before FormatDocument, but
+    // this helps the order of execution be more constant in case there 
+    // are problems that arise. This command will always execute the next
+    // command before doing operations.
+    [Order(After = PredefinedCommandHandlerNames.PasteTrackingPaste)]
+    [Order(Before = PredefinedCommandHandlerNames.FormatDocument)]
+    internal class AddImportsPasteCommandHandler : IChainedCommandHandler<PasteCommandArgs>
+    {
+        public string DisplayName => EditorFeaturesResources.Add_Missing_Imports_On_Paste;
+
+        private readonly IThreadingContext _threadingContext;
+
+        [ImportingConstructor]
+        [Obsolete(MefConstruction.ImportingConstructorMessage, error: true)]
+        public AddImportsPasteCommandHandler(IThreadingContext threadingContext)
+            => _threadingContext = threadingContext;
+
+        public void ExecuteCommand(PasteCommandArgs args, Action nextCommandHandler, CommandExecutionContext executionContext)
+        {
+            // Capture the pre-paste caret position
+            var caretPosition = args.TextView.GetCaretPoint(args.SubjectBuffer);
+            if (!caretPosition.HasValue)
+            {
+                return;
+            }
+
+            nextCommandHandler();
+
+            if (!args.SubjectBuffer.CanApplyChangeDocumentToWorkspace())
+            {
+                return;
+            }
+
+            // Don't perform work if we're inside the interactive window
+            if (args.TextView.IsBufferInInteractiveWindow(args.SubjectBuffer))
+            {
+                return;
+            }
+
+            if (!args.SubjectBuffer.GetFeatureOnOffOption(FeatureOnOffOptions.AddImportsOnPaste))
+            {
+                return;
+            }
+
+            // Create a tracking span from the pre-paste caret position that will grow as text is inserted.
+            var trackingSpan = caretPosition.Value.Snapshot.CreateTrackingSpan(caretPosition.Value.Position, 0, SpanTrackingMode.EdgeInclusive);
+
+            // Applying the post-paste snapshot to the tracking span gives us the span of pasted text.
+            var snapshotSpan = trackingSpan.GetSpan(args.SubjectBuffer.CurrentSnapshot);
+            var textSpan = TextSpan.FromBounds(snapshotSpan.Start, snapshotSpan.End);
+
+            AddMissingImportsForPaste(args, executionContext, textSpan);
+        }
+
+        public CommandState GetCommandState(PasteCommandArgs args, Func<CommandState> nextCommandHandler)
+            => nextCommandHandler();
+
+        private void AddMissingImportsForPaste(PasteCommandArgs args, CommandExecutionContext executionContext, TextSpan textSpan)
+        {
+            var sourceTextContainer = args.SubjectBuffer.AsTextContainer();
+            if (!Workspace.TryGetWorkspace(sourceTextContainer, out var workspace))
+            {
+                return;
+            }
+
+            var document = sourceTextContainer.GetOpenDocumentInCurrentContext();
+            if (document is null)
+            {
+                return;
+            }
+
+            using var _ = executionContext.OperationContext.AddScope(allowCancellation: true, EditorFeaturesResources.Adding_missing_import_directives);
+            var cancellationToken = executionContext.OperationContext.UserCancellationToken;
+
+#pragma warning disable VSTHRD102 // Implement internal logic asynchronously
+            var updatedDocument = _threadingContext.JoinableTaskFactory.Run(() => AddMissingImportsAsync(document, textSpan, cancellationToken));
+#pragma warning restore VSTHRD102 // Implement internal logic asynchronously
+            if (updatedDocument is null)
+            {
+                return;
+            }
+
+            updatedDocument.Project.Solution.Workspace.TryApplyChanges(updatedDocument.Project.Solution);
+        }
+
+        private static async Task<Document?> AddMissingImportsAsync(Document document, TextSpan textSpan, CancellationToken cancellationToken)
+        {
+            var addMissingImportsService = document.GetRequiredLanguageService<IAddMissingImportsFeatureService>();
+            return await addMissingImportsService.AddMissingImportsAsync(document, textSpan, cancellationToken).ConfigureAwait(false);
+        }
+    }
+}

src/EditorFeatures/Core/Implementation/PasteTracking/PasteTrackingPasteCommandHandlercs.cs → src/EditorFeatures/Core/Implementation/PasteTracking/PasteTrackingPasteCommandHandler.cs

@@ -2,8 +2,6 @@
 // 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.ComponentModel.Composition;
 using Microsoft.CodeAnalysis.Editor;
@@ -35,7 +33,9 @@ internal class PasteTrackingPasteCommandHandler : IChainedCommandHandler<PasteCo
         [ImportingConstructor]
         [Obsolete(MefConstruction.ImportingConstructorMessage, error: true)]
         public PasteTrackingPasteCommandHandler(PasteTrackingService pasteTrackingService)
-            => _pasteTrackingService = pasteTrackingService;
+        {
+            _pasteTrackingService = pasteTrackingService;
+        }
 
         public CommandState GetCommandState(PasteCommandArgs args, Func<CommandState> nextCommandHandler)
             => nextCommandHandler();
@@ -44,15 +44,14 @@ public void ExecuteCommand(PasteCommandArgs args, Action nextCommandHandler, Com
         {
             // Capture the pre-paste caret position
             var caretPosition = args.TextView.GetCaretPoint(args.SubjectBuffer);
-
-            // Allow the pasted text to be inserted and formatted.
-            nextCommandHandler();
-
-            if (!args.SubjectBuffer.CanApplyChangeDocumentToWorkspace())
+            if (!caretPosition.HasValue)
             {
                 return;
             }
 
+            // Allow the pasted text to be inserted and formatted.
+            nextCommandHandler();
+
             // Create a tracking span from the pre-paste caret position that will grow as text is inserted.
             var trackingSpan = caretPosition.Value.Snapshot.CreateTrackingSpan(caretPosition.Value.Position, 0, SpanTrackingMode.EdgeInclusive);
 

src/EditorFeatures/Core/Shared/Extensions/ITextViewExtensions.cs

@@ -377,5 +377,12 @@ public static bool TryGetSurfaceBufferSpan(
 
             return span;
         }
+
+        /// <summary>
+        /// Determines if the textbuffer passed in matches the buffer for the textview. If it does
+        /// not, we assume that the textview is in the interactive window.
+        /// </summary>
+        public static bool IsBufferInInteractiveWindow(this ITextView textView, ITextBuffer textBuffer)
+            => textBuffer != textView.TextBuffer;
     }
 }

src/EditorFeatures/Core/Shared/Options/FeatureOnOffOptions.cs

@@ -83,6 +83,9 @@ internal static class FeatureOnOffOptions
         public static readonly Option2<int> UseEnhancedColors = new(
             nameof(FeatureOnOffOptions), nameof(UseEnhancedColors), defaultValue: 1,
             storageLocations: new RoamingProfileStorageLocation("WindowManagement.Options.UseEnhancedColorsForManagedLanguages"));
+
+        public static readonly PerLanguageOption2<bool> AddImportsOnPaste = new(
+            nameof(FeatureOnOffOptions), nameof(AddImportsOnPaste), defaultValue: true);
     }
 
     [ExportOptionProvider, Shared]
@@ -112,6 +115,7 @@ public FeatureOnOffOptionsProvider()
             FeatureOnOffOptions.RefactoringVerification,
             FeatureOnOffOptions.StreamingGoToImplementation,
             FeatureOnOffOptions.NavigateToDecompiledSources,
-            FeatureOnOffOptions.UseEnhancedColors);
+            FeatureOnOffOptions.UseEnhancedColors,
+            FeatureOnOffOptions.AddImportsOnPaste);
     }
 }

src/EditorFeatures/Core/xlf/EditorFeaturesResources.cs.xlf

@@ -2,6 +2,16 @@
 <xliff xmlns="urn:oasis:names:tc:xliff:document:1.2" xmlns:xsi="http://www.w3.org/2001/XMLSchema-instance" version="1.2" xsi:schemaLocation="urn:oasis:names:tc:xliff:document:1.2 xliff-core-1.2-transitional.xsd">
   <file datatype="xml" source-language="en" target-language="cs" original="../EditorFeaturesResources.resx">
     <body>
+      <trans-unit id="Add_Missing_Imports_On_Paste">
+        <source>Add Missing Imports On Paste</source>
+        <target state="new">Add Missing Imports On Paste</target>
+        <note />
+      </trans-unit>
+      <trans-unit id="Adding_missing_import_directives">
+        <source>Adding missing import directives...</source>
+        <target state="new">Adding missing import directives...</target>
+        <note>Used as a thread awaited dialog letting the user know that work is being done on paste. "import" is a language specific term</note>
+      </trans-unit>
       <trans-unit id="An_inline_rename_session_is_active_for_identifier_0">
         <source>An inline rename session is active for identifier '{0}'. Invoke inline rename again to access additional options. You may continue to edit the identifier being renamed at any time.</source>
         <target state="translated">Relace přejmenování na řádku je pro identifikátor {0} aktivní. Pokud chcete získat přístup k dalším možnostem, znovu volejte přejmenování na řádku. Můžete kdykoli pokračovat v úpravách identifikátoru, který se přejmenovává.</target>

src/EditorFeatures/Core/xlf/EditorFeaturesResources.de.xlf

@@ -2,6 +2,16 @@
 <xliff xmlns="urn:oasis:names:tc:xliff:document:1.2" xmlns:xsi="http://www.w3.org/2001/XMLSchema-instance" version="1.2" xsi:schemaLocation="urn:oasis:names:tc:xliff:document:1.2 xliff-core-1.2-transitional.xsd">
   <file datatype="xml" source-language="en" target-language="de" original="../EditorFeaturesResources.resx">
     <body>
+      <trans-unit id="Add_Missing_Imports_On_Paste">
+        <source>Add Missing Imports On Paste</source>
+        <target state="new">Add Missing Imports On Paste</target>
+        <note />
+      </trans-unit>
+      <trans-unit id="Adding_missing_import_directives">
+        <source>Adding missing import directives...</source>
+        <target state="new">Adding missing import directives...</target>
+        <note>Used as a thread awaited dialog letting the user know that work is being done on paste. "import" is a language specific term</note>
+      </trans-unit>
       <trans-unit id="An_inline_rename_session_is_active_for_identifier_0">
         <source>An inline rename session is active for identifier '{0}'. Invoke inline rename again to access additional options. You may continue to edit the identifier being renamed at any time.</source>
         <target state="translated">Für den Bezeichner "{0}" ist eine Inline-Umbenennungssitzung aktiv. Rufen Sie die Inline-Umbenennung erneut auf, um auf zusätzliche Optionen zuzugreifen. Sie können den Bezeichner, der gerade umbenannt wird, jederzeit weiterbearbeiten.</target>

src/EditorFeatures/Core/xlf/EditorFeaturesResources.es.xlf

@@ -2,6 +2,16 @@
 <xliff xmlns="urn:oasis:names:tc:xliff:document:1.2" xmlns:xsi="http://www.w3.org/2001/XMLSchema-instance" version="1.2" xsi:schemaLocation="urn:oasis:names:tc:xliff:document:1.2 xliff-core-1.2-transitional.xsd">
   <file datatype="xml" source-language="en" target-language="es" original="../EditorFeaturesResources.resx">
     <body>
+      <trans-unit id="Add_Missing_Imports_On_Paste">
+        <source>Add Missing Imports On Paste</source>
+        <target state="new">Add Missing Imports On Paste</target>
+        <note />
+      </trans-unit>
+      <trans-unit id="Adding_missing_import_directives">
+        <source>Adding missing import directives...</source>
+        <target state="new">Adding missing import directives...</target>
+        <note>Used as a thread awaited dialog letting the user know that work is being done on paste. "import" is a language specific term</note>
+      </trans-unit>
       <trans-unit id="An_inline_rename_session_is_active_for_identifier_0">
         <source>An inline rename session is active for identifier '{0}'. Invoke inline rename again to access additional options. You may continue to edit the identifier being renamed at any time.</source>
         <target state="translated">Hay una sesión de cambio de nombre insertado que está activa para el identificador "{0}". Vuelva a invocar el cambio de nombre insertado para acceder a más opciones. Puede continuar editando el identificador cuyo nombre se va a cambiar en cualquier momento.</target>

src/EditorFeatures/Core/xlf/EditorFeaturesResources.fr.xlf

@@ -2,6 +2,16 @@
 <xliff xmlns="urn:oasis:names:tc:xliff:document:1.2" xmlns:xsi="http://www.w3.org/2001/XMLSchema-instance" version="1.2" xsi:schemaLocation="urn:oasis:names:tc:xliff:document:1.2 xliff-core-1.2-transitional.xsd">
   <file datatype="xml" source-language="en" target-language="fr" original="../EditorFeaturesResources.resx">
     <body>
+      <trans-unit id="Add_Missing_Imports_On_Paste">
+        <source>Add Missing Imports On Paste</source>
+        <target state="new">Add Missing Imports On Paste</target>
+        <note />
+      </trans-unit>
+      <trans-unit id="Adding_missing_import_directives">
+        <source>Adding missing import directives...</source>
+        <target state="new">Adding missing import directives...</target>
+        <note>Used as a thread awaited dialog letting the user know that work is being done on paste. "import" is a language specific term</note>
+      </trans-unit>
       <trans-unit id="An_inline_rename_session_is_active_for_identifier_0">
         <source>An inline rename session is active for identifier '{0}'. Invoke inline rename again to access additional options. You may continue to edit the identifier being renamed at any time.</source>
         <target state="translated">Une session de renommage inline est active pour l'identificateur '{0}'. Appelez à nouveau le renommage inline pour accéder à des options supplémentaires. Vous pouvez continuer à modifier l'identificateur renommé à tout moment.</target>

src/EditorFeatures/Core/xlf/EditorFeaturesResources.it.xlf

@@ -2,6 +2,16 @@
 <xliff xmlns="urn:oasis:names:tc:xliff:document:1.2" xmlns:xsi="http://www.w3.org/2001/XMLSchema-instance" version="1.2" xsi:schemaLocation="urn:oasis:names:tc:xliff:document:1.2 xliff-core-1.2-transitional.xsd">
   <file datatype="xml" source-language="en" target-language="it" original="../EditorFeaturesResources.resx">
     <body>
+      <trans-unit id="Add_Missing_Imports_On_Paste">
+        <source>Add Missing Imports On Paste</source>
+        <target state="new">Add Missing Imports On Paste</target>
+        <note />
+      </trans-unit>
+      <trans-unit id="Adding_missing_import_directives">
+        <source>Adding missing import directives...</source>
+        <target state="new">Adding missing import directives...</target>
+        <note>Used as a thread awaited dialog letting the user know that work is being done on paste. "import" is a language specific term</note>
+      </trans-unit>
       <trans-unit id="An_inline_rename_session_is_active_for_identifier_0">
         <source>An inline rename session is active for identifier '{0}'. Invoke inline rename again to access additional options. You may continue to edit the identifier being renamed at any time.</source>
         <target state="translated">Per l'identificatore '{0}' è attiva una sessione di ridenominazione inline. Richiamare nuovamente la ridenominazione inline per accedere alle opzioni aggiuntive. È possibile continuare a modificare l'identificatore da rinominare in qualsiasi momento.</target>

src/EditorFeatures/Core/xlf/EditorFeaturesResources.ja.xlf

@@ -2,6 +2,16 @@
 <xliff xmlns="urn:oasis:names:tc:xliff:document:1.2" xmlns:xsi="http://www.w3.org/2001/XMLSchema-instance" version="1.2" xsi:schemaLocation="urn:oasis:names:tc:xliff:document:1.2 xliff-core-1.2-transitional.xsd">
   <file datatype="xml" source-language="en" target-language="ja" original="../EditorFeaturesResources.resx">
     <body>
+      <trans-unit id="Add_Missing_Imports_On_Paste">
+        <source>Add Missing Imports On Paste</source>
+        <target state="new">Add Missing Imports On Paste</target>
+        <note />
+      </trans-unit>
+      <trans-unit id="Adding_missing_import_directives">
+        <source>Adding missing import directives...</source>
+        <target state="new">Adding missing import directives...</target>
+        <note>Used as a thread awaited dialog letting the user know that work is being done on paste. "import" is a language specific term</note>
+      </trans-unit>
       <trans-unit id="An_inline_rename_session_is_active_for_identifier_0">
         <source>An inline rename session is active for identifier '{0}'. Invoke inline rename again to access additional options. You may continue to edit the identifier being renamed at any time.</source>
         <target state="translated">識別子 '{0}' のインラインの名前変更セッションがアクティブです。追加オプションにアクセスするには、インラインの名前変更をもう一度呼び出します。名前を変更する識別子はいつでも引き続き編集できます。</target>