Exit early if the current user has no running NotificationCenter.

Fixes julienXX#29.
mjabbour · Aug 24, 2013 · 04e3148 · 04e3148
1 parent 3ea6ddc
commit 04e3148
Showing 1 changed file with 9 additions and 1 deletion.
diff --git a/Terminal Notifier/AppDelegate.m b/Terminal Notifier/AppDelegate.m
@@ -2,6 +2,8 @@
 #import <ScriptingBridge/ScriptingBridge.h>
 #import <objc/runtime.h>
 
+NSString * const TerminalNotifierBundleID = @"nl.superalloy.oss.terminal-notifier";
+NSString * const NotificationCenterUIBundleID = @"com.apple.notificationcenterui";
 
 NSString *_fakeBundleIdentifier = nil;
 
@@ -12,7 +14,7 @@ @implementation NSBundle (FakeBundleIdentifier)
 - (NSString *)__bundleIdentifier;
 {
   if (self == [NSBundle mainBundle]) {
-    return _fakeBundleIdentifier ? _fakeBundleIdentifier : @"nl.superalloy.oss.terminal-notifier";
+    return _fakeBundleIdentifier ? _fakeBundleIdentifier : TerminalNotifierBundleID;
   } else {
     return [self __bundleIdentifier];
   }
@@ -97,6 +99,12 @@ - (void)applicationDidFinishLaunching:(NSNotification *)notification;
       exit(0);
     }
 
+    NSArray *runningProcesses = [[[NSWorkspace sharedWorkspace] runningApplications] valueForKey:@"bundleIdentifier"];
+    if ([runningProcesses indexOfObject:NotificationCenterUIBundleID] == NSNotFound) {
+      NSLog(@"[!] Unable to post a notification for the current user (%@), as it has no running NotificationCenter instance.", NSUserName());
+      exit(1);
+    }
+
     NSUserDefaults *defaults = [NSUserDefaults standardUserDefaults];
 
     NSString *subtitle = defaults[@"subtitle"];