main.rs 95 KB

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