Initial public release of Warp.

Repo-Sync-Origin: warpdotdev/warp-internal@12af1d983b
2026-04-28 08:43:33 -05:00
commit 0dbd3d567a
4982 changed files with 1431549 additions and 0 deletions
@@ -0,0 +1,42 @@
+use objc2::{rc::Retained, runtime::ProtocolObject};
+use objc2_foundation::{NSActivityOptions, NSObjectProtocol, NSProcessInfo, NSString};
+
+/// A guard object that prevents system sleep while it remains in scope.
+pub struct Guard {
+    process_info: Retained<NSProcessInfo>,
+    activity_token: Retained<ProtocolObject<dyn NSObjectProtocol>>,
+    reason: Retained<NSString>,
+}
+
+// Mark the guard as safe for being sent across threads.  We don't need to worry about thread safety
+// here because the underlying process info and activity marker can be shared across threads, and
+// we are sure that there aren't synchronization issues because we only interact with the activity
+// during creation and drop.
+unsafe impl Send for Guard {}
+unsafe impl Sync for Guard {}
+
+impl Drop for Guard {
+    fn drop(&mut self) {
+        unsafe {
+            self.process_info.endActivity(&self.activity_token);
+        }
+        log::info!("No longer preventing sleep with reason: {}", self.reason);
+    }
+}
+
+/// Returns a guard that prevents system sleep while it remains in scope.
+pub fn prevent_sleep(reason: &'static str) -> Guard {
+    let reason = NSString::from_str(reason);
+
+    let process_info = NSProcessInfo::processInfo();
+    let activity_token =
+        process_info.beginActivityWithOptions_reason(NSActivityOptions::UserInitiated, &reason);
+
+    log::info!("Preventing sleep with reason: {reason}");
+
+    Guard {
+        process_info,
+        activity_token,
+        reason,
+    }
+}