main.rs 77 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586878889909192939495969798991001011021031041051061071081091101111121131141151161171181191201211221231241251261271281291301311321331341351361371381391401411421431441451461471481491501511521531541551561571581591601611621631641651661671681691701711721731741751761771781791801811821831841851861871881891901911921931941951961971981992002012022032042052062072082092102112122132142152162172182192202212222232242252262272282292302312322332342352362372382392402412422432442452462472482492502512522532542552562572582592602612622632642652662672682692702712722732742752762772782792802812822832842852862872882892902912922932942952962972982993003013023033043053063073083093103113123133143153163173183193203213223233243253263273283293303313323333343353363373383393403413423433443453463473483493503513523533543553563573583593603613623633643653663673683693703713723733743753763773783793803813823833843853863873883893903913923933943953963973983994004014024034044054064074084094104114124134144154164174184194204214224234244254264274284294304314324334344354364374384394404414424434444454464474484494504514524534544554564574584594604614624634644654664674684694704714724734744754764774784794804814824834844854864874884894904914924934944954964974984995005015025035045055065075085095105115125135145155165175185195205215225235245255265275285295305315325335345355365375385395405415425435445455465475485495505515525535545555565575585595605615625635645655665675685695705715725735745755765775785795805815825835845855865875885895905915925935945955965975985996006016026036046056066076086096106116126136146156166176186196206216226236246256266276286296306316326336346356366376386396406416426436446456466476486496506516526536546556566576586596606616626636646656666676686696706716726736746756766776786796806816826836846856866876886896906916926936946956966976986997007017027037047057067077087097107117127137147157167177187197207217227237247257267277287297307317327337347357367377387397407417427437447457467477487497507517527537547557567577587597607617627637647657667677687697707717727737747757767777787797807817827837847857867877887897907917927937947957967977987998008018028038048058068078088098108118128138148158168178188198208218228238248258268278288298308318328338348358368378388398408418428438448458468478488498508518528538548558568578588598608618628638648658668678688698708718728738748758768778788798808818828838848858868878888898908918928938948958968978988999009019029039049059069079089099109119129139149159169179189199209219229239249259269279289299309319329339349359369379389399409419429439449459469479489499509519529539549559569579589599609619629639649659669679689699709719729739749759769779789799809819829839849859869879889899909919929939949959969979989991000100110021003100410051006100710081009101010111012101310141015101610171018101910201021102210231024102510261027102810291030103110321033103410351036103710381039104010411042104310441045104610471048104910501051105210531054105510561057105810591060106110621063106410651066106710681069107010711072107310741075107610771078107910801081108210831084108510861087108810891090109110921093109410951096109710981099110011011102110311041105110611071108110911101111111211131114111511161117111811191120112111221123112411251126112711281129113011311132113311341135113611371138113911401141114211431144114511461147114811491150115111521153115411551156115711581159116011611162116311641165116611671168116911701171117211731174117511761177117811791180118111821183118411851186118711881189119011911192119311941195119611971198119912001201120212031204120512061207120812091210121112121213121412151216121712181219122012211222122312241225122612271228122912301231123212331234123512361237123812391240124112421243124412451246124712481249125012511252125312541255125612571258125912601261126212631264126512661267126812691270127112721273127412751276127712781279128012811282128312841285128612871288128912901291129212931294129512961297129812991300130113021303130413051306130713081309131013111312131313141315131613171318131913201321132213231324132513261327132813291330133113321333133413351336133713381339134013411342134313441345134613471348134913501351135213531354135513561357135813591360136113621363136413651366136713681369137013711372137313741375137613771378137913801381138213831384138513861387138813891390139113921393139413951396139713981399140014011402140314041405140614071408140914101411141214131414141514161417141814191420142114221423142414251426142714281429143014311432143314341435143614371438143914401441144214431444144514461447144814491450145114521453145414551456145714581459146014611462146314641465146614671468146914701471147214731474147514761477147814791480148114821483148414851486148714881489149014911492149314941495149614971498149915001501150215031504150515061507150815091510151115121513151415151516151715181519152015211522152315241525152615271528152915301531153215331534153515361537153815391540154115421543154415451546154715481549155015511552155315541555155615571558155915601561156215631564156515661567156815691570157115721573157415751576157715781579158015811582158315841585158615871588158915901591159215931594159515961597159815991600160116021603160416051606160716081609161016111612161316141615161616171618161916201621162216231624162516261627162816291630163116321633163416351636163716381639164016411642164316441645164616471648164916501651165216531654165516561657165816591660166116621663166416651666166716681669167016711672167316741675167616771678167916801681168216831684168516861687168816891690169116921693169416951696169716981699170017011702170317041705170617071708170917101711171217131714171517161717171817191720172117221723172417251726172717281729173017311732173317341735173617371738173917401741174217431744174517461747174817491750175117521753175417551756175717581759176017611762176317641765176617671768176917701771177217731774177517761777177817791780178117821783178417851786178717881789179017911792179317941795179617971798179918001801180218031804180518061807180818091810181118121813181418151816181718181819182018211822182318241825182618271828182918301831183218331834183518361837183818391840184118421843184418451846184718481849185018511852185318541855185618571858185918601861186218631864186518661867186818691870187118721873187418751876187718781879188018811882188318841885188618871888188918901891189218931894189518961897189818991900190119021903190419051906190719081909191019111912191319141915191619171918191919201921192219231924192519261927192819291930193119321933193419351936193719381939194019411942194319441945194619471948194919501951195219531954195519561957195819591960196119621963196419651966196719681969197019711972197319741975197619771978197919801981198219831984198519861987198819891990199119921993199419951996199719981999200020012002200320042005200620072008200920102011201220132014201520162017201820192020202120222023202420252026202720282029203020312032203320342035203620372038203920402041204220432044204520462047204820492050205120522053205420552056205720582059206020612062206320642065206620672068206920702071207220732074207520762077207820792080208120822083208420852086208720882089209020912092209320942095209620972098209921002101210221032104210521062107210821092110211121122113211421152116211721182119212021212122212321242125212621272128212921302131213221332134213521362137213821392140214121422143214421452146214721482149215021512152215321542155215621572158215921602161216221632164216521662167216821692170217121722173217421752176217721782179218021812182218321842185218621872188218921902191219221932194219521962197219821992200220122022203220422052206220722082209221022112212221322142215221622172218221922202221222222232224222522262227222822292230223122322233223422352236223722382239224022412242224322442245224622472248224922502251225222532254225522562257225822592260226122622263226422652266226722682269227022712272227322742275227622772278227922802281228222832284228522862287228822892290229122922293229422952296229722982299230023012302230323042305230623072308230923102311231223132314
  1. mod input;
  2. mod render;
  3. use std::env;
  4. use std::fs;
  5. use std::io::{self, Write};
  6. use std::path::{Path, PathBuf};
  7. use std::time::{SystemTime, UNIX_EPOCH};
  8. use api::{
  9. AnthropicClient, ContentBlockDelta, InputContentBlock, InputMessage, MessageRequest,
  10. MessageResponse, OutputContentBlock, StreamEvent as ApiStreamEvent, ToolChoice, ToolDefinition,
  11. ToolResultContentBlock,
  12. };
  13. use commands::{render_slash_command_help, resume_supported_slash_commands, SlashCommand};
  14. use compat_harness::{extract_manifest, UpstreamPaths};
  15. use render::{Spinner, TerminalRenderer};
  16. use runtime::{
  17. load_system_prompt, ApiClient, ApiRequest, AssistantEvent, CompactionConfig, ConfigLoader,
  18. ConfigSource, ContentBlock, ConversationMessage, ConversationRuntime, MessageRole,
  19. PermissionMode, PermissionPolicy, ProjectContext, RuntimeError, Session, TokenUsage, ToolError,
  20. ToolExecutor, UsageTracker,
  21. };
  22. use serde_json::json;
  23. use tools::{execute_tool, mvp_tool_specs};
  24. const DEFAULT_MODEL: &str = "claude-sonnet-4-20250514";
  25. const DEFAULT_MAX_TOKENS: u32 = 32;
  26. const DEFAULT_DATE: &str = "2026-03-31";
  27. const VERSION: &str = env!("CARGO_PKG_VERSION");
  28. const BUILD_TARGET: Option<&str> = option_env!("TARGET");
  29. const GIT_SHA: Option<&str> = option_env!("GIT_SHA");
  30. fn main() {
  31. if let Err(error) = run() {
  32. eprintln!(
  33. "error: {error}
  34. Run `rusty-claude-cli --help` for usage."
  35. );
  36. std::process::exit(1);
  37. }
  38. }
  39. fn run() -> Result<(), Box<dyn std::error::Error>> {
  40. let args: Vec<String> = env::args().skip(1).collect();
  41. match parse_args(&args)? {
  42. CliAction::DumpManifests => dump_manifests(),
  43. CliAction::BootstrapPlan => print_bootstrap_plan(),
  44. CliAction::PrintSystemPrompt { cwd, date } => print_system_prompt(cwd, date),
  45. CliAction::ResumeSession {
  46. session_path,
  47. commands,
  48. } => resume_session(&session_path, &commands),
  49. CliAction::Prompt {
  50. prompt,
  51. model,
  52. output_format,
  53. } => LiveCli::new(model, false)?.run_turn_with_output(&prompt, output_format)?,
  54. CliAction::Repl { model } => run_repl(model)?,
  55. CliAction::Help => print_help(),
  56. }
  57. Ok(())
  58. }
  59. #[derive(Debug, Clone, PartialEq, Eq)]
  60. enum CliAction {
  61. DumpManifests,
  62. BootstrapPlan,
  63. PrintSystemPrompt {
  64. cwd: PathBuf,
  65. date: String,
  66. },
  67. ResumeSession {
  68. session_path: PathBuf,
  69. commands: Vec<String>,
  70. },
  71. Prompt {
  72. prompt: String,
  73. model: String,
  74. output_format: CliOutputFormat,
  75. },
  76. Repl {
  77. model: String,
  78. },
  79. // prompt-mode formatting is only supported for non-interactive runs
  80. Help,
  81. }
  82. #[derive(Debug, Clone, Copy, PartialEq, Eq)]
  83. enum CliOutputFormat {
  84. Text,
  85. Json,
  86. }
  87. impl CliOutputFormat {
  88. fn parse(value: &str) -> Result<Self, String> {
  89. match value {
  90. "text" => Ok(Self::Text),
  91. "json" => Ok(Self::Json),
  92. other => Err(format!(
  93. "unsupported value for --output-format: {other} (expected text or json)"
  94. )),
  95. }
  96. }
  97. }
  98. fn parse_args(args: &[String]) -> Result<CliAction, String> {
  99. let mut model = DEFAULT_MODEL.to_string();
  100. let mut output_format = CliOutputFormat::Text;
  101. let mut rest = Vec::new();
  102. let mut index = 0;
  103. while index < args.len() {
  104. match args[index].as_str() {
  105. "--model" => {
  106. let value = args
  107. .get(index + 1)
  108. .ok_or_else(|| "missing value for --model".to_string())?;
  109. model.clone_from(value);
  110. index += 2;
  111. }
  112. flag if flag.starts_with("--model=") => {
  113. model = flag[8..].to_string();
  114. index += 1;
  115. }
  116. "--output-format" => {
  117. let value = args
  118. .get(index + 1)
  119. .ok_or_else(|| "missing value for --output-format".to_string())?;
  120. output_format = CliOutputFormat::parse(value)?;
  121. index += 2;
  122. }
  123. flag if flag.starts_with("--output-format=") => {
  124. output_format = CliOutputFormat::parse(&flag[16..])?;
  125. index += 1;
  126. }
  127. other => {
  128. rest.push(other.to_string());
  129. index += 1;
  130. }
  131. }
  132. }
  133. if rest.is_empty() {
  134. return Ok(CliAction::Repl { model });
  135. }
  136. if matches!(rest.first().map(String::as_str), Some("--help" | "-h")) {
  137. return Ok(CliAction::Help);
  138. }
  139. if rest.first().map(String::as_str) == Some("--resume") {
  140. return parse_resume_args(&rest[1..]);
  141. }
  142. match rest[0].as_str() {
  143. "dump-manifests" => Ok(CliAction::DumpManifests),
  144. "bootstrap-plan" => Ok(CliAction::BootstrapPlan),
  145. "system-prompt" => parse_system_prompt_args(&rest[1..]),
  146. "prompt" => {
  147. let prompt = rest[1..].join(" ");
  148. if prompt.trim().is_empty() {
  149. return Err("prompt subcommand requires a prompt string".to_string());
  150. }
  151. Ok(CliAction::Prompt {
  152. prompt,
  153. model,
  154. output_format,
  155. })
  156. }
  157. other if !other.starts_with('/') => Ok(CliAction::Prompt {
  158. prompt: rest.join(" "),
  159. model,
  160. output_format,
  161. }),
  162. other => Err(format!("unknown subcommand: {other}")),
  163. }
  164. }
  165. fn parse_system_prompt_args(args: &[String]) -> Result<CliAction, String> {
  166. let mut cwd = env::current_dir().map_err(|error| error.to_string())?;
  167. let mut date = DEFAULT_DATE.to_string();
  168. let mut index = 0;
  169. while index < args.len() {
  170. match args[index].as_str() {
  171. "--cwd" => {
  172. let value = args
  173. .get(index + 1)
  174. .ok_or_else(|| "missing value for --cwd".to_string())?;
  175. cwd = PathBuf::from(value);
  176. index += 2;
  177. }
  178. "--date" => {
  179. let value = args
  180. .get(index + 1)
  181. .ok_or_else(|| "missing value for --date".to_string())?;
  182. date.clone_from(value);
  183. index += 2;
  184. }
  185. other => return Err(format!("unknown system-prompt option: {other}")),
  186. }
  187. }
  188. Ok(CliAction::PrintSystemPrompt { cwd, date })
  189. }
  190. fn parse_resume_args(args: &[String]) -> Result<CliAction, String> {
  191. let session_path = args
  192. .first()
  193. .ok_or_else(|| "missing session path for --resume".to_string())
  194. .map(PathBuf::from)?;
  195. let commands = args[1..].to_vec();
  196. if commands
  197. .iter()
  198. .any(|command| !command.trim_start().starts_with('/'))
  199. {
  200. return Err("--resume trailing arguments must be slash commands".to_string());
  201. }
  202. Ok(CliAction::ResumeSession {
  203. session_path,
  204. commands,
  205. })
  206. }
  207. fn dump_manifests() {
  208. let workspace_dir = PathBuf::from(env!("CARGO_MANIFEST_DIR")).join("../..");
  209. let paths = UpstreamPaths::from_workspace_dir(&workspace_dir);
  210. match extract_manifest(&paths) {
  211. Ok(manifest) => {
  212. println!("commands: {}", manifest.commands.entries().len());
  213. println!("tools: {}", manifest.tools.entries().len());
  214. println!("bootstrap phases: {}", manifest.bootstrap.phases().len());
  215. }
  216. Err(error) => {
  217. eprintln!("failed to extract manifests: {error}");
  218. std::process::exit(1);
  219. }
  220. }
  221. }
  222. fn print_bootstrap_plan() {
  223. for phase in runtime::BootstrapPlan::claude_code_default().phases() {
  224. println!("- {phase:?}");
  225. }
  226. }
  227. fn print_system_prompt(cwd: PathBuf, date: String) {
  228. match load_system_prompt(cwd, date, env::consts::OS, "unknown") {
  229. Ok(sections) => println!("{}", sections.join("\n\n")),
  230. Err(error) => {
  231. eprintln!("failed to build system prompt: {error}");
  232. std::process::exit(1);
  233. }
  234. }
  235. }
  236. fn resume_session(session_path: &Path, commands: &[String]) {
  237. let session = match Session::load_from_path(session_path) {
  238. Ok(session) => session,
  239. Err(error) => {
  240. eprintln!("failed to restore session: {error}");
  241. std::process::exit(1);
  242. }
  243. };
  244. if commands.is_empty() {
  245. println!(
  246. "Restored session from {} ({} messages).",
  247. session_path.display(),
  248. session.messages.len()
  249. );
  250. return;
  251. }
  252. let mut session = session;
  253. for raw_command in commands {
  254. let Some(command) = SlashCommand::parse(raw_command) else {
  255. eprintln!("unsupported resumed command: {raw_command}");
  256. std::process::exit(2);
  257. };
  258. match run_resume_command(session_path, &session, &command) {
  259. Ok(ResumeCommandOutcome {
  260. session: next_session,
  261. message,
  262. }) => {
  263. session = next_session;
  264. if let Some(message) = message {
  265. println!("{message}");
  266. }
  267. }
  268. Err(error) => {
  269. eprintln!("{error}");
  270. std::process::exit(2);
  271. }
  272. }
  273. }
  274. }
  275. #[derive(Debug, Clone)]
  276. struct ResumeCommandOutcome {
  277. session: Session,
  278. message: Option<String>,
  279. }
  280. #[derive(Debug, Clone)]
  281. struct StatusContext {
  282. cwd: PathBuf,
  283. session_path: Option<PathBuf>,
  284. loaded_config_files: usize,
  285. discovered_config_files: usize,
  286. memory_file_count: usize,
  287. project_root: Option<PathBuf>,
  288. git_branch: Option<String>,
  289. }
  290. #[derive(Debug, Clone, Copy)]
  291. struct StatusUsage {
  292. message_count: usize,
  293. turns: u32,
  294. latest: TokenUsage,
  295. cumulative: TokenUsage,
  296. estimated_tokens: usize,
  297. }
  298. fn format_model_report(model: &str, message_count: usize, turns: u32) -> String {
  299. format!(
  300. "Model
  301. Current model {model}
  302. Session messages {message_count}
  303. Session turns {turns}
  304. Usage
  305. Inspect current model with /model
  306. Switch models with /model <name>"
  307. )
  308. }
  309. fn format_model_switch_report(previous: &str, next: &str, message_count: usize) -> String {
  310. format!(
  311. "Model updated
  312. Previous {previous}
  313. Current {next}
  314. Preserved msgs {message_count}"
  315. )
  316. }
  317. fn format_permissions_report(mode: &str) -> String {
  318. let modes = [
  319. ("read-only", "Read/search tools only", mode == "read-only"),
  320. (
  321. "workspace-write",
  322. "Edit files inside the workspace",
  323. mode == "workspace-write",
  324. ),
  325. (
  326. "danger-full-access",
  327. "Unrestricted tool access",
  328. mode == "danger-full-access",
  329. ),
  330. ]
  331. .into_iter()
  332. .map(|(name, description, is_current)| {
  333. let marker = if is_current {
  334. "● current"
  335. } else {
  336. "○ available"
  337. };
  338. format!(" {name:<18} {marker:<11} {description}")
  339. })
  340. .collect::<Vec<_>>()
  341. .join(
  342. "
  343. ",
  344. );
  345. format!(
  346. "Permissions
  347. Active mode {mode}
  348. Mode status live session default
  349. Modes
  350. {modes}
  351. Usage
  352. Inspect current mode with /permissions
  353. Switch modes with /permissions <mode>"
  354. )
  355. }
  356. fn format_permissions_switch_report(previous: &str, next: &str) -> String {
  357. format!(
  358. "Permissions updated
  359. Result mode switched
  360. Previous mode {previous}
  361. Active mode {next}
  362. Applies to subsequent tool calls
  363. Usage /permissions to inspect current mode"
  364. )
  365. }
  366. fn format_cost_report(usage: TokenUsage) -> String {
  367. format!(
  368. "Cost
  369. Input tokens {}
  370. Output tokens {}
  371. Cache create {}
  372. Cache read {}
  373. Total tokens {}",
  374. usage.input_tokens,
  375. usage.output_tokens,
  376. usage.cache_creation_input_tokens,
  377. usage.cache_read_input_tokens,
  378. usage.total_tokens(),
  379. )
  380. }
  381. fn format_resume_report(session_path: &str, message_count: usize, turns: u32) -> String {
  382. format!(
  383. "Session resumed
  384. Session file {session_path}
  385. Messages {message_count}
  386. Turns {turns}"
  387. )
  388. }
  389. fn format_init_report(path: &Path, created: bool) -> String {
  390. if created {
  391. format!(
  392. "Init
  393. CLAUDE.md {}
  394. Result created
  395. Next step Review and tailor the generated guidance",
  396. path.display()
  397. )
  398. } else {
  399. format!(
  400. "Init
  401. CLAUDE.md {}
  402. Result skipped (already exists)
  403. Next step Edit the existing file intentionally if workflows changed",
  404. path.display()
  405. )
  406. }
  407. }
  408. fn format_compact_report(removed: usize, resulting_messages: usize, skipped: bool) -> String {
  409. if skipped {
  410. format!(
  411. "Compact
  412. Result skipped
  413. Reason session below compaction threshold
  414. Messages kept {resulting_messages}"
  415. )
  416. } else {
  417. format!(
  418. "Compact
  419. Result compacted
  420. Messages removed {removed}
  421. Messages kept {resulting_messages}"
  422. )
  423. }
  424. }
  425. fn parse_git_status_metadata(status: Option<&str>) -> (Option<PathBuf>, Option<String>) {
  426. let Some(status) = status else {
  427. return (None, None);
  428. };
  429. let branch = status.lines().next().and_then(|line| {
  430. line.strip_prefix("## ")
  431. .map(|line| {
  432. line.split(['.', ' '])
  433. .next()
  434. .unwrap_or_default()
  435. .to_string()
  436. })
  437. .filter(|value| !value.is_empty())
  438. });
  439. let project_root = find_git_root().ok();
  440. (project_root, branch)
  441. }
  442. fn find_git_root() -> Result<PathBuf, Box<dyn std::error::Error>> {
  443. let output = std::process::Command::new("git")
  444. .args(["rev-parse", "--show-toplevel"])
  445. .current_dir(env::current_dir()?)
  446. .output()?;
  447. if !output.status.success() {
  448. return Err("not a git repository".into());
  449. }
  450. let path = String::from_utf8(output.stdout)?.trim().to_string();
  451. if path.is_empty() {
  452. return Err("empty git root".into());
  453. }
  454. Ok(PathBuf::from(path))
  455. }
  456. #[allow(clippy::too_many_lines)]
  457. fn run_resume_command(
  458. session_path: &Path,
  459. session: &Session,
  460. command: &SlashCommand,
  461. ) -> Result<ResumeCommandOutcome, Box<dyn std::error::Error>> {
  462. match command {
  463. SlashCommand::Help => Ok(ResumeCommandOutcome {
  464. session: session.clone(),
  465. message: Some(render_repl_help()),
  466. }),
  467. SlashCommand::Compact => {
  468. let result = runtime::compact_session(
  469. session,
  470. CompactionConfig {
  471. max_estimated_tokens: 0,
  472. ..CompactionConfig::default()
  473. },
  474. );
  475. let removed = result.removed_message_count;
  476. let kept = result.compacted_session.messages.len();
  477. let skipped = removed == 0;
  478. result.compacted_session.save_to_path(session_path)?;
  479. Ok(ResumeCommandOutcome {
  480. session: result.compacted_session,
  481. message: Some(format_compact_report(removed, kept, skipped)),
  482. })
  483. }
  484. SlashCommand::Clear { confirm } => {
  485. if !confirm {
  486. return Ok(ResumeCommandOutcome {
  487. session: session.clone(),
  488. message: Some(
  489. "clear: confirmation required; rerun with /clear --confirm".to_string(),
  490. ),
  491. });
  492. }
  493. let cleared = Session::new();
  494. cleared.save_to_path(session_path)?;
  495. Ok(ResumeCommandOutcome {
  496. session: cleared,
  497. message: Some(format!(
  498. "Cleared resumed session file {}.",
  499. session_path.display()
  500. )),
  501. })
  502. }
  503. SlashCommand::Status => {
  504. let tracker = UsageTracker::from_session(session);
  505. let usage = tracker.cumulative_usage();
  506. Ok(ResumeCommandOutcome {
  507. session: session.clone(),
  508. message: Some(format_status_report(
  509. "restored-session",
  510. StatusUsage {
  511. message_count: session.messages.len(),
  512. turns: tracker.turns(),
  513. latest: tracker.current_turn_usage(),
  514. cumulative: usage,
  515. estimated_tokens: 0,
  516. },
  517. permission_mode_label(),
  518. &status_context(Some(session_path))?,
  519. )),
  520. })
  521. }
  522. SlashCommand::Cost => {
  523. let usage = UsageTracker::from_session(session).cumulative_usage();
  524. Ok(ResumeCommandOutcome {
  525. session: session.clone(),
  526. message: Some(format_cost_report(usage)),
  527. })
  528. }
  529. SlashCommand::Config { section } => Ok(ResumeCommandOutcome {
  530. session: session.clone(),
  531. message: Some(render_config_report(section.as_deref())?),
  532. }),
  533. SlashCommand::Memory => Ok(ResumeCommandOutcome {
  534. session: session.clone(),
  535. message: Some(render_memory_report()?),
  536. }),
  537. SlashCommand::Init => Ok(ResumeCommandOutcome {
  538. session: session.clone(),
  539. message: Some(init_claude_md()?),
  540. }),
  541. SlashCommand::Diff => Ok(ResumeCommandOutcome {
  542. session: session.clone(),
  543. message: Some(render_diff_report()?),
  544. }),
  545. SlashCommand::Version => Ok(ResumeCommandOutcome {
  546. session: session.clone(),
  547. message: Some(render_version_report()),
  548. }),
  549. SlashCommand::Export { path } => {
  550. let export_path = resolve_export_path(path.as_deref(), session)?;
  551. fs::write(&export_path, render_export_text(session))?;
  552. Ok(ResumeCommandOutcome {
  553. session: session.clone(),
  554. message: Some(format!(
  555. "Export\n Result wrote transcript\n File {}\n Messages {}",
  556. export_path.display(),
  557. session.messages.len(),
  558. )),
  559. })
  560. }
  561. SlashCommand::Resume { .. }
  562. | SlashCommand::Model { .. }
  563. | SlashCommand::Permissions { .. }
  564. | SlashCommand::Session { .. }
  565. | SlashCommand::Unknown(_) => Err("unsupported resumed slash command".into()),
  566. }
  567. }
  568. fn run_repl(model: String) -> Result<(), Box<dyn std::error::Error>> {
  569. let mut cli = LiveCli::new(model, true)?;
  570. let editor = input::LineEditor::new("› ");
  571. println!("{}", cli.startup_banner());
  572. while let Some(input) = editor.read_line()? {
  573. let trimmed = input.trim();
  574. if trimmed.is_empty() {
  575. continue;
  576. }
  577. if matches!(trimmed, "/exit" | "/quit") {
  578. break;
  579. }
  580. if let Some(command) = SlashCommand::parse(trimmed) {
  581. cli.handle_repl_command(command)?;
  582. continue;
  583. }
  584. cli.run_turn(trimmed)?;
  585. }
  586. Ok(())
  587. }
  588. #[derive(Debug, Clone)]
  589. struct SessionHandle {
  590. id: String,
  591. path: PathBuf,
  592. }
  593. #[derive(Debug, Clone)]
  594. struct ManagedSessionSummary {
  595. id: String,
  596. path: PathBuf,
  597. modified_epoch_secs: u64,
  598. message_count: usize,
  599. }
  600. struct LiveCli {
  601. model: String,
  602. system_prompt: Vec<String>,
  603. runtime: ConversationRuntime<AnthropicRuntimeClient, CliToolExecutor>,
  604. session: SessionHandle,
  605. }
  606. impl LiveCli {
  607. fn new(model: String, enable_tools: bool) -> Result<Self, Box<dyn std::error::Error>> {
  608. let system_prompt = build_system_prompt()?;
  609. let session = create_managed_session_handle()?;
  610. let runtime = build_runtime(
  611. Session::new(),
  612. model.clone(),
  613. system_prompt.clone(),
  614. enable_tools,
  615. )?;
  616. let cli = Self {
  617. model,
  618. system_prompt,
  619. runtime,
  620. session,
  621. };
  622. cli.persist_session()?;
  623. Ok(cli)
  624. }
  625. fn startup_banner(&self) -> String {
  626. format!(
  627. "Rusty Claude CLI\n Model {}\n Working directory {}\n Session {}\n\nType /help for commands. Shift+Enter or Ctrl+J inserts a newline.",
  628. self.model,
  629. env::current_dir().map_or_else(
  630. |_| "<unknown>".to_string(),
  631. |path| path.display().to_string(),
  632. ),
  633. self.session.id,
  634. )
  635. }
  636. fn run_turn(&mut self, input: &str) -> Result<(), Box<dyn std::error::Error>> {
  637. let mut spinner = Spinner::new();
  638. let mut stdout = io::stdout();
  639. spinner.tick(
  640. "Waiting for Claude",
  641. TerminalRenderer::new().color_theme(),
  642. &mut stdout,
  643. )?;
  644. let result = self.runtime.run_turn(input, None);
  645. match result {
  646. Ok(_) => {
  647. spinner.finish(
  648. "Claude response complete",
  649. TerminalRenderer::new().color_theme(),
  650. &mut stdout,
  651. )?;
  652. println!();
  653. self.persist_session()?;
  654. Ok(())
  655. }
  656. Err(error) => {
  657. spinner.fail(
  658. "Claude request failed",
  659. TerminalRenderer::new().color_theme(),
  660. &mut stdout,
  661. )?;
  662. Err(Box::new(error))
  663. }
  664. }
  665. }
  666. fn run_turn_with_output(
  667. &mut self,
  668. input: &str,
  669. output_format: CliOutputFormat,
  670. ) -> Result<(), Box<dyn std::error::Error>> {
  671. match output_format {
  672. CliOutputFormat::Text => self.run_turn(input),
  673. CliOutputFormat::Json => self.run_prompt_json(input),
  674. }
  675. }
  676. fn run_prompt_json(&mut self, input: &str) -> Result<(), Box<dyn std::error::Error>> {
  677. let client = AnthropicClient::from_env()?;
  678. let request = MessageRequest {
  679. model: self.model.clone(),
  680. max_tokens: DEFAULT_MAX_TOKENS,
  681. messages: vec![InputMessage {
  682. role: "user".to_string(),
  683. content: vec![InputContentBlock::Text {
  684. text: input.to_string(),
  685. }],
  686. }],
  687. system: (!self.system_prompt.is_empty()).then(|| self.system_prompt.join("\n\n")),
  688. tools: None,
  689. tool_choice: None,
  690. stream: false,
  691. };
  692. let runtime = tokio::runtime::Runtime::new()?;
  693. let response = runtime.block_on(client.send_message(&request))?;
  694. let text = response
  695. .content
  696. .iter()
  697. .filter_map(|block| match block {
  698. OutputContentBlock::Text { text } => Some(text.as_str()),
  699. OutputContentBlock::ToolUse { .. } => None,
  700. })
  701. .collect::<Vec<_>>()
  702. .join("");
  703. println!(
  704. "{}",
  705. json!({
  706. "message": text,
  707. "model": self.model,
  708. "usage": {
  709. "input_tokens": response.usage.input_tokens,
  710. "output_tokens": response.usage.output_tokens,
  711. "cache_creation_input_tokens": response.usage.cache_creation_input_tokens,
  712. "cache_read_input_tokens": response.usage.cache_read_input_tokens,
  713. }
  714. })
  715. );
  716. Ok(())
  717. }
  718. fn handle_repl_command(
  719. &mut self,
  720. command: SlashCommand,
  721. ) -> Result<(), Box<dyn std::error::Error>> {
  722. match command {
  723. SlashCommand::Help => println!("{}", render_repl_help()),
  724. SlashCommand::Status => self.print_status(),
  725. SlashCommand::Compact => self.compact()?,
  726. SlashCommand::Model { model } => self.set_model(model)?,
  727. SlashCommand::Permissions { mode } => self.set_permissions(mode)?,
  728. SlashCommand::Clear { confirm } => self.clear_session(confirm)?,
  729. SlashCommand::Cost => self.print_cost(),
  730. SlashCommand::Resume { session_path } => self.resume_session(session_path)?,
  731. SlashCommand::Config { section } => Self::print_config(section.as_deref())?,
  732. SlashCommand::Memory => Self::print_memory()?,
  733. SlashCommand::Init => Self::run_init()?,
  734. SlashCommand::Diff => Self::print_diff()?,
  735. SlashCommand::Version => Self::print_version(),
  736. SlashCommand::Export { path } => self.export_session(path.as_deref())?,
  737. SlashCommand::Session { action, target } => {
  738. self.handle_session_command(action.as_deref(), target.as_deref())?;
  739. }
  740. SlashCommand::Unknown(name) => eprintln!("unknown slash command: /{name}"),
  741. }
  742. Ok(())
  743. }
  744. fn persist_session(&self) -> Result<(), Box<dyn std::error::Error>> {
  745. self.runtime.session().save_to_path(&self.session.path)?;
  746. Ok(())
  747. }
  748. fn print_status(&self) {
  749. let cumulative = self.runtime.usage().cumulative_usage();
  750. let latest = self.runtime.usage().current_turn_usage();
  751. println!(
  752. "{}",
  753. format_status_report(
  754. &self.model,
  755. StatusUsage {
  756. message_count: self.runtime.session().messages.len(),
  757. turns: self.runtime.usage().turns(),
  758. latest,
  759. cumulative,
  760. estimated_tokens: self.runtime.estimated_tokens(),
  761. },
  762. permission_mode_label(),
  763. &status_context(Some(&self.session.path)).expect("status context should load"),
  764. )
  765. );
  766. }
  767. fn set_model(&mut self, model: Option<String>) -> Result<(), Box<dyn std::error::Error>> {
  768. let Some(model) = model else {
  769. println!(
  770. "{}",
  771. format_model_report(
  772. &self.model,
  773. self.runtime.session().messages.len(),
  774. self.runtime.usage().turns(),
  775. )
  776. );
  777. return Ok(());
  778. };
  779. if model == self.model {
  780. println!(
  781. "{}",
  782. format_model_report(
  783. &self.model,
  784. self.runtime.session().messages.len(),
  785. self.runtime.usage().turns(),
  786. )
  787. );
  788. return Ok(());
  789. }
  790. let previous = self.model.clone();
  791. let session = self.runtime.session().clone();
  792. let message_count = session.messages.len();
  793. self.runtime = build_runtime(session, model.clone(), self.system_prompt.clone(), true)?;
  794. self.model.clone_from(&model);
  795. self.persist_session()?;
  796. println!(
  797. "{}",
  798. format_model_switch_report(&previous, &model, message_count)
  799. );
  800. Ok(())
  801. }
  802. fn set_permissions(&mut self, mode: Option<String>) -> Result<(), Box<dyn std::error::Error>> {
  803. let Some(mode) = mode else {
  804. println!("{}", format_permissions_report(permission_mode_label()));
  805. return Ok(());
  806. };
  807. let normalized = normalize_permission_mode(&mode).ok_or_else(|| {
  808. format!(
  809. "unsupported permission mode '{mode}'. Use read-only, workspace-write, or danger-full-access."
  810. )
  811. })?;
  812. if normalized == permission_mode_label() {
  813. println!("{}", format_permissions_report(normalized));
  814. return Ok(());
  815. }
  816. let previous = permission_mode_label().to_string();
  817. let session = self.runtime.session().clone();
  818. self.runtime = build_runtime_with_permission_mode(
  819. session,
  820. self.model.clone(),
  821. self.system_prompt.clone(),
  822. true,
  823. normalized,
  824. )?;
  825. self.persist_session()?;
  826. println!(
  827. "{}",
  828. format_permissions_switch_report(&previous, normalized)
  829. );
  830. Ok(())
  831. }
  832. fn clear_session(&mut self, confirm: bool) -> Result<(), Box<dyn std::error::Error>> {
  833. if !confirm {
  834. println!(
  835. "clear: confirmation required; run /clear --confirm to start a fresh session."
  836. );
  837. return Ok(());
  838. }
  839. self.session = create_managed_session_handle()?;
  840. self.runtime = build_runtime_with_permission_mode(
  841. Session::new(),
  842. self.model.clone(),
  843. self.system_prompt.clone(),
  844. true,
  845. permission_mode_label(),
  846. )?;
  847. self.persist_session()?;
  848. println!(
  849. "Session cleared\n Mode fresh session\n Preserved model {}\n Permission mode {}\n Session {}",
  850. self.model,
  851. permission_mode_label(),
  852. self.session.id,
  853. );
  854. Ok(())
  855. }
  856. fn print_cost(&self) {
  857. let cumulative = self.runtime.usage().cumulative_usage();
  858. println!("{}", format_cost_report(cumulative));
  859. }
  860. fn resume_session(
  861. &mut self,
  862. session_path: Option<String>,
  863. ) -> Result<(), Box<dyn std::error::Error>> {
  864. let Some(session_ref) = session_path else {
  865. println!("Usage: /resume <session-path>");
  866. return Ok(());
  867. };
  868. let handle = resolve_session_reference(&session_ref)?;
  869. let session = Session::load_from_path(&handle.path)?;
  870. let message_count = session.messages.len();
  871. self.runtime = build_runtime_with_permission_mode(
  872. session,
  873. self.model.clone(),
  874. self.system_prompt.clone(),
  875. true,
  876. permission_mode_label(),
  877. )?;
  878. self.session = handle;
  879. self.persist_session()?;
  880. println!(
  881. "{}",
  882. format_resume_report(
  883. &self.session.path.display().to_string(),
  884. message_count,
  885. self.runtime.usage().turns(),
  886. )
  887. );
  888. Ok(())
  889. }
  890. fn print_config(section: Option<&str>) -> Result<(), Box<dyn std::error::Error>> {
  891. println!("{}", render_config_report(section)?);
  892. Ok(())
  893. }
  894. fn print_memory() -> Result<(), Box<dyn std::error::Error>> {
  895. println!("{}", render_memory_report()?);
  896. Ok(())
  897. }
  898. fn run_init() -> Result<(), Box<dyn std::error::Error>> {
  899. println!("{}", init_claude_md()?);
  900. Ok(())
  901. }
  902. fn print_diff() -> Result<(), Box<dyn std::error::Error>> {
  903. println!("{}", render_diff_report()?);
  904. Ok(())
  905. }
  906. fn print_version() {
  907. println!("{}", render_version_report());
  908. }
  909. fn export_session(
  910. &self,
  911. requested_path: Option<&str>,
  912. ) -> Result<(), Box<dyn std::error::Error>> {
  913. let export_path = resolve_export_path(requested_path, self.runtime.session())?;
  914. fs::write(&export_path, render_export_text(self.runtime.session()))?;
  915. println!(
  916. "Export\n Result wrote transcript\n File {}\n Messages {}",
  917. export_path.display(),
  918. self.runtime.session().messages.len(),
  919. );
  920. Ok(())
  921. }
  922. fn handle_session_command(
  923. &mut self,
  924. action: Option<&str>,
  925. target: Option<&str>,
  926. ) -> Result<(), Box<dyn std::error::Error>> {
  927. match action {
  928. None | Some("list") => {
  929. println!("{}", render_session_list(&self.session.id)?);
  930. Ok(())
  931. }
  932. Some("switch") => {
  933. let Some(target) = target else {
  934. println!("Usage: /session switch <session-id>");
  935. return Ok(());
  936. };
  937. let handle = resolve_session_reference(target)?;
  938. let session = Session::load_from_path(&handle.path)?;
  939. let message_count = session.messages.len();
  940. self.runtime = build_runtime_with_permission_mode(
  941. session,
  942. self.model.clone(),
  943. self.system_prompt.clone(),
  944. true,
  945. permission_mode_label(),
  946. )?;
  947. self.session = handle;
  948. self.persist_session()?;
  949. println!(
  950. "Session switched\n Active session {}\n File {}\n Messages {}",
  951. self.session.id,
  952. self.session.path.display(),
  953. message_count,
  954. );
  955. Ok(())
  956. }
  957. Some(other) => {
  958. println!("Unknown /session action '{other}'. Use /session list or /session switch <session-id>.");
  959. Ok(())
  960. }
  961. }
  962. }
  963. fn compact(&mut self) -> Result<(), Box<dyn std::error::Error>> {
  964. let result = self.runtime.compact(CompactionConfig::default());
  965. let removed = result.removed_message_count;
  966. let kept = result.compacted_session.messages.len();
  967. let skipped = removed == 0;
  968. self.runtime = build_runtime_with_permission_mode(
  969. result.compacted_session,
  970. self.model.clone(),
  971. self.system_prompt.clone(),
  972. true,
  973. permission_mode_label(),
  974. )?;
  975. self.persist_session()?;
  976. println!("{}", format_compact_report(removed, kept, skipped));
  977. Ok(())
  978. }
  979. }
  980. fn sessions_dir() -> Result<PathBuf, Box<dyn std::error::Error>> {
  981. let cwd = env::current_dir()?;
  982. let path = cwd.join(".claude").join("sessions");
  983. fs::create_dir_all(&path)?;
  984. Ok(path)
  985. }
  986. fn create_managed_session_handle() -> Result<SessionHandle, Box<dyn std::error::Error>> {
  987. let id = generate_session_id();
  988. let path = sessions_dir()?.join(format!("{id}.json"));
  989. Ok(SessionHandle { id, path })
  990. }
  991. fn generate_session_id() -> String {
  992. let millis = SystemTime::now()
  993. .duration_since(UNIX_EPOCH)
  994. .map(|duration| duration.as_millis())
  995. .unwrap_or_default();
  996. format!("session-{millis}")
  997. }
  998. fn resolve_session_reference(reference: &str) -> Result<SessionHandle, Box<dyn std::error::Error>> {
  999. let direct = PathBuf::from(reference);
  1000. let path = if direct.exists() {
  1001. direct
  1002. } else {
  1003. sessions_dir()?.join(format!("{reference}.json"))
  1004. };
  1005. if !path.exists() {
  1006. return Err(format!("session not found: {reference}").into());
  1007. }
  1008. let id = path
  1009. .file_stem()
  1010. .and_then(|value| value.to_str())
  1011. .unwrap_or(reference)
  1012. .to_string();
  1013. Ok(SessionHandle { id, path })
  1014. }
  1015. fn list_managed_sessions() -> Result<Vec<ManagedSessionSummary>, Box<dyn std::error::Error>> {
  1016. let mut sessions = Vec::new();
  1017. for entry in fs::read_dir(sessions_dir()?)? {
  1018. let entry = entry?;
  1019. let path = entry.path();
  1020. if path.extension().and_then(|ext| ext.to_str()) != Some("json") {
  1021. continue;
  1022. }
  1023. let metadata = entry.metadata()?;
  1024. let modified_epoch_secs = metadata
  1025. .modified()
  1026. .ok()
  1027. .and_then(|time| time.duration_since(UNIX_EPOCH).ok())
  1028. .map(|duration| duration.as_secs())
  1029. .unwrap_or_default();
  1030. let message_count = Session::load_from_path(&path)
  1031. .map(|session| session.messages.len())
  1032. .unwrap_or_default();
  1033. let id = path
  1034. .file_stem()
  1035. .and_then(|value| value.to_str())
  1036. .unwrap_or("unknown")
  1037. .to_string();
  1038. sessions.push(ManagedSessionSummary {
  1039. id,
  1040. path,
  1041. modified_epoch_secs,
  1042. message_count,
  1043. });
  1044. }
  1045. sessions.sort_by(|left, right| right.modified_epoch_secs.cmp(&left.modified_epoch_secs));
  1046. Ok(sessions)
  1047. }
  1048. fn render_session_list(active_session_id: &str) -> Result<String, Box<dyn std::error::Error>> {
  1049. let sessions = list_managed_sessions()?;
  1050. let mut lines = vec![
  1051. "Sessions".to_string(),
  1052. format!(" Directory {}", sessions_dir()?.display()),
  1053. ];
  1054. if sessions.is_empty() {
  1055. lines.push(" No managed sessions saved yet.".to_string());
  1056. return Ok(lines.join("\n"));
  1057. }
  1058. for session in sessions {
  1059. let marker = if session.id == active_session_id {
  1060. "● current"
  1061. } else {
  1062. "○ saved"
  1063. };
  1064. lines.push(format!(
  1065. " {id:<20} {marker:<10} msgs={msgs:<4} modified={modified} path={path}",
  1066. id = session.id,
  1067. msgs = session.message_count,
  1068. modified = session.modified_epoch_secs,
  1069. path = session.path.display(),
  1070. ));
  1071. }
  1072. Ok(lines.join("\n"))
  1073. }
  1074. fn render_repl_help() -> String {
  1075. [
  1076. "REPL".to_string(),
  1077. " /exit Quit the REPL".to_string(),
  1078. " /quit Quit the REPL".to_string(),
  1079. String::new(),
  1080. render_slash_command_help(),
  1081. ]
  1082. .join(
  1083. "
  1084. ",
  1085. )
  1086. }
  1087. fn status_context(
  1088. session_path: Option<&Path>,
  1089. ) -> Result<StatusContext, Box<dyn std::error::Error>> {
  1090. let cwd = env::current_dir()?;
  1091. let loader = ConfigLoader::default_for(&cwd);
  1092. let discovered_config_files = loader.discover().len();
  1093. let runtime_config = loader.load()?;
  1094. let project_context = ProjectContext::discover_with_git(&cwd, DEFAULT_DATE)?;
  1095. let (project_root, git_branch) =
  1096. parse_git_status_metadata(project_context.git_status.as_deref());
  1097. Ok(StatusContext {
  1098. cwd,
  1099. session_path: session_path.map(Path::to_path_buf),
  1100. loaded_config_files: runtime_config.loaded_entries().len(),
  1101. discovered_config_files,
  1102. memory_file_count: project_context.instruction_files.len(),
  1103. project_root,
  1104. git_branch,
  1105. })
  1106. }
  1107. fn format_status_report(
  1108. model: &str,
  1109. usage: StatusUsage,
  1110. permission_mode: &str,
  1111. context: &StatusContext,
  1112. ) -> String {
  1113. [
  1114. format!(
  1115. "Status
  1116. Model {model}
  1117. Permission mode {permission_mode}
  1118. Messages {}
  1119. Turns {}
  1120. Estimated tokens {}",
  1121. usage.message_count, usage.turns, usage.estimated_tokens,
  1122. ),
  1123. format!(
  1124. "Usage
  1125. Latest total {}
  1126. Cumulative input {}
  1127. Cumulative output {}
  1128. Cumulative total {}",
  1129. usage.latest.total_tokens(),
  1130. usage.cumulative.input_tokens,
  1131. usage.cumulative.output_tokens,
  1132. usage.cumulative.total_tokens(),
  1133. ),
  1134. format!(
  1135. "Workspace
  1136. Cwd {}
  1137. Project root {}
  1138. Git branch {}
  1139. Session {}
  1140. Config files loaded {}/{}
  1141. Memory files {}",
  1142. context.cwd.display(),
  1143. context
  1144. .project_root
  1145. .as_ref()
  1146. .map_or_else(|| "unknown".to_string(), |path| path.display().to_string()),
  1147. context.git_branch.as_deref().unwrap_or("unknown"),
  1148. context.session_path.as_ref().map_or_else(
  1149. || "live-repl".to_string(),
  1150. |path| path.display().to_string()
  1151. ),
  1152. context.loaded_config_files,
  1153. context.discovered_config_files,
  1154. context.memory_file_count,
  1155. ),
  1156. ]
  1157. .join(
  1158. "
  1159. ",
  1160. )
  1161. }
  1162. fn render_config_report(section: Option<&str>) -> Result<String, Box<dyn std::error::Error>> {
  1163. let cwd = env::current_dir()?;
  1164. let loader = ConfigLoader::default_for(&cwd);
  1165. let discovered = loader.discover();
  1166. let runtime_config = loader.load()?;
  1167. let mut lines = vec![
  1168. format!(
  1169. "Config
  1170. Working directory {}
  1171. Loaded files {}
  1172. Merged keys {}",
  1173. cwd.display(),
  1174. runtime_config.loaded_entries().len(),
  1175. runtime_config.merged().len()
  1176. ),
  1177. "Discovered files".to_string(),
  1178. ];
  1179. for entry in discovered {
  1180. let source = match entry.source {
  1181. ConfigSource::User => "user",
  1182. ConfigSource::Project => "project",
  1183. ConfigSource::Local => "local",
  1184. };
  1185. let status = if runtime_config
  1186. .loaded_entries()
  1187. .iter()
  1188. .any(|loaded_entry| loaded_entry.path == entry.path)
  1189. {
  1190. "loaded"
  1191. } else {
  1192. "missing"
  1193. };
  1194. lines.push(format!(
  1195. " {source:<7} {status:<7} {}",
  1196. entry.path.display()
  1197. ));
  1198. }
  1199. if let Some(section) = section {
  1200. lines.push(format!("Merged section: {section}"));
  1201. let value = match section {
  1202. "env" => runtime_config.get("env"),
  1203. "hooks" => runtime_config.get("hooks"),
  1204. "model" => runtime_config.get("model"),
  1205. other => {
  1206. lines.push(format!(
  1207. " Unsupported config section '{other}'. Use env, hooks, or model."
  1208. ));
  1209. return Ok(lines.join(
  1210. "
  1211. ",
  1212. ));
  1213. }
  1214. };
  1215. lines.push(format!(
  1216. " {}",
  1217. match value {
  1218. Some(value) => value.render(),
  1219. None => "<unset>".to_string(),
  1220. }
  1221. ));
  1222. return Ok(lines.join(
  1223. "
  1224. ",
  1225. ));
  1226. }
  1227. lines.push("Merged JSON".to_string());
  1228. lines.push(format!(" {}", runtime_config.as_json().render()));
  1229. Ok(lines.join(
  1230. "
  1231. ",
  1232. ))
  1233. }
  1234. fn render_memory_report() -> Result<String, Box<dyn std::error::Error>> {
  1235. let cwd = env::current_dir()?;
  1236. let project_context = ProjectContext::discover(&cwd, DEFAULT_DATE)?;
  1237. let mut lines = vec![format!(
  1238. "Memory
  1239. Working directory {}
  1240. Instruction files {}",
  1241. cwd.display(),
  1242. project_context.instruction_files.len()
  1243. )];
  1244. if project_context.instruction_files.is_empty() {
  1245. lines.push("Discovered files".to_string());
  1246. lines.push(
  1247. " No CLAUDE instruction files discovered in the current directory ancestry."
  1248. .to_string(),
  1249. );
  1250. } else {
  1251. lines.push("Discovered files".to_string());
  1252. for (index, file) in project_context.instruction_files.iter().enumerate() {
  1253. let preview = file.content.lines().next().unwrap_or("").trim();
  1254. let preview = if preview.is_empty() {
  1255. "<empty>"
  1256. } else {
  1257. preview
  1258. };
  1259. lines.push(format!(" {}. {}", index + 1, file.path.display(),));
  1260. lines.push(format!(
  1261. " lines={} preview={}",
  1262. file.content.lines().count(),
  1263. preview
  1264. ));
  1265. }
  1266. }
  1267. Ok(lines.join(
  1268. "
  1269. ",
  1270. ))
  1271. }
  1272. fn init_claude_md() -> Result<String, Box<dyn std::error::Error>> {
  1273. let cwd = env::current_dir()?;
  1274. let claude_md = cwd.join("CLAUDE.md");
  1275. if claude_md.exists() {
  1276. return Ok(format_init_report(&claude_md, false));
  1277. }
  1278. let content = render_init_claude_md(&cwd);
  1279. fs::write(&claude_md, content)?;
  1280. Ok(format_init_report(&claude_md, true))
  1281. }
  1282. fn render_init_claude_md(cwd: &Path) -> String {
  1283. let mut lines = vec![
  1284. "# CLAUDE.md".to_string(),
  1285. String::new(),
  1286. "This file provides guidance to Claude Code (claude.ai/code) when working with code in this repository.".to_string(),
  1287. String::new(),
  1288. ];
  1289. let mut command_lines = Vec::new();
  1290. if cwd.join("rust").join("Cargo.toml").is_file() {
  1291. command_lines.push("- Run Rust verification from `rust/`: `cargo fmt`, `cargo clippy --workspace --all-targets -- -D warnings`, `cargo test --workspace`".to_string());
  1292. } else if cwd.join("Cargo.toml").is_file() {
  1293. command_lines.push("- Run Rust verification from the repo root: `cargo fmt`, `cargo clippy --workspace --all-targets -- -D warnings`, `cargo test --workspace`".to_string());
  1294. }
  1295. if cwd.join("tests").is_dir() && cwd.join("src").is_dir() {
  1296. command_lines.push("- `src/` and `tests/` are also present; check those surfaces before removing or renaming Python-era compatibility assets.".to_string());
  1297. }
  1298. if !command_lines.is_empty() {
  1299. lines.push("## Verification".to_string());
  1300. lines.extend(command_lines);
  1301. lines.push(String::new());
  1302. }
  1303. let mut structure_lines = Vec::new();
  1304. if cwd.join("rust").is_dir() {
  1305. structure_lines.push(
  1306. "- `rust/` contains the Rust workspace and the active CLI/runtime implementation."
  1307. .to_string(),
  1308. );
  1309. }
  1310. if cwd.join("src").is_dir() {
  1311. structure_lines.push("- `src/` contains the older Python-first workspace artifacts referenced by the repo history and tests.".to_string());
  1312. }
  1313. if cwd.join("tests").is_dir() {
  1314. structure_lines.push("- `tests/` exercises compatibility and porting behavior across the repository surfaces.".to_string());
  1315. }
  1316. if !structure_lines.is_empty() {
  1317. lines.push("## Repository shape".to_string());
  1318. lines.extend(structure_lines);
  1319. lines.push(String::new());
  1320. }
  1321. lines.push("## Working agreement".to_string());
  1322. lines.push("- Prefer small, reviewable Rust changes and keep slash-command behavior aligned between the shared command registry and the CLI entrypoints.".to_string());
  1323. lines.push("- Do not overwrite existing CLAUDE.md content automatically; update it intentionally when repo workflows change.".to_string());
  1324. lines.push(String::new());
  1325. lines.join(
  1326. "
  1327. ",
  1328. )
  1329. }
  1330. fn normalize_permission_mode(mode: &str) -> Option<&'static str> {
  1331. match mode.trim() {
  1332. "read-only" => Some("read-only"),
  1333. "workspace-write" => Some("workspace-write"),
  1334. "danger-full-access" => Some("danger-full-access"),
  1335. _ => None,
  1336. }
  1337. }
  1338. fn permission_mode_label() -> &'static str {
  1339. match env::var("RUSTY_CLAUDE_PERMISSION_MODE") {
  1340. Ok(value) if value == "read-only" => "read-only",
  1341. Ok(value) if value == "danger-full-access" => "danger-full-access",
  1342. _ => "workspace-write",
  1343. }
  1344. }
  1345. fn render_diff_report() -> Result<String, Box<dyn std::error::Error>> {
  1346. let output = std::process::Command::new("git")
  1347. .args(["diff", "--", ":(exclude).omx"])
  1348. .current_dir(env::current_dir()?)
  1349. .output()?;
  1350. if !output.status.success() {
  1351. let stderr = String::from_utf8_lossy(&output.stderr).trim().to_string();
  1352. return Err(format!("git diff failed: {stderr}").into());
  1353. }
  1354. let diff = String::from_utf8(output.stdout)?;
  1355. if diff.trim().is_empty() {
  1356. return Ok(
  1357. "Diff\n Result clean working tree\n Detail no current changes"
  1358. .to_string(),
  1359. );
  1360. }
  1361. Ok(format!("Diff\n\n{}", diff.trim_end()))
  1362. }
  1363. fn render_version_report() -> String {
  1364. let git_sha = GIT_SHA.unwrap_or("unknown");
  1365. let target = BUILD_TARGET.unwrap_or("unknown");
  1366. format!(
  1367. "Version\n Version {VERSION}\n Git SHA {git_sha}\n Target {target}\n Build date {DEFAULT_DATE}"
  1368. )
  1369. }
  1370. fn render_export_text(session: &Session) -> String {
  1371. let mut lines = vec!["# Conversation Export".to_string(), String::new()];
  1372. for (index, message) in session.messages.iter().enumerate() {
  1373. let role = match message.role {
  1374. MessageRole::System => "system",
  1375. MessageRole::User => "user",
  1376. MessageRole::Assistant => "assistant",
  1377. MessageRole::Tool => "tool",
  1378. };
  1379. lines.push(format!("## {}. {role}", index + 1));
  1380. for block in &message.blocks {
  1381. match block {
  1382. ContentBlock::Text { text } => lines.push(text.clone()),
  1383. ContentBlock::ToolUse { id, name, input } => {
  1384. lines.push(format!("[tool_use id={id} name={name}] {input}"));
  1385. }
  1386. ContentBlock::ToolResult {
  1387. tool_use_id,
  1388. tool_name,
  1389. output,
  1390. is_error,
  1391. } => {
  1392. lines.push(format!(
  1393. "[tool_result id={tool_use_id} name={tool_name} error={is_error}] {output}"
  1394. ));
  1395. }
  1396. }
  1397. }
  1398. lines.push(String::new());
  1399. }
  1400. lines.join("\n")
  1401. }
  1402. fn default_export_filename(session: &Session) -> String {
  1403. let stem = session
  1404. .messages
  1405. .iter()
  1406. .find_map(|message| match message.role {
  1407. MessageRole::User => message.blocks.iter().find_map(|block| match block {
  1408. ContentBlock::Text { text } => Some(text.as_str()),
  1409. _ => None,
  1410. }),
  1411. _ => None,
  1412. })
  1413. .map_or("conversation", |text| {
  1414. text.lines().next().unwrap_or("conversation")
  1415. })
  1416. .chars()
  1417. .map(|ch| {
  1418. if ch.is_ascii_alphanumeric() {
  1419. ch.to_ascii_lowercase()
  1420. } else {
  1421. '-'
  1422. }
  1423. })
  1424. .collect::<String>()
  1425. .split('-')
  1426. .filter(|part| !part.is_empty())
  1427. .take(8)
  1428. .collect::<Vec<_>>()
  1429. .join("-");
  1430. let fallback = if stem.is_empty() {
  1431. "conversation"
  1432. } else {
  1433. &stem
  1434. };
  1435. format!("{fallback}.txt")
  1436. }
  1437. fn resolve_export_path(
  1438. requested_path: Option<&str>,
  1439. session: &Session,
  1440. ) -> Result<PathBuf, Box<dyn std::error::Error>> {
  1441. let cwd = env::current_dir()?;
  1442. let file_name =
  1443. requested_path.map_or_else(|| default_export_filename(session), ToOwned::to_owned);
  1444. let final_name = if Path::new(&file_name)
  1445. .extension()
  1446. .is_some_and(|ext| ext.eq_ignore_ascii_case("txt"))
  1447. {
  1448. file_name
  1449. } else {
  1450. format!("{file_name}.txt")
  1451. };
  1452. Ok(cwd.join(final_name))
  1453. }
  1454. fn build_system_prompt() -> Result<Vec<String>, Box<dyn std::error::Error>> {
  1455. Ok(load_system_prompt(
  1456. env::current_dir()?,
  1457. DEFAULT_DATE,
  1458. env::consts::OS,
  1459. "unknown",
  1460. )?)
  1461. }
  1462. fn build_runtime(
  1463. session: Session,
  1464. model: String,
  1465. system_prompt: Vec<String>,
  1466. enable_tools: bool,
  1467. ) -> Result<ConversationRuntime<AnthropicRuntimeClient, CliToolExecutor>, Box<dyn std::error::Error>>
  1468. {
  1469. build_runtime_with_permission_mode(
  1470. session,
  1471. model,
  1472. system_prompt,
  1473. enable_tools,
  1474. permission_mode_label(),
  1475. )
  1476. }
  1477. fn build_runtime_with_permission_mode(
  1478. session: Session,
  1479. model: String,
  1480. system_prompt: Vec<String>,
  1481. enable_tools: bool,
  1482. permission_mode: &str,
  1483. ) -> Result<ConversationRuntime<AnthropicRuntimeClient, CliToolExecutor>, Box<dyn std::error::Error>>
  1484. {
  1485. Ok(ConversationRuntime::new(
  1486. session,
  1487. AnthropicRuntimeClient::new(model, enable_tools)?,
  1488. CliToolExecutor::new(),
  1489. permission_policy(permission_mode),
  1490. system_prompt,
  1491. ))
  1492. }
  1493. struct AnthropicRuntimeClient {
  1494. runtime: tokio::runtime::Runtime,
  1495. client: AnthropicClient,
  1496. model: String,
  1497. enable_tools: bool,
  1498. }
  1499. impl AnthropicRuntimeClient {
  1500. fn new(model: String, enable_tools: bool) -> Result<Self, Box<dyn std::error::Error>> {
  1501. Ok(Self {
  1502. runtime: tokio::runtime::Runtime::new()?,
  1503. client: AnthropicClient::from_env()?,
  1504. model,
  1505. enable_tools,
  1506. })
  1507. }
  1508. }
  1509. impl ApiClient for AnthropicRuntimeClient {
  1510. #[allow(clippy::too_many_lines)]
  1511. fn stream(&mut self, request: ApiRequest) -> Result<Vec<AssistantEvent>, RuntimeError> {
  1512. let message_request = MessageRequest {
  1513. model: self.model.clone(),
  1514. max_tokens: DEFAULT_MAX_TOKENS,
  1515. messages: convert_messages(&request.messages),
  1516. system: (!request.system_prompt.is_empty()).then(|| request.system_prompt.join("\n\n")),
  1517. tools: self.enable_tools.then(|| {
  1518. mvp_tool_specs()
  1519. .into_iter()
  1520. .map(|spec| ToolDefinition {
  1521. name: spec.name.to_string(),
  1522. description: Some(spec.description.to_string()),
  1523. input_schema: spec.input_schema,
  1524. })
  1525. .collect()
  1526. }),
  1527. tool_choice: self.enable_tools.then_some(ToolChoice::Auto),
  1528. stream: true,
  1529. };
  1530. self.runtime.block_on(async {
  1531. let mut stream = self
  1532. .client
  1533. .stream_message(&message_request)
  1534. .await
  1535. .map_err(|error| RuntimeError::new(error.to_string()))?;
  1536. let mut stdout = io::stdout();
  1537. let mut events = Vec::new();
  1538. let mut pending_tool: Option<(String, String, String)> = None;
  1539. let mut saw_stop = false;
  1540. while let Some(event) = stream
  1541. .next_event()
  1542. .await
  1543. .map_err(|error| RuntimeError::new(error.to_string()))?
  1544. {
  1545. match event {
  1546. ApiStreamEvent::MessageStart(start) => {
  1547. for block in start.message.content {
  1548. push_output_block(block, &mut stdout, &mut events, &mut pending_tool)?;
  1549. }
  1550. }
  1551. ApiStreamEvent::ContentBlockStart(start) => {
  1552. push_output_block(
  1553. start.content_block,
  1554. &mut stdout,
  1555. &mut events,
  1556. &mut pending_tool,
  1557. )?;
  1558. }
  1559. ApiStreamEvent::ContentBlockDelta(delta) => match delta.delta {
  1560. ContentBlockDelta::TextDelta { text } => {
  1561. if !text.is_empty() {
  1562. write!(stdout, "{text}")
  1563. .and_then(|()| stdout.flush())
  1564. .map_err(|error| RuntimeError::new(error.to_string()))?;
  1565. events.push(AssistantEvent::TextDelta(text));
  1566. }
  1567. }
  1568. ContentBlockDelta::InputJsonDelta { partial_json } => {
  1569. if let Some((_, _, input)) = &mut pending_tool {
  1570. input.push_str(&partial_json);
  1571. }
  1572. }
  1573. },
  1574. ApiStreamEvent::ContentBlockStop(_) => {
  1575. if let Some((id, name, input)) = pending_tool.take() {
  1576. events.push(AssistantEvent::ToolUse { id, name, input });
  1577. }
  1578. }
  1579. ApiStreamEvent::MessageDelta(delta) => {
  1580. events.push(AssistantEvent::Usage(TokenUsage {
  1581. input_tokens: delta.usage.input_tokens,
  1582. output_tokens: delta.usage.output_tokens,
  1583. cache_creation_input_tokens: 0,
  1584. cache_read_input_tokens: 0,
  1585. }));
  1586. }
  1587. ApiStreamEvent::MessageStop(_) => {
  1588. saw_stop = true;
  1589. events.push(AssistantEvent::MessageStop);
  1590. }
  1591. }
  1592. }
  1593. if !saw_stop
  1594. && events.iter().any(|event| {
  1595. matches!(event, AssistantEvent::TextDelta(text) if !text.is_empty())
  1596. || matches!(event, AssistantEvent::ToolUse { .. })
  1597. })
  1598. {
  1599. events.push(AssistantEvent::MessageStop);
  1600. }
  1601. if events
  1602. .iter()
  1603. .any(|event| matches!(event, AssistantEvent::MessageStop))
  1604. {
  1605. return Ok(events);
  1606. }
  1607. let response = self
  1608. .client
  1609. .send_message(&MessageRequest {
  1610. stream: false,
  1611. ..message_request.clone()
  1612. })
  1613. .await
  1614. .map_err(|error| RuntimeError::new(error.to_string()))?;
  1615. response_to_events(response, &mut stdout)
  1616. })
  1617. }
  1618. }
  1619. fn push_output_block(
  1620. block: OutputContentBlock,
  1621. out: &mut impl Write,
  1622. events: &mut Vec<AssistantEvent>,
  1623. pending_tool: &mut Option<(String, String, String)>,
  1624. ) -> Result<(), RuntimeError> {
  1625. match block {
  1626. OutputContentBlock::Text { text } => {
  1627. if !text.is_empty() {
  1628. write!(out, "{text}")
  1629. .and_then(|()| out.flush())
  1630. .map_err(|error| RuntimeError::new(error.to_string()))?;
  1631. events.push(AssistantEvent::TextDelta(text));
  1632. }
  1633. }
  1634. OutputContentBlock::ToolUse { id, name, input } => {
  1635. *pending_tool = Some((id, name, input.to_string()));
  1636. }
  1637. }
  1638. Ok(())
  1639. }
  1640. fn response_to_events(
  1641. response: MessageResponse,
  1642. out: &mut impl Write,
  1643. ) -> Result<Vec<AssistantEvent>, RuntimeError> {
  1644. let mut events = Vec::new();
  1645. let mut pending_tool = None;
  1646. for block in response.content {
  1647. push_output_block(block, out, &mut events, &mut pending_tool)?;
  1648. if let Some((id, name, input)) = pending_tool.take() {
  1649. events.push(AssistantEvent::ToolUse { id, name, input });
  1650. }
  1651. }
  1652. events.push(AssistantEvent::Usage(TokenUsage {
  1653. input_tokens: response.usage.input_tokens,
  1654. output_tokens: response.usage.output_tokens,
  1655. cache_creation_input_tokens: response.usage.cache_creation_input_tokens,
  1656. cache_read_input_tokens: response.usage.cache_read_input_tokens,
  1657. }));
  1658. events.push(AssistantEvent::MessageStop);
  1659. Ok(events)
  1660. }
  1661. struct CliToolExecutor {
  1662. renderer: TerminalRenderer,
  1663. }
  1664. impl CliToolExecutor {
  1665. fn new() -> Self {
  1666. Self {
  1667. renderer: TerminalRenderer::new(),
  1668. }
  1669. }
  1670. }
  1671. impl ToolExecutor for CliToolExecutor {
  1672. fn execute(&mut self, tool_name: &str, input: &str) -> Result<String, ToolError> {
  1673. let value = serde_json::from_str(input)
  1674. .map_err(|error| ToolError::new(format!("invalid tool input JSON: {error}")))?;
  1675. match execute_tool(tool_name, &value) {
  1676. Ok(output) => {
  1677. let markdown = format!("### Tool `{tool_name}`\n\n```json\n{output}\n```\n");
  1678. self.renderer
  1679. .stream_markdown(&markdown, &mut io::stdout())
  1680. .map_err(|error| ToolError::new(error.to_string()))?;
  1681. Ok(output)
  1682. }
  1683. Err(error) => Err(ToolError::new(error)),
  1684. }
  1685. }
  1686. }
  1687. fn permission_policy(mode: &str) -> PermissionPolicy {
  1688. if normalize_permission_mode(mode) == Some("read-only") {
  1689. PermissionPolicy::new(PermissionMode::Deny)
  1690. .with_tool_mode("read_file", PermissionMode::Allow)
  1691. .with_tool_mode("glob_search", PermissionMode::Allow)
  1692. .with_tool_mode("grep_search", PermissionMode::Allow)
  1693. } else {
  1694. PermissionPolicy::new(PermissionMode::Allow)
  1695. }
  1696. }
  1697. fn convert_messages(messages: &[ConversationMessage]) -> Vec<InputMessage> {
  1698. messages
  1699. .iter()
  1700. .filter_map(|message| {
  1701. let role = match message.role {
  1702. MessageRole::System | MessageRole::User | MessageRole::Tool => "user",
  1703. MessageRole::Assistant => "assistant",
  1704. };
  1705. let content = message
  1706. .blocks
  1707. .iter()
  1708. .map(|block| match block {
  1709. ContentBlock::Text { text } => InputContentBlock::Text { text: text.clone() },
  1710. ContentBlock::ToolUse { id, name, input } => InputContentBlock::ToolUse {
  1711. id: id.clone(),
  1712. name: name.clone(),
  1713. input: serde_json::from_str(input)
  1714. .unwrap_or_else(|_| serde_json::json!({ "raw": input })),
  1715. },
  1716. ContentBlock::ToolResult {
  1717. tool_use_id,
  1718. output,
  1719. is_error,
  1720. ..
  1721. } => InputContentBlock::ToolResult {
  1722. tool_use_id: tool_use_id.clone(),
  1723. content: vec![ToolResultContentBlock::Text {
  1724. text: output.clone(),
  1725. }],
  1726. is_error: *is_error,
  1727. },
  1728. })
  1729. .collect::<Vec<_>>();
  1730. (!content.is_empty()).then(|| InputMessage {
  1731. role: role.to_string(),
  1732. content,
  1733. })
  1734. })
  1735. .collect()
  1736. }
  1737. fn print_help() {
  1738. println!("rusty-claude-cli v{VERSION}");
  1739. println!();
  1740. println!("Usage:");
  1741. println!(" rusty-claude-cli [--model MODEL]");
  1742. println!(" Start the interactive REPL");
  1743. println!(" rusty-claude-cli [--model MODEL] [--output-format text|json] prompt TEXT");
  1744. println!(" Send one prompt and exit");
  1745. println!(" rusty-claude-cli [--model MODEL] [--output-format text|json] TEXT");
  1746. println!(" Shorthand non-interactive prompt mode");
  1747. println!(" rusty-claude-cli --resume SESSION.json [/status] [/compact] [...]");
  1748. println!(" Inspect or maintain a saved session without entering the REPL");
  1749. println!(" rusty-claude-cli dump-manifests");
  1750. println!(" rusty-claude-cli bootstrap-plan");
  1751. println!(" rusty-claude-cli system-prompt [--cwd PATH] [--date YYYY-MM-DD]");
  1752. println!();
  1753. println!("Flags:");
  1754. println!(" --model MODEL Override the active model");
  1755. println!(" --output-format FORMAT Non-interactive output format: text or json");
  1756. println!();
  1757. println!("Interactive slash commands:");
  1758. println!("{}", render_slash_command_help());
  1759. println!();
  1760. let resume_commands = resume_supported_slash_commands()
  1761. .into_iter()
  1762. .map(|spec| match spec.argument_hint {
  1763. Some(argument_hint) => format!("/{} {}", spec.name, argument_hint),
  1764. None => format!("/{}", spec.name),
  1765. })
  1766. .collect::<Vec<_>>()
  1767. .join(", ");
  1768. println!("Resume-safe commands: {resume_commands}");
  1769. println!("Examples:");
  1770. println!(" rusty-claude-cli --model claude-opus \"summarize this repo\"");
  1771. println!(" rusty-claude-cli --output-format json prompt \"explain src/main.rs\"");
  1772. println!(" rusty-claude-cli --resume session.json /status /diff /export notes.txt");
  1773. }
  1774. #[cfg(test)]
  1775. mod tests {
  1776. use super::{
  1777. format_compact_report, format_cost_report, format_init_report, format_model_report,
  1778. format_model_switch_report, format_permissions_report, format_permissions_switch_report,
  1779. format_resume_report, format_status_report, normalize_permission_mode, parse_args,
  1780. parse_git_status_metadata, render_config_report, render_init_claude_md,
  1781. render_memory_report, render_repl_help, resume_supported_slash_commands, status_context,
  1782. CliAction, CliOutputFormat, SlashCommand, StatusUsage, DEFAULT_MODEL,
  1783. };
  1784. use runtime::{ContentBlock, ConversationMessage, MessageRole};
  1785. use std::path::{Path, PathBuf};
  1786. #[test]
  1787. fn defaults_to_repl_when_no_args() {
  1788. assert_eq!(
  1789. parse_args(&[]).expect("args should parse"),
  1790. CliAction::Repl {
  1791. model: DEFAULT_MODEL.to_string(),
  1792. }
  1793. );
  1794. }
  1795. #[test]
  1796. fn parses_prompt_subcommand() {
  1797. let args = vec![
  1798. "prompt".to_string(),
  1799. "hello".to_string(),
  1800. "world".to_string(),
  1801. ];
  1802. assert_eq!(
  1803. parse_args(&args).expect("args should parse"),
  1804. CliAction::Prompt {
  1805. prompt: "hello world".to_string(),
  1806. model: DEFAULT_MODEL.to_string(),
  1807. output_format: CliOutputFormat::Text,
  1808. }
  1809. );
  1810. }
  1811. #[test]
  1812. fn parses_bare_prompt_and_json_output_flag() {
  1813. let args = vec![
  1814. "--output-format=json".to_string(),
  1815. "--model".to_string(),
  1816. "claude-opus".to_string(),
  1817. "explain".to_string(),
  1818. "this".to_string(),
  1819. ];
  1820. assert_eq!(
  1821. parse_args(&args).expect("args should parse"),
  1822. CliAction::Prompt {
  1823. prompt: "explain this".to_string(),
  1824. model: "claude-opus".to_string(),
  1825. output_format: CliOutputFormat::Json,
  1826. }
  1827. );
  1828. }
  1829. #[test]
  1830. fn parses_system_prompt_options() {
  1831. let args = vec![
  1832. "system-prompt".to_string(),
  1833. "--cwd".to_string(),
  1834. "/tmp/project".to_string(),
  1835. "--date".to_string(),
  1836. "2026-04-01".to_string(),
  1837. ];
  1838. assert_eq!(
  1839. parse_args(&args).expect("args should parse"),
  1840. CliAction::PrintSystemPrompt {
  1841. cwd: PathBuf::from("/tmp/project"),
  1842. date: "2026-04-01".to_string(),
  1843. }
  1844. );
  1845. }
  1846. #[test]
  1847. fn parses_resume_flag_with_slash_command() {
  1848. let args = vec![
  1849. "--resume".to_string(),
  1850. "session.json".to_string(),
  1851. "/compact".to_string(),
  1852. ];
  1853. assert_eq!(
  1854. parse_args(&args).expect("args should parse"),
  1855. CliAction::ResumeSession {
  1856. session_path: PathBuf::from("session.json"),
  1857. commands: vec!["/compact".to_string()],
  1858. }
  1859. );
  1860. }
  1861. #[test]
  1862. fn parses_resume_flag_with_multiple_slash_commands() {
  1863. let args = vec![
  1864. "--resume".to_string(),
  1865. "session.json".to_string(),
  1866. "/status".to_string(),
  1867. "/compact".to_string(),
  1868. "/cost".to_string(),
  1869. ];
  1870. assert_eq!(
  1871. parse_args(&args).expect("args should parse"),
  1872. CliAction::ResumeSession {
  1873. session_path: PathBuf::from("session.json"),
  1874. commands: vec![
  1875. "/status".to_string(),
  1876. "/compact".to_string(),
  1877. "/cost".to_string(),
  1878. ],
  1879. }
  1880. );
  1881. }
  1882. #[test]
  1883. fn shared_help_uses_resume_annotation_copy() {
  1884. let help = commands::render_slash_command_help();
  1885. assert!(help.contains("Slash commands"));
  1886. assert!(help.contains("works with --resume SESSION.json"));
  1887. }
  1888. #[test]
  1889. fn repl_help_includes_shared_commands_and_exit() {
  1890. let help = render_repl_help();
  1891. assert!(help.contains("REPL"));
  1892. assert!(help.contains("/help"));
  1893. assert!(help.contains("/status"));
  1894. assert!(help.contains("/model [model]"));
  1895. assert!(help.contains("/permissions [read-only|workspace-write|danger-full-access]"));
  1896. assert!(help.contains("/clear [--confirm]"));
  1897. assert!(help.contains("/cost"));
  1898. assert!(help.contains("/resume <session-path>"));
  1899. assert!(help.contains("/config [env|hooks|model]"));
  1900. assert!(help.contains("/memory"));
  1901. assert!(help.contains("/init"));
  1902. assert!(help.contains("/diff"));
  1903. assert!(help.contains("/version"));
  1904. assert!(help.contains("/export [file]"));
  1905. assert!(help.contains("/session [list|switch <session-id>]"));
  1906. assert!(help.contains("/exit"));
  1907. }
  1908. #[test]
  1909. fn resume_supported_command_list_matches_expected_surface() {
  1910. let names = resume_supported_slash_commands()
  1911. .into_iter()
  1912. .map(|spec| spec.name)
  1913. .collect::<Vec<_>>();
  1914. assert_eq!(
  1915. names,
  1916. vec![
  1917. "help", "status", "compact", "clear", "cost", "config", "memory", "init", "diff",
  1918. "version", "export",
  1919. ]
  1920. );
  1921. }
  1922. #[test]
  1923. fn resume_report_uses_sectioned_layout() {
  1924. let report = format_resume_report("session.json", 14, 6);
  1925. assert!(report.contains("Session resumed"));
  1926. assert!(report.contains("Session file session.json"));
  1927. assert!(report.contains("Messages 14"));
  1928. assert!(report.contains("Turns 6"));
  1929. }
  1930. #[test]
  1931. fn compact_report_uses_structured_output() {
  1932. let compacted = format_compact_report(8, 5, false);
  1933. assert!(compacted.contains("Compact"));
  1934. assert!(compacted.contains("Result compacted"));
  1935. assert!(compacted.contains("Messages removed 8"));
  1936. let skipped = format_compact_report(0, 3, true);
  1937. assert!(skipped.contains("Result skipped"));
  1938. }
  1939. #[test]
  1940. fn cost_report_uses_sectioned_layout() {
  1941. let report = format_cost_report(runtime::TokenUsage {
  1942. input_tokens: 20,
  1943. output_tokens: 8,
  1944. cache_creation_input_tokens: 3,
  1945. cache_read_input_tokens: 1,
  1946. });
  1947. assert!(report.contains("Cost"));
  1948. assert!(report.contains("Input tokens 20"));
  1949. assert!(report.contains("Output tokens 8"));
  1950. assert!(report.contains("Cache create 3"));
  1951. assert!(report.contains("Cache read 1"));
  1952. assert!(report.contains("Total tokens 32"));
  1953. }
  1954. #[test]
  1955. fn permissions_report_uses_sectioned_layout() {
  1956. let report = format_permissions_report("workspace-write");
  1957. assert!(report.contains("Permissions"));
  1958. assert!(report.contains("Active mode workspace-write"));
  1959. assert!(report.contains("Modes"));
  1960. assert!(report.contains("read-only ○ available Read/search tools only"));
  1961. assert!(report.contains("workspace-write ● current Edit files inside the workspace"));
  1962. assert!(report.contains("danger-full-access ○ available Unrestricted tool access"));
  1963. }
  1964. #[test]
  1965. fn permissions_switch_report_is_structured() {
  1966. let report = format_permissions_switch_report("read-only", "workspace-write");
  1967. assert!(report.contains("Permissions updated"));
  1968. assert!(report.contains("Result mode switched"));
  1969. assert!(report.contains("Previous mode read-only"));
  1970. assert!(report.contains("Active mode workspace-write"));
  1971. assert!(report.contains("Applies to subsequent tool calls"));
  1972. }
  1973. #[test]
  1974. fn init_report_uses_structured_output() {
  1975. let created = format_init_report(Path::new("/tmp/CLAUDE.md"), true);
  1976. assert!(created.contains("Init"));
  1977. assert!(created.contains("Result created"));
  1978. let skipped = format_init_report(Path::new("/tmp/CLAUDE.md"), false);
  1979. assert!(skipped.contains("skipped (already exists)"));
  1980. }
  1981. #[test]
  1982. fn model_report_uses_sectioned_layout() {
  1983. let report = format_model_report("claude-sonnet", 12, 4);
  1984. assert!(report.contains("Model"));
  1985. assert!(report.contains("Current model claude-sonnet"));
  1986. assert!(report.contains("Session messages 12"));
  1987. assert!(report.contains("Switch models with /model <name>"));
  1988. }
  1989. #[test]
  1990. fn model_switch_report_preserves_context_summary() {
  1991. let report = format_model_switch_report("claude-sonnet", "claude-opus", 9);
  1992. assert!(report.contains("Model updated"));
  1993. assert!(report.contains("Previous claude-sonnet"));
  1994. assert!(report.contains("Current claude-opus"));
  1995. assert!(report.contains("Preserved msgs 9"));
  1996. }
  1997. #[test]
  1998. fn status_line_reports_model_and_token_totals() {
  1999. let status = format_status_report(
  2000. "claude-sonnet",
  2001. StatusUsage {
  2002. message_count: 7,
  2003. turns: 3,
  2004. latest: runtime::TokenUsage {
  2005. input_tokens: 5,
  2006. output_tokens: 4,
  2007. cache_creation_input_tokens: 1,
  2008. cache_read_input_tokens: 0,
  2009. },
  2010. cumulative: runtime::TokenUsage {
  2011. input_tokens: 20,
  2012. output_tokens: 8,
  2013. cache_creation_input_tokens: 2,
  2014. cache_read_input_tokens: 1,
  2015. },
  2016. estimated_tokens: 128,
  2017. },
  2018. "workspace-write",
  2019. &super::StatusContext {
  2020. cwd: PathBuf::from("/tmp/project"),
  2021. session_path: Some(PathBuf::from("session.json")),
  2022. loaded_config_files: 2,
  2023. discovered_config_files: 3,
  2024. memory_file_count: 4,
  2025. project_root: Some(PathBuf::from("/tmp")),
  2026. git_branch: Some("main".to_string()),
  2027. },
  2028. );
  2029. assert!(status.contains("Status"));
  2030. assert!(status.contains("Model claude-sonnet"));
  2031. assert!(status.contains("Permission mode workspace-write"));
  2032. assert!(status.contains("Messages 7"));
  2033. assert!(status.contains("Latest total 10"));
  2034. assert!(status.contains("Cumulative total 31"));
  2035. assert!(status.contains("Cwd /tmp/project"));
  2036. assert!(status.contains("Project root /tmp"));
  2037. assert!(status.contains("Git branch main"));
  2038. assert!(status.contains("Session session.json"));
  2039. assert!(status.contains("Config files loaded 2/3"));
  2040. assert!(status.contains("Memory files 4"));
  2041. }
  2042. #[test]
  2043. fn config_report_supports_section_views() {
  2044. let report = render_config_report(Some("env")).expect("config report should render");
  2045. assert!(report.contains("Merged section: env"));
  2046. }
  2047. #[test]
  2048. fn memory_report_uses_sectioned_layout() {
  2049. let report = render_memory_report().expect("memory report should render");
  2050. assert!(report.contains("Memory"));
  2051. assert!(report.contains("Working directory"));
  2052. assert!(report.contains("Instruction files"));
  2053. assert!(report.contains("Discovered files"));
  2054. }
  2055. #[test]
  2056. fn config_report_uses_sectioned_layout() {
  2057. let report = render_config_report(None).expect("config report should render");
  2058. assert!(report.contains("Config"));
  2059. assert!(report.contains("Discovered files"));
  2060. assert!(report.contains("Merged JSON"));
  2061. }
  2062. #[test]
  2063. fn parses_git_status_metadata() {
  2064. let (root, branch) = parse_git_status_metadata(Some(
  2065. "## rcc/cli...origin/rcc/cli
  2066. M src/main.rs",
  2067. ));
  2068. assert_eq!(branch.as_deref(), Some("rcc/cli"));
  2069. let _ = root;
  2070. }
  2071. #[test]
  2072. fn status_context_reads_real_workspace_metadata() {
  2073. let context = status_context(None).expect("status context should load");
  2074. assert!(context.cwd.is_absolute());
  2075. assert_eq!(context.discovered_config_files, 3);
  2076. assert!(context.loaded_config_files <= context.discovered_config_files);
  2077. }
  2078. #[test]
  2079. fn normalizes_supported_permission_modes() {
  2080. assert_eq!(normalize_permission_mode("read-only"), Some("read-only"));
  2081. assert_eq!(
  2082. normalize_permission_mode("workspace-write"),
  2083. Some("workspace-write")
  2084. );
  2085. assert_eq!(
  2086. normalize_permission_mode("danger-full-access"),
  2087. Some("danger-full-access")
  2088. );
  2089. assert_eq!(normalize_permission_mode("unknown"), None);
  2090. }
  2091. #[test]
  2092. fn clear_command_requires_explicit_confirmation_flag() {
  2093. assert_eq!(
  2094. SlashCommand::parse("/clear"),
  2095. Some(SlashCommand::Clear { confirm: false })
  2096. );
  2097. assert_eq!(
  2098. SlashCommand::parse("/clear --confirm"),
  2099. Some(SlashCommand::Clear { confirm: true })
  2100. );
  2101. }
  2102. #[test]
  2103. fn parses_resume_and_config_slash_commands() {
  2104. assert_eq!(
  2105. SlashCommand::parse("/resume saved-session.json"),
  2106. Some(SlashCommand::Resume {
  2107. session_path: Some("saved-session.json".to_string())
  2108. })
  2109. );
  2110. assert_eq!(
  2111. SlashCommand::parse("/clear --confirm"),
  2112. Some(SlashCommand::Clear { confirm: true })
  2113. );
  2114. assert_eq!(
  2115. SlashCommand::parse("/config"),
  2116. Some(SlashCommand::Config { section: None })
  2117. );
  2118. assert_eq!(
  2119. SlashCommand::parse("/config env"),
  2120. Some(SlashCommand::Config {
  2121. section: Some("env".to_string())
  2122. })
  2123. );
  2124. assert_eq!(SlashCommand::parse("/memory"), Some(SlashCommand::Memory));
  2125. assert_eq!(SlashCommand::parse("/init"), Some(SlashCommand::Init));
  2126. }
  2127. #[test]
  2128. fn init_template_mentions_detected_rust_workspace() {
  2129. let rendered = render_init_claude_md(Path::new("."));
  2130. assert!(rendered.contains("# CLAUDE.md"));
  2131. assert!(rendered.contains("cargo clippy --workspace --all-targets -- -D warnings"));
  2132. }
  2133. #[test]
  2134. fn converts_tool_roundtrip_messages() {
  2135. let messages = vec![
  2136. ConversationMessage::user_text("hello"),
  2137. ConversationMessage::assistant(vec![ContentBlock::ToolUse {
  2138. id: "tool-1".to_string(),
  2139. name: "bash".to_string(),
  2140. input: "{\"command\":\"pwd\"}".to_string(),
  2141. }]),
  2142. ConversationMessage {
  2143. role: MessageRole::Tool,
  2144. blocks: vec![ContentBlock::ToolResult {
  2145. tool_use_id: "tool-1".to_string(),
  2146. tool_name: "bash".to_string(),
  2147. output: "ok".to_string(),
  2148. is_error: false,
  2149. }],
  2150. usage: None,
  2151. },
  2152. ];
  2153. let converted = super::convert_messages(&messages);
  2154. assert_eq!(converted.len(), 3);
  2155. assert_eq!(converted[1].role, "assistant");
  2156. assert_eq!(converted[2].role, "user");
  2157. }
  2158. }