feat: added another QOL change to prevent sleep while codify apply or destroy is running

Author: kevinwang5658

src/commands/apply.ts

@@ -34,6 +34,10 @@ For more information, visit: https://codifycli.com/docs/commands/apply
       char: 'v',
       description: 'Print plugin output (stdout/stderr) to the terminal.',
     }),
+    'allow-sleep': Flags.boolean({
+      description: 'Allow the system to sleep during apply operations.',
+      default: false,
+    }),
   }
 
   static args = {
@@ -58,6 +62,7 @@ For more information, visit: https://codifycli.com/docs/commands/apply
       path: flags.path ?? args.pathArgs,
       verbosityLevel: flags.debug || flags.verbose ? 3 : 0,
       autoApprove: flags.yes,
+      allowSleep: flags['allow-sleep'],
       // secure: flags.secure,
     }, this.reporter);
 

src/commands/destroy.ts

@@ -42,6 +42,10 @@ For more information, visit: https://codifycli.com/docs/commands/destory`
       char: 'v',
       description: 'Print plugin output (stdout/stderr) to the terminal.',
     }),
+    'allow-sleep': Flags.boolean({
+      description: 'Allow the system to sleep during destroy operations.',
+      default: false,
+    }),
   }
 
   public async run(): Promise<void> {
@@ -56,6 +60,7 @@ For more information, visit: https://codifycli.com/docs/commands/destory`
       typeIds: args,
       path: flags.path,
       autoApprove: flags.yes,
+      allowSleep: flags['allow-sleep'],
     }, this.reporter)
 
     process.exit(0);

src/orchestrators/apply.ts

@@ -1,6 +1,7 @@
 import { ProcessName, ctx } from '../events/context.js';
 import { DefaultReporter } from '../ui/reporters/default-reporter.js';
 import { Reporter } from '../ui/reporters/reporter.js';
+import { SleepInhibitor } from '../utils/sleep-inhibitor.js';
 import { VerbosityLevel } from '../utils/verbosity-level.js';
 import { PlanOrchestrator } from './plan.js';
 
@@ -10,6 +11,7 @@ export interface ApplyArgs {
   verbosityLevel?: number;
   noProgress?: boolean;
   autoApprove?: boolean;
+  allowSleep?: boolean;
 }
 
 export const ApplyOrchestrator = {
@@ -40,17 +42,27 @@ export const ApplyOrchestrator = {
       });
     }
 
-    if (!args.noProgress) ctx.processStarted(ProcessName.APPLY);
-    if (!args.noProgress) await reporter.displayProgress();
+    const inhibitor = args.allowSleep ? null : SleepInhibitor.start();
+    if (inhibitor && reporter instanceof DefaultReporter) {
+      reporter.setSleepPrevented(true);
+    }
+
+    try {
+      if (!args.noProgress) ctx.processStarted(ProcessName.APPLY);
+      if (!args.noProgress) await reporter.displayProgress();
 
-    const applyResult = await pluginManager.apply(project, filteredPlan);
+      const applyResult = await pluginManager.apply(project, filteredPlan);
 
-    if (!args.noProgress) ctx.processFinished(ProcessName.APPLY);
+      if (!args.noProgress) ctx.processFinished(ProcessName.APPLY);
 
-    await reporter.displayApplyComplete(applyResult);
+      await reporter.displayApplyComplete(applyResult);
 
-    if (applyResult.isPartialFailure()) {
-      process.exit(1);
+      if (applyResult.isPartialFailure()) {
+        process.exit(1);
+      }
+    } finally {
+      inhibitor?.stop();
     }
+
   },
 };

src/orchestrators/destroy.ts

@@ -7,6 +7,7 @@ import { ProcessName, SubProcessName, ctx } from '../events/context.js';
 import { PluginManager, ResourceDefinitionMap } from '../plugins/plugin-manager.js';
 import { DefaultReporter } from '../ui/reporters/default-reporter.js';
 import { PromptType, Reporter } from '../ui/reporters/reporter.js';
+import { SleepInhibitor } from '../utils/sleep-inhibitor.js';
 import { wildCardMatch } from '../utils/wild-card-match.js';
 
 export interface DestroyArgs {
@@ -15,6 +16,7 @@ export interface DestroyArgs {
   secureMode?: boolean;
   verbosityLevel?: number;
   autoApprove?: boolean;
+  allowSleep?: boolean;
 }
 
 export class DestroyOrchestrator {
@@ -69,15 +71,24 @@ export class DestroyOrchestrator {
       });
     }
 
-    await reporter.displayProgress();
-    const applyResult = await ctx.process(ProcessName.DESTROY, () =>
-      pluginManager.apply(destroyProject, filteredPlan)
-    )
+    const inhibitor = args.allowSleep ? null : SleepInhibitor.start();
+    if (inhibitor && reporter instanceof DefaultReporter) {
+      reporter.setSleepPrevented(true);
+    }
 
-    await reporter.displayApplyComplete(applyResult);
+    try {
+      await reporter.displayProgress();
+      const applyResult = await ctx.process(ProcessName.DESTROY, () =>
+        pluginManager.apply(destroyProject, filteredPlan)
+      )
 
-    if (applyResult.isPartialFailure()) {
-      process.exit(1);
+      await reporter.displayApplyComplete(applyResult);
+
+      if (applyResult.isPartialFailure()) {
+        process.exit(1);
+      }
+    } finally {
+      inhibitor?.stop();
     }
   }
 

src/ui/components/progress/progress-display.tsx

@@ -30,6 +30,7 @@ export function ProgressDisplay(props: { emitter: EventEmitter }) {
   const [progress] = useAtom(store.progressState);
   const [isVerbose, setIsVerbose] = useState(() => VerbosityLevel.get() > 0);
   const [passwordSaved] = useAtom(store.isSudoPasswordCached);
+  const [sleepPrevented] = useAtom(store.isSleepPrevented);
 
   const isApplyOrDestroy = progress?.name === ProcessName.APPLY || progress?.name === ProcessName.DESTROY;
 
@@ -66,6 +67,7 @@ export function ProgressDisplay(props: { emitter: EventEmitter }) {
           ? <Text color="green">✓ sudo password</Text>
           : <Text dimColor>[p] Enter sudo password</Text>
         }
+        {sleepPrevented && <Text color="green">✓ sleep prevented</Text>}
       </Box>
     )}
   </Box>

src/ui/reporters/default-reporter.tsx

@@ -85,6 +85,10 @@ export class DefaultReporter implements Reporter {
     store.set(store.isSudoPasswordCached, true);
   }
 
+  setSleepPrevented(value: boolean): void {
+    store.set(store.isSleepPrevented, value);
+  }
+
   async promptPressKeyToContinue(message?: string): Promise<void> {
     const previousRenderState = this.getRenderState();
 

src/ui/store/index.ts

@@ -36,6 +36,7 @@ export const store = new class {
 
   progressState = atom(null as ProgressState | null)
   isSudoPasswordCached = atom(false)
+  isSleepPrevented = atom(false)
 
   get<Value>(atom: Atom<Value>): Value {
     return this.internal.get(atom);

src/utils/sleep-inhibitor.ts

@@ -0,0 +1,42 @@
+import { spawn, type ChildProcess } from 'node:child_process';
+
+import { OsUtils } from './os-utils.js';
+
+export class SleepInhibitor {
+  private process: ChildProcess;
+
+  private constructor(process: ChildProcess) {
+    this.process = process;
+  }
+
+  static start(): SleepInhibitor | null {
+    let child: ChildProcess;
+
+    if (OsUtils.isMacOS()) {
+      child = spawn('caffeinate', ['-i'], { stdio: 'ignore', detached: false });
+    } else if (OsUtils.isLinux()) {
+      child = spawn(
+        'systemd-inhibit',
+        ['--what=sleep', '--why=Codify apply in progress', '--mode=block', 'sleep', 'infinity'],
+        { stdio: 'ignore', detached: false }
+      );
+    } else {
+      return null;
+    }
+
+    child.unref();
+
+    const inhibitor = new SleepInhibitor(child);
+    for (const sig of ['exit', 'SIGINT', 'SIGTERM', 'uncaughtException']) {
+      process.on(sig, () => inhibitor.stop());
+    }
+
+    return inhibitor;
+  }
+
+  stop(): void {
+    if (!this.process.killed) {
+      this.process.kill();
+    }
+  }
+}