main.rs 87 KB

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