Revert "Register view helper when frame is shown"

This reverts commit c372e2b.

Author: adamint

src/Microsoft.VisualStudio.Editors/DesignerFramework/BaseDesignerLoader.vb

@@ -684,14 +684,6 @@ Namespace Microsoft.VisualStudio.Editors.DesignerFramework
         Protected Overridable Sub OnDesignerWindowActivated(Activated As Boolean)
         End Sub
 
-        ''' <summary>
-        ''' Called before the frame that's hosting the designer is shown.
-        ''' </summary>
-        ''' <param name="FirstShow">Indicates whether this is the first time the designer is being shown.</param>
-        ''' <param name="Frame">The frame hosting the designer.</param>
-        Protected Overridable Sub OnBeforeDesignerWindowShow(FirstShow As Boolean, Frame As IVsWindowFrame)
-        End Sub
-
 #End Region
 
         ''' <summary>
@@ -826,16 +818,6 @@ Namespace Microsoft.VisualStudio.Editors.DesignerFramework
             End If
         End Function
 
-        Private Function OnBeforeDocumentWindowShow(docCookie As UInteger, firstShow As Integer, frame As IVsWindowFrame) As Integer _
-            Implements IVsRunningDocTableEvents.OnBeforeDocumentWindowShow, IVsRunningDocTableEvents2.OnBeforeDocumentWindowShow
-
-            ' Only notify if our frame showing
-            If CType(GetService(GetType(IVsWindowFrame)), IVsWindowFrame) Is frame Then
-                OnBeforeDesignerWindowShow(firstShow = 1, frame)
-            End If
-
-        End Function
-
 #Region "RDT events we don't care about"
 
         Private Function OnAfterAttributeChange(docCookie As UInteger, attributes As UInteger) As Integer _
@@ -858,6 +840,10 @@ Namespace Microsoft.VisualStudio.Editors.DesignerFramework
             Implements IVsRunningDocTableEvents.OnAfterSave, IVsRunningDocTableEvents2.OnAfterSave
         End Function
 
+        Private Function OnBeforeDocumentWindowShow(docCookie As UInteger, firstShow As Integer, frame As IVsWindowFrame) As Integer _
+            Implements IVsRunningDocTableEvents.OnBeforeDocumentWindowShow, IVsRunningDocTableEvents2.OnBeforeDocumentWindowShow
+        End Function
+
 #End Region
 #End Region
 

src/Microsoft.VisualStudio.Editors/ResourceEditor/ResourceEditorDesignerLoader.vb

@@ -141,6 +141,8 @@ Namespace Microsoft.VisualStudio.Editors.ResourceEditor
                         'Try to restore the editor state from before the last reload, if any.
                         NewResourceEditorRoot.RootDesigner.TryDepersistSavedEditorState()
 
+                        'Now that we know the load succeeded, we can try registering our view helper
+                        NewResourceEditorRoot.RootDesigner.RegisterViewHelper()
                     Catch ex As Exception When ReportWithoutCrash(ex, NameOf(HandleLoad), NameOf(ResourceEditorDesignerLoader))
                         _rootComponent = Nothing
 
@@ -256,12 +258,6 @@ Namespace Microsoft.VisualStudio.Editors.ResourceEditor
             End If
         End Sub
 
-        Protected Overrides Sub OnBeforeDesignerWindowShow(FirstShow As Boolean, Frame As IVsWindowFrame)
-            If FirstShow Then
-                _rootComponent.RootDesigner.RegisterViewHelper(Frame)
-            End If
-        End Sub
-
         ''' <summary>
         ''' ManualCheckOut without the ProjectReloaded flag because it is not needed in
         '''   the resource editor.

src/Microsoft.VisualStudio.Editors/ResourceEditor/ResourceEditorRootDesigner.vb

@@ -410,12 +410,34 @@ Namespace Microsoft.VisualStudio.Editors.ResourceEditor
         ''' Register this root designer as a view helper with the current frame so the shell will can find our
         '''   implementations of IVsFindTarget, IOleCommandTarget, etc.
         ''' </summary>
-        Public Sub RegisterViewHelper(VsWindowFrame As IVsWindowFrame)
+        Public Sub RegisterViewHelper()
             Try
+                Dim VsWindowFrame As IVsWindowFrame = CType(GetService(GetType(IVsWindowFrame)), IVsWindowFrame)
+
                 If VsWindowFrame IsNot Nothing Then
                     VSErrorHandler.ThrowOnFailure(VsWindowFrame.SetProperty(__VSFPROPID.VSFPROPID_ViewHelper, New UnknownWrapper(Me)))
                 Else
-                    Debug.Fail("Unable to register our view helper")
+                    If _view IsNot Nothing Then
+                        'We don't have a window frame yet.  Need to delay this registration until we do.
+                        '  Easiest way is to use BeginInvoke.
+                        If _delayRegisteringViewHelper Then
+                            'This is already our second try
+                            Debug.Fail("Unable to delay-register our view helper")
+                            _delayRegisteringViewHelper = False
+                        Else
+                            'Try again, delayed.
+                            _delayRegisteringViewHelper = True
+
+                            ' VS Whidbey #260046 -- Make sure the control is created before calling Invoke/BeginInvoke                                                      
+                            If _view.Created = False Then
+                                _view.CreateControl()
+                            End If
+
+                            _view.BeginInvoke(New System.Windows.Forms.MethodInvoker(AddressOf RegisterViewHelper))
+                        End If
+                    Else
+                        Debug.Fail("View not set in RegisterViewHelper() - can't delay-register view helper")
+                    End If
                 End If
             Catch ex As Exception When ReportWithoutCrash(ex, NameOf(RegisterViewHelper), NameOf(ResourceEditorRootDesigner))
             End Try