Added functionality to activate keeping the mac awake when connected to AC power

KeepingYouAwake/KYAAppController/KYAAppController.m

@@ -14,6 +14,11 @@
 #import "KYABatteryCapacityThreshold.h"
 #import "KYAActivationDurationsMenuController.h"
 #import "KYAActivationUserNotification.h"
+#import <IOKit/ps/IOPowerSources.h>
+
+#define KYA_POWER_SOURCE_STATE "Power Source State"
+#define KYA_AC_POWER "AC Power"
+#define KYA_CHECK_POWER_SOURCE_CHANGE 10.0
 
 // Deprecated!
 #define KYA_MINUTES(m) (m * 60.0f)
@@ -57,6 +62,13 @@ - (instancetype)init
                    selector:@selector(batteryCapacityThresholdDidChange:)
                        name:kKYABatteryCapacityThresholdDidChangeNotification
                      object:nil];
+        // TODO search for a better way to check for power source change
+        [NSTimer scheduledTimerWithTimeInterval:KYA_CHECK_POWER_SOURCE_CHANGE
+                                         target:self
+                                       selector:@selector(checkPowerSource)
+                                       userInfo:nil
+                                        repeats:YES];
+
     }
     return self;
 }
@@ -176,6 +188,20 @@ - (void)configureUserNotificationCenter
 
 #pragma mark - Device Power Monitoring
 
+- (void)checkPowerSource
+{
+    CFTypeRef powerSourceInfo = IOPSCopyPowerSourcesInfo();
+    CFArrayRef powerSources = IOPSCopyPowerSourcesList(powerSourceInfo);
+
+    NSString *powerSourceState = [(NSDictionary *)CFBridgingRelease(IOPSGetPowerSourceDescription(powerSourceInfo, CFArrayGetValueAtIndex(powerSources, 0))) objectForKey:@KYA_POWER_SOURCE_STATE];
+    if([powerSourceState isEqualToString:@KYA_AC_POWER] && [NSUserDefaults.standardUserDefaults kya_isActivateOnPowerEnabled])
+    {
+        [self activateTimerWithTimeInterval:KYASleepWakeTimeIntervalIndefinite];
+    } else {
+        [self terminateTimer];
+    }
+}
+
 - (void)checkAndEnableBatteryOverride
 {
     Auto batteryMonitor = KYADevice.currentDevice.batteryMonitor;

KeepingYouAwake/KYALocalizedStrings.m

@@ -52,6 +52,7 @@
 #define KYA_L10N_DISABLE_MENU_BAR_ICON_HIGHLIGHT_COLOR NSLocalizedString(@"Disable menu bar icon highlight color", @"Disable menu bar icon highlight color")
 #define KYA_L10N_QUIT_ON_TIMER_EXPIRATION NSLocalizedString(@"Quit when activation duration is over", @"Quit when activation duration is over")
 #define KYA_L10N_ALLOW_DISPLAY_SLEEP NSLocalizedString(@"Allow the display to sleep", @"Allow the display to sleep")
+#define KYA_L10N_ACTIVATE_ON_POWER NSLocalizedString(@"Activate when power source is connected", @"Activate when power source is connected")
 
 #define KYA_L10N_DURATIONS_ALERT_REALLY_RESET_TITLE NSLocalizedString(@"Reset Activation Durations", @"Reset Activation Durations")
 #define KYA_L10N_DURATIONS_ALERT_REALLY_RESET_MESSAGE NSLocalizedString(@"Do you really want to reset the activation durations to the default values?", @"Do you really want to reset the activation durations to the default values?")

KeepingYouAwake/Preferences/KYAAdvancedPreferencesViewController/KYAAdvancedPreferencesViewController.m

@@ -67,6 +67,9 @@ - (void)configureAdvancedPreferences
     [preferences addObject:[[KYAPreference alloc] initWithTitle:KYA_L10N_ALLOW_DISPLAY_SLEEP
                                                     defaultsKey:KYAUserDefaultsKeyAllowDisplaySleep
                             ]];
+    [preferences addObject:[[KYAPreference alloc] initWithTitle:KYA_L10N_ACTIVATE_ON_POWER
+                                                    defaultsKey:KYAUserDefaultsKeyisActivateOnPowerEnabled
+                            ]];
 
     self.preferences = [preferences copy];
 }

Packages/KYAApplicationSupport/Sources/KYAApplicationSupport/NSUserDefaults+Keys.m

@@ -12,6 +12,7 @@
 // User Default Keys
 NSString * const KYAUserDefaultsKeyActivateOnLaunch = @"info.marcel-dierkes.KeepingYouAwake.ActivateOnLaunch";
 NSString * const KYAUserDefaultsKeyAllowDisplaySleep = @"info.marcel-dierkes.KeepingYouAwake.AllowDisplaySleep";
+NSString * const KYAUserDefaultsKeyisActivateOnPowerEnabled = @"info.marcel-dierkes.KeepingYouAwake.ActivateOnPower";
 NSString * const KYAUserDefaultsKeyMenuBarIconHighlightDisabled = @"info.marcel-dierkes.KeepingYouAwake.MenuBarIconHighlightDisabled";
 NSString * const KYAUserDefaultsKeyIsQuitOnTimerExpirationEnabled = @"info.marcel-dierkes.KeepingYouAwake.QuitOnTimerExpirationEnabled";
 
@@ -23,8 +24,10 @@
 @implementation NSUserDefaults (Keys)
 @dynamic kya_activateOnLaunch;
 @dynamic kya_allowDisplaySleep;
+@dynamic kya_activateOnPowerEnabled;
 @dynamic kya_menuBarIconHighlightDisabled;
-@dynamic kya_batteryCapacityThresholdEnabled, kya_batteryCapacityThreshold;
+@dynamic kya_batteryCapacityThresholdEnabled;
+@dynamic kya_batteryCapacityThreshold;
 @dynamic kya_lowPowerModeMonitoringEnabled;
 @dynamic kya_preReleaseUpdatesEnabled;
 @dynamic kya_quitOnTimerExpirationEnabled;
@@ -125,4 +128,16 @@ - (void)setKya_quitOnTimerExpirationEnabled:(BOOL)quitOnTimerExpirationEnabled
     [self setBool:quitOnTimerExpirationEnabled forKey:KYAUserDefaultsKeyIsQuitOnTimerExpirationEnabled];
 }
 
+#pragma mark - Activate on AC Power Enabled
+
+- (BOOL)kya_isActivateOnPowerEnabled
+{
+    return [self boolForKey:KYAUserDefaultsKeyisActivateOnPowerEnabled];
+}
+
+- (void)setKya_activateOnPowerEnabled:(BOOL)activateOnPowerEnabled
+{
+    [self setBool:activateOnPowerEnabled forKey:KYAUserDefaultsKeyisActivateOnPowerEnabled];
+}
+
 @end

Packages/KYAApplicationSupport/Sources/KYAApplicationSupport/include/KYAApplicationSupport/NSUserDefaults+Keys.h

@@ -13,6 +13,7 @@ NS_ASSUME_NONNULL_BEGIN
 // User Default Keys
 FOUNDATION_EXPORT NSString * const KYAUserDefaultsKeyActivateOnLaunch;
 FOUNDATION_EXPORT NSString * const KYAUserDefaultsKeyAllowDisplaySleep;
+FOUNDATION_EXPORT NSString * const KYAUserDefaultsKeyisActivateOnPowerEnabled;
 FOUNDATION_EXPORT NSString * const KYAUserDefaultsKeyMenuBarIconHighlightDisabled;
 FOUNDATION_EXPORT NSString * const KYAUserDefaultsKeyIsQuitOnTimerExpirationEnabled;
 FOUNDATION_EXPORT NSString * const KYAUserDefaultsKeyBatteryCapacityThresholdEnabled;
@@ -57,6 +58,10 @@ FOUNDATION_EXPORT NSString * const KYAUserDefaultsKeyPreReleaseUpdatesEnabled;
  */
 @property (nonatomic, getter=kya_isQuitOnTimerExpirationEnabled) BOOL kya_quitOnTimerExpirationEnabled;
 
+/// Returns YES if the app should activate when AC power is connected
+
+@property (nonatomic, getter=kya_isActivateOnPowerEnabled) BOOL kya_activateOnPowerEnabled;
+
 @end
 
 NS_ASSUME_NONNULL_END