temp: try to walk the control center looking for "Focus"

Author: cmorten

src/macOS/enableDoNotDisturb.ts

@@ -37,68 +37,11 @@ set command to "
 my withTimeout(command, timeoutSeconds)
 `;
 
-const enableFocusModeVenturaAppleScript = (
-  locale: string,
-  platformMajor: number
-) => {
+const enableFocusModeVenturaAppleScript = (locale: string) => {
   // TODO: attempt to rewrite scripts without any locale specific instructions
   // so this setup can be used regardless of user locale preferences.
   const center = locale.trim() === "en_GB" ? "Centre" : "Center";
 
-  if (platformMajor >= 25) {
-    return `-- Startup delay to reduce chance of "Application isn't running (-600)" errors
-delay 1
-
-set timeoutSeconds to 30.0
-
-tell application "System Events"
-  -- open Control Center using the same keystroke used in your script
-  key down 63
-  keystroke "c"
-  delay 0.3
-  key up "c"
-  key up 63
-  delay 0.5
-  try
-    tell application process "Control Center"
-      set root to window 1
-      my dumpElements(root, 0)
-    end tell
-  on error errMsg
-    log "ERROR: " & errMsg
-  end try
-end tell
-
-on dumpElements(el, indent)
-  try
-    set r to (role of el) as string
-  on error
-    set r to ""
-  end try
-  try
-    set n to (name of el) as string
-  on error
-    set n to ""
-  end try
-  try
-    set idx to (index of el) as string
-  on error
-    set idx to ""
-  end try
-  set prefix to ""
-  repeat indent times
-    set prefix to prefix & "  "
-  end repeat
-  log prefix & r & " | " & n & " | index:" & idx
-  try
-    set children to UI elements of el
-    repeat with c in children
-      my dumpElements(c, indent + 1)
-    end repeat
-  end try
-end dumpElements`;
-  }
-
   return `-- Startup delay to reduce chance of "Application isn't running (-600)" errors
 delay 1
 
@@ -144,6 +87,77 @@ my withTimeout(command, timeoutSeconds)
 `;
 };
 
+const enableFocusModeTahoeAppleScript = (locale: string) => {
+  // TODO: attempt to rewrite scripts without any locale specific instructions
+  // so this setup can be used regardless of user locale preferences.
+  const center = locale.trim() === "en_GB" ? "Centre" : "Center";
+
+  return `-- Startup delay to reduce chance of "Application isn't running (-600)" errors
+delay 1
+
+on findDoNotDisturb(e)
+  tell application "System Events"
+    try
+      log ("Checking element: " & (name of e) & " (" & (role of e) & ")")
+
+      if name of e is "Focus" then
+        return e
+      end if
+    end try
+
+    try
+      repeat with k in UI elements of e
+        set found to my findDoNotDisturb(k)
+        if found is not missing value then return found
+      end repeat
+    end try
+  end tell
+
+  return missing value
+end findDoNotDisturb
+
+-- Open Control Center drop down
+tell application "System Events"
+  key down 63
+  keystroke "c"
+  delay 0.5
+  key up "c"
+  key up 63
+end tell
+
+tell application "System Events"
+  tell process "Control ${center}"
+    set roots to UI elements of window 1
+
+    repeat with r in roots
+      set doNotDisturbCheckbox to my findDoNotDisturb(r)
+      if doNotDisturbCheckbox is not missing value then exit repeat
+    end repeat
+
+    -- Toggle Do Not Disturb
+    if doNotDisturbCheckbox is not missing value then
+      set doNotDisturbCheckboxStatus to value of doNotDisturbCheckbox as boolean
+
+      tell doNotDisturbCheckbox
+        if doNotDisturbCheckboxStatus is false then
+          perform action 1 of doNotDisturbCheckbox
+        end if
+      end tell
+    end if
+  end tell
+end tell
+
+-- Close Control Center drop down
+tell application "System Events"
+  key down 63
+  key down "c"
+  delay 0.5
+  key up "c"
+  key up 63
+end tell
+`;
+};
+
 export async function enableDoNotDisturb() {
   const platformMajor = Number(os.version().split("Version ")[1].split(".")[0]);
 
@@ -153,12 +167,21 @@ export async function enableDoNotDisturb() {
     } else if (platformMajor === 21) {
       // From MacOS 12 Monterey (Darwin 21) there is no known way to enable DND via system defaults
       await retryOnError(() => runAppleScript(enableFocusModeAppleScript));
-    } else {
+    } else if (platformMajor < 25) {
       const { stdout: locale } = await promisify(exec)(getLocale);
 
       // From MacOS 13 Ventura (Darwin 22) there is no known way to enable DND via system settings
       await retryOnError(() =>
-        runAppleScript(enableFocusModeVenturaAppleScript(locale, platformMajor))
+        runAppleScript(enableFocusModeVenturaAppleScript(locale))
+      );
+    } else {
+      const { stdout: locale } = await promisify(exec)(getLocale);
+
+      // From MacOS 26 Tahoe (Darwin 25) there is no known way to enable DND via system settings
+      await retryOnError(() =>
+        runAppleScript(enableFocusModeTahoeAppleScript(locale), {
+          wrapped: false,
+        })
       );
     }
   } catch (e) {

src/macOS/runAppleScript.ts

@@ -5,7 +5,13 @@ export const DEFAULT_MAX_BUFFER = 1000 * 1000 * 100;
 
 export async function runAppleScript<T = string | void>(
   script: string,
-  { timeout = DEFAULT_TIMEOUT } = { timeout: DEFAULT_TIMEOUT }
+  {
+    timeout = DEFAULT_TIMEOUT,
+    wrapped = true,
+  }: { timeout?: number; wrapped?: boolean } = {
+    timeout: DEFAULT_TIMEOUT,
+    wrapped: true,
+  }
 ): Promise<T> {
   const scriptWithTimeout = `
 with timeout of ${timeout} seconds
@@ -38,7 +44,11 @@ end doWithTimeout
       {
         maxBuffer: DEFAULT_MAX_BUFFER,
       },
-      (error, stdout) => {
+      (error, stdout, stderr) => {
+        console.log(error);
+        console.log(stdout);
+        console.log(stderr);
+
         if (error) {
           return reject(error);
         }
@@ -51,7 +61,7 @@ end doWithTimeout
       }
     );
 
-    child.stdin.write(scriptWithTimeout);
+    child.stdin.write(wrapped ? scriptWithTimeout : script);
     child.stdin.end();
   })) as unknown as T;
 }