Catch SIGTERM signal and fix brightness value storage.

Bensge · Jan 14, 2017 · d2bc94c · d2bc94c
1 parent ac664f4
commit d2bc94c
Showing 1 changed file with 19 additions and 0 deletions.
diff --git a/NativeDisplayBrightness/AppDelegate.m b/NativeDisplayBrightness/AppDelegate.m
@@ -152,6 +152,7 @@ - (void)_loadBrightness
 
 - (void)applicationDidFinishLaunching:(NSNotification *)aNotification
 {
+    delegateInstance = self;
     if (![self _loadBezelServices])
     {
         [self _loadOSDFramework];
@@ -160,10 +161,28 @@ - (void)applicationDidFinishLaunching:(NSNotification *)aNotification
     [self _checkTrusted];
     [self _registerGlobalKeyboardEvents];
     [self _loadBrightness];
+    [self _registerSignalHandling];
+}
+
+__weak AppDelegate *delegateInstance = nil;
+
+void shutdownSignalHandler(int signal) {
+    NSLog(@"Caught SIGTERM");
+    [delegateInstance _willTerminate];
+}
+
+- (void)_registerSignalHandling
+{
+    signal(SIGTERM, shutdownSignalHandler);
 }
 
 
 - (void)applicationWillTerminate:(NSNotification *)aNotification
+{
+    [self _willTerminate];
+}
+
+- (void)_willTerminate
 {
     NSLog(@"willTerminate");
     [self _saveBrightness];