initialize before using message pump

Author: justcoding121

EventHook/ApplicationWatcher.cs

@@ -56,7 +56,8 @@ public static void Start()
                     _prevTimeApp = DateTime.Now;
 
                     appQueue = new AsyncCollection<object>();
-
+                   
+                    SharedMessagePump.Initialize();
                     Task.Factory.StartNew(() => { }).ContinueWith(x =>
                       {
                           WindowHook.WindowCreated += new GeneralShellHookEventHandler(WindowCreated);

EventHook/ClipboardWatcher.cs

@@ -46,6 +46,7 @@ public static void Start()
                     {
                         _clipQueue = new AsyncCollection<object>();
 
+                        SharedMessagePump.Initialize();
                         Task.Factory.StartNew(() => { }).ContinueWith(x =>
                         {
                             _clip = new ClipBoardHook();

EventHook/Helpers/SharedMessagePump.cs

@@ -14,14 +14,18 @@ internal class SharedMessagePump
     {
         static Lazy<TaskScheduler> _scheduler;
         static Lazy<MessageHandler> _messageHandler;
-
+  
         static SharedMessagePump()
         {
             _scheduler = new Lazy<TaskScheduler>(() =>
             {
-                          
-               if(SynchronizationContext.Current!=null)
-                    return TaskScheduler.FromCurrentSynchronizationContext();
+                Dispatcher dispatcher = Dispatcher.FromThread(Thread.CurrentThread);
+                if (dispatcher != null)
+                {
+                    if (SynchronizationContext.Current != null)
+                        return TaskScheduler.FromCurrentSynchronizationContext();
+                }           
+            
 
                 TaskScheduler current = null;
 
@@ -66,6 +70,12 @@ static SharedMessagePump()
                 });
         }
 
+        internal static void Initialize()
+        {
+            GetTaskScheduler();
+            GetHandle();
+        }
+
         internal static TaskScheduler GetTaskScheduler()
         {
             return _scheduler.Value;

EventHook/KeyboardWatcher.cs

@@ -48,7 +48,8 @@ public static void Start()
                     _kh = new KeyboardHook();
                     _kh.KeyDown += new RawKeyEventHandler(KListener);
                     _kh.KeyUp += new RawKeyEventHandler(KListener);
-                    var handle = SharedMessagePump.GetHandle();
+
+                    SharedMessagePump.Initialize();
                     Task.Factory.StartNew(() => { }).ContinueWith(x =>
                     {
                         _kh.Start();

EventHook/MouseWatcher.cs

@@ -36,6 +36,7 @@ public static void Start()
                     _mh = new MouseHook();
                     _mh.MouseAction += MListener;
 
+                    SharedMessagePump.Initialize();
                     Task.Factory.StartNew(() => { }).ContinueWith(x =>
                     {
                         _mh.Start();