Support osx idle on screensaver, screenlock

Add fah-screen-agent
Support idle/not idle notifications
Add tools.scons

Author: kbernhagen

.gitignore

@@ -21,6 +21,7 @@
 *.app
 
 # Build byproducts
+/bin
 /build
 /package.txt
 /package-description.txt

SConstruct

@@ -64,12 +64,15 @@ if env['PLATFORM'] == 'win32' or int(env.get('cross_mingw', 0)):
                               duplicate = 0)
     Default(hide_console)
 
+tools = SConscript('src/tools.scons', variant_dir = 'build', duplicate = 0)
+Default(tools)
+Depends(tools, client)
 # Clean
-Clean(client, ['build', 'config.log'])
+Clean(client, ['bin', 'build', 'config.log'])
 
 # Dist
 docs = ['README.md', 'CHANGELOG.md', 'LICENSE']
-distfiles = docs + [client, 'images/fahlogo.png']
+distfiles = docs + client + tools + ['images/fahlogo.png']
 if env['PLATFORM'] == 'posix':
     distfiles.append('install/lin/fah-client.service')
 if hide_console is not None: distfiles.append(hide_console)
@@ -122,14 +125,18 @@ if 'package' in COMMAND_LINE_TARGETS:
         # Specify components for the osx distribution pkg
         client_home = '.'
         client_root = client_home + '/build/pkg/root'
-        pkg_files = [[str(client[0]), 'usr/local/bin/', 0o755],
-                     ['build/install/osx/fahclient.url',
+        pkg_files = [['build/install/osx/fahclient.url',
                       'Applications/Folding@home/fahclient.url', 0o644],
                      ['build/install/osx/uninstall.url',
                       'Applications/Folding@home/uninstall.url', 0o644],
                      ['build/install/osx/launchd.plist',
                       'Library/LaunchDaemons/' +
-                      'org.foldingathome.fahclient.plist', 0o644]]
+                      'org.foldingathome.fahclient.plist', 0o644],
+                     ['build/install/osx/fah-screen-agent.plist','Library/LaunchAgents/' +
+                      'org.foldingathome.fah-screen-agent.plist', 0o644]]
+        for tool in client + tools:
+            pkg_files += [[str(tool), 'usr/local/bin/', 0o755]]
+
         pkg_components = [
             {
                 # name is component pkg file name and name shown in installer
@@ -145,7 +152,7 @@ if 'package' in COMMAND_LINE_TARGETS:
                 # default build/pkg/root, as per cbang config pkg module
                 'root'        : client_root,
                 # relative to root
-                'sign_tools'  : ['usr/local/bin/fah-client'],
+                'sign_tools'  : ['usr/local/bin/*'],
                 'must_close_apps': [
                     'org.foldingathome.fahviewer',
                     'org.foldingathome.fahcontrol',
@@ -220,7 +227,6 @@ if 'package' in COMMAND_LINE_TARGETS:
         pkg_background     = 'fah-opacity-50.png',
         pkg_customize      = 'always',
         pkg_target         = pkg_target,
-        pkg_arch           = env.get('package_arch', 'x86_64'),
         pkg_components     = pkg_components,
     )
 

install/osx/fah-screen-agent.plist

@@ -0,0 +1,25 @@
+<?xml version="1.0" encoding="UTF-8"?>
+<!DOCTYPE plist PUBLIC "-//Apple//DTD PLIST 1.0//EN" "http://www.apple.com/DTDs/PropertyList-1.0.dtd">
+<plist version="1.0">
+<dict>
+	<key>KeepAlive</key>
+	<dict>
+		<key>SuccessfulExit</key>
+		<false/>
+	</dict>
+	<key>Label</key>
+	<string>org.foldingathome.fah-screen-agent</string>
+	<key>LimitLoadToSessionType</key>
+	<string>Aqua</string>
+	<key>LowPriorityIO</key>
+	<true/>
+	<key>Program</key>
+	<string>/usr/local/bin/fah-screen-agent</string>
+	<key>RunAtLoad</key>
+	<true/>
+	<key>StandardOutPath</key>
+	<string>/dev/null</string>
+	<key>Umask</key>
+	<integer>18</integer>
+</dict>
+</plist>

install/osx/scripts/postinstall

@@ -11,16 +11,28 @@ chmod -R u+rwX,go-w "$RUN_DIR"
 chown -R nobody:nobody "$RUN_DIR"
 
 PLIST=/Library/LaunchDaemons/org.foldingathome.fahclient.plist
+AGENT_PLIST="/Library/LaunchAgents/org.foldingathome.fah-screen-agent.plist"
 
 "$SCRIPTS"/organize-credits.sh &
 
 # Start service
-chmod 0644 "$PLIST"
+chmod 0644 "$PLIST" "$AGENT_PLIST"
 launchctl load -w "$PLIST"
 
 # start, in case RunAtLoad is false
 launchctl start org.foldingathome.fahclient || true
 
+# restart any running agents in other sessions
+# note that this does not reload their launchd jobs
+killall -HUP fah-screen-agent || true
+
+# start agent in user's gui session
+conuser=$(/usr/bin/stat -f "%Su" /dev/console) || conuser="root"
+conuid=$(/usr/bin/id -u "$conuser") || conuid=0
+if [[ $conuid != 0 ]]; then
+  launchctl bootstrap gui/$conuid "$AGENT_PLIST" || true
+fi
+
 # Don't launch GUI if CLI install
 [ "$COMMAND_LINE_INSTALL" == "1" ] && exit 0
 

install/osx/scripts/preinstall

@@ -10,6 +10,7 @@ SCRIPTS="$(dirname "$0")"
 OLD_LAUNCHD="/Library/LaunchDaemons/FAHClient.plist"
 OLD_LAUNCHD2="/Library/LaunchDaemons/edu.stanford.folding.fahclient.plist"
 NEW_LAUNCHD="/Library/LaunchDaemons/org.foldingathome.fahclient.plist"
+AGENT_PLIST="/Library/LaunchAgents/org.foldingathome.fah-screen-agent.plist"
 
 if [ -f "$NEW_LAUNCHD" ]; then
     launchctl unload -w "$NEW_LAUNCHD" || true
@@ -23,6 +24,15 @@ if [ -f "$OLD_LAUNCHD" ]; then
     rm -f "$OLD_LAUNCHD" || true
 fi
 
+if [ -f "$AGENT_PLIST" ]; then
+  # stop agent running in console user's session and unload job
+  conuser=$(/usr/bin/stat -f "%Su" /dev/console) || conuser="root"
+  conuid=$(/usr/bin/id -u "$conuser") || conuid=0
+  if [[ $conuid != 0 ]]; then
+    launchctl bootout gui/$conuid "$AGENT_PLIST" || true
+  fi
+fi
+
 # Assuming upgrade, remove old stuff
 F1="/Applications/FAHClient.url"
 F2="/Applications/Folding@home/Web Control.url"

src/fah/client/osx/OSXOSImpl.cpp

@@ -49,6 +49,8 @@
 #include <pwd.h>
 #include <unistd.h>
 
+#include <fah/screen-agent/defines.h>
+
 using namespace FAH::Client;
 using namespace cb;
 using namespace std;
@@ -66,6 +68,9 @@ enum {
 
 
 namespace {
+  CFStringRef kScreenIdle    = CFSTR(SCREEN_IDLE_NOTIFICATION);
+  CFStringRef kScreenNotIdle = CFSTR(SCREEN_NOT_IDLE_NOTIFICATION);
+
 #pragma mark c callbacks
 
   void consoleUserCB(SCDynamicStoreRef s, CFArrayRef keys, void *info) {
@@ -94,6 +99,17 @@ namespace {
     OSXOSImpl::instance().requestExit();
   }
 
+  void screenIdleCB(CFNotificationCenterRef center, void *observer,
+                  CFNotificationName name, const void *object,
+                  CFDictionaryRef info) {
+    OSXOSImpl::instance().noteScreenIdle();
+  }
+
+  void screenNotIdleCB(CFNotificationCenterRef center, void *observer,
+                  CFNotificationName name, const void *object,
+                  CFDictionaryRef info) {
+    OSXOSImpl::instance().noteScreenNotIdle();
+  }
 }
 
 
@@ -204,7 +220,7 @@ void OSXOSImpl::run() {
 
 
 void OSXOSImpl::finishInit() {
-    LOG_DEBUG(5, "OSXOSImpl::finishInit() on thread "  << pthread_self() <<
+  LOG_DEBUG(5, "OSXOSImpl::finishInit() on thread "  << pthread_self() <<
               (pthread_main_np() ? " main" : ""));
 
   // Init display power state if registration succeeded
@@ -222,8 +238,10 @@ void OSXOSImpl::finishInit() {
 
 
 void OSXOSImpl::updateSystemIdle() {
-  bool shouldBeIdle = displayPower == kDisplayPowerOff || loginwindowIsActive ||
-    screensaverIsActive || screenIsLocked;
+  bool shouldBeIdle;
+  if (gotScreenNotIdleRecently()) shouldBeIdle = false;
+  else shouldBeIdle = displayPower == kDisplayPowerOff || loginwindowIsActive ||
+    gotScreenIdleRecently();
   if (shouldBeIdle == systemIsIdle) return;
   systemIsIdle = shouldBeIdle;
   event->activate();
@@ -483,6 +501,45 @@ bool OSXOSImpl::registerForConsoleUserNotifications() {
 }
 
 
+void OSXOSImpl::noteScreenIdle() {
+  screenIdleExpiry =
+   dispatch_time(DISPATCH_TIME_NOW, SCREEN_NOTIFICATION_EXPIRES * NSEC_PER_SEC);
+  bool wasIdle = screenIdle;
+  screenIdle = true;
+  dispatch_after(screenIdleExpiry + NSEC_PER_SEC, dispatch_get_main_queue(), ^{
+    updateSystemIdle();
+  });
+  if (!wasIdle) delayedUpdateSystemIdle(5);
+}
+
+
+void OSXOSImpl::noteScreenNotIdle() {
+  screenNotIdleExpiry =
+   dispatch_time(DISPATCH_TIME_NOW, SCREEN_NOTIFICATION_EXPIRES * NSEC_PER_SEC);
+  screenNotIdle = true;
+  dispatch_after(screenNotIdleExpiry + NSEC_PER_SEC,dispatch_get_main_queue(),^{
+    updateSystemIdle();
+  });
+  updateSystemIdle();
+}
+
+
+bool OSXOSImpl::gotScreenIdleRecently() {
+  if (!screenIdle) return false;
+  if (dispatch_time(DISPATCH_TIME_NOW, 0) < screenIdleExpiry) return true;
+  screenIdle = false;
+  return false;
+}
+
+
+bool OSXOSImpl::gotScreenNotIdleRecently() {
+  if (!screenNotIdle) return false;
+  if (dispatch_time(DISPATCH_TIME_NOW, 0) < screenNotIdleExpiry) return true;
+  screenNotIdle = false;
+  return false;
+}
+
+
 bool OSXOSImpl::registerForDarwinNotifications() {
   CFNotificationCenterRef nc = CFNotificationCenterGetDarwinNotifyCenter();
 
@@ -496,16 +553,18 @@ bool OSXOSImpl::registerForDarwinNotifications() {
   CFStringRef name =
     CFStringCreateWithCString(0, key.c_str(), kCFStringEncodingUTF8);
 
-  if (name) {
-    CFNotificationCenterAddObserver(
-      nc, (void *)this, &noteQuitCB, name, 0,
-      CFNotificationSuspensionBehaviorCoalesce);
-    CFRelease(name);
+  if (!name) return false;
+  CFNotificationCenterAddObserver(nc, (void *)this, &noteQuitCB,
+    name, 0, CFNotificationSuspensionBehaviorCoalesce);
+  CFRelease(name);
 
-    return true;
-  }
+  CFNotificationCenterAddObserver(nc, (void *)this, &screenIdleCB,
+    kScreenIdle, 0, CFNotificationSuspensionBehaviorCoalesce);
 
-  return false;
+  CFNotificationCenterAddObserver(nc, (void *)this, &screenNotIdleCB,
+    kScreenNotIdle, 0, CFNotificationSuspensionBehaviorCoalesce);
+
+  return true;
 }
 
 

src/fah/client/osx/OSXOSImpl.h

@@ -46,10 +46,15 @@ namespace FAH {
       static OSXOSImpl *singleton;
 
       std::atomic<bool> systemIsIdle = false;
-      bool screensaverIsActive = false;
-      bool screenIsLocked = false;
+
       bool loginwindowIsActive = false;
 
+      // these are notification received bools; both are needed
+      bool screenIdle = false;
+      bool screenNotIdle = false;
+      dispatch_time_t screenIdleExpiry = 0;
+      dispatch_time_t screenNotIdleExpiry = 0;
+
       io_service_t displayWrangler = 0;
       IONotificationPortRef displayNotePort = 0;
       CFRunLoopSourceRef displayNoteSource = 0;
@@ -87,11 +92,15 @@ namespace FAH {
       (void *context, io_service_t service, natural_t mtype, void *marg);
       void finishInit();
       void updateTimerFired(CFRunLoopTimerRef timer, void *info);
+      void noteScreenIdle();
+      void noteScreenNotIdle();
 
     protected:
       void initialize();
       void addHeartbeatTimerToRunLoop(CFRunLoopRef loop);
       bool registerForConsoleUserNotifications();
+      bool gotScreenIdleRecently();
+      bool gotScreenNotIdleRecently();
       bool registerForDarwinNotifications();
       bool registerForDisplayPowerNotifications();
       bool registerForLaunchEvents();

src/fah/screen-agent/defines.h

@@ -0,0 +1,39 @@
+/******************************************************************************\
+
+                  This file is part of the Folding@home Client.
+
+          The fah-client runs Folding@home protein folding simulations.
+                    Copyright (c) 2001-2023, foldingathome.org
+                               All rights reserved.
+
+       This program is free software; you can redistribute it and/or modify
+       it under the terms of the GNU General Public License as published by
+        the Free Software Foundation; either version 3 of the License, or
+                       (at your option) any later version.
+
+         This program is distributed in the hope that it will be useful,
+          but WITHOUT ANY WARRANTY; without even the implied warranty of
+          MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+                   GNU General Public License for more details.
+
+     You should have received a copy of the GNU General Public License along
+     with this program; if not, write to the Free Software Foundation, Inc.,
+           51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
+
+                  For information regarding this software email:
+                                 Joseph Coffland
+                          joseph@cauldrondevelopment.com
+
+\******************************************************************************/
+
+#pragma once
+
+// Defines shared with fah-client
+
+#define SCREEN_IDLE_NOTIFICATION     "org.foldingathome.screen.idle"
+#define SCREEN_NOT_IDLE_NOTIFICATION "org.foldingathome.screen.notidle"
+
+#define SCREEN_NOTIFICATION_INTERVAL 60
+#define SCREEN_NOTIFICATION_LEEWAY   2
+#define SCREEN_NOTIFICATION_EXPIRES  (\
+  SCREEN_NOTIFICATION_INTERVAL + 2 * SCREEN_NOTIFICATION_LEEWAY + 1)