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:
@@ -0,0 +1,177 @@
|
||||
//! `Tool::should_list` predicate + `ToolDyn` blanket forwarding.
|
||||
|
||||
use std::sync::Arc;
|
||||
|
||||
use schemars::JsonSchema;
|
||||
use serde::{Deserialize, Serialize};
|
||||
|
||||
use kigi_tool_protocol::ToolId;
|
||||
use kigi_tool_runtime::{
|
||||
ArcTool, Cwd, ListToolsContext, Tool, ToolCallContext, ToolDyn, ToolError, ToolOutput,
|
||||
};
|
||||
use kigi_tool_types::ToolDescription;
|
||||
|
||||
#[derive(Deserialize, JsonSchema)]
|
||||
struct NoArgs {}
|
||||
|
||||
#[derive(Serialize)]
|
||||
struct Unit;
|
||||
|
||||
impl ToolOutput for Unit {}
|
||||
struct AlwaysTool;
|
||||
|
||||
impl Tool for AlwaysTool {
|
||||
type Args = NoArgs;
|
||||
type Output = Unit;
|
||||
fn id(&self) -> ToolId {
|
||||
ToolId::new("always").unwrap()
|
||||
}
|
||||
fn description(&self, _ctx: &::kigi_tool_runtime::ListToolsContext) -> ToolDescription {
|
||||
ToolDescription::new("always", "a")
|
||||
}
|
||||
async fn run(&self, _: ToolCallContext, _: NoArgs) -> Result<Unit, ToolError> {
|
||||
Ok(Unit)
|
||||
}
|
||||
}
|
||||
|
||||
struct NeedsCwdTool;
|
||||
|
||||
impl Tool for NeedsCwdTool {
|
||||
type Args = NoArgs;
|
||||
type Output = Unit;
|
||||
fn id(&self) -> ToolId {
|
||||
ToolId::new("needs_cwd").unwrap()
|
||||
}
|
||||
fn description(&self, _ctx: &::kigi_tool_runtime::ListToolsContext) -> ToolDescription {
|
||||
ToolDescription::new("needs_cwd", "a")
|
||||
}
|
||||
fn should_list(&self, ctx: &ListToolsContext) -> bool {
|
||||
ctx.extensions.contains::<Cwd>()
|
||||
}
|
||||
async fn run(&self, _: ToolCallContext, _: NoArgs) -> Result<Unit, ToolError> {
|
||||
Ok(Unit)
|
||||
}
|
||||
}
|
||||
|
||||
#[derive(Clone, Debug)]
|
||||
struct AttachmentCount(usize);
|
||||
|
||||
struct NeedsAttachmentTool;
|
||||
|
||||
impl Tool for NeedsAttachmentTool {
|
||||
type Args = NoArgs;
|
||||
type Output = Unit;
|
||||
fn id(&self) -> ToolId {
|
||||
ToolId::new("needs_attachment").unwrap()
|
||||
}
|
||||
fn description(&self, _ctx: &::kigi_tool_runtime::ListToolsContext) -> ToolDescription {
|
||||
ToolDescription::new("needs_attachment", "a")
|
||||
}
|
||||
fn should_list(&self, ctx: &ListToolsContext) -> bool {
|
||||
ctx.extensions
|
||||
.get::<AttachmentCount>()
|
||||
.is_some_and(|c| c.0 > 0)
|
||||
}
|
||||
async fn run(&self, _: ToolCallContext, _: NoArgs) -> Result<Unit, ToolError> {
|
||||
Ok(Unit)
|
||||
}
|
||||
}
|
||||
|
||||
// Tool::should_list (typed)
|
||||
|
||||
#[test]
|
||||
fn default_returns_true() {
|
||||
assert!(Tool::should_list(&AlwaysTool, &ListToolsContext::default()));
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn reads_extensions() {
|
||||
let tool = NeedsCwdTool;
|
||||
assert!(!Tool::should_list(&tool, &ListToolsContext::default()));
|
||||
|
||||
let mut ctx = ListToolsContext::default();
|
||||
ctx.extensions
|
||||
.insert(Cwd(std::path::PathBuf::from("/work")));
|
||||
assert!(Tool::should_list(&tool, &ctx));
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn reads_custom_extension() {
|
||||
let tool = NeedsAttachmentTool;
|
||||
assert!(!Tool::should_list(&tool, &ListToolsContext::default()));
|
||||
|
||||
let mut zero = ListToolsContext::default();
|
||||
zero.extensions.insert(AttachmentCount(0));
|
||||
assert!(!Tool::should_list(&tool, &zero));
|
||||
|
||||
let mut some = ListToolsContext::default();
|
||||
some.extensions.insert(AttachmentCount(3));
|
||||
assert!(Tool::should_list(&tool, &some));
|
||||
}
|
||||
|
||||
// ToolDyn blanket forwarding
|
||||
|
||||
#[test]
|
||||
fn dyn_forwards_default() {
|
||||
let tool: ArcTool = Arc::new(AlwaysTool);
|
||||
assert!(tool.should_list(&ListToolsContext::default()));
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn dyn_forwards_custom() {
|
||||
let tool: ArcTool = Arc::new(NeedsCwdTool);
|
||||
assert!(!tool.should_list(&ListToolsContext::default()));
|
||||
|
||||
let mut ctx = ListToolsContext::default();
|
||||
ctx.extensions
|
||||
.insert(Cwd(std::path::PathBuf::from("/home")));
|
||||
assert!(tool.should_list(&ctx));
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn arc_dyn_callable() {
|
||||
let tool: Arc<dyn ToolDyn> = Arc::new(NeedsAttachmentTool);
|
||||
let mut ctx = ListToolsContext::default();
|
||||
ctx.extensions.insert(AttachmentCount(1));
|
||||
assert!(tool.should_list(&ctx));
|
||||
}
|
||||
|
||||
// ListToolsContext
|
||||
|
||||
#[test]
|
||||
fn list_ctx_default_is_empty() {
|
||||
let ctx = ListToolsContext::default();
|
||||
assert!(ctx.extensions.is_empty());
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn list_ctx_insert_and_get() {
|
||||
let mut ctx = ListToolsContext::default();
|
||||
ctx.extensions.insert(Cwd(std::path::PathBuf::from("/a")));
|
||||
assert_eq!(
|
||||
ctx.extensions.get::<Cwd>().unwrap().0,
|
||||
std::path::PathBuf::from("/a")
|
||||
);
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn list_ctx_clone_is_independent() {
|
||||
let mut ctx = ListToolsContext::default();
|
||||
ctx.extensions.insert(AttachmentCount(5));
|
||||
let mut copy = ctx.clone();
|
||||
copy.extensions.remove::<AttachmentCount>();
|
||||
assert!(ctx.extensions.contains::<AttachmentCount>());
|
||||
assert!(!copy.extensions.contains::<AttachmentCount>());
|
||||
}
|
||||
|
||||
// TypedExtensions standalone
|
||||
|
||||
#[test]
|
||||
fn typed_extensions_insert_get_remove() {
|
||||
let mut ext = kigi_tool_runtime::TypedExtensions::new();
|
||||
assert!(ext.is_empty());
|
||||
ext.insert(42_u32);
|
||||
assert_eq!(*ext.get::<u32>().unwrap(), 42);
|
||||
ext.remove::<u32>();
|
||||
assert!(ext.get::<u32>().is_none());
|
||||
}
|
||||
Reference in New Issue
Block a user