public void OnConnection(object Application, ext_ConnectMode ConnectMode, object AddInInst, ref Array custom)
{
try
{
_vbe = RootComWrapperFactory.GetVbeWrapper(Application);
_addin = RootComWrapperFactory.GetAddInWrapper(AddInInst);
_addin.Object = this;
VbeProvider.Initialize(_vbe);
VbeNativeServices.HookEvents(_vbe);
#if DEBUG
// FOR DEBUGGING/DEVELOPMENT PURPOSES, ALLOW ACCESS TO SOME VBETypeLibsAPI FEATURES FROM VBA
_addin.Object = new VBEditor.ComManagement.TypeLibsAPI.VBETypeLibsAPI_Object(_vbe);
#endif
switch (ConnectMode)
{
case ext_ConnectMode.ext_cm_Startup:
// normal execution path - don't initialize just yet, wait for OnStartupComplete to be called by the host.
break;
case ext_ConnectMode.ext_cm_AfterStartup:
_isBeginShutdownExecuted = false; //When we reconnect after having been unloaded, the variable might no longer have its initial value.
InitializeAddIn();
break;
}
}
catch (Exception e)
{
Console.WriteLine(e);
}
}
public void OnConnection(object Application, ext_ConnectMode ConnectMode, object AddInInst, ref Array custom)
{
try
{
_vbe = RootComWrapperFactory.GetVbeWrapper(Application);
_addin = RootComWrapperFactory.GetAddInWrapper(AddInInst);
_addin.Object = this;
_vbeNativeApi = new VbeNativeApiAccessor();
_beepInterceptor = new BeepInterceptor(_vbeNativeApi);
VbeProvider.Initialize(_vbe, _vbeNativeApi, _beepInterceptor);
VbeNativeServices.HookEvents(_vbe);
SetAddInObject();
switch (ConnectMode)
{
case ext_ConnectMode.ext_cm_Startup:
// normal execution path - don't initialize just yet, wait for OnStartupComplete to be called by the host.
break;
case ext_ConnectMode.ext_cm_AfterStartup:
_isBeginShutdownExecuted = false; //When we reconnect after having been unloaded, the variable might no longer have its initial value.
InitializeAddIn();
break;
}
}
catch (Exception e)
{
Console.WriteLine(e);
}
}
private static void AssociateCodePane(object sender, EventArgs eventArgs)
{
var subclass = (CodePaneSubclass)sender;
subclass.VbeObject = VbeNativeServices.GetCodePaneFromHwnd(subclass.Hwnd);
SubclassLogger.Trace($"CodePane subclass for hWnd 0x{subclass.Hwnd.ToInt64():X8} associated itself with its VBE object.");
}
private void OnVbeFocusChanged(object sender, WindowChangedEventArgs e)
{
if (e.EventType == FocusType.GotFocus)
{
switch (e.Hwnd.ToWindowType())
{
case WindowType.DesignerWindow:
Task.Run(() =>
{
using (var component = _vbe.SelectedVBComponent)
{
DispatchSelectedDesignerDeclaration(component);
}
});
break;
case WindowType.CodePane:
//Caret changed in a code pane.
Task.Run(() =>
{
using (var pane = VbeNativeServices.GetCodePaneFromHwnd(e.Hwnd))
{
var selection = pane.GetQualifiedSelection();
if (!selection.HasValue)
{
return;
}
var selectedDeclaration = _selectedDeclarationProvider.SelectedDeclaration(selection.Value);
var eventArgs = new DeclarationChangedEventArgs(_vbe, selectedDeclaration);
DispatchSelectedDeclaration(eventArgs);
}
});
break;
}
}
else if (e.EventType == FocusType.ChildFocus)
{
//Treeview selection changed in project window.
Task.Run(() =>
{
using (var component = _vbe.SelectedVBComponent)
{
DispatchSelectedProjectNodeDeclaration(component);
}
});
}
}
private void OnVbeFocusChanged(object sender, WindowChangedEventArgs e)
{
if (e.EventType == FocusType.GotFocus)
{
switch (e.Hwnd.ToWindowType())
{
case WindowType.DesignerWindow:
Task.Run(() =>
{
using (var component = _vbe.SelectedVBComponent)
{
DispatchSelectedDesignerDeclaration(component);
}
});
break;
case WindowType.CodePane:
//Caret changed in a code pane.
Task.Run(() =>
{
using (var pane = VbeNativeServices.GetCodePaneFromHwnd(e.Hwnd))
{
DispatchSelectedDeclaration(
new DeclarationChangedEventArgs(_vbe, _parser.State.FindSelectedDeclaration(pane)));
}
});
break;
}
}
else if (e.EventType == FocusType.ChildFocus)
{
//Treeview selection changed in project window.
Task.Run(() =>
{
using (var component = _vbe.SelectedVBComponent)
{
DispatchSelectedProjectNodeDeclaration(component);
}
});
}
}
private void ShutdownAddIn()
{
var currentDomain = AppDomain.CurrentDomain;
try
{
_logger.Log(LogLevel.Info, "Rubberduck is shutting down.");
_logger.Log(LogLevel.Trace, "Unhooking VBENativeServices events...");
VbeNativeServices.UnhookEvents();
VbeProvider.Terminate();
_logger.Log(LogLevel.Trace, "Releasing dockable hosts...");
using (var windows = _vbe.Windows)
{
windows.ReleaseDockableHosts();
}
if (_app != null)
{
_logger.Log(LogLevel.Trace, "Initiating App.Shutdown...");
_app.Shutdown();
_app = null;
}
if (_container != null)
{
_logger.Log(LogLevel.Trace, "Disposing IoC container...");
_container.Dispose();
_container = null;
}
}
catch (Exception e)
{
_logger.Error(e);
_logger.Log(LogLevel.Warn, "Exception is swallowed.");
//throw; // <<~ uncomment to crash the process
}
finally
{
try
{
_logger.Log(LogLevel.Trace, "Disposing COM safe...");
ComSafeManager.DisposeAndResetComSafe();
_addin = null;
_vbe = null;
_isInitialized = false;
_logger.Log(LogLevel.Info, "No exceptions were thrown.");
}
catch (Exception e)
{
_logger.Error(e);
_logger.Log(LogLevel.Warn, "Exception disposing the ComSafe has been swallowed.");
//throw; // <<~ uncomment to crash the process
}
finally
{
_logger.Log(LogLevel.Trace, "Unregistering AppDomain handlers....");
currentDomain.AssemblyResolve -= LoadFromSameFolder;
currentDomain.UnhandledException -= HandlAppDomainException;
_logger.Log(LogLevel.Trace, "Done. Main Shutdown completed. Toolwindows follow. Quack!");
_isInitialized = false;
}
}
}
