main.rs 110 KB

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