mirror of
https://github.com/instructkr/claw-code.git
synced 2026-04-03 17:14:48 +08:00
Compare commits
1 Commits
rcc/runtim
...
rcc/image
| Author | SHA1 | Date | |
|---|---|---|---|
|
|
5b046836b9 |
@@ -133,7 +133,6 @@ Inside the REPL, useful commands include:
|
||||
/diff
|
||||
/version
|
||||
/export notes.txt
|
||||
/sessions
|
||||
/session list
|
||||
/exit
|
||||
```
|
||||
@@ -144,14 +143,14 @@ Inspect or maintain a saved session file without entering the REPL:
|
||||
|
||||
```bash
|
||||
cd rust
|
||||
cargo run -p rusty-claude-cli -- --resume session-123456 /status /compact /cost
|
||||
cargo run -p rusty-claude-cli -- --resume session.json /status /compact /cost
|
||||
```
|
||||
|
||||
You can also inspect memory/config state for a restored session:
|
||||
|
||||
```bash
|
||||
cd rust
|
||||
cargo run -p rusty-claude-cli -- --resume ~/.claude/sessions/session-123456.json /memory /config
|
||||
cargo run -p rusty-claude-cli -- --resume session.json /memory /config
|
||||
```
|
||||
|
||||
## Available commands
|
||||
@@ -159,7 +158,7 @@ cargo run -p rusty-claude-cli -- --resume ~/.claude/sessions/session-123456.json
|
||||
### Top-level CLI commands
|
||||
|
||||
- `prompt <text...>` — run one prompt non-interactively
|
||||
- `--resume <session-id-or-path> [/commands...]` — inspect or maintain a saved session stored under `~/.claude/sessions/`
|
||||
- `--resume <session.json> [/commands...]` — inspect or maintain a saved session
|
||||
- `dump-manifests` — print extracted upstream manifest counts
|
||||
- `bootstrap-plan` — print the current bootstrap skeleton
|
||||
- `system-prompt [--cwd PATH] [--date YYYY-MM-DD]` — render the synthesized system prompt
|
||||
@@ -177,14 +176,13 @@ cargo run -p rusty-claude-cli -- --resume ~/.claude/sessions/session-123456.json
|
||||
- `/permissions [read-only|workspace-write|danger-full-access]` — inspect or switch permissions
|
||||
- `/clear [--confirm]` — clear the current local session
|
||||
- `/cost` — show token usage totals
|
||||
- `/resume <session-id-or-path>` — load a saved session into the REPL
|
||||
- `/resume <session-path>` — load a saved session into the REPL
|
||||
- `/config [env|hooks|model]` — inspect discovered Claude config
|
||||
- `/memory` — inspect loaded instruction memory files
|
||||
- `/init` — create a starter `CLAUDE.md`
|
||||
- `/diff` — show the current git diff for the workspace
|
||||
- `/version` — print version and build metadata locally
|
||||
- `/export [file]` — export the current conversation transcript
|
||||
- `/sessions` — list recent managed local sessions from `~/.claude/sessions/`
|
||||
- `/session [list|switch <session-id>]` — inspect or switch managed local sessions
|
||||
- `/exit` — leave the REPL
|
||||
|
||||
|
||||
@@ -11,7 +11,7 @@ pub use error::ApiError;
|
||||
pub use sse::{parse_frame, SseParser};
|
||||
pub use types::{
|
||||
ContentBlockDelta, ContentBlockDeltaEvent, ContentBlockStartEvent, ContentBlockStopEvent,
|
||||
InputContentBlock, InputMessage, MessageDelta, MessageDeltaEvent, MessageRequest,
|
||||
ImageSource, InputContentBlock, InputMessage, MessageDelta, MessageDeltaEvent, MessageRequest,
|
||||
MessageResponse, MessageStartEvent, MessageStopEvent, OutputContentBlock, StreamEvent,
|
||||
ToolChoice, ToolDefinition, ToolResultContentBlock, Usage,
|
||||
};
|
||||
|
||||
@@ -64,6 +64,9 @@ pub enum InputContentBlock {
|
||||
Text {
|
||||
text: String,
|
||||
},
|
||||
Image {
|
||||
source: ImageSource,
|
||||
},
|
||||
ToolUse {
|
||||
id: String,
|
||||
name: String,
|
||||
@@ -77,6 +80,14 @@ pub enum InputContentBlock {
|
||||
},
|
||||
}
|
||||
|
||||
#[derive(Debug, Clone, PartialEq, Eq, Serialize, Deserialize)]
|
||||
pub struct ImageSource {
|
||||
#[serde(rename = "type")]
|
||||
pub kind: String,
|
||||
pub media_type: String,
|
||||
pub data: String,
|
||||
}
|
||||
|
||||
#[derive(Debug, Clone, PartialEq, Serialize, Deserialize)]
|
||||
#[serde(tag = "type", rename_all = "snake_case")]
|
||||
pub enum ToolResultContentBlock {
|
||||
|
||||
@@ -4,8 +4,8 @@ use std::time::Duration;
|
||||
|
||||
use api::{
|
||||
AnthropicClient, ApiError, ContentBlockDelta, ContentBlockDeltaEvent, ContentBlockStartEvent,
|
||||
InputContentBlock, InputMessage, MessageDeltaEvent, MessageRequest, OutputContentBlock,
|
||||
StreamEvent, ToolChoice, ToolDefinition,
|
||||
ImageSource, InputContentBlock, InputMessage, MessageDeltaEvent, MessageRequest,
|
||||
OutputContentBlock, StreamEvent, ToolChoice, ToolDefinition,
|
||||
};
|
||||
use serde_json::json;
|
||||
use tokio::io::{AsyncReadExt, AsyncWriteExt};
|
||||
@@ -75,6 +75,39 @@ async fn send_message_posts_json_and_parses_response() {
|
||||
assert_eq!(body["tool_choice"]["type"], json!("auto"));
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn image_content_blocks_serialize_with_base64_source() {
|
||||
let request = MessageRequest {
|
||||
model: "claude-3-7-sonnet-latest".to_string(),
|
||||
max_tokens: 64,
|
||||
messages: vec![InputMessage {
|
||||
role: "user".to_string(),
|
||||
content: vec![InputContentBlock::Image {
|
||||
source: ImageSource {
|
||||
kind: "base64".to_string(),
|
||||
media_type: "image/png".to_string(),
|
||||
data: "AQID".to_string(),
|
||||
},
|
||||
}],
|
||||
}],
|
||||
system: None,
|
||||
tools: None,
|
||||
tool_choice: None,
|
||||
stream: false,
|
||||
};
|
||||
|
||||
let json = serde_json::to_value(request).expect("request should serialize");
|
||||
assert_eq!(json["messages"][0]["content"][0]["type"], json!("image"));
|
||||
assert_eq!(
|
||||
json["messages"][0]["content"][0]["source"],
|
||||
json!({
|
||||
"type": "base64",
|
||||
"media_type": "image/png",
|
||||
"data": "AQID"
|
||||
})
|
||||
);
|
||||
}
|
||||
|
||||
#[tokio::test]
|
||||
async fn stream_message_parses_sse_events_with_tool_use() {
|
||||
let state = Arc::new(Mutex::new(Vec::<CapturedRequest>::new()));
|
||||
|
||||
@@ -84,7 +84,7 @@ const SLASH_COMMAND_SPECS: &[SlashCommandSpec] = &[
|
||||
SlashCommandSpec {
|
||||
name: "resume",
|
||||
summary: "Load a saved session into the REPL",
|
||||
argument_hint: Some("<session-id-or-path>"),
|
||||
argument_hint: Some("<session-path>"),
|
||||
resume_supported: false,
|
||||
},
|
||||
SlashCommandSpec {
|
||||
@@ -129,12 +129,6 @@ const SLASH_COMMAND_SPECS: &[SlashCommandSpec] = &[
|
||||
argument_hint: Some("[list|switch <session-id>]"),
|
||||
resume_supported: false,
|
||||
},
|
||||
SlashCommandSpec {
|
||||
name: "sessions",
|
||||
summary: "List recent managed local sessions",
|
||||
argument_hint: None,
|
||||
resume_supported: false,
|
||||
},
|
||||
];
|
||||
|
||||
#[derive(Debug, Clone, PartialEq, Eq)]
|
||||
@@ -169,7 +163,6 @@ pub enum SlashCommand {
|
||||
action: Option<String>,
|
||||
target: Option<String>,
|
||||
},
|
||||
Sessions,
|
||||
Unknown(String),
|
||||
}
|
||||
|
||||
@@ -214,7 +207,6 @@ impl SlashCommand {
|
||||
action: parts.next().map(ToOwned::to_owned),
|
||||
target: parts.next().map(ToOwned::to_owned),
|
||||
},
|
||||
"sessions" => Self::Sessions,
|
||||
other => Self::Unknown(other.to_string()),
|
||||
})
|
||||
}
|
||||
@@ -299,7 +291,6 @@ pub fn handle_slash_command(
|
||||
| SlashCommand::Version
|
||||
| SlashCommand::Export { .. }
|
||||
| SlashCommand::Session { .. }
|
||||
| SlashCommand::Sessions
|
||||
| SlashCommand::Unknown(_) => None,
|
||||
}
|
||||
}
|
||||
@@ -374,10 +365,6 @@ mod tests {
|
||||
target: Some("abc123".to_string())
|
||||
})
|
||||
);
|
||||
assert_eq!(
|
||||
SlashCommand::parse("/sessions"),
|
||||
Some(SlashCommand::Sessions)
|
||||
);
|
||||
}
|
||||
|
||||
#[test]
|
||||
@@ -391,7 +378,7 @@ mod tests {
|
||||
assert!(help.contains("/permissions [read-only|workspace-write|danger-full-access]"));
|
||||
assert!(help.contains("/clear [--confirm]"));
|
||||
assert!(help.contains("/cost"));
|
||||
assert!(help.contains("/resume <session-id-or-path>"));
|
||||
assert!(help.contains("/resume <session-path>"));
|
||||
assert!(help.contains("/config [env|hooks|model]"));
|
||||
assert!(help.contains("/memory"));
|
||||
assert!(help.contains("/init"));
|
||||
@@ -399,8 +386,7 @@ mod tests {
|
||||
assert!(help.contains("/version"));
|
||||
assert!(help.contains("/export [file]"));
|
||||
assert!(help.contains("/session [list|switch <session-id>]"));
|
||||
assert!(help.contains("/sessions"));
|
||||
assert_eq!(slash_command_specs().len(), 16);
|
||||
assert_eq!(slash_command_specs().len(), 15);
|
||||
assert_eq!(resume_supported_slash_commands().len(), 11);
|
||||
}
|
||||
|
||||
@@ -418,7 +404,6 @@ mod tests {
|
||||
text: "recent".to_string(),
|
||||
}]),
|
||||
],
|
||||
metadata: None,
|
||||
};
|
||||
|
||||
let result = handle_slash_command(
|
||||
@@ -483,6 +468,5 @@ mod tests {
|
||||
assert!(
|
||||
handle_slash_command("/session list", &session, CompactionConfig::default()).is_none()
|
||||
);
|
||||
assert!(handle_slash_command("/sessions", &session, CompactionConfig::default()).is_none());
|
||||
}
|
||||
}
|
||||
|
||||
@@ -105,7 +105,6 @@ pub fn compact_session(session: &Session, config: CompactionConfig) -> Compactio
|
||||
compacted_session: Session {
|
||||
version: session.version,
|
||||
messages: compacted_messages,
|
||||
metadata: session.metadata.clone(),
|
||||
},
|
||||
removed_message_count: removed.len(),
|
||||
}
|
||||
@@ -394,7 +393,6 @@ mod tests {
|
||||
let session = Session {
|
||||
version: 1,
|
||||
messages: vec![ConversationMessage::user_text("hello")],
|
||||
metadata: None,
|
||||
};
|
||||
|
||||
let result = compact_session(&session, CompactionConfig::default());
|
||||
@@ -422,7 +420,6 @@ mod tests {
|
||||
usage: None,
|
||||
},
|
||||
],
|
||||
metadata: None,
|
||||
};
|
||||
|
||||
let result = compact_session(
|
||||
|
||||
@@ -73,9 +73,7 @@ pub use remote::{
|
||||
RemoteSessionContext, UpstreamProxyBootstrap, UpstreamProxyState, DEFAULT_REMOTE_BASE_URL,
|
||||
DEFAULT_SESSION_TOKEN_PATH, DEFAULT_SYSTEM_CA_BUNDLE, NO_PROXY_HOSTS, UPSTREAM_PROXY_ENV_KEYS,
|
||||
};
|
||||
pub use session::{
|
||||
ContentBlock, ConversationMessage, MessageRole, Session, SessionError, SessionMetadata,
|
||||
};
|
||||
pub use session::{ContentBlock, ConversationMessage, MessageRole, Session, SessionError};
|
||||
pub use usage::{
|
||||
format_usd, pricing_for_model, ModelPricing, TokenUsage, UsageCostEstimate, UsageTracker,
|
||||
};
|
||||
|
||||
@@ -39,19 +39,10 @@ pub struct ConversationMessage {
|
||||
pub usage: Option<TokenUsage>,
|
||||
}
|
||||
|
||||
#[derive(Debug, Clone, PartialEq, Eq)]
|
||||
pub struct SessionMetadata {
|
||||
pub started_at: String,
|
||||
pub model: String,
|
||||
pub message_count: u32,
|
||||
pub last_prompt: Option<String>,
|
||||
}
|
||||
|
||||
#[derive(Debug, Clone, PartialEq, Eq)]
|
||||
pub struct Session {
|
||||
pub version: u32,
|
||||
pub messages: Vec<ConversationMessage>,
|
||||
pub metadata: Option<SessionMetadata>,
|
||||
}
|
||||
|
||||
#[derive(Debug)]
|
||||
@@ -91,7 +82,6 @@ impl Session {
|
||||
Self {
|
||||
version: 1,
|
||||
messages: Vec::new(),
|
||||
metadata: None,
|
||||
}
|
||||
}
|
||||
|
||||
@@ -121,9 +111,6 @@ impl Session {
|
||||
.collect(),
|
||||
),
|
||||
);
|
||||
if let Some(metadata) = &self.metadata {
|
||||
object.insert("metadata".to_string(), metadata.to_json());
|
||||
}
|
||||
JsonValue::Object(object)
|
||||
}
|
||||
|
||||
@@ -144,15 +131,7 @@ impl Session {
|
||||
.iter()
|
||||
.map(ConversationMessage::from_json)
|
||||
.collect::<Result<Vec<_>, _>>()?;
|
||||
let metadata = object
|
||||
.get("metadata")
|
||||
.map(SessionMetadata::from_json)
|
||||
.transpose()?;
|
||||
Ok(Self {
|
||||
version,
|
||||
messages,
|
||||
metadata,
|
||||
})
|
||||
Ok(Self { version, messages })
|
||||
}
|
||||
}
|
||||
|
||||
@@ -162,41 +141,6 @@ impl Default for Session {
|
||||
}
|
||||
}
|
||||
|
||||
impl SessionMetadata {
|
||||
#[must_use]
|
||||
pub fn to_json(&self) -> JsonValue {
|
||||
let mut object = BTreeMap::new();
|
||||
object.insert(
|
||||
"started_at".to_string(),
|
||||
JsonValue::String(self.started_at.clone()),
|
||||
);
|
||||
object.insert("model".to_string(), JsonValue::String(self.model.clone()));
|
||||
object.insert(
|
||||
"message_count".to_string(),
|
||||
JsonValue::Number(i64::from(self.message_count)),
|
||||
);
|
||||
if let Some(last_prompt) = &self.last_prompt {
|
||||
object.insert(
|
||||
"last_prompt".to_string(),
|
||||
JsonValue::String(last_prompt.clone()),
|
||||
);
|
||||
}
|
||||
JsonValue::Object(object)
|
||||
}
|
||||
|
||||
fn from_json(value: &JsonValue) -> Result<Self, SessionError> {
|
||||
let object = value.as_object().ok_or_else(|| {
|
||||
SessionError::Format("session metadata must be an object".to_string())
|
||||
})?;
|
||||
Ok(Self {
|
||||
started_at: required_string(object, "started_at")?,
|
||||
model: required_string(object, "model")?,
|
||||
message_count: required_u32(object, "message_count")?,
|
||||
last_prompt: optional_string(object, "last_prompt"),
|
||||
})
|
||||
}
|
||||
}
|
||||
|
||||
impl ConversationMessage {
|
||||
#[must_use]
|
||||
pub fn user_text(text: impl Into<String>) -> Self {
|
||||
@@ -424,13 +368,6 @@ fn required_string(
|
||||
.ok_or_else(|| SessionError::Format(format!("missing {key}")))
|
||||
}
|
||||
|
||||
fn optional_string(object: &BTreeMap<String, JsonValue>, key: &str) -> Option<String> {
|
||||
object
|
||||
.get(key)
|
||||
.and_then(JsonValue::as_str)
|
||||
.map(ToOwned::to_owned)
|
||||
}
|
||||
|
||||
fn required_u32(object: &BTreeMap<String, JsonValue>, key: &str) -> Result<u32, SessionError> {
|
||||
let value = object
|
||||
.get(key)
|
||||
@@ -441,8 +378,7 @@ fn required_u32(object: &BTreeMap<String, JsonValue>, key: &str) -> Result<u32,
|
||||
|
||||
#[cfg(test)]
|
||||
mod tests {
|
||||
use super::{ContentBlock, ConversationMessage, MessageRole, Session, SessionMetadata};
|
||||
use crate::json::JsonValue;
|
||||
use super::{ContentBlock, ConversationMessage, MessageRole, Session};
|
||||
use crate::usage::TokenUsage;
|
||||
use std::fs;
|
||||
use std::time::{SystemTime, UNIX_EPOCH};
|
||||
@@ -450,12 +386,6 @@ mod tests {
|
||||
#[test]
|
||||
fn persists_and_restores_session_json() {
|
||||
let mut session = Session::new();
|
||||
session.metadata = Some(SessionMetadata {
|
||||
started_at: "2026-04-01T00:00:00Z".to_string(),
|
||||
model: "claude-sonnet".to_string(),
|
||||
message_count: 3,
|
||||
last_prompt: Some("hello".to_string()),
|
||||
});
|
||||
session
|
||||
.messages
|
||||
.push(ConversationMessage::user_text("hello"));
|
||||
@@ -498,23 +428,5 @@ mod tests {
|
||||
restored.messages[1].usage.expect("usage").total_tokens(),
|
||||
17
|
||||
);
|
||||
assert_eq!(restored.metadata, session.metadata);
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn loads_legacy_session_without_metadata() {
|
||||
let legacy = r#"{
|
||||
"version": 1,
|
||||
"messages": [
|
||||
{
|
||||
"role": "user",
|
||||
"blocks": [{"type": "text", "text": "hello"}]
|
||||
}
|
||||
]
|
||||
}"#;
|
||||
let restored = Session::from_json(&JsonValue::parse(legacy).expect("legacy json"))
|
||||
.expect("legacy session should parse");
|
||||
assert_eq!(restored.messages.len(), 1);
|
||||
assert!(restored.metadata.is_none());
|
||||
}
|
||||
}
|
||||
|
||||
@@ -300,7 +300,6 @@ mod tests {
|
||||
cache_read_input_tokens: 0,
|
||||
}),
|
||||
}],
|
||||
metadata: None,
|
||||
};
|
||||
|
||||
let tracker = UsageTracker::from_session(&session);
|
||||
|
||||
@@ -11,8 +11,8 @@ use std::process::Command;
|
||||
use std::time::{SystemTime, UNIX_EPOCH};
|
||||
|
||||
use api::{
|
||||
resolve_startup_auth_source, AnthropicClient, AuthSource, ContentBlockDelta, InputContentBlock,
|
||||
InputMessage, MessageRequest, MessageResponse, OutputContentBlock,
|
||||
resolve_startup_auth_source, AnthropicClient, AuthSource, ContentBlockDelta, ImageSource,
|
||||
InputContentBlock, InputMessage, MessageRequest, MessageResponse, OutputContentBlock,
|
||||
StreamEvent as ApiStreamEvent, ToolChoice, ToolDefinition, ToolResultContentBlock,
|
||||
};
|
||||
|
||||
@@ -27,7 +27,7 @@ use runtime::{
|
||||
AssistantEvent, CompactionConfig, ConfigLoader, ConfigSource, ContentBlock,
|
||||
ConversationMessage, ConversationRuntime, MessageRole, OAuthAuthorizationRequest,
|
||||
OAuthTokenExchangeRequest, PermissionMode, PermissionPolicy, ProjectContext, RuntimeError,
|
||||
Session, SessionMetadata, TokenUsage, ToolError, ToolExecutor, UsageTracker,
|
||||
Session, TokenUsage, ToolError, ToolExecutor, UsageTracker,
|
||||
};
|
||||
use serde_json::json;
|
||||
use tools::{execute_tool, mvp_tool_specs, ToolSpec};
|
||||
@@ -37,11 +37,11 @@ const DEFAULT_MAX_TOKENS: u32 = 32;
|
||||
const DEFAULT_DATE: &str = "2026-03-31";
|
||||
const DEFAULT_OAUTH_CALLBACK_PORT: u16 = 4545;
|
||||
const VERSION: &str = env!("CARGO_PKG_VERSION");
|
||||
const OLD_SESSION_COMPACTION_AGE_SECS: u64 = 60 * 60 * 24;
|
||||
const BUILD_TARGET: Option<&str> = option_env!("TARGET");
|
||||
const GIT_SHA: Option<&str> = option_env!("GIT_SHA");
|
||||
|
||||
type AllowedToolSet = BTreeSet<String>;
|
||||
const IMAGE_REF_PREFIX: &str = "@";
|
||||
|
||||
fn main() {
|
||||
if let Err(error) = run() {
|
||||
@@ -536,14 +536,7 @@ fn print_version() {
|
||||
}
|
||||
|
||||
fn resume_session(session_path: &Path, commands: &[String]) {
|
||||
let handle = match resolve_session_reference(&session_path.display().to_string()) {
|
||||
Ok(handle) => handle,
|
||||
Err(error) => {
|
||||
eprintln!("failed to resolve session: {error}");
|
||||
std::process::exit(1);
|
||||
}
|
||||
};
|
||||
let session = match Session::load_from_path(&handle.path) {
|
||||
let session = match Session::load_from_path(session_path) {
|
||||
Ok(session) => session,
|
||||
Err(error) => {
|
||||
eprintln!("failed to restore session: {error}");
|
||||
@@ -554,7 +547,7 @@ fn resume_session(session_path: &Path, commands: &[String]) {
|
||||
if commands.is_empty() {
|
||||
println!(
|
||||
"Restored session from {} ({} messages).",
|
||||
handle.path.display(),
|
||||
session_path.display(),
|
||||
session.messages.len()
|
||||
);
|
||||
return;
|
||||
@@ -566,7 +559,7 @@ fn resume_session(session_path: &Path, commands: &[String]) {
|
||||
eprintln!("unsupported resumed command: {raw_command}");
|
||||
std::process::exit(2);
|
||||
};
|
||||
match run_resume_command(&handle.path, &session, &command) {
|
||||
match run_resume_command(session_path, &session, &command) {
|
||||
Ok(ResumeCommandOutcome {
|
||||
session: next_session,
|
||||
message,
|
||||
@@ -891,7 +884,6 @@ fn run_resume_command(
|
||||
| SlashCommand::Model { .. }
|
||||
| SlashCommand::Permissions { .. }
|
||||
| SlashCommand::Session { .. }
|
||||
| SlashCommand::Sessions
|
||||
| SlashCommand::Unknown(_) => Err("unsupported resumed slash command".into()),
|
||||
}
|
||||
}
|
||||
@@ -948,9 +940,6 @@ struct ManagedSessionSummary {
|
||||
path: PathBuf,
|
||||
modified_epoch_secs: u64,
|
||||
message_count: usize,
|
||||
model: Option<String>,
|
||||
started_at: Option<String>,
|
||||
last_prompt: Option<String>,
|
||||
}
|
||||
|
||||
struct LiveCli {
|
||||
@@ -971,7 +960,6 @@ impl LiveCli {
|
||||
) -> Result<Self, Box<dyn std::error::Error>> {
|
||||
let system_prompt = build_system_prompt()?;
|
||||
let session = create_managed_session_handle()?;
|
||||
auto_compact_inactive_sessions(&session.id)?;
|
||||
let runtime = build_runtime(
|
||||
Session::new(),
|
||||
model.clone(),
|
||||
@@ -1055,9 +1043,7 @@ impl LiveCli {
|
||||
max_tokens: DEFAULT_MAX_TOKENS,
|
||||
messages: vec![InputMessage {
|
||||
role: "user".to_string(),
|
||||
content: vec![InputContentBlock::Text {
|
||||
text: input.to_string(),
|
||||
}],
|
||||
content: prompt_to_content_blocks(input, &env::current_dir()?)?,
|
||||
}],
|
||||
system: (!self.system_prompt.is_empty()).then(|| self.system_prompt.join("\n\n")),
|
||||
tools: None,
|
||||
@@ -1143,10 +1129,6 @@ impl LiveCli {
|
||||
SlashCommand::Session { action, target } => {
|
||||
self.handle_session_command(action.as_deref(), target.as_deref())?
|
||||
}
|
||||
SlashCommand::Sessions => {
|
||||
println!("{}", render_session_list(&self.session.id)?);
|
||||
false
|
||||
}
|
||||
SlashCommand::Unknown(name) => {
|
||||
eprintln!("unknown slash command: /{name}");
|
||||
false
|
||||
@@ -1155,10 +1137,7 @@ impl LiveCli {
|
||||
}
|
||||
|
||||
fn persist_session(&self) -> Result<(), Box<dyn std::error::Error>> {
|
||||
let mut session = self.runtime.session().clone();
|
||||
session.metadata = Some(derive_session_metadata(&session, &self.model));
|
||||
session.save_to_path(&self.session.path)?;
|
||||
auto_compact_inactive_sessions(&self.session.id)?;
|
||||
self.runtime.session().save_to_path(&self.session.path)?;
|
||||
Ok(())
|
||||
}
|
||||
|
||||
@@ -1303,20 +1282,13 @@ impl LiveCli {
|
||||
session_path: Option<String>,
|
||||
) -> Result<bool, Box<dyn std::error::Error>> {
|
||||
let Some(session_ref) = session_path else {
|
||||
println!("Usage: /resume <session-id-or-path>");
|
||||
println!("Usage: /resume <session-path>");
|
||||
return Ok(false);
|
||||
};
|
||||
|
||||
let handle = resolve_session_reference(&session_ref)?;
|
||||
let session = Session::load_from_path(&handle.path)?;
|
||||
let message_count = session.messages.len();
|
||||
if let Some(model) = session
|
||||
.metadata
|
||||
.as_ref()
|
||||
.map(|metadata| metadata.model.clone())
|
||||
{
|
||||
self.model = model;
|
||||
}
|
||||
self.runtime = build_runtime(
|
||||
session,
|
||||
self.model.clone(),
|
||||
@@ -1393,13 +1365,6 @@ impl LiveCli {
|
||||
let handle = resolve_session_reference(target)?;
|
||||
let session = Session::load_from_path(&handle.path)?;
|
||||
let message_count = session.messages.len();
|
||||
if let Some(model) = session
|
||||
.metadata
|
||||
.as_ref()
|
||||
.map(|metadata| metadata.model.clone())
|
||||
{
|
||||
self.model = model;
|
||||
}
|
||||
self.runtime = build_runtime(
|
||||
session,
|
||||
self.model.clone(),
|
||||
@@ -1444,10 +1409,8 @@ impl LiveCli {
|
||||
}
|
||||
|
||||
fn sessions_dir() -> Result<PathBuf, Box<dyn std::error::Error>> {
|
||||
let home = env::var_os("HOME")
|
||||
.map(PathBuf::from)
|
||||
.ok_or_else(|| io::Error::new(io::ErrorKind::NotFound, "HOME is not set"))?;
|
||||
let path = home.join(".claude").join("sessions");
|
||||
let cwd = env::current_dir()?;
|
||||
let path = cwd.join(".claude").join("sessions");
|
||||
fs::create_dir_all(&path)?;
|
||||
Ok(path)
|
||||
}
|
||||
@@ -1468,19 +1431,8 @@ fn generate_session_id() -> String {
|
||||
|
||||
fn resolve_session_reference(reference: &str) -> Result<SessionHandle, Box<dyn std::error::Error>> {
|
||||
let direct = PathBuf::from(reference);
|
||||
let expanded = if let Some(stripped) = reference.strip_prefix("~/") {
|
||||
sessions_dir()?
|
||||
.parent()
|
||||
.and_then(|claude| claude.parent())
|
||||
.map(|home| home.join(stripped))
|
||||
.unwrap_or(direct.clone())
|
||||
} else {
|
||||
direct.clone()
|
||||
};
|
||||
let path = if direct.exists() {
|
||||
direct
|
||||
} else if expanded.exists() {
|
||||
expanded
|
||||
} else {
|
||||
sessions_dir()?.join(format!("{reference}.json"))
|
||||
};
|
||||
@@ -1510,11 +1462,9 @@ fn list_managed_sessions() -> Result<Vec<ManagedSessionSummary>, Box<dyn std::er
|
||||
.and_then(|time| time.duration_since(UNIX_EPOCH).ok())
|
||||
.map(|duration| duration.as_secs())
|
||||
.unwrap_or_default();
|
||||
let session = Session::load_from_path(&path).ok();
|
||||
let derived_message_count = session.as_ref().map_or(0, |session| session.messages.len());
|
||||
let stored = session
|
||||
.as_ref()
|
||||
.and_then(|session| session.metadata.as_ref());
|
||||
let message_count = Session::load_from_path(&path)
|
||||
.map(|session| session.messages.len())
|
||||
.unwrap_or_default();
|
||||
let id = path
|
||||
.file_stem()
|
||||
.and_then(|value| value.to_str())
|
||||
@@ -1524,12 +1474,7 @@ fn list_managed_sessions() -> Result<Vec<ManagedSessionSummary>, Box<dyn std::er
|
||||
id,
|
||||
path,
|
||||
modified_epoch_secs,
|
||||
message_count: stored.map_or(derived_message_count, |metadata| {
|
||||
metadata.message_count as usize
|
||||
}),
|
||||
model: stored.map(|metadata| metadata.model.clone()),
|
||||
started_at: stored.map(|metadata| metadata.started_at.clone()),
|
||||
last_prompt: stored.and_then(|metadata| metadata.last_prompt.clone()),
|
||||
message_count,
|
||||
});
|
||||
}
|
||||
sessions.sort_by(|left, right| right.modified_epoch_secs.cmp(&left.modified_epoch_secs));
|
||||
@@ -1552,99 +1497,17 @@ fn render_session_list(active_session_id: &str) -> Result<String, Box<dyn std::e
|
||||
} else {
|
||||
"○ saved"
|
||||
};
|
||||
let model = session.model.as_deref().unwrap_or("unknown");
|
||||
let started = session.started_at.as_deref().unwrap_or("unknown");
|
||||
let last_prompt = session.last_prompt.as_deref().map_or_else(
|
||||
|| "-".to_string(),
|
||||
|prompt| truncate_for_summary(prompt, 36),
|
||||
);
|
||||
lines.push(format!(
|
||||
" {id:<20} {marker:<10} msgs={msgs:<4} model={model:<24} started={started} modified={modified} last={last_prompt} path={path}",
|
||||
" {id:<20} {marker:<10} msgs={msgs:<4} modified={modified} path={path}",
|
||||
id = session.id,
|
||||
msgs = session.message_count,
|
||||
model = model,
|
||||
started = started,
|
||||
modified = session.modified_epoch_secs,
|
||||
last_prompt = last_prompt,
|
||||
path = session.path.display(),
|
||||
));
|
||||
}
|
||||
Ok(lines.join("\n"))
|
||||
}
|
||||
|
||||
fn current_epoch_secs() -> u64 {
|
||||
SystemTime::now()
|
||||
.duration_since(UNIX_EPOCH)
|
||||
.map(|duration| duration.as_secs())
|
||||
.unwrap_or_default()
|
||||
}
|
||||
|
||||
fn current_timestamp_rfc3339ish() -> String {
|
||||
format!("{}Z", current_epoch_secs())
|
||||
}
|
||||
|
||||
fn last_prompt_from_session(session: &Session) -> Option<String> {
|
||||
session
|
||||
.messages
|
||||
.iter()
|
||||
.rev()
|
||||
.find(|message| message.role == MessageRole::User)
|
||||
.and_then(|message| {
|
||||
message.blocks.iter().find_map(|block| match block {
|
||||
ContentBlock::Text { text } => Some(text.trim().to_string()),
|
||||
_ => None,
|
||||
})
|
||||
})
|
||||
.filter(|text| !text.is_empty())
|
||||
}
|
||||
|
||||
fn derive_session_metadata(session: &Session, model: &str) -> SessionMetadata {
|
||||
let started_at = session
|
||||
.metadata
|
||||
.as_ref()
|
||||
.map_or_else(current_timestamp_rfc3339ish, |metadata| {
|
||||
metadata.started_at.clone()
|
||||
});
|
||||
SessionMetadata {
|
||||
started_at,
|
||||
model: model.to_string(),
|
||||
message_count: session.messages.len().try_into().unwrap_or(u32::MAX),
|
||||
last_prompt: last_prompt_from_session(session),
|
||||
}
|
||||
}
|
||||
|
||||
fn session_age_secs(modified_epoch_secs: u64) -> u64 {
|
||||
current_epoch_secs().saturating_sub(modified_epoch_secs)
|
||||
}
|
||||
|
||||
fn auto_compact_inactive_sessions(
|
||||
active_session_id: &str,
|
||||
) -> Result<(), Box<dyn std::error::Error>> {
|
||||
for summary in list_managed_sessions()? {
|
||||
if summary.id == active_session_id
|
||||
|| session_age_secs(summary.modified_epoch_secs) < OLD_SESSION_COMPACTION_AGE_SECS
|
||||
{
|
||||
continue;
|
||||
}
|
||||
let path = summary.path.clone();
|
||||
let Ok(session) = Session::load_from_path(&path) else {
|
||||
continue;
|
||||
};
|
||||
if !runtime::should_compact(&session, CompactionConfig::default()) {
|
||||
continue;
|
||||
}
|
||||
let mut compacted =
|
||||
runtime::compact_session(&session, CompactionConfig::default()).compacted_session;
|
||||
let model = compacted.metadata.as_ref().map_or_else(
|
||||
|| DEFAULT_MODEL.to_string(),
|
||||
|metadata| metadata.model.clone(),
|
||||
);
|
||||
compacted.metadata = Some(derive_session_metadata(&compacted, &model));
|
||||
compacted.save_to_path(&path)?;
|
||||
}
|
||||
Ok(())
|
||||
}
|
||||
|
||||
fn render_repl_help() -> String {
|
||||
[
|
||||
"REPL".to_string(),
|
||||
@@ -2157,7 +2020,7 @@ impl ApiClient for AnthropicRuntimeClient {
|
||||
let message_request = MessageRequest {
|
||||
model: self.model.clone(),
|
||||
max_tokens: DEFAULT_MAX_TOKENS,
|
||||
messages: convert_messages(&request.messages),
|
||||
messages: convert_messages(&request.messages)?,
|
||||
system: (!request.system_prompt.is_empty()).then(|| request.system_prompt.join("\n\n")),
|
||||
tools: self.enable_tools.then(|| {
|
||||
filter_tool_specs(self.allowed_tools.as_ref())
|
||||
@@ -2436,7 +2299,10 @@ fn tool_permission_specs() -> Vec<ToolSpec> {
|
||||
mvp_tool_specs()
|
||||
}
|
||||
|
||||
fn convert_messages(messages: &[ConversationMessage]) -> Vec<InputMessage> {
|
||||
fn convert_messages(messages: &[ConversationMessage]) -> Result<Vec<InputMessage>, RuntimeError> {
|
||||
let cwd = env::current_dir().map_err(|error| {
|
||||
RuntimeError::new(format!("failed to resolve current directory: {error}"))
|
||||
})?;
|
||||
messages
|
||||
.iter()
|
||||
.filter_map(|message| {
|
||||
@@ -2447,36 +2313,224 @@ fn convert_messages(messages: &[ConversationMessage]) -> Vec<InputMessage> {
|
||||
let content = message
|
||||
.blocks
|
||||
.iter()
|
||||
.map(|block| match block {
|
||||
ContentBlock::Text { text } => InputContentBlock::Text { text: text.clone() },
|
||||
ContentBlock::ToolUse { id, name, input } => InputContentBlock::ToolUse {
|
||||
id: id.clone(),
|
||||
name: name.clone(),
|
||||
input: serde_json::from_str(input)
|
||||
.unwrap_or_else(|_| serde_json::json!({ "raw": input })),
|
||||
},
|
||||
ContentBlock::ToolResult {
|
||||
tool_use_id,
|
||||
output,
|
||||
is_error,
|
||||
..
|
||||
} => InputContentBlock::ToolResult {
|
||||
tool_use_id: tool_use_id.clone(),
|
||||
content: vec![ToolResultContentBlock::Text {
|
||||
text: output.clone(),
|
||||
}],
|
||||
is_error: *is_error,
|
||||
},
|
||||
})
|
||||
.collect::<Vec<_>>();
|
||||
(!content.is_empty()).then(|| InputMessage {
|
||||
role: role.to_string(),
|
||||
content,
|
||||
})
|
||||
.try_fold(Vec::new(), |mut acc, block| {
|
||||
match block {
|
||||
ContentBlock::Text { text } => {
|
||||
if message.role == MessageRole::User {
|
||||
acc.extend(
|
||||
prompt_to_content_blocks(text, &cwd)
|
||||
.map_err(RuntimeError::new)?,
|
||||
);
|
||||
} else {
|
||||
acc.push(InputContentBlock::Text { text: text.clone() });
|
||||
}
|
||||
}
|
||||
ContentBlock::ToolUse { id, name, input } => {
|
||||
acc.push(InputContentBlock::ToolUse {
|
||||
id: id.clone(),
|
||||
name: name.clone(),
|
||||
input: serde_json::from_str(input)
|
||||
.unwrap_or_else(|_| serde_json::json!({ "raw": input })),
|
||||
});
|
||||
}
|
||||
ContentBlock::ToolResult {
|
||||
tool_use_id,
|
||||
output,
|
||||
is_error,
|
||||
..
|
||||
} => acc.push(InputContentBlock::ToolResult {
|
||||
tool_use_id: tool_use_id.clone(),
|
||||
content: vec![ToolResultContentBlock::Text {
|
||||
text: output.clone(),
|
||||
}],
|
||||
is_error: *is_error,
|
||||
}),
|
||||
}
|
||||
Ok::<_, RuntimeError>(acc)
|
||||
});
|
||||
match content {
|
||||
Ok(content) if !content.is_empty() => Some(Ok(InputMessage {
|
||||
role: role.to_string(),
|
||||
content,
|
||||
})),
|
||||
Ok(_) => None,
|
||||
Err(error) => Some(Err(error)),
|
||||
}
|
||||
})
|
||||
.collect()
|
||||
}
|
||||
|
||||
fn prompt_to_content_blocks(input: &str, cwd: &Path) -> Result<Vec<InputContentBlock>, String> {
|
||||
let mut blocks = Vec::new();
|
||||
let mut text_buffer = String::new();
|
||||
let mut chars = input.char_indices().peekable();
|
||||
|
||||
while let Some((index, ch)) = chars.next() {
|
||||
if ch == '!' && input[index..].starts_with("![") {
|
||||
if let Some((alt_end, path_start, path_end)) = parse_markdown_image_ref(input, index) {
|
||||
let _ = alt_end;
|
||||
flush_text_block(&mut blocks, &mut text_buffer);
|
||||
let path = &input[path_start..path_end];
|
||||
blocks.push(load_image_block(path, cwd)?);
|
||||
while let Some((next_index, _)) = chars.peek() {
|
||||
if *next_index < path_end + 1 {
|
||||
let _ = chars.next();
|
||||
} else {
|
||||
break;
|
||||
}
|
||||
}
|
||||
continue;
|
||||
}
|
||||
}
|
||||
|
||||
if ch == '@' && is_ref_boundary(input[..index].chars().next_back()) {
|
||||
let path_end = find_path_end(input, index + 1);
|
||||
if path_end > index + 1 {
|
||||
let candidate = &input[index + 1..path_end];
|
||||
if looks_like_image_ref(candidate, cwd) {
|
||||
flush_text_block(&mut blocks, &mut text_buffer);
|
||||
blocks.push(load_image_block(candidate, cwd)?);
|
||||
while let Some((next_index, _)) = chars.peek() {
|
||||
if *next_index < path_end {
|
||||
let _ = chars.next();
|
||||
} else {
|
||||
break;
|
||||
}
|
||||
}
|
||||
continue;
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
text_buffer.push(ch);
|
||||
}
|
||||
|
||||
flush_text_block(&mut blocks, &mut text_buffer);
|
||||
if blocks.is_empty() {
|
||||
blocks.push(InputContentBlock::Text {
|
||||
text: input.to_string(),
|
||||
});
|
||||
}
|
||||
Ok(blocks)
|
||||
}
|
||||
|
||||
fn parse_markdown_image_ref(input: &str, start: usize) -> Option<(usize, usize, usize)> {
|
||||
let after_bang = input.get(start + 2..)?;
|
||||
let alt_end_offset = after_bang.find("](")?;
|
||||
let path_start = start + 2 + alt_end_offset + 2;
|
||||
let remainder = input.get(path_start..)?;
|
||||
let path_end_offset = remainder.find(')')?;
|
||||
let path_end = path_start + path_end_offset;
|
||||
Some((start + 2 + alt_end_offset, path_start, path_end))
|
||||
}
|
||||
|
||||
fn is_ref_boundary(ch: Option<char>) -> bool {
|
||||
ch.is_none_or(char::is_whitespace)
|
||||
}
|
||||
|
||||
fn find_path_end(input: &str, start: usize) -> usize {
|
||||
input[start..]
|
||||
.char_indices()
|
||||
.find_map(|(offset, ch)| (ch.is_whitespace()).then_some(start + offset))
|
||||
.unwrap_or(input.len())
|
||||
}
|
||||
|
||||
fn looks_like_image_ref(candidate: &str, cwd: &Path) -> bool {
|
||||
let resolved = resolve_prompt_path(candidate, cwd);
|
||||
media_type_for_path(Path::new(candidate)).is_some()
|
||||
|| resolved.is_file()
|
||||
|| candidate.contains(std::path::MAIN_SEPARATOR)
|
||||
|| candidate.starts_with("./")
|
||||
|| candidate.starts_with("../")
|
||||
}
|
||||
|
||||
fn flush_text_block(blocks: &mut Vec<InputContentBlock>, text_buffer: &mut String) {
|
||||
if text_buffer.is_empty() {
|
||||
return;
|
||||
}
|
||||
blocks.push(InputContentBlock::Text {
|
||||
text: std::mem::take(text_buffer),
|
||||
});
|
||||
}
|
||||
|
||||
fn load_image_block(path_ref: &str, cwd: &Path) -> Result<InputContentBlock, String> {
|
||||
let resolved = resolve_prompt_path(path_ref, cwd);
|
||||
let media_type = media_type_for_path(&resolved).ok_or_else(|| {
|
||||
format!(
|
||||
"unsupported image format for reference {IMAGE_REF_PREFIX}{path_ref}; supported: png, jpg, jpeg, gif, webp"
|
||||
)
|
||||
})?;
|
||||
let bytes = fs::read(&resolved).map_err(|error| {
|
||||
format!(
|
||||
"failed to read image reference {}: {error}",
|
||||
resolved.display()
|
||||
)
|
||||
})?;
|
||||
Ok(InputContentBlock::Image {
|
||||
source: ImageSource {
|
||||
kind: "base64".to_string(),
|
||||
media_type: media_type.to_string(),
|
||||
data: encode_base64(&bytes),
|
||||
},
|
||||
})
|
||||
}
|
||||
|
||||
fn resolve_prompt_path(path_ref: &str, cwd: &Path) -> PathBuf {
|
||||
let path = Path::new(path_ref);
|
||||
if path.is_absolute() {
|
||||
path.to_path_buf()
|
||||
} else {
|
||||
cwd.join(path)
|
||||
}
|
||||
}
|
||||
|
||||
fn media_type_for_path(path: &Path) -> Option<&'static str> {
|
||||
let extension = path.extension()?.to_str()?.to_ascii_lowercase();
|
||||
match extension.as_str() {
|
||||
"png" => Some("image/png"),
|
||||
"jpg" | "jpeg" => Some("image/jpeg"),
|
||||
"gif" => Some("image/gif"),
|
||||
"webp" => Some("image/webp"),
|
||||
_ => None,
|
||||
}
|
||||
}
|
||||
|
||||
fn encode_base64(bytes: &[u8]) -> String {
|
||||
const TABLE: &[u8; 64] = b"ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz0123456789+/";
|
||||
let mut output = String::new();
|
||||
let mut index = 0;
|
||||
while index + 3 <= bytes.len() {
|
||||
let block = (u32::from(bytes[index]) << 16)
|
||||
| (u32::from(bytes[index + 1]) << 8)
|
||||
| u32::from(bytes[index + 2]);
|
||||
output.push(TABLE[((block >> 18) & 0x3F) as usize] as char);
|
||||
output.push(TABLE[((block >> 12) & 0x3F) as usize] as char);
|
||||
output.push(TABLE[((block >> 6) & 0x3F) as usize] as char);
|
||||
output.push(TABLE[(block & 0x3F) as usize] as char);
|
||||
index += 3;
|
||||
}
|
||||
|
||||
match bytes.len().saturating_sub(index) {
|
||||
1 => {
|
||||
let block = u32::from(bytes[index]) << 16;
|
||||
output.push(TABLE[((block >> 18) & 0x3F) as usize] as char);
|
||||
output.push(TABLE[((block >> 12) & 0x3F) as usize] as char);
|
||||
output.push('=');
|
||||
output.push('=');
|
||||
}
|
||||
2 => {
|
||||
let block = (u32::from(bytes[index]) << 16) | (u32::from(bytes[index + 1]) << 8);
|
||||
output.push(TABLE[((block >> 18) & 0x3F) as usize] as char);
|
||||
output.push(TABLE[((block >> 12) & 0x3F) as usize] as char);
|
||||
output.push(TABLE[((block >> 6) & 0x3F) as usize] as char);
|
||||
output.push('=');
|
||||
}
|
||||
_ => {}
|
||||
}
|
||||
|
||||
output
|
||||
}
|
||||
|
||||
fn print_help() {
|
||||
println!("rusty-claude-cli v{VERSION}");
|
||||
println!();
|
||||
@@ -2525,72 +2579,18 @@ fn print_help() {
|
||||
#[cfg(test)]
|
||||
mod tests {
|
||||
use super::{
|
||||
derive_session_metadata, filter_tool_specs, format_compact_report, format_cost_report,
|
||||
format_init_report, format_model_report, format_model_switch_report,
|
||||
format_permissions_report, format_permissions_switch_report, format_resume_report,
|
||||
format_status_report, format_tool_call_start, format_tool_result, list_managed_sessions,
|
||||
normalize_permission_mode, parse_args, parse_git_status_metadata, render_config_report,
|
||||
render_init_claude_md, render_memory_report, render_repl_help,
|
||||
resume_supported_slash_commands, sessions_dir, status_context, CliAction, CliOutputFormat,
|
||||
SlashCommand, StatusUsage, DEFAULT_MODEL,
|
||||
filter_tool_specs, format_compact_report, format_cost_report, format_init_report,
|
||||
format_model_report, format_model_switch_report, format_permissions_report,
|
||||
format_permissions_switch_report, format_resume_report, format_status_report,
|
||||
format_tool_call_start, format_tool_result, normalize_permission_mode, parse_args,
|
||||
parse_git_status_metadata, render_config_report, render_init_claude_md,
|
||||
render_memory_report, render_repl_help, resume_supported_slash_commands, status_context,
|
||||
CliAction, CliOutputFormat, SlashCommand, StatusUsage, DEFAULT_MODEL,
|
||||
};
|
||||
use runtime::{ContentBlock, ConversationMessage, MessageRole, PermissionMode, Session};
|
||||
use std::fs;
|
||||
use api::InputContentBlock;
|
||||
use runtime::{ContentBlock, ConversationMessage, MessageRole, PermissionMode};
|
||||
use std::path::{Path, PathBuf};
|
||||
|
||||
#[test]
|
||||
fn derive_session_metadata_recomputes_prompt_and_count() {
|
||||
let mut session = Session::new();
|
||||
session
|
||||
.messages
|
||||
.push(ConversationMessage::user_text("first prompt"));
|
||||
session
|
||||
.messages
|
||||
.push(ConversationMessage::assistant(vec![ContentBlock::Text {
|
||||
text: "reply".to_string(),
|
||||
}]));
|
||||
let metadata = derive_session_metadata(&session, "claude-test");
|
||||
assert_eq!(metadata.model, "claude-test");
|
||||
assert_eq!(metadata.message_count, 2);
|
||||
assert_eq!(metadata.last_prompt.as_deref(), Some("first prompt"));
|
||||
assert!(metadata.started_at.ends_with('Z'));
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn managed_sessions_use_home_directory_and_list_metadata() {
|
||||
let temp =
|
||||
std::env::temp_dir().join(format!("rusty-claude-cli-home-{}", std::process::id()));
|
||||
let _ = fs::remove_dir_all(&temp);
|
||||
fs::create_dir_all(&temp).expect("temp home should exist");
|
||||
let previous_home = std::env::var_os("HOME");
|
||||
std::env::set_var("HOME", &temp);
|
||||
|
||||
let dir = sessions_dir().expect("sessions dir");
|
||||
assert_eq!(dir, temp.join(".claude").join("sessions"));
|
||||
|
||||
let mut session = Session::new();
|
||||
session
|
||||
.messages
|
||||
.push(ConversationMessage::user_text("persist me"));
|
||||
session.metadata = Some(derive_session_metadata(&session, "claude-home"));
|
||||
let file = dir.join("session-test.json");
|
||||
session.save_to_path(&file).expect("session save");
|
||||
|
||||
let listed = list_managed_sessions().expect("session list");
|
||||
let found = listed
|
||||
.into_iter()
|
||||
.find(|entry| entry.id == "session-test")
|
||||
.expect("saved session should be listed");
|
||||
assert_eq!(found.message_count, 1);
|
||||
assert_eq!(found.model.as_deref(), Some("claude-home"));
|
||||
assert_eq!(found.last_prompt.as_deref(), Some("persist me"));
|
||||
|
||||
fs::remove_file(file).ok();
|
||||
if let Some(previous_home) = previous_home {
|
||||
std::env::set_var("HOME", previous_home);
|
||||
}
|
||||
fs::remove_dir_all(temp).ok();
|
||||
}
|
||||
use std::time::{SystemTime, UNIX_EPOCH};
|
||||
|
||||
#[test]
|
||||
fn defaults_to_repl_when_no_args() {
|
||||
@@ -2797,8 +2797,7 @@ mod tests {
|
||||
assert!(help.contains("/permissions [read-only|workspace-write|danger-full-access]"));
|
||||
assert!(help.contains("/clear [--confirm]"));
|
||||
assert!(help.contains("/cost"));
|
||||
assert!(help.contains("/resume <session-id-or-path>"));
|
||||
assert!(help.contains("/sessions"));
|
||||
assert!(help.contains("/resume <session-path>"));
|
||||
assert!(help.contains("/config [env|hooks|model]"));
|
||||
assert!(help.contains("/memory"));
|
||||
assert!(help.contains("/init"));
|
||||
@@ -3074,11 +3073,110 @@ mod tests {
|
||||
},
|
||||
];
|
||||
|
||||
let converted = super::convert_messages(&messages);
|
||||
let converted = super::convert_messages(&messages).expect("messages should convert");
|
||||
assert_eq!(converted.len(), 3);
|
||||
assert_eq!(converted[1].role, "assistant");
|
||||
assert_eq!(converted[2].role, "user");
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn prompt_to_content_blocks_keeps_text_only_prompt() {
|
||||
let blocks = super::prompt_to_content_blocks("hello world", Path::new("."))
|
||||
.expect("text prompt should parse");
|
||||
assert_eq!(
|
||||
blocks,
|
||||
vec![InputContentBlock::Text {
|
||||
text: "hello world".to_string()
|
||||
}]
|
||||
);
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn prompt_to_content_blocks_embeds_at_image_refs() {
|
||||
let temp = temp_fixture_dir("at-image-ref");
|
||||
let image_path = temp.join("sample.png");
|
||||
std::fs::write(&image_path, [1_u8, 2, 3]).expect("fixture write");
|
||||
let prompt = format!("describe @{} please", image_path.display());
|
||||
|
||||
let blocks = super::prompt_to_content_blocks(&prompt, Path::new("."))
|
||||
.expect("image ref should parse");
|
||||
|
||||
assert!(matches!(
|
||||
&blocks[0],
|
||||
InputContentBlock::Text { text } if text == "describe "
|
||||
));
|
||||
assert!(matches!(
|
||||
&blocks[1],
|
||||
InputContentBlock::Image { source }
|
||||
if source.kind == "base64"
|
||||
&& source.media_type == "image/png"
|
||||
&& source.data == "AQID"
|
||||
));
|
||||
assert!(matches!(
|
||||
&blocks[2],
|
||||
InputContentBlock::Text { text } if text == " please"
|
||||
));
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn prompt_to_content_blocks_embeds_markdown_image_refs() {
|
||||
let temp = temp_fixture_dir("markdown-image-ref");
|
||||
let image_path = temp.join("sample.webp");
|
||||
std::fs::write(&image_path, [255_u8]).expect("fixture write");
|
||||
let prompt = format!("see  now", image_path.display());
|
||||
|
||||
let blocks = super::prompt_to_content_blocks(&prompt, Path::new("."))
|
||||
.expect("markdown image ref should parse");
|
||||
|
||||
assert!(matches!(
|
||||
&blocks[1],
|
||||
InputContentBlock::Image { source }
|
||||
if source.media_type == "image/webp" && source.data == "/w=="
|
||||
));
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn prompt_to_content_blocks_rejects_unsupported_formats() {
|
||||
let temp = temp_fixture_dir("unsupported-image-ref");
|
||||
let image_path = temp.join("sample.bmp");
|
||||
std::fs::write(&image_path, [1_u8]).expect("fixture write");
|
||||
let prompt = format!("describe @{}", image_path.display());
|
||||
|
||||
let error = super::prompt_to_content_blocks(&prompt, Path::new("."))
|
||||
.expect_err("unsupported image ref should fail");
|
||||
|
||||
assert!(error.contains("unsupported image format"));
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn convert_messages_expands_user_text_image_refs() {
|
||||
let temp = temp_fixture_dir("convert-message-image-ref");
|
||||
let image_path = temp.join("sample.gif");
|
||||
std::fs::write(&image_path, [71_u8, 73, 70]).expect("fixture write");
|
||||
let messages = vec![ConversationMessage::user_text(format!(
|
||||
"inspect @{}",
|
||||
image_path.display()
|
||||
))];
|
||||
|
||||
let converted = super::convert_messages(&messages).expect("messages should convert");
|
||||
|
||||
assert_eq!(converted.len(), 1);
|
||||
assert!(matches!(
|
||||
&converted[0].content[1],
|
||||
InputContentBlock::Image { source }
|
||||
if source.media_type == "image/gif" && source.data == "R0lG"
|
||||
));
|
||||
}
|
||||
|
||||
fn temp_fixture_dir(label: &str) -> PathBuf {
|
||||
let unique = SystemTime::now()
|
||||
.duration_since(UNIX_EPOCH)
|
||||
.expect("clock should advance")
|
||||
.as_nanos();
|
||||
let path = std::env::temp_dir().join(format!("rusty-claude-cli-{label}-{unique}"));
|
||||
std::fs::create_dir_all(&path).expect("temp dir should exist");
|
||||
path
|
||||
}
|
||||
#[test]
|
||||
fn repl_help_mentions_history_completion_and_multiline() {
|
||||
let help = render_repl_help();
|
||||
|
||||
Reference in New Issue
Block a user