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,148 @@
+//! Actor-internal state.
+//!
+//! All fields are touched only from the actor task, so no mutex /
+//! atomic synchronization is needed -- the actor's command-loop
+//! serialization gives us a "single-threaded with shared state"
+//! discipline matching the hunk-tracker pattern.
+
+use std::collections::HashMap;
+
+use tokio_util::sync::CancellationToken;
+
+use crate::config::{RetryPolicy, SamplerConfig};
+use crate::types::RequestId;
+
+/// In-flight request bookkeeping.
+///
+/// `cancel_token` is owned by the actor (cloned into the spawned
+/// per-request task). The completion oneshot is moved into the
+/// per-request task at spawn time and is therefore not stored here.
+pub(crate) struct ActiveRequest {
+    pub(crate) cancel_token: CancellationToken,
+}
+
+/// Actor-owned state.
+pub(crate) struct ActorState {
+    pub(crate) active_requests: HashMap<RequestId, ActiveRequest>,
+    pub(crate) config: SamplerConfig,
+    pub(crate) retry_policy: RetryPolicy,
+}
+
+impl ActorState {
+    pub(crate) fn new(config: SamplerConfig, retry_policy: RetryPolicy) -> Self {
+        Self {
+            active_requests: HashMap::new(),
+            config,
+            retry_policy,
+        }
+    }
+
+    /// Register a newly-spawned request. Returns the previous entry if
+    /// the same `request_id` was already in flight (callers should
+    /// cancel the previous token before overwriting).
+    pub(crate) fn register(
+        &mut self,
+        request_id: RequestId,
+        active: ActiveRequest,
+    ) -> Option<ActiveRequest> {
+        self.active_requests.insert(request_id, active)
+    }
+
+    /// Remove a request from the active set without cancelling its
+    /// token. Used by the cleanup signal sent from per-request tasks
+    /// when they exit normally.
+    pub(crate) fn remove(&mut self, request_id: &RequestId) -> Option<ActiveRequest> {
+        self.active_requests.remove(request_id)
+    }
+
+    /// Cancel and remove an in-flight request.
+    pub(crate) fn cancel(&mut self, request_id: &RequestId) -> bool {
+        if let Some(active) = self.active_requests.remove(request_id) {
+            active.cancel_token.cancel();
+            true
+        } else {
+            false
+        }
+    }
+
+    /// Replace the default config. The next request submitted without
+    /// an override will use this.
+    pub(crate) fn update_config(&mut self, config: SamplerConfig) {
+        self.config = config;
+    }
+}
+
+#[cfg(test)]
+mod tests {
+    use super::*;
+    use crate::client::ApiBackend;
+    use indexmap::IndexMap;
+
+    /// Minimal config builder for tests in this module.
+    fn cfg() -> SamplerConfig {
+        SamplerConfig {
+            api_key: None,
+            base_url: "https://example.test".into(),
+            model: "test-model".into(),
+            max_completion_tokens: None,
+            temperature: None,
+            top_p: None,
+            api_backend: ApiBackend::ChatCompletions,
+            auth_scheme: Default::default(),
+            extra_headers: IndexMap::new(),
+            context_window: 8192,
+            force_http1: false,
+            max_retries: None,
+            stream_tool_calls: false,
+            idle_timeout_secs: None,
+            reasoning_effort: None,
+            origin_client: None,
+            client_identifier: None,
+            deployment_id: None,
+            user_id: None,
+            client_version: None,
+            attribution_callback: None,
+            bearer_resolver: None,
+            supports_backend_search: false,
+            compactions_remaining: None,
+            compaction_at_tokens: None,
+            doom_loop_recovery: None,
+            header_injector: None,
+        }
+    }
+
+    #[test]
+    fn cancel_unknown_request_returns_false() {
+        let mut state = ActorState::new(cfg(), RetryPolicy::default());
+        assert!(!state.cancel(&RequestId::from("unknown")));
+    }
+
+    #[test]
+    fn register_then_cancel_removes() {
+        let mut state = ActorState::new(cfg(), RetryPolicy::default());
+        let id = RequestId::from("req-1");
+        state.register(
+            id.clone(),
+            ActiveRequest {
+                cancel_token: CancellationToken::new(),
+            },
+        );
+        assert_eq!(state.active_requests.len(), 1);
+        assert!(state.cancel(&id));
+        assert_eq!(state.active_requests.len(), 0);
+    }
+
+    #[test]
+    fn register_returns_previous_when_same_id() {
+        let mut state = ActorState::new(cfg(), RetryPolicy::default());
+        let id = RequestId::from("req-1");
+        let first = ActiveRequest {
+            cancel_token: CancellationToken::new(),
+        };
+        let second = ActiveRequest {
+            cancel_token: CancellationToken::new(),
+        };
+        assert!(state.register(id.clone(), first).is_none());
+        assert!(state.register(id.clone(), second).is_some());
+    }
+}