|
@@ -27,7 +27,7 @@ use runtime::{
|
|
|
AssistantEvent, CompactionConfig, ConfigLoader, ConfigSource, ContentBlock,
|
|
AssistantEvent, CompactionConfig, ConfigLoader, ConfigSource, ContentBlock,
|
|
|
ConversationMessage, ConversationRuntime, MessageRole, OAuthAuthorizationRequest,
|
|
ConversationMessage, ConversationRuntime, MessageRole, OAuthAuthorizationRequest,
|
|
|
OAuthTokenExchangeRequest, PermissionMode, PermissionPolicy, ProjectContext, RuntimeError,
|
|
OAuthTokenExchangeRequest, PermissionMode, PermissionPolicy, ProjectContext, RuntimeError,
|
|
|
- Session, TokenUsage, ToolError, ToolExecutor, UsageTracker,
|
|
|
|
|
|
|
+ Session, SessionMetadata, TokenUsage, ToolError, ToolExecutor, UsageTracker,
|
|
|
};
|
|
};
|
|
|
use serde_json::json;
|
|
use serde_json::json;
|
|
|
use tools::{execute_tool, mvp_tool_specs, ToolSpec};
|
|
use tools::{execute_tool, mvp_tool_specs, ToolSpec};
|
|
@@ -37,6 +37,7 @@ const DEFAULT_MAX_TOKENS: u32 = 32;
|
|
|
const DEFAULT_DATE: &str = "2026-03-31";
|
|
const DEFAULT_DATE: &str = "2026-03-31";
|
|
|
const DEFAULT_OAUTH_CALLBACK_PORT: u16 = 4545;
|
|
const DEFAULT_OAUTH_CALLBACK_PORT: u16 = 4545;
|
|
|
const VERSION: &str = env!("CARGO_PKG_VERSION");
|
|
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 BUILD_TARGET: Option<&str> = option_env!("TARGET");
|
|
|
const GIT_SHA: Option<&str> = option_env!("GIT_SHA");
|
|
const GIT_SHA: Option<&str> = option_env!("GIT_SHA");
|
|
|
|
|
|
|
@@ -70,8 +71,7 @@ fn run() -> Result<(), Box<dyn std::error::Error>> {
|
|
|
output_format,
|
|
output_format,
|
|
|
allowed_tools,
|
|
allowed_tools,
|
|
|
permission_mode,
|
|
permission_mode,
|
|
|
- color,
|
|
|
|
|
- } => LiveCli::new(model, false, allowed_tools, permission_mode, color)?
|
|
|
|
|
|
|
+ } => LiveCli::new(model, false, allowed_tools, permission_mode)?
|
|
|
.run_turn_with_output(&prompt, output_format)?,
|
|
.run_turn_with_output(&prompt, output_format)?,
|
|
|
CliAction::Login => run_login()?,
|
|
CliAction::Login => run_login()?,
|
|
|
CliAction::Logout => run_logout()?,
|
|
CliAction::Logout => run_logout()?,
|
|
@@ -79,8 +79,7 @@ fn run() -> Result<(), Box<dyn std::error::Error>> {
|
|
|
model,
|
|
model,
|
|
|
allowed_tools,
|
|
allowed_tools,
|
|
|
permission_mode,
|
|
permission_mode,
|
|
|
- color,
|
|
|
|
|
- } => run_repl(model, allowed_tools, permission_mode, color)?,
|
|
|
|
|
|
|
+ } => run_repl(model, allowed_tools, permission_mode)?,
|
|
|
CliAction::Help => print_help(),
|
|
CliAction::Help => print_help(),
|
|
|
}
|
|
}
|
|
|
Ok(())
|
|
Ok(())
|
|
@@ -105,7 +104,6 @@ enum CliAction {
|
|
|
output_format: CliOutputFormat,
|
|
output_format: CliOutputFormat,
|
|
|
allowed_tools: Option<AllowedToolSet>,
|
|
allowed_tools: Option<AllowedToolSet>,
|
|
|
permission_mode: PermissionMode,
|
|
permission_mode: PermissionMode,
|
|
|
- color: bool,
|
|
|
|
|
},
|
|
},
|
|
|
Login,
|
|
Login,
|
|
|
Logout,
|
|
Logout,
|
|
@@ -113,7 +111,6 @@ enum CliAction {
|
|
|
model: String,
|
|
model: String,
|
|
|
allowed_tools: Option<AllowedToolSet>,
|
|
allowed_tools: Option<AllowedToolSet>,
|
|
|
permission_mode: PermissionMode,
|
|
permission_mode: PermissionMode,
|
|
|
- color: bool,
|
|
|
|
|
},
|
|
},
|
|
|
// prompt-mode formatting is only supported for non-interactive runs
|
|
// prompt-mode formatting is only supported for non-interactive runs
|
|
|
Help,
|
|
Help,
|
|
@@ -144,7 +141,6 @@ fn parse_args(args: &[String]) -> Result<CliAction, String> {
|
|
|
let mut permission_mode = default_permission_mode();
|
|
let mut permission_mode = default_permission_mode();
|
|
|
let mut wants_version = false;
|
|
let mut wants_version = false;
|
|
|
let mut allowed_tool_values = Vec::new();
|
|
let mut allowed_tool_values = Vec::new();
|
|
|
- let mut color = true;
|
|
|
|
|
let mut rest = Vec::new();
|
|
let mut rest = Vec::new();
|
|
|
let mut index = 0;
|
|
let mut index = 0;
|
|
|
|
|
|
|
@@ -154,10 +150,6 @@ fn parse_args(args: &[String]) -> Result<CliAction, String> {
|
|
|
wants_version = true;
|
|
wants_version = true;
|
|
|
index += 1;
|
|
index += 1;
|
|
|
}
|
|
}
|
|
|
- "--no-color" => {
|
|
|
|
|
- color = false;
|
|
|
|
|
- index += 1;
|
|
|
|
|
- }
|
|
|
|
|
"--model" => {
|
|
"--model" => {
|
|
|
let value = args
|
|
let value = args
|
|
|
.get(index + 1)
|
|
.get(index + 1)
|
|
@@ -224,7 +216,6 @@ fn parse_args(args: &[String]) -> Result<CliAction, String> {
|
|
|
model,
|
|
model,
|
|
|
allowed_tools,
|
|
allowed_tools,
|
|
|
permission_mode,
|
|
permission_mode,
|
|
|
- color,
|
|
|
|
|
});
|
|
});
|
|
|
}
|
|
}
|
|
|
if matches!(rest.first().map(String::as_str), Some("--help" | "-h")) {
|
|
if matches!(rest.first().map(String::as_str), Some("--help" | "-h")) {
|
|
@@ -251,7 +242,6 @@ fn parse_args(args: &[String]) -> Result<CliAction, String> {
|
|
|
output_format,
|
|
output_format,
|
|
|
allowed_tools,
|
|
allowed_tools,
|
|
|
permission_mode,
|
|
permission_mode,
|
|
|
- color,
|
|
|
|
|
})
|
|
})
|
|
|
}
|
|
}
|
|
|
other if !other.starts_with('/') => Ok(CliAction::Prompt {
|
|
other if !other.starts_with('/') => Ok(CliAction::Prompt {
|
|
@@ -260,7 +250,6 @@ fn parse_args(args: &[String]) -> Result<CliAction, String> {
|
|
|
output_format,
|
|
output_format,
|
|
|
allowed_tools,
|
|
allowed_tools,
|
|
|
permission_mode,
|
|
permission_mode,
|
|
|
- color,
|
|
|
|
|
}),
|
|
}),
|
|
|
other => Err(format!("unknown subcommand: {other}")),
|
|
other => Err(format!("unknown subcommand: {other}")),
|
|
|
}
|
|
}
|
|
@@ -547,7 +536,14 @@ fn print_version() {
|
|
|
}
|
|
}
|
|
|
|
|
|
|
|
fn resume_session(session_path: &Path, commands: &[String]) {
|
|
fn resume_session(session_path: &Path, commands: &[String]) {
|
|
|
- let session = match Session::load_from_path(session_path) {
|
|
|
|
|
|
|
+ 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) {
|
|
|
Ok(session) => session,
|
|
Ok(session) => session,
|
|
|
Err(error) => {
|
|
Err(error) => {
|
|
|
eprintln!("failed to restore session: {error}");
|
|
eprintln!("failed to restore session: {error}");
|
|
@@ -558,7 +554,7 @@ fn resume_session(session_path: &Path, commands: &[String]) {
|
|
|
if commands.is_empty() {
|
|
if commands.is_empty() {
|
|
|
println!(
|
|
println!(
|
|
|
"Restored session from {} ({} messages).",
|
|
"Restored session from {} ({} messages).",
|
|
|
- session_path.display(),
|
|
|
|
|
|
|
+ handle.path.display(),
|
|
|
session.messages.len()
|
|
session.messages.len()
|
|
|
);
|
|
);
|
|
|
return;
|
|
return;
|
|
@@ -570,7 +566,7 @@ fn resume_session(session_path: &Path, commands: &[String]) {
|
|
|
eprintln!("unsupported resumed command: {raw_command}");
|
|
eprintln!("unsupported resumed command: {raw_command}");
|
|
|
std::process::exit(2);
|
|
std::process::exit(2);
|
|
|
};
|
|
};
|
|
|
- match run_resume_command(session_path, &session, &command) {
|
|
|
|
|
|
|
+ match run_resume_command(&handle.path, &session, &command) {
|
|
|
Ok(ResumeCommandOutcome {
|
|
Ok(ResumeCommandOutcome {
|
|
|
session: next_session,
|
|
session: next_session,
|
|
|
message,
|
|
message,
|
|
@@ -895,6 +891,7 @@ fn run_resume_command(
|
|
|
| SlashCommand::Model { .. }
|
|
| SlashCommand::Model { .. }
|
|
|
| SlashCommand::Permissions { .. }
|
|
| SlashCommand::Permissions { .. }
|
|
|
| SlashCommand::Session { .. }
|
|
| SlashCommand::Session { .. }
|
|
|
|
|
+ | SlashCommand::Sessions
|
|
|
| SlashCommand::Unknown(_) => Err("unsupported resumed slash command".into()),
|
|
| SlashCommand::Unknown(_) => Err("unsupported resumed slash command".into()),
|
|
|
}
|
|
}
|
|
|
}
|
|
}
|
|
@@ -903,9 +900,8 @@ fn run_repl(
|
|
|
model: String,
|
|
model: String,
|
|
|
allowed_tools: Option<AllowedToolSet>,
|
|
allowed_tools: Option<AllowedToolSet>,
|
|
|
permission_mode: PermissionMode,
|
|
permission_mode: PermissionMode,
|
|
|
- color: bool,
|
|
|
|
|
) -> Result<(), Box<dyn std::error::Error>> {
|
|
) -> Result<(), Box<dyn std::error::Error>> {
|
|
|
- let mut cli = LiveCli::new(model, true, allowed_tools, permission_mode, color)?;
|
|
|
|
|
|
|
+ let mut cli = LiveCli::new(model, true, allowed_tools, permission_mode)?;
|
|
|
let mut editor = input::LineEditor::new("› ", slash_command_completion_candidates());
|
|
let mut editor = input::LineEditor::new("› ", slash_command_completion_candidates());
|
|
|
println!("{}", cli.startup_banner());
|
|
println!("{}", cli.startup_banner());
|
|
|
|
|
|
|
@@ -952,17 +948,18 @@ struct ManagedSessionSummary {
|
|
|
path: PathBuf,
|
|
path: PathBuf,
|
|
|
modified_epoch_secs: u64,
|
|
modified_epoch_secs: u64,
|
|
|
message_count: usize,
|
|
message_count: usize,
|
|
|
|
|
+ model: Option<String>,
|
|
|
|
|
+ started_at: Option<String>,
|
|
|
|
|
+ last_prompt: Option<String>,
|
|
|
}
|
|
}
|
|
|
|
|
|
|
|
struct LiveCli {
|
|
struct LiveCli {
|
|
|
model: String,
|
|
model: String,
|
|
|
allowed_tools: Option<AllowedToolSet>,
|
|
allowed_tools: Option<AllowedToolSet>,
|
|
|
permission_mode: PermissionMode,
|
|
permission_mode: PermissionMode,
|
|
|
- color: bool,
|
|
|
|
|
system_prompt: Vec<String>,
|
|
system_prompt: Vec<String>,
|
|
|
runtime: ConversationRuntime<AnthropicRuntimeClient, CliToolExecutor>,
|
|
runtime: ConversationRuntime<AnthropicRuntimeClient, CliToolExecutor>,
|
|
|
session: SessionHandle,
|
|
session: SessionHandle,
|
|
|
- renderer: TerminalRenderer,
|
|
|
|
|
}
|
|
}
|
|
|
|
|
|
|
|
impl LiveCli {
|
|
impl LiveCli {
|
|
@@ -971,10 +968,10 @@ impl LiveCli {
|
|
|
enable_tools: bool,
|
|
enable_tools: bool,
|
|
|
allowed_tools: Option<AllowedToolSet>,
|
|
allowed_tools: Option<AllowedToolSet>,
|
|
|
permission_mode: PermissionMode,
|
|
permission_mode: PermissionMode,
|
|
|
- color: bool,
|
|
|
|
|
) -> Result<Self, Box<dyn std::error::Error>> {
|
|
) -> Result<Self, Box<dyn std::error::Error>> {
|
|
|
let system_prompt = build_system_prompt()?;
|
|
let system_prompt = build_system_prompt()?;
|
|
|
let session = create_managed_session_handle()?;
|
|
let session = create_managed_session_handle()?;
|
|
|
|
|
+ auto_compact_inactive_sessions(&session.id)?;
|
|
|
let runtime = build_runtime(
|
|
let runtime = build_runtime(
|
|
|
Session::new(),
|
|
Session::new(),
|
|
|
model.clone(),
|
|
model.clone(),
|
|
@@ -982,17 +979,14 @@ impl LiveCli {
|
|
|
enable_tools,
|
|
enable_tools,
|
|
|
allowed_tools.clone(),
|
|
allowed_tools.clone(),
|
|
|
permission_mode,
|
|
permission_mode,
|
|
|
- color,
|
|
|
|
|
)?;
|
|
)?;
|
|
|
let cli = Self {
|
|
let cli = Self {
|
|
|
model,
|
|
model,
|
|
|
allowed_tools,
|
|
allowed_tools,
|
|
|
permission_mode,
|
|
permission_mode,
|
|
|
- color,
|
|
|
|
|
system_prompt,
|
|
system_prompt,
|
|
|
runtime,
|
|
runtime,
|
|
|
session,
|
|
session,
|
|
|
- renderer: TerminalRenderer::with_color(color),
|
|
|
|
|
};
|
|
};
|
|
|
cli.persist_session()?;
|
|
cli.persist_session()?;
|
|
|
Ok(cli)
|
|
Ok(cli)
|
|
@@ -1016,33 +1010,26 @@ impl LiveCli {
|
|
|
let mut stdout = io::stdout();
|
|
let mut stdout = io::stdout();
|
|
|
spinner.tick(
|
|
spinner.tick(
|
|
|
"Waiting for Claude",
|
|
"Waiting for Claude",
|
|
|
- self.renderer.color_theme(),
|
|
|
|
|
|
|
+ TerminalRenderer::new().color_theme(),
|
|
|
&mut stdout,
|
|
&mut stdout,
|
|
|
)?;
|
|
)?;
|
|
|
let mut permission_prompter = CliPermissionPrompter::new(self.permission_mode);
|
|
let mut permission_prompter = CliPermissionPrompter::new(self.permission_mode);
|
|
|
let result = self.runtime.run_turn(input, Some(&mut permission_prompter));
|
|
let result = self.runtime.run_turn(input, Some(&mut permission_prompter));
|
|
|
match result {
|
|
match result {
|
|
|
- Ok(summary) => {
|
|
|
|
|
|
|
+ Ok(_) => {
|
|
|
spinner.finish(
|
|
spinner.finish(
|
|
|
"Claude response complete",
|
|
"Claude response complete",
|
|
|
- self.renderer.color_theme(),
|
|
|
|
|
|
|
+ TerminalRenderer::new().color_theme(),
|
|
|
&mut stdout,
|
|
&mut stdout,
|
|
|
)?;
|
|
)?;
|
|
|
println!();
|
|
println!();
|
|
|
- println!(
|
|
|
|
|
- "{}",
|
|
|
|
|
- self.renderer.token_usage_summary(
|
|
|
|
|
- u64::from(summary.usage.input_tokens),
|
|
|
|
|
- u64::from(summary.usage.output_tokens)
|
|
|
|
|
- )
|
|
|
|
|
- );
|
|
|
|
|
self.persist_session()?;
|
|
self.persist_session()?;
|
|
|
Ok(())
|
|
Ok(())
|
|
|
}
|
|
}
|
|
|
Err(error) => {
|
|
Err(error) => {
|
|
|
spinner.fail(
|
|
spinner.fail(
|
|
|
"Claude request failed",
|
|
"Claude request failed",
|
|
|
- self.renderer.color_theme(),
|
|
|
|
|
|
|
+ TerminalRenderer::new().color_theme(),
|
|
|
&mut stdout,
|
|
&mut stdout,
|
|
|
)?;
|
|
)?;
|
|
|
Err(Box::new(error))
|
|
Err(Box::new(error))
|
|
@@ -1156,6 +1143,10 @@ impl LiveCli {
|
|
|
SlashCommand::Session { action, target } => {
|
|
SlashCommand::Session { action, target } => {
|
|
|
self.handle_session_command(action.as_deref(), target.as_deref())?
|
|
self.handle_session_command(action.as_deref(), target.as_deref())?
|
|
|
}
|
|
}
|
|
|
|
|
+ SlashCommand::Sessions => {
|
|
|
|
|
+ println!("{}", render_session_list(&self.session.id)?);
|
|
|
|
|
+ false
|
|
|
|
|
+ }
|
|
|
SlashCommand::Unknown(name) => {
|
|
SlashCommand::Unknown(name) => {
|
|
|
eprintln!("unknown slash command: /{name}");
|
|
eprintln!("unknown slash command: /{name}");
|
|
|
false
|
|
false
|
|
@@ -1164,7 +1155,10 @@ impl LiveCli {
|
|
|
}
|
|
}
|
|
|
|
|
|
|
|
fn persist_session(&self) -> Result<(), Box<dyn std::error::Error>> {
|
|
fn persist_session(&self) -> Result<(), Box<dyn std::error::Error>> {
|
|
|
- self.runtime.session().save_to_path(&self.session.path)?;
|
|
|
|
|
|
|
+ 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)?;
|
|
|
Ok(())
|
|
Ok(())
|
|
|
}
|
|
}
|
|
|
|
|
|
|
@@ -1223,7 +1217,6 @@ impl LiveCli {
|
|
|
true,
|
|
true,
|
|
|
self.allowed_tools.clone(),
|
|
self.allowed_tools.clone(),
|
|
|
self.permission_mode,
|
|
self.permission_mode,
|
|
|
- self.color,
|
|
|
|
|
)?;
|
|
)?;
|
|
|
self.model.clone_from(&model);
|
|
self.model.clone_from(&model);
|
|
|
println!(
|
|
println!(
|
|
@@ -1266,7 +1259,6 @@ impl LiveCli {
|
|
|
true,
|
|
true,
|
|
|
self.allowed_tools.clone(),
|
|
self.allowed_tools.clone(),
|
|
|
self.permission_mode,
|
|
self.permission_mode,
|
|
|
- self.color,
|
|
|
|
|
)?;
|
|
)?;
|
|
|
println!(
|
|
println!(
|
|
|
"{}",
|
|
"{}",
|
|
@@ -1291,7 +1283,6 @@ impl LiveCli {
|
|
|
true,
|
|
true,
|
|
|
self.allowed_tools.clone(),
|
|
self.allowed_tools.clone(),
|
|
|
self.permission_mode,
|
|
self.permission_mode,
|
|
|
- self.color,
|
|
|
|
|
)?;
|
|
)?;
|
|
|
println!(
|
|
println!(
|
|
|
"Session cleared\n Mode fresh session\n Preserved model {}\n Permission mode {}\n Session {}",
|
|
"Session cleared\n Mode fresh session\n Preserved model {}\n Permission mode {}\n Session {}",
|
|
@@ -1312,13 +1303,20 @@ impl LiveCli {
|
|
|
session_path: Option<String>,
|
|
session_path: Option<String>,
|
|
|
) -> Result<bool, Box<dyn std::error::Error>> {
|
|
) -> Result<bool, Box<dyn std::error::Error>> {
|
|
|
let Some(session_ref) = session_path else {
|
|
let Some(session_ref) = session_path else {
|
|
|
- println!("Usage: /resume <session-path>");
|
|
|
|
|
|
|
+ println!("Usage: /resume <session-id-or-path>");
|
|
|
return Ok(false);
|
|
return Ok(false);
|
|
|
};
|
|
};
|
|
|
|
|
|
|
|
let handle = resolve_session_reference(&session_ref)?;
|
|
let handle = resolve_session_reference(&session_ref)?;
|
|
|
let session = Session::load_from_path(&handle.path)?;
|
|
let session = Session::load_from_path(&handle.path)?;
|
|
|
let message_count = session.messages.len();
|
|
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(
|
|
self.runtime = build_runtime(
|
|
|
session,
|
|
session,
|
|
|
self.model.clone(),
|
|
self.model.clone(),
|
|
@@ -1326,7 +1324,6 @@ impl LiveCli {
|
|
|
true,
|
|
true,
|
|
|
self.allowed_tools.clone(),
|
|
self.allowed_tools.clone(),
|
|
|
self.permission_mode,
|
|
self.permission_mode,
|
|
|
- self.color,
|
|
|
|
|
)?;
|
|
)?;
|
|
|
self.session = handle;
|
|
self.session = handle;
|
|
|
println!(
|
|
println!(
|
|
@@ -1396,6 +1393,13 @@ impl LiveCli {
|
|
|
let handle = resolve_session_reference(target)?;
|
|
let handle = resolve_session_reference(target)?;
|
|
|
let session = Session::load_from_path(&handle.path)?;
|
|
let session = Session::load_from_path(&handle.path)?;
|
|
|
let message_count = session.messages.len();
|
|
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(
|
|
self.runtime = build_runtime(
|
|
|
session,
|
|
session,
|
|
|
self.model.clone(),
|
|
self.model.clone(),
|
|
@@ -1403,7 +1407,6 @@ impl LiveCli {
|
|
|
true,
|
|
true,
|
|
|
self.allowed_tools.clone(),
|
|
self.allowed_tools.clone(),
|
|
|
self.permission_mode,
|
|
self.permission_mode,
|
|
|
- self.color,
|
|
|
|
|
)?;
|
|
)?;
|
|
|
self.session = handle;
|
|
self.session = handle;
|
|
|
println!(
|
|
println!(
|
|
@@ -1433,7 +1436,6 @@ impl LiveCli {
|
|
|
true,
|
|
true,
|
|
|
self.allowed_tools.clone(),
|
|
self.allowed_tools.clone(),
|
|
|
self.permission_mode,
|
|
self.permission_mode,
|
|
|
- self.color,
|
|
|
|
|
)?;
|
|
)?;
|
|
|
self.persist_session()?;
|
|
self.persist_session()?;
|
|
|
println!("{}", format_compact_report(removed, kept, skipped));
|
|
println!("{}", format_compact_report(removed, kept, skipped));
|
|
@@ -1442,8 +1444,10 @@ impl LiveCli {
|
|
|
}
|
|
}
|
|
|
|
|
|
|
|
fn sessions_dir() -> Result<PathBuf, Box<dyn std::error::Error>> {
|
|
fn sessions_dir() -> Result<PathBuf, Box<dyn std::error::Error>> {
|
|
|
- let cwd = env::current_dir()?;
|
|
|
|
|
- let path = cwd.join(".claude").join("sessions");
|
|
|
|
|
|
|
+ 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");
|
|
|
fs::create_dir_all(&path)?;
|
|
fs::create_dir_all(&path)?;
|
|
|
Ok(path)
|
|
Ok(path)
|
|
|
}
|
|
}
|
|
@@ -1464,8 +1468,19 @@ fn generate_session_id() -> String {
|
|
|
|
|
|
|
|
fn resolve_session_reference(reference: &str) -> Result<SessionHandle, Box<dyn std::error::Error>> {
|
|
fn resolve_session_reference(reference: &str) -> Result<SessionHandle, Box<dyn std::error::Error>> {
|
|
|
let direct = PathBuf::from(reference);
|
|
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() {
|
|
let path = if direct.exists() {
|
|
|
direct
|
|
direct
|
|
|
|
|
+ } else if expanded.exists() {
|
|
|
|
|
+ expanded
|
|
|
} else {
|
|
} else {
|
|
|
sessions_dir()?.join(format!("{reference}.json"))
|
|
sessions_dir()?.join(format!("{reference}.json"))
|
|
|
};
|
|
};
|
|
@@ -1495,9 +1510,11 @@ fn list_managed_sessions() -> Result<Vec<ManagedSessionSummary>, Box<dyn std::er
|
|
|
.and_then(|time| time.duration_since(UNIX_EPOCH).ok())
|
|
.and_then(|time| time.duration_since(UNIX_EPOCH).ok())
|
|
|
.map(|duration| duration.as_secs())
|
|
.map(|duration| duration.as_secs())
|
|
|
.unwrap_or_default();
|
|
.unwrap_or_default();
|
|
|
- let message_count = Session::load_from_path(&path)
|
|
|
|
|
- .map(|session| session.messages.len())
|
|
|
|
|
- .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 id = path
|
|
let id = path
|
|
|
.file_stem()
|
|
.file_stem()
|
|
|
.and_then(|value| value.to_str())
|
|
.and_then(|value| value.to_str())
|
|
@@ -1507,7 +1524,12 @@ fn list_managed_sessions() -> Result<Vec<ManagedSessionSummary>, Box<dyn std::er
|
|
|
id,
|
|
id,
|
|
|
path,
|
|
path,
|
|
|
modified_epoch_secs,
|
|
modified_epoch_secs,
|
|
|
- message_count,
|
|
|
|
|
|
|
+ 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()),
|
|
|
});
|
|
});
|
|
|
}
|
|
}
|
|
|
sessions.sort_by(|left, right| right.modified_epoch_secs.cmp(&left.modified_epoch_secs));
|
|
sessions.sort_by(|left, right| right.modified_epoch_secs.cmp(&left.modified_epoch_secs));
|
|
@@ -1530,17 +1552,99 @@ fn render_session_list(active_session_id: &str) -> Result<String, Box<dyn std::e
|
|
|
} else {
|
|
} else {
|
|
|
"○ saved"
|
|
"○ 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!(
|
|
lines.push(format!(
|
|
|
- " {id:<20} {marker:<10} msgs={msgs:<4} modified={modified} path={path}",
|
|
|
|
|
|
|
+ " {id:<20} {marker:<10} msgs={msgs:<4} model={model:<24} started={started} modified={modified} last={last_prompt} path={path}",
|
|
|
id = session.id,
|
|
id = session.id,
|
|
|
msgs = session.message_count,
|
|
msgs = session.message_count,
|
|
|
|
|
+ model = model,
|
|
|
|
|
+ started = started,
|
|
|
modified = session.modified_epoch_secs,
|
|
modified = session.modified_epoch_secs,
|
|
|
|
|
+ last_prompt = last_prompt,
|
|
|
path = session.path.display(),
|
|
path = session.path.display(),
|
|
|
));
|
|
));
|
|
|
}
|
|
}
|
|
|
Ok(lines.join("\n"))
|
|
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 {
|
|
fn render_repl_help() -> String {
|
|
|
[
|
|
[
|
|
|
"REPL".to_string(),
|
|
"REPL".to_string(),
|
|
@@ -1956,13 +2060,12 @@ fn build_runtime(
|
|
|
enable_tools: bool,
|
|
enable_tools: bool,
|
|
|
allowed_tools: Option<AllowedToolSet>,
|
|
allowed_tools: Option<AllowedToolSet>,
|
|
|
permission_mode: PermissionMode,
|
|
permission_mode: PermissionMode,
|
|
|
- color: bool,
|
|
|
|
|
) -> Result<ConversationRuntime<AnthropicRuntimeClient, CliToolExecutor>, Box<dyn std::error::Error>>
|
|
) -> Result<ConversationRuntime<AnthropicRuntimeClient, CliToolExecutor>, Box<dyn std::error::Error>>
|
|
|
{
|
|
{
|
|
|
Ok(ConversationRuntime::new(
|
|
Ok(ConversationRuntime::new(
|
|
|
session,
|
|
session,
|
|
|
- AnthropicRuntimeClient::new(model, enable_tools, allowed_tools.clone(), color)?,
|
|
|
|
|
- CliToolExecutor::new(allowed_tools, color),
|
|
|
|
|
|
|
+ AnthropicRuntimeClient::new(model, enable_tools, allowed_tools.clone())?,
|
|
|
|
|
+ CliToolExecutor::new(allowed_tools),
|
|
|
permission_policy(permission_mode),
|
|
permission_policy(permission_mode),
|
|
|
system_prompt,
|
|
system_prompt,
|
|
|
))
|
|
))
|
|
@@ -2020,7 +2123,6 @@ struct AnthropicRuntimeClient {
|
|
|
model: String,
|
|
model: String,
|
|
|
enable_tools: bool,
|
|
enable_tools: bool,
|
|
|
allowed_tools: Option<AllowedToolSet>,
|
|
allowed_tools: Option<AllowedToolSet>,
|
|
|
- color: bool,
|
|
|
|
|
}
|
|
}
|
|
|
|
|
|
|
|
impl AnthropicRuntimeClient {
|
|
impl AnthropicRuntimeClient {
|
|
@@ -2028,7 +2130,6 @@ impl AnthropicRuntimeClient {
|
|
|
model: String,
|
|
model: String,
|
|
|
enable_tools: bool,
|
|
enable_tools: bool,
|
|
|
allowed_tools: Option<AllowedToolSet>,
|
|
allowed_tools: Option<AllowedToolSet>,
|
|
|
- color: bool,
|
|
|
|
|
) -> Result<Self, Box<dyn std::error::Error>> {
|
|
) -> Result<Self, Box<dyn std::error::Error>> {
|
|
|
Ok(Self {
|
|
Ok(Self {
|
|
|
runtime: tokio::runtime::Runtime::new()?,
|
|
runtime: tokio::runtime::Runtime::new()?,
|
|
@@ -2036,7 +2137,6 @@ impl AnthropicRuntimeClient {
|
|
|
model,
|
|
model,
|
|
|
enable_tools,
|
|
enable_tools,
|
|
|
allowed_tools,
|
|
allowed_tools,
|
|
|
- color,
|
|
|
|
|
})
|
|
})
|
|
|
}
|
|
}
|
|
|
}
|
|
}
|
|
@@ -2073,7 +2173,6 @@ impl ApiClient for AnthropicRuntimeClient {
|
|
|
stream: true,
|
|
stream: true,
|
|
|
};
|
|
};
|
|
|
|
|
|
|
|
- let renderer = TerminalRenderer::with_color(self.color);
|
|
|
|
|
self.runtime.block_on(async {
|
|
self.runtime.block_on(async {
|
|
|
let mut stream = self
|
|
let mut stream = self
|
|
|
.client
|
|
.client
|
|
@@ -2093,18 +2192,11 @@ impl ApiClient for AnthropicRuntimeClient {
|
|
|
match event {
|
|
match event {
|
|
|
ApiStreamEvent::MessageStart(start) => {
|
|
ApiStreamEvent::MessageStart(start) => {
|
|
|
for block in start.message.content {
|
|
for block in start.message.content {
|
|
|
- push_output_block(
|
|
|
|
|
- &TerminalRenderer::with_color(true),
|
|
|
|
|
- block,
|
|
|
|
|
- &mut stdout,
|
|
|
|
|
- &mut events,
|
|
|
|
|
- &mut pending_tool,
|
|
|
|
|
- )?;
|
|
|
|
|
|
|
+ push_output_block(block, &mut stdout, &mut events, &mut pending_tool)?;
|
|
|
}
|
|
}
|
|
|
}
|
|
}
|
|
|
ApiStreamEvent::ContentBlockStart(start) => {
|
|
ApiStreamEvent::ContentBlockStart(start) => {
|
|
|
push_output_block(
|
|
push_output_block(
|
|
|
- &renderer,
|
|
|
|
|
start.content_block,
|
|
start.content_block,
|
|
|
&mut stdout,
|
|
&mut stdout,
|
|
|
&mut events,
|
|
&mut events,
|
|
@@ -2170,7 +2262,7 @@ impl ApiClient for AnthropicRuntimeClient {
|
|
|
})
|
|
})
|
|
|
.await
|
|
.await
|
|
|
.map_err(|error| RuntimeError::new(error.to_string()))?;
|
|
.map_err(|error| RuntimeError::new(error.to_string()))?;
|
|
|
- response_to_events(&renderer, response, &mut stdout)
|
|
|
|
|
|
|
+ response_to_events(response, &mut stdout)
|
|
|
})
|
|
})
|
|
|
}
|
|
}
|
|
|
}
|
|
}
|
|
@@ -2182,29 +2274,19 @@ fn slash_command_completion_candidates() -> Vec<String> {
|
|
|
.collect()
|
|
.collect()
|
|
|
}
|
|
}
|
|
|
|
|
|
|
|
-fn format_tool_call_start(renderer: &TerminalRenderer, name: &str, input: &str) -> String {
|
|
|
|
|
|
|
+fn format_tool_call_start(name: &str, input: &str) -> String {
|
|
|
format!(
|
|
format!(
|
|
|
- "{} {} {} {}",
|
|
|
|
|
- renderer.warning("Tool call:"),
|
|
|
|
|
- renderer.info(name),
|
|
|
|
|
- renderer.warning("args="),
|
|
|
|
|
|
|
+ "Tool call
|
|
|
|
|
+ Name {name}
|
|
|
|
|
+ Input {}",
|
|
|
summarize_tool_payload(input)
|
|
summarize_tool_payload(input)
|
|
|
)
|
|
)
|
|
|
}
|
|
}
|
|
|
|
|
|
|
|
-fn format_tool_result(
|
|
|
|
|
- renderer: &TerminalRenderer,
|
|
|
|
|
- name: &str,
|
|
|
|
|
- output: &str,
|
|
|
|
|
- is_error: bool,
|
|
|
|
|
-) -> String {
|
|
|
|
|
- let status = if is_error {
|
|
|
|
|
- renderer.error("error")
|
|
|
|
|
- } else {
|
|
|
|
|
- renderer.success("ok")
|
|
|
|
|
- };
|
|
|
|
|
|
|
+fn format_tool_result(name: &str, output: &str, is_error: bool) -> String {
|
|
|
|
|
+ let status = if is_error { "error" } else { "ok" };
|
|
|
format!(
|
|
format!(
|
|
|
- "### {} {}
|
|
|
|
|
|
|
+ "### Tool `{name}`
|
|
|
|
|
|
|
|
- Status: {status}
|
|
- Status: {status}
|
|
|
- Output:
|
|
- Output:
|
|
@@ -2213,8 +2295,6 @@ fn format_tool_result(
|
|
|
{}
|
|
{}
|
|
|
```
|
|
```
|
|
|
",
|
|
",
|
|
|
- renderer.warning("Tool"),
|
|
|
|
|
- renderer.info(format!("`{name}`")),
|
|
|
|
|
prettify_tool_payload(output)
|
|
prettify_tool_payload(output)
|
|
|
)
|
|
)
|
|
|
}
|
|
}
|
|
@@ -2245,7 +2325,6 @@ fn truncate_for_summary(value: &str, limit: usize) -> String {
|
|
|
}
|
|
}
|
|
|
|
|
|
|
|
fn push_output_block(
|
|
fn push_output_block(
|
|
|
- renderer: &TerminalRenderer,
|
|
|
|
|
block: OutputContentBlock,
|
|
block: OutputContentBlock,
|
|
|
out: &mut impl Write,
|
|
out: &mut impl Write,
|
|
|
events: &mut Vec<AssistantEvent>,
|
|
events: &mut Vec<AssistantEvent>,
|
|
@@ -2265,7 +2344,7 @@ fn push_output_block(
|
|
|
out,
|
|
out,
|
|
|
"
|
|
"
|
|
|
{}",
|
|
{}",
|
|
|
- format_tool_call_start(renderer, &name, &input.to_string())
|
|
|
|
|
|
|
+ format_tool_call_start(&name, &input.to_string())
|
|
|
)
|
|
)
|
|
|
.and_then(|()| out.flush())
|
|
.and_then(|()| out.flush())
|
|
|
.map_err(|error| RuntimeError::new(error.to_string()))?;
|
|
.map_err(|error| RuntimeError::new(error.to_string()))?;
|
|
@@ -2276,7 +2355,6 @@ fn push_output_block(
|
|
|
}
|
|
}
|
|
|
|
|
|
|
|
fn response_to_events(
|
|
fn response_to_events(
|
|
|
- renderer: &TerminalRenderer,
|
|
|
|
|
response: MessageResponse,
|
|
response: MessageResponse,
|
|
|
out: &mut impl Write,
|
|
out: &mut impl Write,
|
|
|
) -> Result<Vec<AssistantEvent>, RuntimeError> {
|
|
) -> Result<Vec<AssistantEvent>, RuntimeError> {
|
|
@@ -2284,7 +2362,7 @@ fn response_to_events(
|
|
|
let mut pending_tool = None;
|
|
let mut pending_tool = None;
|
|
|
|
|
|
|
|
for block in response.content {
|
|
for block in response.content {
|
|
|
- push_output_block(renderer, block, out, &mut events, &mut pending_tool)?;
|
|
|
|
|
|
|
+ push_output_block(block, out, &mut events, &mut pending_tool)?;
|
|
|
if let Some((id, name, input)) = pending_tool.take() {
|
|
if let Some((id, name, input)) = pending_tool.take() {
|
|
|
events.push(AssistantEvent::ToolUse { id, name, input });
|
|
events.push(AssistantEvent::ToolUse { id, name, input });
|
|
|
}
|
|
}
|
|
@@ -2306,9 +2384,9 @@ struct CliToolExecutor {
|
|
|
}
|
|
}
|
|
|
|
|
|
|
|
impl CliToolExecutor {
|
|
impl CliToolExecutor {
|
|
|
- fn new(allowed_tools: Option<AllowedToolSet>, color: bool) -> Self {
|
|
|
|
|
|
|
+ fn new(allowed_tools: Option<AllowedToolSet>) -> Self {
|
|
|
Self {
|
|
Self {
|
|
|
- renderer: TerminalRenderer::with_color(color),
|
|
|
|
|
|
|
+ renderer: TerminalRenderer::new(),
|
|
|
allowed_tools,
|
|
allowed_tools,
|
|
|
}
|
|
}
|
|
|
}
|
|
}
|
|
@@ -2329,14 +2407,14 @@ impl ToolExecutor for CliToolExecutor {
|
|
|
.map_err(|error| ToolError::new(format!("invalid tool input JSON: {error}")))?;
|
|
.map_err(|error| ToolError::new(format!("invalid tool input JSON: {error}")))?;
|
|
|
match execute_tool(tool_name, &value) {
|
|
match execute_tool(tool_name, &value) {
|
|
|
Ok(output) => {
|
|
Ok(output) => {
|
|
|
- let markdown = format_tool_result(&self.renderer, tool_name, &output, false);
|
|
|
|
|
|
|
+ let markdown = format_tool_result(tool_name, &output, false);
|
|
|
self.renderer
|
|
self.renderer
|
|
|
.stream_markdown(&markdown, &mut io::stdout())
|
|
.stream_markdown(&markdown, &mut io::stdout())
|
|
|
.map_err(|error| ToolError::new(error.to_string()))?;
|
|
.map_err(|error| ToolError::new(error.to_string()))?;
|
|
|
Ok(output)
|
|
Ok(output)
|
|
|
}
|
|
}
|
|
|
Err(error) => {
|
|
Err(error) => {
|
|
|
- let markdown = format_tool_result(&self.renderer, tool_name, &error, true);
|
|
|
|
|
|
|
+ let markdown = format_tool_result(tool_name, &error, true);
|
|
|
self.renderer
|
|
self.renderer
|
|
|
.stream_markdown(&markdown, &mut io::stdout())
|
|
.stream_markdown(&markdown, &mut io::stdout())
|
|
|
.map_err(|stream_error| ToolError::new(stream_error.to_string()))?;
|
|
.map_err(|stream_error| ToolError::new(stream_error.to_string()))?;
|
|
@@ -2422,7 +2500,6 @@ fn print_help() {
|
|
|
println!(" --output-format FORMAT Non-interactive output format: text or json");
|
|
println!(" --output-format FORMAT Non-interactive output format: text or json");
|
|
|
println!(" --permission-mode MODE Set read-only, workspace-write, or danger-full-access");
|
|
println!(" --permission-mode MODE Set read-only, workspace-write, or danger-full-access");
|
|
|
println!(" --allowedTools TOOLS Restrict enabled tools (repeatable; comma-separated aliases supported)");
|
|
println!(" --allowedTools TOOLS Restrict enabled tools (repeatable; comma-separated aliases supported)");
|
|
|
- println!(" --no-color Disable ANSI color output");
|
|
|
|
|
println!(" --version, -V Print version and build information locally");
|
|
println!(" --version, -V Print version and build information locally");
|
|
|
println!();
|
|
println!();
|
|
|
println!("Interactive slash commands:");
|
|
println!("Interactive slash commands:");
|
|
@@ -2445,92 +2522,76 @@ fn print_help() {
|
|
|
println!(" rusty-claude-cli login");
|
|
println!(" rusty-claude-cli login");
|
|
|
}
|
|
}
|
|
|
|
|
|
|
|
-#[cfg(test)]
|
|
|
|
|
-fn print_help_text_for_test() -> String {
|
|
|
|
|
- use std::fmt::Write as _;
|
|
|
|
|
-
|
|
|
|
|
- let mut output = String::new();
|
|
|
|
|
- let _ = writeln!(
|
|
|
|
|
- output,
|
|
|
|
|
- "rusty-claude-cli v{VERSION}
|
|
|
|
|
-"
|
|
|
|
|
- );
|
|
|
|
|
- let _ = writeln!(output, "Usage:");
|
|
|
|
|
- let _ = writeln!(
|
|
|
|
|
- output,
|
|
|
|
|
- " rusty-claude-cli [--model MODEL] [--allowedTools TOOL[,TOOL...]]"
|
|
|
|
|
- );
|
|
|
|
|
- let _ = writeln!(output, " Start the interactive REPL");
|
|
|
|
|
- let _ = writeln!(
|
|
|
|
|
- output,
|
|
|
|
|
- " rusty-claude-cli [--model MODEL] [--output-format text|json] prompt TEXT"
|
|
|
|
|
- );
|
|
|
|
|
- let _ = writeln!(output, " Send one prompt and exit");
|
|
|
|
|
- let _ = writeln!(
|
|
|
|
|
- output,
|
|
|
|
|
- " rusty-claude-cli [--model MODEL] [--output-format text|json] TEXT"
|
|
|
|
|
- );
|
|
|
|
|
- let _ = writeln!(output, " Shorthand non-interactive prompt mode");
|
|
|
|
|
- let _ = writeln!(
|
|
|
|
|
- output,
|
|
|
|
|
- " rusty-claude-cli --resume SESSION.json [/status] [/compact] [...]"
|
|
|
|
|
- );
|
|
|
|
|
- let _ = writeln!(
|
|
|
|
|
- output,
|
|
|
|
|
- " Inspect or maintain a saved session without entering the REPL"
|
|
|
|
|
- );
|
|
|
|
|
- let _ = writeln!(output, " rusty-claude-cli dump-manifests");
|
|
|
|
|
- let _ = writeln!(output, " rusty-claude-cli bootstrap-plan");
|
|
|
|
|
- let _ = writeln!(
|
|
|
|
|
- output,
|
|
|
|
|
- " rusty-claude-cli system-prompt [--cwd PATH] [--date YYYY-MM-DD]"
|
|
|
|
|
- );
|
|
|
|
|
- let _ = writeln!(output, " rusty-claude-cli login");
|
|
|
|
|
- let _ = writeln!(
|
|
|
|
|
- output,
|
|
|
|
|
- " rusty-claude-cli logout
|
|
|
|
|
-"
|
|
|
|
|
- );
|
|
|
|
|
- let _ = writeln!(output, "Flags:");
|
|
|
|
|
- let _ = writeln!(
|
|
|
|
|
- output,
|
|
|
|
|
- " --model MODEL Override the active model"
|
|
|
|
|
- );
|
|
|
|
|
- let _ = writeln!(
|
|
|
|
|
- output,
|
|
|
|
|
- " --output-format FORMAT Non-interactive output format: text or json"
|
|
|
|
|
- );
|
|
|
|
|
- let _ = writeln!(
|
|
|
|
|
- output,
|
|
|
|
|
- " --permission-mode MODE Set read-only, workspace-write, or danger-full-access"
|
|
|
|
|
- );
|
|
|
|
|
- let _ = writeln!(output, " --allowedTools TOOLS Restrict enabled tools (repeatable; comma-separated aliases supported)");
|
|
|
|
|
- let _ = writeln!(
|
|
|
|
|
- output,
|
|
|
|
|
- " --no-color Disable ANSI color output"
|
|
|
|
|
- );
|
|
|
|
|
- let _ = writeln!(
|
|
|
|
|
- output,
|
|
|
|
|
- " --version, -V Print version and build information locally"
|
|
|
|
|
- );
|
|
|
|
|
- output
|
|
|
|
|
-}
|
|
|
|
|
-
|
|
|
|
|
#[cfg(test)]
|
|
#[cfg(test)]
|
|
|
mod tests {
|
|
mod tests {
|
|
|
use super::{
|
|
use super::{
|
|
|
- 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,
|
|
|
|
|
|
|
+ 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,
|
|
|
};
|
|
};
|
|
|
- use crate::{print_help_text_for_test, render::TerminalRenderer};
|
|
|
|
|
- use runtime::{ContentBlock, ConversationMessage, MessageRole, PermissionMode};
|
|
|
|
|
|
|
+ use runtime::{ContentBlock, ConversationMessage, MessageRole, PermissionMode, Session};
|
|
|
|
|
+ use std::fs;
|
|
|
use std::path::{Path, PathBuf};
|
|
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();
|
|
|
|
|
+ }
|
|
|
|
|
+
|
|
|
#[test]
|
|
#[test]
|
|
|
fn defaults_to_repl_when_no_args() {
|
|
fn defaults_to_repl_when_no_args() {
|
|
|
assert_eq!(
|
|
assert_eq!(
|
|
@@ -2539,7 +2600,6 @@ mod tests {
|
|
|
model: DEFAULT_MODEL.to_string(),
|
|
model: DEFAULT_MODEL.to_string(),
|
|
|
allowed_tools: None,
|
|
allowed_tools: None,
|
|
|
permission_mode: PermissionMode::WorkspaceWrite,
|
|
permission_mode: PermissionMode::WorkspaceWrite,
|
|
|
- color: true,
|
|
|
|
|
}
|
|
}
|
|
|
);
|
|
);
|
|
|
}
|
|
}
|
|
@@ -2559,7 +2619,6 @@ mod tests {
|
|
|
output_format: CliOutputFormat::Text,
|
|
output_format: CliOutputFormat::Text,
|
|
|
allowed_tools: None,
|
|
allowed_tools: None,
|
|
|
permission_mode: PermissionMode::WorkspaceWrite,
|
|
permission_mode: PermissionMode::WorkspaceWrite,
|
|
|
- color: true,
|
|
|
|
|
}
|
|
}
|
|
|
);
|
|
);
|
|
|
}
|
|
}
|
|
@@ -2581,27 +2640,6 @@ mod tests {
|
|
|
output_format: CliOutputFormat::Json,
|
|
output_format: CliOutputFormat::Json,
|
|
|
allowed_tools: None,
|
|
allowed_tools: None,
|
|
|
permission_mode: PermissionMode::WorkspaceWrite,
|
|
permission_mode: PermissionMode::WorkspaceWrite,
|
|
|
- color: true,
|
|
|
|
|
- }
|
|
|
|
|
- );
|
|
|
|
|
- }
|
|
|
|
|
-
|
|
|
|
|
- #[test]
|
|
|
|
|
- fn parses_no_color_flag() {
|
|
|
|
|
- let args = vec![
|
|
|
|
|
- "--no-color".to_string(),
|
|
|
|
|
- "prompt".to_string(),
|
|
|
|
|
- "hello".to_string(),
|
|
|
|
|
- ];
|
|
|
|
|
- assert_eq!(
|
|
|
|
|
- parse_args(&args).expect("args should parse"),
|
|
|
|
|
- CliAction::Prompt {
|
|
|
|
|
- prompt: "hello".to_string(),
|
|
|
|
|
- model: DEFAULT_MODEL.to_string(),
|
|
|
|
|
- output_format: CliOutputFormat::Text,
|
|
|
|
|
- allowed_tools: None,
|
|
|
|
|
- permission_mode: PermissionMode::WorkspaceWrite,
|
|
|
|
|
- color: false,
|
|
|
|
|
}
|
|
}
|
|
|
);
|
|
);
|
|
|
}
|
|
}
|
|
@@ -2627,7 +2665,6 @@ mod tests {
|
|
|
model: DEFAULT_MODEL.to_string(),
|
|
model: DEFAULT_MODEL.to_string(),
|
|
|
allowed_tools: None,
|
|
allowed_tools: None,
|
|
|
permission_mode: PermissionMode::ReadOnly,
|
|
permission_mode: PermissionMode::ReadOnly,
|
|
|
- color: true,
|
|
|
|
|
}
|
|
}
|
|
|
);
|
|
);
|
|
|
}
|
|
}
|
|
@@ -2650,7 +2687,6 @@ mod tests {
|
|
|
.collect()
|
|
.collect()
|
|
|
),
|
|
),
|
|
|
permission_mode: PermissionMode::WorkspaceWrite,
|
|
permission_mode: PermissionMode::WorkspaceWrite,
|
|
|
- color: true,
|
|
|
|
|
}
|
|
}
|
|
|
);
|
|
);
|
|
|
}
|
|
}
|
|
@@ -2761,7 +2797,8 @@ mod tests {
|
|
|
assert!(help.contains("/permissions [read-only|workspace-write|danger-full-access]"));
|
|
assert!(help.contains("/permissions [read-only|workspace-write|danger-full-access]"));
|
|
|
assert!(help.contains("/clear [--confirm]"));
|
|
assert!(help.contains("/clear [--confirm]"));
|
|
|
assert!(help.contains("/cost"));
|
|
assert!(help.contains("/cost"));
|
|
|
- assert!(help.contains("/resume <session-path>"));
|
|
|
|
|
|
|
+ assert!(help.contains("/resume <session-id-or-path>"));
|
|
|
|
|
+ assert!(help.contains("/sessions"));
|
|
|
assert!(help.contains("/config [env|hooks|model]"));
|
|
assert!(help.contains("/config [env|hooks|model]"));
|
|
|
assert!(help.contains("/memory"));
|
|
assert!(help.contains("/memory"));
|
|
|
assert!(help.contains("/init"));
|
|
assert!(help.contains("/init"));
|
|
@@ -3047,21 +3084,17 @@ mod tests {
|
|
|
let help = render_repl_help();
|
|
let help = render_repl_help();
|
|
|
assert!(help.contains("Up/Down"));
|
|
assert!(help.contains("Up/Down"));
|
|
|
assert!(help.contains("Tab"));
|
|
assert!(help.contains("Tab"));
|
|
|
- assert!(print_help_text_for_test().contains("--no-color"));
|
|
|
|
|
assert!(help.contains("Shift+Enter/Ctrl+J"));
|
|
assert!(help.contains("Shift+Enter/Ctrl+J"));
|
|
|
}
|
|
}
|
|
|
|
|
|
|
|
#[test]
|
|
#[test]
|
|
|
fn tool_rendering_helpers_compact_output() {
|
|
fn tool_rendering_helpers_compact_output() {
|
|
|
- let renderer = TerminalRenderer::with_color(false);
|
|
|
|
|
- let start = format_tool_call_start(&renderer, "read_file", r#"{"path":"src/main.rs"}"#);
|
|
|
|
|
- assert!(start.contains("Tool call:"));
|
|
|
|
|
- assert!(start.contains("read_file"));
|
|
|
|
|
|
|
+ let start = format_tool_call_start("read_file", r#"{"path":"src/main.rs"}"#);
|
|
|
|
|
+ assert!(start.contains("Tool call"));
|
|
|
assert!(start.contains("src/main.rs"));
|
|
assert!(start.contains("src/main.rs"));
|
|
|
|
|
|
|
|
- let done = format_tool_result(&renderer, "read_file", r#"{"contents":"hello"}"#, false);
|
|
|
|
|
- assert!(done.contains("Tool"));
|
|
|
|
|
- assert!(done.contains("`read_file`"));
|
|
|
|
|
|
|
+ let done = format_tool_result("read_file", r#"{"contents":"hello"}"#, false);
|
|
|
|
|
+ assert!(done.contains("Tool `read_file`"));
|
|
|
assert!(done.contains("contents"));
|
|
assert!(done.contains("contents"));
|
|
|
}
|
|
}
|
|
|
}
|
|
}
|