remove unnecessary code duplication

cathyjf · Oct 7, 2022 · 24164d5 · 24164d5
1 parent a162c19
commit 24164d5
Showing 1 changed file with 4 additions and 11 deletions.
diff --git a/MacPass/MPLockDaemon.m b/MacPass/MPLockDaemon.m
@@ -77,7 +77,7 @@ - (void)setLockOnLogout:(BOOL)lockOnLogout {
   if(_lockOnLogout != lockOnLogout) {
     _lockOnLogout = lockOnLogout;
     if(_lockOnLogout) {
-      [NSWorkspace.sharedWorkspace.notificationCenter addObserver:self selector:@selector(_willLogOutNotification:) name:NSWorkspaceSessionDidResignActiveNotification object:nil];
+      [NSWorkspace.sharedWorkspace.notificationCenter addObserver:self selector:@selector(_lockDocumentsForNotification:) name:NSWorkspaceSessionDidResignActiveNotification object:nil];
     }
     else {
       [NSWorkspace.sharedWorkspace.notificationCenter removeObserver:self name:@"" object:nil];
@@ -89,7 +89,7 @@ - (void)setLockOnSleep:(BOOL)lockOnSleep {
   if(_lockOnSleep != lockOnSleep) {
     _lockOnSleep = lockOnSleep;
     if(_lockOnSleep) {
-      [NSWorkspace.sharedWorkspace.notificationCenter addObserver:self selector:@selector(_willSleepNotification:) name:NSWorkspaceWillSleepNotification object:nil];
+      [NSWorkspace.sharedWorkspace.notificationCenter addObserver:self selector:@selector(_lockDocumentsForNotification:) name:NSWorkspaceWillSleepNotification object:nil];
     }
     else {
       [NSWorkspace.sharedWorkspace.notificationCenter removeObserver:self name:NSWorkspaceWillSleepNotification object:nil];
@@ -101,7 +101,7 @@ - (void)setLockOnScreenSleep:(BOOL)lockOnScreenSleep {
   if(_lockOnScreenSleep != lockOnScreenSleep) {
     _lockOnScreenSleep = lockOnScreenSleep;
     if(_lockOnScreenSleep) {
-      [NSWorkspace.sharedWorkspace.notificationCenter addObserver:self selector:@selector(_willScreenSleepNotification:) name:NSWorkspaceScreensDidSleepNotification object:nil];
+      [NSWorkspace.sharedWorkspace.notificationCenter addObserver:self selector:@selector(_lockDocumentsForNotification:) name:NSWorkspaceScreensDidSleepNotification object:nil];
     }
     else {
       [NSWorkspace.sharedWorkspace.notificationCenter removeObserver:self name:NSWorkspaceScreensDidSleepNotification object:nil];
@@ -120,14 +120,7 @@ - (void)setIdleLockTime:(NSUInteger)idleLockTime {
     }
   }
 }
-
-- (void)_willLogOutNotification:(NSNotification *)notification {
-  [((MPAppDelegate *)NSApp.delegate) lockAllDocuments];
-}
-- (void)_willSleepNotification:(NSNotification *)notification {
-  [((MPAppDelegate *)NSApp.delegate) lockAllDocuments];
-}
-- (void)_willScreenSleepNotification:(NSNotification *)notification {
+- (void)_lockDocumentsForNotification:(NSNotification *)notification {
   [((MPAppDelegate *)NSApp.delegate) lockAllDocuments];
 }