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).
This commit is contained in:
2026-07-17 05:31:01 -04:00
commit d6c20fc13f
2612 changed files with 1353757 additions and 0 deletions
@@ -0,0 +1,317 @@
//! `InnerDispatchForResolver` coverage. Verifies the cycle-safe `Weak`
//! resolver semantics and the session-bound resolution path.
use std::sync::Arc;
use dashmap::DashMap;
use async_trait::async_trait;
use futures::StreamExt;
use serde::{Deserialize, Serialize};
use kigi_computer_hub_core::{
CompoundResolver, ConnectionCleanupReport, ErasedTool, InnerDispatchForResolver, ResolvedTool,
SessionCleanupReport, ToolHandle, ToolRegistry, ToolSessionBindOutcome,
ToolSessionUnbindOutcome,
};
use kigi_tool_protocol::{
ConnectionId, RegistrationOutcome, ServerId, SessionId, ToolDefinitionMode, ToolId,
ToolRegistration, ToolServerRegistration, TransportKind, UserId,
};
use kigi_tool_runtime::{
SearchSnapshot, ServerSummary, Tool, ToolCallContext, ToolDispatch, ToolError, ToolStreamItem,
};
use kigi_tool_types::ToolDescription;
#[derive(Debug, Default, Clone, Serialize, Deserialize, schemars::JsonSchema)]
struct EchoArgs {
payload: String,
}
#[derive(Debug)]
struct EchoTool;
impl Tool for EchoTool {
type Args = EchoArgs;
type Output = serde_json::Value;
fn id(&self) -> ToolId {
ToolId::new("echo").expect("tool id")
}
fn description(&self, _ctx: &::kigi_tool_runtime::ListToolsContext) -> ToolDescription {
ToolDescription::new("echo", "Echoes its input.")
}
async fn run(
&self,
_ctx: ToolCallContext,
args: Self::Args,
) -> Result<Self::Output, ToolError> {
Ok(serde_json::json!({"echoed": args.payload}))
}
}
type RegistryEntry = (ToolRegistration, Arc<dyn ToolHandle>);
#[derive(Debug, Default)]
struct InMemRegistry {
entries: DashMap<(SessionId, ToolId), RegistryEntry>,
}
impl InMemRegistry {
fn install(&self, session: SessionId, tool: ToolId, handle: Arc<dyn ToolHandle>) {
let registration = ToolRegistration {
tool_id: tool.clone(),
sessions: Some(vec![session.clone()]),
user_id: UserId::new("alice").expect("user id"),
server_id: None,
description: handle.description(&kigi_tool_runtime::ListToolsContext::default()),
input_schema: None,
capabilities: Some(handle.capabilities()),
notification_schemas: None,
transport_kind: TransportKind::Local,
if_match_generation: None,
metadata: None,
};
self.entries.insert((session, tool), (registration, handle));
}
}
#[async_trait]
impl ToolRegistry for InMemRegistry {
async fn register_tool(
&self,
_connection_id: ConnectionId,
_reg: ToolRegistration,
) -> RegistrationOutcome {
unreachable!()
}
async fn register_server(
&self,
_connection_id: ConnectionId,
_reg: ToolServerRegistration,
) -> Vec<RegistrationOutcome> {
unreachable!()
}
async fn unregister_tool(&self, _connection_id: &ConnectionId, _tool: &ToolId) -> bool {
unreachable!()
}
async fn unregister_server(&self, _connection_id: &ConnectionId, _server: &ServerId) -> usize {
unreachable!()
}
async fn bind_tool_session(
&self,
_connection_id: &ConnectionId,
_tool: &ToolId,
_session_id: &SessionId,
) -> ToolSessionBindOutcome {
unreachable!()
}
async fn unbind_tool_session(
&self,
_connection_id: &ConnectionId,
_tool: &ToolId,
_session_id: &SessionId,
) -> ToolSessionUnbindOutcome {
unreachable!()
}
async fn drop_connection(&self, _connection_id: &ConnectionId) -> ConnectionCleanupReport {
ConnectionCleanupReport::default()
}
fn find_tool(&self, session: &SessionId, tool: &ToolId) -> Option<ResolvedTool> {
let (registration, handle) = self
.entries
.get(&(session.clone(), tool.clone()))?
.value()
.clone();
Some(ResolvedTool::Local {
tool: handle,
registration,
})
}
fn list_tools(&self, _session: &SessionId, _mode: &ToolDefinitionMode) -> Vec<ToolDescription> {
vec![]
}
fn list_servers(&self, _session: &SessionId) -> Vec<ServerSummary> {
vec![]
}
fn search(&self, _session: &SessionId, _query: &str, _limit: usize) -> SearchSnapshot {
SearchSnapshot {
results: vec![],
total_hidden_tools: 0,
is_ready: true,
}
}
async fn unregister_session(&self, _session: &SessionId) -> SessionCleanupReport {
SessionCleanupReport::default()
}
fn tool_sessions(
&self,
_connection_id: &ConnectionId,
_tool: &ToolId,
) -> std::collections::HashSet<SessionId> {
std::collections::HashSet::new()
}
fn list_servers_for_user(
&self,
_user_id: &kigi_tool_protocol::UserId,
) -> Vec<kigi_computer_hub_core::registry::ServerRecord> {
Vec::new()
}
fn get_server_record(
&self,
_connection_id: &ConnectionId,
) -> Option<kigi_computer_hub_core::registry::ServerRecord> {
None
}
}
fn sid(s: &str) -> SessionId {
SessionId::new(s).expect("session id")
}
fn tid(s: &str) -> ToolId {
ToolId::new(s).expect("tool id")
}
#[tokio::test]
async fn inner_dispatch_resolves_through_bound_session() {
let registry = Arc::new(InMemRegistry::default());
registry.install(
sid("sess-1"),
tid("echo"),
Arc::new(ErasedTool::new(EchoTool)),
);
let resolver = Arc::new(CompoundResolver::local_only(
registry as Arc<dyn ToolRegistry>,
));
let inner = InnerDispatchForResolver::new(Arc::downgrade(&resolver), sid("sess-1"));
assert_eq!(inner.session_id(), &sid("sess-1"));
let result = inner
.call_terminal(
tid("echo"),
serde_json::json!({"payload": "x"}),
ToolCallContext::default(),
)
.await
.expect("terminal ok");
assert_eq!(result.value, serde_json::json!({"echoed": "x"}));
}
#[tokio::test]
async fn inner_dispatch_returns_not_found_when_tool_absent() {
let registry = Arc::new(InMemRegistry::default());
let resolver = Arc::new(CompoundResolver::local_only(
registry as Arc<dyn ToolRegistry>,
));
let inner = InnerDispatchForResolver::new(Arc::downgrade(&resolver), sid("sess-1"));
let mut stream = inner
.call(
tid("ghost"),
serde_json::json!(null),
ToolCallContext::default(),
)
.await;
let item = stream.next().await.expect("terminal");
match item {
ToolStreamItem::Terminal(Err(ref e))
if e.kind == kigi_tool_runtime::ToolErrorKind::NotFound =>
{
assert!(
e.detail.contains("ghost"),
"detail should mention tool id: {}",
e.detail
);
}
other => panic!("expected Terminal(NotFound), got {other:?}"),
}
}
#[tokio::test]
async fn inner_dispatch_uses_bound_session_not_context_session() {
// Even if the context were to carry a different session, the inner
// dispatch handle resolves against its construction-time session.
let registry = Arc::new(InMemRegistry::default());
registry.install(
sid("sess-A"),
tid("echo"),
Arc::new(ErasedTool::new(EchoTool)),
);
let resolver = Arc::new(CompoundResolver::local_only(
registry as Arc<dyn ToolRegistry>,
));
let inner = InnerDispatchForResolver::new(Arc::downgrade(&resolver), sid("sess-B"));
let mut stream = inner
.call(
tid("echo"),
serde_json::json!({"payload": "x"}),
ToolCallContext::default(),
)
.await;
let item = stream.next().await.expect("terminal");
match item {
ToolStreamItem::Terminal(Err(ref e))
if e.kind == kigi_tool_runtime::ToolErrorKind::NotFound => {}
other => panic!("session-A registration must not be visible from session-B, got {other:?}"),
}
}
#[tokio::test]
async fn inner_dispatch_after_resolver_drop_returns_computer_hub_dropped() {
let registry = Arc::new(InMemRegistry::default());
let resolver = Arc::new(CompoundResolver::local_only(
registry as Arc<dyn ToolRegistry>,
));
let weak = Arc::downgrade(&resolver);
let inner = InnerDispatchForResolver::new(weak, sid("sess-1"));
drop(resolver);
let mut stream = inner
.call(
tid("echo"),
serde_json::json!(null),
ToolCallContext::default(),
)
.await;
let item = stream.next().await.expect("terminal");
match item {
ToolStreamItem::Terminal(Err(ref e))
if e.kind == kigi_tool_runtime::ToolErrorKind::Custom =>
{
assert!(
e.detail.contains("computer_hub_dropped")
|| e.details
.as_ref()
.and_then(|d| d.get("code"))
.and_then(|c| c.as_str())
== Some("computer_hub_dropped"),
"expected computer_hub_dropped code, got: {:?}",
e
);
}
other => panic!("expected Terminal(Custom(computer_hub_dropped)), got {other:?}"),
}
}
#[tokio::test]
async fn inner_dispatch_implements_object_safe_tool_dispatch() {
let registry = Arc::new(InMemRegistry::default());
let resolver = Arc::new(CompoundResolver::local_only(
registry as Arc<dyn ToolRegistry>,
));
let inner: Arc<dyn ToolDispatch> = Arc::new(InnerDispatchForResolver::new(
Arc::downgrade(&resolver),
sid("sess-1"),
));
let result = inner
.call_terminal(
tid("ghost"),
serde_json::json!(null),
ToolCallContext::default(),
)
.await;
assert!(matches!(result, Err(ref e) if e.kind == kigi_tool_runtime::ToolErrorKind::NotFound));
}