nightytidy 0.2.6 → 0.2.7

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "nightytidy",
-  "version": "0.2.6",
+  "version": "0.2.7",
   "description": "Automated overnight codebase improvement through Claude Code",
   "license": "MIT",
   "author": "Dorian Spitz",

package/src/agent/index.js

@@ -12,6 +12,7 @@ import { WebhookDispatcher } from './webhook-dispatcher.js';
 import { CliBridge } from './cli-bridge.js';
 import { AgentGit } from './git-integration.js';
 import { FirebaseAuth } from './firebase-auth.js';
+import { acquireKeepAwake, releaseKeepAwake } from './keep-awake.js';
 
 const FIREBASE_WEBHOOK_URL = 'https://webhookingest-24h6taciuq-uc.a.run.app';
 
@@ -479,7 +480,10 @@ export async function startAgent() {
 
   async function processQueue() {
     const run = runQueue.dequeue();
-    if (!run) return;
+    if (!run) {
+      releaseKeepAwake();
+      return;
+    }
 
     const project = projectManager.getProject(run.projectId);
     if (!project) {
@@ -497,6 +501,7 @@ export async function startAgent() {
     runOutputBuffer = '';
     runProgress = { stepList: [], completedCount: 0, failedCount: 0, totalCost: 0, currentStepNum: null };
 
+    acquireKeepAwake();
     info(`\n━━━ Run started: ${project.name} ━━━`);
     info(`  Steps: [${run.steps.join(', ')}] (${run.steps.length} total)`);
     info(`  Project: ${project.path}`);
@@ -1036,6 +1041,7 @@ export async function startAgent() {
   // Graceful shutdown
   const shutdown = async () => {
     info('Agent shutting down...');
+    releaseKeepAwake();
     saveInterruptedState();
     scheduler.stopAll();
     await wsServer.stop();

package/src/agent/keep-awake.js

@@ -0,0 +1,60 @@
+/**
+ * Prevents the OS from sleeping while NightyTidy runs are active.
+ *
+ * Windows: Uses PowerShell to call SetThreadExecutionState with
+ * ES_CONTINUOUS | ES_SYSTEM_REQUIRED (0x80000001). This tells Windows
+ * "don't sleep, this process needs the system." The flag is automatically
+ * cleared when the process exits or when releaseKeepAwake() is called.
+ *
+ * macOS/Linux: Uses caffeinate / systemd-inhibit respectively.
+ *
+ * No admin privileges required on any platform.
+ */
+import { execSync, spawn } from 'node:child_process';
+import { debug, warn } from '../logger.js';
+
+let keepAwakeProcess = null;
+
+export function acquireKeepAwake() {
+  if (keepAwakeProcess) return; // already held
+
+  try {
+    if (process.platform === 'win32') {
+      // PowerShell script that sets ES_CONTINUOUS | ES_SYSTEM_REQUIRED
+      // and then sleeps forever. When we kill this process, the flag clears.
+      keepAwakeProcess = spawn('powershell', [
+        '-NoProfile', '-WindowStyle', 'Hidden', '-Command',
+        `Add-Type -TypeDefinition 'using System; using System.Runtime.InteropServices; public class SleepPreventer { [DllImport("kernel32.dll")] public static extern uint SetThreadExecutionState(uint esFlags); }'; [SleepPreventer]::SetThreadExecutionState(0x80000001); while($true) { Start-Sleep -Seconds 3600 }`,
+      ], { stdio: 'ignore', detached: false });
+      keepAwakeProcess.unref();
+      keepAwakeProcess.on('error', () => { keepAwakeProcess = null; });
+      debug('Sleep prevention acquired (Windows SetThreadExecutionState)');
+    } else if (process.platform === 'darwin') {
+      // macOS: caffeinate prevents sleep, -i = idle sleep, -s = system sleep
+      keepAwakeProcess = spawn('caffeinate', ['-is'], { stdio: 'ignore' });
+      keepAwakeProcess.unref();
+      keepAwakeProcess.on('error', () => { keepAwakeProcess = null; });
+      debug('Sleep prevention acquired (macOS caffeinate)');
+    } else {
+      // Linux: systemd-inhibit (may not exist on all distros)
+      keepAwakeProcess = spawn('systemd-inhibit', [
+        '--what=idle:sleep', '--who=NightyTidy', '--why=Running codebase improvement',
+        'sleep', 'infinity',
+      ], { stdio: 'ignore' });
+      keepAwakeProcess.unref();
+      keepAwakeProcess.on('error', () => { keepAwakeProcess = null; });
+      debug('Sleep prevention acquired (Linux systemd-inhibit)');
+    }
+  } catch {
+    warn('Could not acquire sleep prevention — system may sleep during runs');
+  }
+}
+
+export function releaseKeepAwake() {
+  if (!keepAwakeProcess) return;
+  try {
+    keepAwakeProcess.kill();
+  } catch { /* already dead */ }
+  keepAwakeProcess = null;
+  debug('Sleep prevention released');
+}