M0: compilable skeleton — Kigi 0.1.0 fork surgery

Hard fork of xai-org/grok-build (Apache-2.0) re-targeted as Kigi, an unofficial Kimi Code CLI community build. Rename & identity - 72 xai-*/xai-grok-* crates -> kigi-* (explicit: xai-grok-pager-bin -> kigi-bin [binary `kigi`], xai-grok-pager -> kigi-tui; rest mechanical); ptyctl, ptyctl-cli, third_party/ unchanged; proto package xai.grok.tools.v1 -> kigi.tools.v1 - Config home ~/.kigi (KIGI_SHARE_DIR override), env prefix GROK_* -> KIGI_*, `kigi --version` carries the unofficial-community-build notice - clap identity, help text, startup banner, prompt templates rebranded (templates re-encrypted) Deletions (PRD removal list #5/#6/#7/#9/#10) - voice input (xai-grok-voice) and all TUI wiring - telemetry: Mixpanel client, external OTel stream, Sentry, OTLP layers, trace/GCS/S3 upload queues (kigi-file-utils halved), workspace upload module & dc_log, heap-profile uploader, auth-diagnostics uploader, session-analytics halves of feedback; local zero-egress observability preserved in new kigi-log crate (unified log, --debug firehose, subsystem file logs, opt-in instrumentation) - announcements (crate, remote-settings fields, TUI surfaces) - plugin marketplace (crate, sources/browse/CTA/extensions-modal tab); direct plugin install/uninstall/update via kigi-agent git_install kept - relay/gateway/assets endpoints and features (agent relay, headless relay transport, gateway bridge, LeaderEnvUrls); leader IPC socket now ~/.kigi/leader.sock + KIGI_LEADER_SOCKET, no ws-url derivation - functional types rehomed instead of deleted: PermissionMode -> kigi-config-types, McpInitStrategy -> kigi-mcp, PrCreationSource -> session signals, TerminalDiagnostics -> kigi-pager-render, agent_id -> shell util Endpoints - kigi-env rewritten: single production KigiEndpoints {coding_api_base_url https://api.kimi.com/coding/v1 (KIGI_CODE_BASE_URL), oauth_host https://auth.kimi.com (KIGI_OAUTH_HOST), update_base_url (GitHub Releases API), upgrade_page_url}; GrokBuildEnvironment enum deleted Toolchain & workspace hygiene - Rust 1.97.0 pinned; edition 2024; full cargo update; git2 hoisted to workspace at 0.21 (Option->Result API migration), quick-xml 0.41 - Root Cargo.toml hand-maintained (PRD §8.1): version 0.1.0 inherited by all members, members sorted, unused deps pruned - cargo-deny advisories gate (deny.toml with documented transitive exceptions); CI workflow (check/clippy/fmt/deny/test, macOS+Linux) - cross-crate test seams re-gated behind `test-support` cargo feature; insta snapshot baselines renamed to the kigi_tui prefix - clippy --workspace --all-targets: zero warnings; fmt clean Fixes surfaced by the port - updater probe/installer divergence (bin/kigi vs bin/grok symlink set) - idle model-metadata refresh dead under KIGI_CODE_BASE_URL override (new is_effective_coding_endpoint_url, loopback+override aware) - macOS symlinked-TMPDIR fixture canonicalization (foreign_sessions, fast-worktree); RSS measurement tests serialized via serial_test Docs & legal (Apache §4) - NOTICE added (upstream attribution + change statement); THIRD-PARTY notices sustained; kigi-tools ported-code notices extended; README, CONTRIBUTING, SECURITY, AGENTS.md rewritten Out of scope for M0 (tracked): Kimi auth/inference (M1), search/fetch, command parity, config import (M2), Computer Hub excision & final brand-token sweep (M2), distribution & self-update rewrite (M3).
2026-07-17 05:31:01 -04:00
commit d6c20fc13f
2612 changed files with 1353757 additions and 0 deletions
@@ -0,0 +1,275 @@
+//! Platform-specific sleep prevention.
+//!
+//! Prevents the machine from idle-sleeping while an agent turn is in progress.
+//! macOS uses IOKit power assertions; Linux spawns `systemd-inhibit`.
+//!
+//! Threading: lives on `AppView` (single-threaded, `!Send`).
+//! Uses `Cell`/`RefCell` instead of atomics/mutexes.
+
+use std::cell::Cell;
+
+/// Prevents idle sleep while an agent turn is running.
+///
+/// Calls are idempotent: repeated `inhibit()` or `release()` calls are no-ops
+/// when already in the requested state. On `Drop`, any held assertion is released.
+pub struct SleepInhibitor {
+    #[cfg(target_os = "macos")]
+    assertion_id: Cell<Option<u32>>,
+    #[cfg(target_os = "linux")]
+    child: std::cell::RefCell<Option<std::process::Child>>,
+    active: Cell<bool>,
+    /// Set on first `platform_inhibit` failure to avoid repeated spawn
+    /// attempts on platforms where the inhibitor is unavailable (e.g.
+    /// containers without systemd-inhibit).
+    platform_unavailable: Cell<bool>,
+    enabled: bool,
+}
+
+impl SleepInhibitor {
+    pub fn new(enabled: bool) -> Self {
+        Self {
+            #[cfg(target_os = "macos")]
+            assertion_id: Cell::new(None),
+            #[cfg(target_os = "linux")]
+            child: std::cell::RefCell::new(None),
+            active: Cell::new(false),
+            platform_unavailable: Cell::new(false),
+            enabled,
+        }
+    }
+
+    /// Prevent idle sleep. No-op if already inhibiting, disabled, or
+    /// platform support was already determined to be unavailable.
+    pub fn inhibit(&self) {
+        if !self.enabled || self.active.get() || self.platform_unavailable.get() {
+            return;
+        }
+        if self.platform_inhibit() {
+            self.active.set(true);
+        } else {
+            self.platform_unavailable.set(true);
+        }
+    }
+
+    /// Allow idle sleep again. No-op if not currently inhibiting.
+    pub fn release(&self) {
+        if !self.active.get() {
+            return;
+        }
+        self.platform_release();
+        self.active.set(false);
+    }
+
+    #[cfg(target_os = "macos")]
+    fn platform_inhibit(&self) -> bool {
+        let mut assertion_id: u32 = 0;
+        let reason = core_foundation::string::CFString::new("grok: agent turn in progress");
+        let assertion_type =
+            core_foundation::string::CFString::from_static_string("NoIdleSleepAssertion");
+
+        // IOPMAssertionCreateWithName returns kIOReturnSuccess (0) on success.
+        let result = unsafe {
+            IOPMAssertionCreateWithName(
+                assertion_type.as_concrete_TypeRef(),
+                255, // kIOPMAssertionLevelOn
+                reason.as_concrete_TypeRef(),
+                &mut assertion_id,
+            )
+        };
+
+        if result == 0 {
+            self.assertion_id.set(Some(assertion_id));
+            true
+        } else {
+            tracing::warn!(error_code = result, "failed to create IOPMAssertion");
+            false
+        }
+    }
+
+    #[cfg(target_os = "macos")]
+    fn platform_release(&self) {
+        if let Some(id) = self.assertion_id.get() {
+            let result = unsafe { IOPMAssertionRelease(id) };
+            if result != 0 {
+                tracing::warn!(error_code = result, "failed to release IOPMAssertion");
+            }
+            self.assertion_id.set(None);
+        }
+    }
+
+    #[cfg(target_os = "linux")]
+    fn platform_inhibit(&self) -> bool {
+        let mut cmd = std::process::Command::new("systemd-inhibit");
+        cmd.args([
+            "--what=idle",
+            "--who=grok",
+            "--why=agent turn in progress",
+            "sleep",
+            "infinity",
+        ])
+        .stdin(std::process::Stdio::null())
+        .stdout(std::process::Stdio::null())
+        .stderr(std::process::Stdio::null());
+        kigi_tty_utils::detach_std_command(&mut cmd);
+        let result = cmd.spawn();
+
+        match result {
+            Ok(child) => {
+                *self.child.borrow_mut() = Some(child);
+                true
+            }
+            Err(e) => {
+                tracing::debug!(error = %e, "systemd-inhibit not available");
+                false
+            }
+        }
+    }
+
+    #[cfg(target_os = "linux")]
+    fn platform_release(&self) {
+        if let Some(mut child) = self.child.borrow_mut().take() {
+            let _ = nix::sys::signal::kill(
+                nix::unistd::Pid::from_raw(child.id() as i32),
+                nix::sys::signal::Signal::SIGTERM,
+            );
+            let _ = child.wait();
+        }
+    }
+
+    #[cfg(not(any(target_os = "macos", target_os = "linux")))]
+    fn platform_inhibit(&self) -> bool {
+        false
+    }
+
+    #[cfg(not(any(target_os = "macos", target_os = "linux")))]
+    fn platform_release(&self) {}
+}
+
+impl Drop for SleepInhibitor {
+    fn drop(&mut self) {
+        self.release();
+    }
+}
+
+// -- macOS IOKit FFI ---------------------------------------------------------
+
+#[cfg(target_os = "macos")]
+use core_foundation::base::TCFType;
+
+#[cfg(target_os = "macos")]
+#[link(name = "IOKit", kind = "framework")]
+unsafe extern "C" {
+    fn IOPMAssertionCreateWithName(
+        assertion_type: core_foundation::string::CFStringRef,
+        assertion_level: u32,
+        reason_for_activity: core_foundation::string::CFStringRef,
+        assertion_id: *mut u32,
+    ) -> i32;
+
+    fn IOPMAssertionRelease(assertion_id: u32) -> i32;
+}
+
+#[cfg(test)]
+mod tests {
+    use super::*;
+
+    #[test]
+    fn disabled_inhibitor_is_noop() {
+        let inhibitor = SleepInhibitor::new(false);
+        inhibitor.inhibit();
+        assert!(!inhibitor.active.get());
+        inhibitor.release();
+        assert!(!inhibitor.active.get());
+    }
+
+    /// On unsupported platforms (not macOS/Linux), platform_inhibit returns
+    /// false and `platform_unavailable` latches to prevent retry spam.
+    #[cfg(not(any(target_os = "macos", target_os = "linux")))]
+    #[test]
+    fn platform_unavailable_prevents_retries() {
+        let inhibitor = SleepInhibitor::new(true);
+        // First attempt: platform_inhibit fails, flag latches.
+        inhibitor.inhibit();
+        assert!(!inhibitor.active.get());
+        assert!(inhibitor.platform_unavailable.get());
+        // Subsequent calls are short-circuited by the flag.
+        inhibitor.inhibit();
+        assert!(!inhibitor.active.get());
+    }
+
+    #[test]
+    fn inhibit_is_idempotent() {
+        let inhibitor = SleepInhibitor::new(true);
+        inhibitor.inhibit();
+        let was_active = inhibitor.active.get();
+        // Second call should not change state (and not spawn a second child/assertion).
+        inhibitor.inhibit();
+        assert_eq!(inhibitor.active.get(), was_active);
+        inhibitor.release();
+    }
+
+    #[test]
+    fn release_is_idempotent() {
+        let inhibitor = SleepInhibitor::new(true);
+        // Release without prior inhibit is a no-op.
+        inhibitor.release();
+        assert!(!inhibitor.active.get());
+        // Double release after inhibit is safe.
+        inhibitor.inhibit();
+        inhibitor.release();
+        assert!(!inhibitor.active.get());
+        inhibitor.release();
+        assert!(!inhibitor.active.get());
+    }
+
+    #[test]
+    fn inhibit_release_cycle() {
+        let inhibitor = SleepInhibitor::new(true);
+        inhibitor.inhibit();
+        // On Linux, this spawns systemd-inhibit; on other platforms it's a no-op.
+        // Either way, the active flag tracks the intent.
+        let was_active = inhibitor.active.get();
+        inhibitor.release();
+        assert!(!inhibitor.active.get());
+        // Can re-inhibit after release.
+        inhibitor.inhibit();
+        assert_eq!(inhibitor.active.get(), was_active);
+        inhibitor.release();
+    }
+
+    #[test]
+    fn drop_releases() {
+        let inhibitor = SleepInhibitor::new(true);
+        inhibitor.inhibit();
+        drop(inhibitor);
+        // No assertion — just verify no panic/leak.
+    }
+
+    #[cfg(target_os = "linux")]
+    #[test]
+    fn linux_inhibit_spawns_child() {
+        let inhibitor = SleepInhibitor::new(true);
+        inhibitor.inhibit();
+        if inhibitor.active.get() {
+            assert!(inhibitor.child.borrow().is_some());
+            inhibitor.release();
+            assert!(inhibitor.child.borrow().is_none());
+        }
+        // If systemd-inhibit isn't available, active stays false — that's fine.
+    }
+
+    #[cfg(target_os = "linux")]
+    #[test]
+    fn linux_release_kills_child() {
+        let inhibitor = SleepInhibitor::new(true);
+        inhibitor.inhibit();
+        if inhibitor.active.get() {
+            // Grab the pid before release.
+            let pid = inhibitor.child.borrow().as_ref().map(|c| c.id());
+            assert!(pid.is_some());
+            inhibitor.release();
+            assert!(inhibitor.child.borrow().is_none());
+            assert!(!inhibitor.active.get());
+        }
+    }
+}