main.rs 93 KB

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