main.rs 266 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586878889909192939495969798991001011021031041051061071081091101111121131141151161171181191201211221231241251261271281291301311321331341351361371381391401411421431441451461471481491501511521531541551561571581591601611621631641651661671681691701711721731741751761771781791801811821831841851861871881891901911921931941951961971981992002012022032042052062072082092102112122132142152162172182192202212222232242252262272282292302312322332342352362372382392402412422432442452462472482492502512522532542552562572582592602612622632642652662672682692702712722732742752762772782792802812822832842852862872882892902912922932942952962972982993003013023033043053063073083093103113123133143153163173183193203213223233243253263273283293303313323333343353363373383393403413423433443453463473483493503513523533543553563573583593603613623633643653663673683693703713723733743753763773783793803813823833843853863873883893903913923933943953963973983994004014024034044054064074084094104114124134144154164174184194204214224234244254264274284294304314324334344354364374384394404414424434444454464474484494504514524534544554564574584594604614624634644654664674684694704714724734744754764774784794804814824834844854864874884894904914924934944954964974984995005015025035045055065075085095105115125135145155165175185195205215225235245255265275285295305315325335345355365375385395405415425435445455465475485495505515525535545555565575585595605615625635645655665675685695705715725735745755765775785795805815825835845855865875885895905915925935945955965975985996006016026036046056066076086096106116126136146156166176186196206216226236246256266276286296306316326336346356366376386396406416426436446456466476486496506516526536546556566576586596606616626636646656666676686696706716726736746756766776786796806816826836846856866876886896906916926936946956966976986997007017027037047057067077087097107117127137147157167177187197207217227237247257267277287297307317327337347357367377387397407417427437447457467477487497507517527537547557567577587597607617627637647657667677687697707717727737747757767777787797807817827837847857867877887897907917927937947957967977987998008018028038048058068078088098108118128138148158168178188198208218228238248258268278288298308318328338348358368378388398408418428438448458468478488498508518528538548558568578588598608618628638648658668678688698708718728738748758768778788798808818828838848858868878888898908918928938948958968978988999009019029039049059069079089099109119129139149159169179189199209219229239249259269279289299309319329339349359369379389399409419429439449459469479489499509519529539549559569579589599609619629639649659669679689699709719729739749759769779789799809819829839849859869879889899909919929939949959969979989991000100110021003100410051006100710081009101010111012101310141015101610171018101910201021102210231024102510261027102810291030103110321033103410351036103710381039104010411042104310441045104610471048104910501051105210531054105510561057105810591060106110621063106410651066106710681069107010711072107310741075107610771078107910801081108210831084108510861087108810891090109110921093109410951096109710981099110011011102110311041105110611071108110911101111111211131114111511161117111811191120112111221123112411251126112711281129113011311132113311341135113611371138113911401141114211431144114511461147114811491150115111521153115411551156115711581159116011611162116311641165116611671168116911701171117211731174117511761177117811791180118111821183118411851186118711881189119011911192119311941195119611971198119912001201120212031204120512061207120812091210121112121213121412151216121712181219122012211222122312241225122612271228122912301231123212331234123512361237123812391240124112421243124412451246124712481249125012511252125312541255125612571258125912601261126212631264126512661267126812691270127112721273127412751276127712781279128012811282128312841285128612871288128912901291129212931294129512961297129812991300130113021303130413051306130713081309131013111312131313141315131613171318131913201321132213231324132513261327132813291330133113321333133413351336133713381339134013411342134313441345134613471348134913501351135213531354135513561357135813591360136113621363136413651366136713681369137013711372137313741375137613771378137913801381138213831384138513861387138813891390139113921393139413951396139713981399140014011402140314041405140614071408140914101411141214131414141514161417141814191420142114221423142414251426142714281429143014311432143314341435143614371438143914401441144214431444144514461447144814491450145114521453145414551456145714581459146014611462146314641465146614671468146914701471147214731474147514761477147814791480148114821483148414851486148714881489149014911492149314941495149614971498149915001501150215031504150515061507150815091510151115121513151415151516151715181519152015211522152315241525152615271528152915301531153215331534153515361537153815391540154115421543154415451546154715481549155015511552155315541555155615571558155915601561156215631564156515661567156815691570157115721573157415751576157715781579158015811582158315841585158615871588158915901591159215931594159515961597159815991600160116021603160416051606160716081609161016111612161316141615161616171618161916201621162216231624162516261627162816291630163116321633163416351636163716381639164016411642164316441645164616471648164916501651165216531654165516561657165816591660166116621663166416651666166716681669167016711672167316741675167616771678167916801681168216831684168516861687168816891690169116921693169416951696169716981699170017011702170317041705170617071708170917101711171217131714171517161717171817191720172117221723172417251726172717281729173017311732173317341735173617371738173917401741174217431744174517461747174817491750175117521753175417551756175717581759176017611762176317641765176617671768176917701771177217731774177517761777177817791780178117821783178417851786178717881789179017911792179317941795179617971798179918001801180218031804180518061807180818091810181118121813181418151816181718181819182018211822182318241825182618271828182918301831183218331834183518361837183818391840184118421843184418451846184718481849185018511852185318541855185618571858185918601861186218631864186518661867186818691870187118721873187418751876187718781879188018811882188318841885188618871888188918901891189218931894189518961897189818991900190119021903190419051906190719081909191019111912191319141915191619171918191919201921192219231924192519261927192819291930193119321933193419351936193719381939194019411942194319441945194619471948194919501951195219531954195519561957195819591960196119621963196419651966196719681969197019711972197319741975197619771978197919801981198219831984198519861987198819891990199119921993199419951996199719981999200020012002200320042005200620072008200920102011201220132014201520162017201820192020202120222023202420252026202720282029203020312032203320342035203620372038203920402041204220432044204520462047204820492050205120522053205420552056205720582059206020612062206320642065206620672068206920702071207220732074207520762077207820792080208120822083208420852086208720882089209020912092209320942095209620972098209921002101210221032104210521062107210821092110211121122113211421152116211721182119212021212122212321242125212621272128212921302131213221332134213521362137213821392140214121422143214421452146214721482149215021512152215321542155215621572158215921602161216221632164216521662167216821692170217121722173217421752176217721782179218021812182218321842185218621872188218921902191219221932194219521962197219821992200220122022203220422052206220722082209221022112212221322142215221622172218221922202221222222232224222522262227222822292230223122322233223422352236223722382239224022412242224322442245224622472248224922502251225222532254225522562257225822592260226122622263226422652266226722682269227022712272227322742275227622772278227922802281228222832284228522862287228822892290229122922293229422952296229722982299230023012302230323042305230623072308230923102311231223132314231523162317231823192320232123222323232423252326232723282329233023312332233323342335233623372338233923402341234223432344234523462347234823492350235123522353235423552356235723582359236023612362236323642365236623672368236923702371237223732374237523762377237823792380238123822383238423852386238723882389239023912392239323942395239623972398239924002401240224032404240524062407240824092410241124122413241424152416241724182419242024212422242324242425242624272428242924302431243224332434243524362437243824392440244124422443244424452446244724482449245024512452245324542455245624572458245924602461246224632464246524662467246824692470247124722473247424752476247724782479248024812482248324842485248624872488248924902491249224932494249524962497249824992500250125022503250425052506250725082509251025112512251325142515251625172518251925202521252225232524252525262527252825292530253125322533253425352536253725382539254025412542254325442545254625472548254925502551255225532554255525562557255825592560256125622563256425652566256725682569257025712572257325742575257625772578257925802581258225832584258525862587258825892590259125922593259425952596259725982599260026012602260326042605260626072608260926102611261226132614261526162617261826192620262126222623262426252626262726282629263026312632263326342635263626372638263926402641264226432644264526462647264826492650265126522653265426552656265726582659266026612662266326642665266626672668266926702671267226732674267526762677267826792680268126822683268426852686268726882689269026912692269326942695269626972698269927002701270227032704270527062707270827092710271127122713271427152716271727182719272027212722272327242725272627272728272927302731273227332734273527362737273827392740274127422743274427452746274727482749275027512752275327542755275627572758275927602761276227632764276527662767276827692770277127722773277427752776277727782779278027812782278327842785278627872788278927902791279227932794279527962797279827992800280128022803280428052806280728082809281028112812281328142815281628172818281928202821282228232824282528262827282828292830283128322833283428352836283728382839284028412842284328442845284628472848284928502851285228532854285528562857285828592860286128622863286428652866286728682869287028712872287328742875287628772878287928802881288228832884288528862887288828892890289128922893289428952896289728982899290029012902290329042905290629072908290929102911291229132914291529162917291829192920292129222923292429252926292729282929293029312932293329342935293629372938293929402941294229432944294529462947294829492950295129522953295429552956295729582959296029612962296329642965296629672968296929702971297229732974297529762977297829792980298129822983298429852986298729882989299029912992299329942995299629972998299930003001300230033004300530063007300830093010301130123013301430153016301730183019302030213022302330243025302630273028302930303031303230333034303530363037303830393040304130423043304430453046304730483049305030513052305330543055305630573058305930603061306230633064306530663067306830693070307130723073307430753076307730783079308030813082308330843085308630873088308930903091309230933094309530963097309830993100310131023103310431053106310731083109311031113112311331143115311631173118311931203121312231233124312531263127312831293130313131323133313431353136313731383139314031413142314331443145314631473148314931503151315231533154315531563157315831593160316131623163316431653166316731683169317031713172317331743175317631773178317931803181318231833184318531863187318831893190319131923193319431953196319731983199320032013202320332043205320632073208320932103211321232133214321532163217321832193220322132223223322432253226322732283229323032313232323332343235323632373238323932403241324232433244324532463247324832493250325132523253325432553256325732583259326032613262326332643265326632673268326932703271327232733274327532763277327832793280328132823283328432853286328732883289329032913292329332943295329632973298329933003301330233033304330533063307330833093310331133123313331433153316331733183319332033213322332333243325332633273328332933303331333233333334333533363337333833393340334133423343334433453346334733483349335033513352335333543355335633573358335933603361336233633364336533663367336833693370337133723373337433753376337733783379338033813382338333843385338633873388338933903391339233933394339533963397339833993400340134023403340434053406340734083409341034113412341334143415341634173418341934203421342234233424342534263427342834293430343134323433343434353436343734383439344034413442344334443445344634473448344934503451345234533454345534563457345834593460346134623463346434653466346734683469347034713472347334743475347634773478347934803481348234833484348534863487348834893490349134923493349434953496349734983499350035013502350335043505350635073508350935103511351235133514351535163517351835193520352135223523352435253526352735283529353035313532353335343535353635373538353935403541354235433544354535463547354835493550355135523553355435553556355735583559356035613562356335643565356635673568356935703571357235733574357535763577357835793580358135823583358435853586358735883589359035913592359335943595359635973598359936003601360236033604360536063607360836093610361136123613361436153616361736183619362036213622362336243625362636273628362936303631363236333634363536363637363836393640364136423643364436453646364736483649365036513652365336543655365636573658365936603661366236633664366536663667366836693670367136723673367436753676367736783679368036813682368336843685368636873688368936903691369236933694369536963697369836993700370137023703370437053706370737083709371037113712371337143715371637173718371937203721372237233724372537263727372837293730373137323733373437353736373737383739374037413742374337443745374637473748374937503751375237533754375537563757375837593760376137623763376437653766376737683769377037713772377337743775377637773778377937803781378237833784378537863787378837893790379137923793379437953796379737983799380038013802380338043805380638073808380938103811381238133814381538163817381838193820382138223823382438253826382738283829383038313832383338343835383638373838383938403841384238433844384538463847384838493850385138523853385438553856385738583859386038613862386338643865386638673868386938703871387238733874387538763877387838793880388138823883388438853886388738883889389038913892389338943895389638973898389939003901390239033904390539063907390839093910391139123913391439153916391739183919392039213922392339243925392639273928392939303931393239333934393539363937393839393940394139423943394439453946394739483949395039513952395339543955395639573958395939603961396239633964396539663967396839693970397139723973397439753976397739783979398039813982398339843985398639873988398939903991399239933994399539963997399839994000400140024003400440054006400740084009401040114012401340144015401640174018401940204021402240234024402540264027402840294030403140324033403440354036403740384039404040414042404340444045404640474048404940504051405240534054405540564057405840594060406140624063406440654066406740684069407040714072407340744075407640774078407940804081408240834084408540864087408840894090409140924093409440954096409740984099410041014102410341044105410641074108410941104111411241134114411541164117411841194120412141224123412441254126412741284129413041314132413341344135413641374138413941404141414241434144414541464147414841494150415141524153415441554156415741584159416041614162416341644165416641674168416941704171417241734174417541764177417841794180418141824183418441854186418741884189419041914192419341944195419641974198419942004201420242034204420542064207420842094210421142124213421442154216421742184219422042214222422342244225422642274228422942304231423242334234423542364237423842394240424142424243424442454246424742484249425042514252425342544255425642574258425942604261426242634264426542664267426842694270427142724273427442754276427742784279428042814282428342844285428642874288428942904291429242934294429542964297429842994300430143024303430443054306430743084309431043114312431343144315431643174318431943204321432243234324432543264327432843294330433143324333433443354336433743384339434043414342434343444345434643474348434943504351435243534354435543564357435843594360436143624363436443654366436743684369437043714372437343744375437643774378437943804381438243834384438543864387438843894390439143924393439443954396439743984399440044014402440344044405440644074408440944104411441244134414441544164417441844194420442144224423442444254426442744284429443044314432443344344435443644374438443944404441444244434444444544464447444844494450445144524453445444554456445744584459446044614462446344644465446644674468446944704471447244734474447544764477447844794480448144824483448444854486448744884489449044914492449344944495449644974498449945004501450245034504450545064507450845094510451145124513451445154516451745184519452045214522452345244525452645274528452945304531453245334534453545364537453845394540454145424543454445454546454745484549455045514552455345544555455645574558455945604561456245634564456545664567456845694570457145724573457445754576457745784579458045814582458345844585458645874588458945904591459245934594459545964597459845994600460146024603460446054606460746084609461046114612461346144615461646174618461946204621462246234624462546264627462846294630463146324633463446354636463746384639464046414642464346444645464646474648464946504651465246534654465546564657465846594660466146624663466446654666466746684669467046714672467346744675467646774678467946804681468246834684468546864687468846894690469146924693469446954696469746984699470047014702470347044705470647074708470947104711471247134714471547164717471847194720472147224723472447254726472747284729473047314732473347344735473647374738473947404741474247434744474547464747474847494750475147524753475447554756475747584759476047614762476347644765476647674768476947704771477247734774477547764777477847794780478147824783478447854786478747884789479047914792479347944795479647974798479948004801480248034804480548064807480848094810481148124813481448154816481748184819482048214822482348244825482648274828482948304831483248334834483548364837483848394840484148424843484448454846484748484849485048514852485348544855485648574858485948604861486248634864486548664867486848694870487148724873487448754876487748784879488048814882488348844885488648874888488948904891489248934894489548964897489848994900490149024903490449054906490749084909491049114912491349144915491649174918491949204921492249234924492549264927492849294930493149324933493449354936493749384939494049414942494349444945494649474948494949504951495249534954495549564957495849594960496149624963496449654966496749684969497049714972497349744975497649774978497949804981498249834984498549864987498849894990499149924993499449954996499749984999500050015002500350045005500650075008500950105011501250135014501550165017501850195020502150225023502450255026502750285029503050315032503350345035503650375038503950405041504250435044504550465047504850495050505150525053505450555056505750585059506050615062506350645065506650675068506950705071507250735074507550765077507850795080508150825083508450855086508750885089509050915092509350945095509650975098509951005101510251035104510551065107510851095110511151125113511451155116511751185119512051215122512351245125512651275128512951305131513251335134513551365137513851395140514151425143514451455146514751485149515051515152515351545155515651575158515951605161516251635164516551665167516851695170517151725173517451755176517751785179518051815182518351845185518651875188518951905191519251935194519551965197519851995200520152025203520452055206520752085209521052115212521352145215521652175218521952205221522252235224522552265227522852295230523152325233523452355236523752385239524052415242524352445245524652475248524952505251525252535254525552565257525852595260526152625263526452655266526752685269527052715272527352745275527652775278527952805281528252835284528552865287528852895290529152925293529452955296529752985299530053015302530353045305530653075308530953105311531253135314531553165317531853195320532153225323532453255326532753285329533053315332533353345335533653375338533953405341534253435344534553465347534853495350535153525353535453555356535753585359536053615362536353645365536653675368536953705371537253735374537553765377537853795380538153825383538453855386538753885389539053915392539353945395539653975398539954005401540254035404540554065407540854095410541154125413541454155416541754185419542054215422542354245425542654275428542954305431543254335434543554365437543854395440544154425443544454455446544754485449545054515452545354545455545654575458545954605461546254635464546554665467546854695470547154725473547454755476547754785479548054815482548354845485548654875488548954905491549254935494549554965497549854995500550155025503550455055506550755085509551055115512551355145515551655175518551955205521552255235524552555265527552855295530553155325533553455355536553755385539554055415542554355445545554655475548554955505551555255535554555555565557555855595560556155625563556455655566556755685569557055715572557355745575557655775578557955805581558255835584558555865587558855895590559155925593559455955596559755985599560056015602560356045605560656075608560956105611561256135614561556165617561856195620562156225623562456255626562756285629563056315632563356345635563656375638563956405641564256435644564556465647564856495650565156525653565456555656565756585659566056615662566356645665566656675668566956705671567256735674567556765677567856795680568156825683568456855686568756885689569056915692569356945695569656975698569957005701570257035704570557065707570857095710571157125713571457155716571757185719572057215722572357245725572657275728572957305731573257335734573557365737573857395740574157425743574457455746574757485749575057515752575357545755575657575758575957605761576257635764576557665767576857695770577157725773577457755776577757785779578057815782578357845785578657875788578957905791579257935794579557965797579857995800580158025803580458055806580758085809581058115812581358145815581658175818581958205821582258235824582558265827582858295830583158325833583458355836583758385839584058415842584358445845584658475848584958505851585258535854585558565857585858595860586158625863586458655866586758685869587058715872587358745875587658775878587958805881588258835884588558865887588858895890589158925893589458955896589758985899590059015902590359045905590659075908590959105911591259135914591559165917591859195920592159225923592459255926592759285929593059315932593359345935593659375938593959405941594259435944594559465947594859495950595159525953595459555956595759585959596059615962596359645965596659675968596959705971597259735974597559765977597859795980598159825983598459855986598759885989599059915992599359945995599659975998599960006001600260036004600560066007600860096010601160126013601460156016601760186019602060216022602360246025602660276028602960306031603260336034603560366037603860396040604160426043604460456046604760486049605060516052605360546055605660576058605960606061606260636064606560666067606860696070607160726073607460756076607760786079608060816082608360846085608660876088608960906091609260936094609560966097609860996100610161026103610461056106610761086109611061116112611361146115611661176118611961206121612261236124612561266127612861296130613161326133613461356136613761386139614061416142614361446145614661476148614961506151615261536154615561566157615861596160616161626163616461656166616761686169617061716172617361746175617661776178617961806181618261836184618561866187618861896190619161926193619461956196619761986199620062016202620362046205620662076208620962106211621262136214621562166217621862196220622162226223622462256226622762286229623062316232623362346235623662376238623962406241624262436244624562466247624862496250625162526253625462556256625762586259626062616262626362646265626662676268626962706271627262736274627562766277627862796280628162826283628462856286628762886289629062916292629362946295629662976298629963006301630263036304630563066307630863096310631163126313631463156316631763186319632063216322632363246325632663276328632963306331633263336334633563366337633863396340634163426343634463456346634763486349635063516352635363546355635663576358635963606361636263636364636563666367636863696370637163726373637463756376637763786379638063816382638363846385638663876388638963906391639263936394639563966397639863996400640164026403640464056406640764086409641064116412641364146415641664176418641964206421642264236424642564266427642864296430643164326433643464356436643764386439644064416442644364446445644664476448644964506451645264536454645564566457645864596460646164626463646464656466646764686469647064716472647364746475647664776478647964806481648264836484648564866487648864896490649164926493649464956496649764986499650065016502650365046505650665076508650965106511651265136514651565166517651865196520652165226523652465256526652765286529653065316532653365346535653665376538653965406541654265436544654565466547654865496550655165526553655465556556655765586559656065616562656365646565656665676568656965706571657265736574657565766577657865796580658165826583658465856586658765886589659065916592659365946595659665976598659966006601660266036604660566066607660866096610661166126613661466156616661766186619662066216622662366246625662666276628662966306631663266336634663566366637663866396640664166426643664466456646664766486649665066516652665366546655665666576658665966606661666266636664666566666667666866696670667166726673667466756676667766786679668066816682668366846685668666876688668966906691669266936694669566966697669866996700670167026703670467056706670767086709671067116712671367146715671667176718671967206721672267236724672567266727672867296730673167326733673467356736673767386739674067416742674367446745674667476748674967506751675267536754675567566757675867596760676167626763676467656766676767686769677067716772677367746775677667776778677967806781678267836784678567866787678867896790679167926793679467956796679767986799680068016802680368046805680668076808680968106811681268136814681568166817681868196820682168226823682468256826682768286829683068316832683368346835683668376838683968406841684268436844684568466847684868496850685168526853685468556856685768586859686068616862686368646865686668676868686968706871687268736874687568766877687868796880688168826883688468856886688768886889689068916892689368946895689668976898689969006901690269036904690569066907690869096910691169126913691469156916691769186919692069216922692369246925692669276928692969306931693269336934693569366937693869396940694169426943694469456946694769486949695069516952695369546955695669576958695969606961696269636964696569666967696869696970697169726973697469756976697769786979698069816982698369846985698669876988698969906991699269936994699569966997699869997000700170027003700470057006700770087009701070117012701370147015701670177018701970207021702270237024702570267027702870297030703170327033703470357036703770387039704070417042704370447045704670477048704970507051705270537054705570567057705870597060706170627063706470657066706770687069707070717072707370747075707670777078707970807081708270837084708570867087708870897090709170927093709470957096709770987099710071017102710371047105710671077108710971107111711271137114711571167117711871197120712171227123712471257126712771287129713071317132713371347135713671377138713971407141714271437144714571467147714871497150715171527153715471557156715771587159716071617162716371647165716671677168716971707171717271737174717571767177717871797180718171827183718471857186718771887189719071917192719371947195719671977198719972007201720272037204720572067207720872097210721172127213721472157216721772187219722072217222722372247225722672277228722972307231723272337234723572367237723872397240724172427243724472457246724772487249725072517252725372547255725672577258725972607261726272637264726572667267726872697270727172727273727472757276727772787279728072817282728372847285728672877288728972907291729272937294729572967297729872997300730173027303730473057306730773087309731073117312731373147315731673177318731973207321732273237324732573267327732873297330733173327333733473357336733773387339734073417342734373447345734673477348734973507351735273537354735573567357735873597360736173627363736473657366736773687369737073717372737373747375737673777378737973807381738273837384738573867387738873897390739173927393739473957396739773987399740074017402740374047405740674077408740974107411741274137414741574167417741874197420742174227423742474257426742774287429743074317432743374347435743674377438743974407441744274437444744574467447744874497450745174527453745474557456745774587459746074617462746374647465746674677468746974707471747274737474747574767477747874797480748174827483748474857486748774887489749074917492749374947495749674977498749975007501750275037504750575067507750875097510751175127513751475157516751775187519752075217522752375247525752675277528752975307531753275337534753575367537753875397540754175427543754475457546754775487549755075517552755375547555755675577558755975607561756275637564756575667567756875697570757175727573757475757576757775787579758075817582758375847585758675877588758975907591759275937594759575967597759875997600760176027603760476057606760776087609761076117612761376147615761676177618761976207621762276237624762576267627762876297630763176327633763476357636763776387639764076417642764376447645764676477648764976507651765276537654765576567657765876597660766176627663766476657666766776687669767076717672767376747675767676777678767976807681768276837684768576867687768876897690769176927693769476957696769776987699770077017702770377047705770677077708770977107711771277137714771577167717771877197720772177227723772477257726772777287729773077317732773377347735
  1. #![allow(
  2. dead_code,
  3. unused_imports,
  4. unused_variables,
  5. clippy::unneeded_struct_pattern,
  6. clippy::unnecessary_wraps,
  7. clippy::unused_self
  8. )]
  9. mod init;
  10. mod input;
  11. mod render;
  12. use std::collections::BTreeSet;
  13. use std::env;
  14. use std::fs;
  15. use std::io::{self, Read, Write};
  16. use std::net::TcpListener;
  17. use std::ops::{Deref, DerefMut};
  18. use std::path::{Path, PathBuf};
  19. use std::process::Command;
  20. use std::sync::mpsc::{self, Receiver, RecvTimeoutError, Sender};
  21. use std::sync::{Arc, Mutex};
  22. use std::thread::{self, JoinHandle};
  23. use std::time::{Duration, Instant, UNIX_EPOCH};
  24. use api::{
  25. resolve_startup_auth_source, AnthropicClient, AuthSource, ContentBlockDelta, InputContentBlock,
  26. InputMessage, MessageRequest, MessageResponse, OutputContentBlock, PromptCache,
  27. StreamEvent as ApiStreamEvent, ToolChoice, ToolDefinition, ToolResultContentBlock,
  28. };
  29. use commands::{
  30. handle_agents_slash_command, handle_mcp_slash_command, handle_plugins_slash_command,
  31. handle_skills_slash_command, render_slash_command_help, resume_supported_slash_commands,
  32. slash_command_specs, validate_slash_command_input, SlashCommand,
  33. };
  34. use compat_harness::{extract_manifest, UpstreamPaths};
  35. use init::initialize_repo;
  36. use plugins::{PluginHooks, PluginManager, PluginManagerConfig, PluginRegistry};
  37. use render::{MarkdownStreamState, Spinner, TerminalRenderer};
  38. use runtime::{
  39. clear_oauth_credentials, format_usd, generate_pkce_pair, generate_state, load_system_prompt,
  40. parse_oauth_callback_request_target, pricing_for_model, resolve_sandbox_status,
  41. save_oauth_credentials, ApiClient, ApiRequest, AssistantEvent, CompactionConfig, ConfigLoader,
  42. ConfigSource, ContentBlock, ConversationMessage, ConversationRuntime, McpServerManager,
  43. McpTool, MessageRole, ModelPricing, OAuthAuthorizationRequest, OAuthConfig,
  44. OAuthTokenExchangeRequest, PermissionMode, PermissionPolicy, ProjectContext, PromptCacheEvent,
  45. ResolvedPermissionMode, RuntimeError, Session, TokenUsage, ToolError, ToolExecutor,
  46. UsageTracker,
  47. };
  48. use serde::Deserialize;
  49. use serde_json::json;
  50. use tools::{GlobalToolRegistry, RuntimeToolDefinition, ToolSearchOutput};
  51. const DEFAULT_MODEL: &str = "claude-opus-4-6";
  52. fn max_tokens_for_model(model: &str) -> u32 {
  53. if model.contains("opus") {
  54. 32_000
  55. } else {
  56. 64_000
  57. }
  58. }
  59. const DEFAULT_DATE: &str = "2026-03-31";
  60. const DEFAULT_OAUTH_CALLBACK_PORT: u16 = 4545;
  61. const VERSION: &str = env!("CARGO_PKG_VERSION");
  62. const BUILD_TARGET: Option<&str> = option_env!("TARGET");
  63. const GIT_SHA: Option<&str> = option_env!("GIT_SHA");
  64. const INTERNAL_PROGRESS_HEARTBEAT_INTERVAL: Duration = Duration::from_secs(3);
  65. const PRIMARY_SESSION_EXTENSION: &str = "jsonl";
  66. const LEGACY_SESSION_EXTENSION: &str = "json";
  67. const LATEST_SESSION_REFERENCE: &str = "latest";
  68. const SESSION_REFERENCE_ALIASES: &[&str] = &[LATEST_SESSION_REFERENCE, "last", "recent"];
  69. const CLI_OPTION_SUGGESTIONS: &[&str] = &[
  70. "--help",
  71. "-h",
  72. "--version",
  73. "-V",
  74. "--model",
  75. "--output-format",
  76. "--permission-mode",
  77. "--dangerously-skip-permissions",
  78. "--allowedTools",
  79. "--allowed-tools",
  80. "--resume",
  81. "--print",
  82. "-p",
  83. ];
  84. type AllowedToolSet = BTreeSet<String>;
  85. fn main() {
  86. if let Err(error) = run() {
  87. let message = error.to_string();
  88. if message.contains("`claw --help`") {
  89. eprintln!("error: {message}");
  90. } else {
  91. eprintln!(
  92. "error: {message}
  93. Run `claw --help` for usage."
  94. );
  95. }
  96. std::process::exit(1);
  97. }
  98. }
  99. fn run() -> Result<(), Box<dyn std::error::Error>> {
  100. let args: Vec<String> = env::args().skip(1).collect();
  101. match parse_args(&args)? {
  102. CliAction::DumpManifests => dump_manifests(),
  103. CliAction::BootstrapPlan => print_bootstrap_plan(),
  104. CliAction::Agents { args } => LiveCli::print_agents(args.as_deref())?,
  105. CliAction::Mcp { args } => LiveCli::print_mcp(args.as_deref())?,
  106. CliAction::Skills { args } => LiveCli::print_skills(args.as_deref())?,
  107. CliAction::PrintSystemPrompt { cwd, date } => print_system_prompt(cwd, date),
  108. CliAction::Version => print_version(),
  109. CliAction::ResumeSession {
  110. session_path,
  111. commands,
  112. } => resume_session(&session_path, &commands),
  113. CliAction::Status {
  114. model,
  115. permission_mode,
  116. } => print_status_snapshot(&model, permission_mode)?,
  117. CliAction::Sandbox => print_sandbox_status_snapshot()?,
  118. CliAction::Prompt {
  119. prompt,
  120. model,
  121. output_format,
  122. allowed_tools,
  123. permission_mode,
  124. } => LiveCli::new(model, true, allowed_tools, permission_mode)?
  125. .run_turn_with_output(&prompt, output_format)?,
  126. CliAction::Login => run_login()?,
  127. CliAction::Logout => run_logout()?,
  128. CliAction::Init => run_init()?,
  129. CliAction::Repl {
  130. model,
  131. allowed_tools,
  132. permission_mode,
  133. } => run_repl(model, allowed_tools, permission_mode)?,
  134. CliAction::Help => print_help(),
  135. }
  136. Ok(())
  137. }
  138. #[derive(Debug, Clone, PartialEq, Eq)]
  139. enum CliAction {
  140. DumpManifests,
  141. BootstrapPlan,
  142. Agents {
  143. args: Option<String>,
  144. },
  145. Mcp {
  146. args: Option<String>,
  147. },
  148. Skills {
  149. args: Option<String>,
  150. },
  151. PrintSystemPrompt {
  152. cwd: PathBuf,
  153. date: String,
  154. },
  155. Version,
  156. ResumeSession {
  157. session_path: PathBuf,
  158. commands: Vec<String>,
  159. },
  160. Status {
  161. model: String,
  162. permission_mode: PermissionMode,
  163. },
  164. Sandbox,
  165. Prompt {
  166. prompt: String,
  167. model: String,
  168. output_format: CliOutputFormat,
  169. allowed_tools: Option<AllowedToolSet>,
  170. permission_mode: PermissionMode,
  171. },
  172. Login,
  173. Logout,
  174. Init,
  175. Repl {
  176. model: String,
  177. allowed_tools: Option<AllowedToolSet>,
  178. permission_mode: PermissionMode,
  179. },
  180. // prompt-mode formatting is only supported for non-interactive runs
  181. Help,
  182. }
  183. #[derive(Debug, Clone, Copy, PartialEq, Eq)]
  184. enum CliOutputFormat {
  185. Text,
  186. Json,
  187. }
  188. impl CliOutputFormat {
  189. fn parse(value: &str) -> Result<Self, String> {
  190. match value {
  191. "text" => Ok(Self::Text),
  192. "json" => Ok(Self::Json),
  193. other => Err(format!(
  194. "unsupported value for --output-format: {other} (expected text or json)"
  195. )),
  196. }
  197. }
  198. }
  199. #[allow(clippy::too_many_lines)]
  200. fn parse_args(args: &[String]) -> Result<CliAction, String> {
  201. let mut model = DEFAULT_MODEL.to_string();
  202. let mut output_format = CliOutputFormat::Text;
  203. let mut permission_mode = default_permission_mode();
  204. let mut wants_help = false;
  205. let mut wants_version = false;
  206. let mut allowed_tool_values = Vec::new();
  207. let mut rest = Vec::new();
  208. let mut index = 0;
  209. while index < args.len() {
  210. match args[index].as_str() {
  211. "--help" | "-h" if rest.is_empty() => {
  212. wants_help = true;
  213. index += 1;
  214. }
  215. "--version" | "-V" => {
  216. wants_version = true;
  217. index += 1;
  218. }
  219. "--model" => {
  220. let value = args
  221. .get(index + 1)
  222. .ok_or_else(|| "missing value for --model".to_string())?;
  223. model = resolve_model_alias(value).to_string();
  224. index += 2;
  225. }
  226. flag if flag.starts_with("--model=") => {
  227. model = resolve_model_alias(&flag[8..]).to_string();
  228. index += 1;
  229. }
  230. "--output-format" => {
  231. let value = args
  232. .get(index + 1)
  233. .ok_or_else(|| "missing value for --output-format".to_string())?;
  234. output_format = CliOutputFormat::parse(value)?;
  235. index += 2;
  236. }
  237. "--permission-mode" => {
  238. let value = args
  239. .get(index + 1)
  240. .ok_or_else(|| "missing value for --permission-mode".to_string())?;
  241. permission_mode = parse_permission_mode_arg(value)?;
  242. index += 2;
  243. }
  244. flag if flag.starts_with("--output-format=") => {
  245. output_format = CliOutputFormat::parse(&flag[16..])?;
  246. index += 1;
  247. }
  248. flag if flag.starts_with("--permission-mode=") => {
  249. permission_mode = parse_permission_mode_arg(&flag[18..])?;
  250. index += 1;
  251. }
  252. "--dangerously-skip-permissions" => {
  253. permission_mode = PermissionMode::DangerFullAccess;
  254. index += 1;
  255. }
  256. "-p" => {
  257. // Claw Code compat: -p "prompt" = one-shot prompt
  258. let prompt = args[index + 1..].join(" ");
  259. if prompt.trim().is_empty() {
  260. return Err("-p requires a prompt string".to_string());
  261. }
  262. return Ok(CliAction::Prompt {
  263. prompt,
  264. model: resolve_model_alias(&model).to_string(),
  265. output_format,
  266. allowed_tools: normalize_allowed_tools(&allowed_tool_values)?,
  267. permission_mode,
  268. });
  269. }
  270. "--print" => {
  271. // Claw Code compat: --print makes output non-interactive
  272. output_format = CliOutputFormat::Text;
  273. index += 1;
  274. }
  275. "--resume" if rest.is_empty() => {
  276. rest.push("--resume".to_string());
  277. index += 1;
  278. }
  279. flag if rest.is_empty() && flag.starts_with("--resume=") => {
  280. rest.push("--resume".to_string());
  281. rest.push(flag[9..].to_string());
  282. index += 1;
  283. }
  284. "--allowedTools" | "--allowed-tools" => {
  285. let value = args
  286. .get(index + 1)
  287. .ok_or_else(|| "missing value for --allowedTools".to_string())?;
  288. allowed_tool_values.push(value.clone());
  289. index += 2;
  290. }
  291. flag if flag.starts_with("--allowedTools=") => {
  292. allowed_tool_values.push(flag[15..].to_string());
  293. index += 1;
  294. }
  295. flag if flag.starts_with("--allowed-tools=") => {
  296. allowed_tool_values.push(flag[16..].to_string());
  297. index += 1;
  298. }
  299. other if rest.is_empty() && other.starts_with('-') => {
  300. return Err(format_unknown_option(other))
  301. }
  302. other => {
  303. rest.push(other.to_string());
  304. index += 1;
  305. }
  306. }
  307. }
  308. if wants_help {
  309. return Ok(CliAction::Help);
  310. }
  311. if wants_version {
  312. return Ok(CliAction::Version);
  313. }
  314. let allowed_tools = normalize_allowed_tools(&allowed_tool_values)?;
  315. if rest.is_empty() {
  316. return Ok(CliAction::Repl {
  317. model,
  318. allowed_tools,
  319. permission_mode,
  320. });
  321. }
  322. if rest.first().map(String::as_str) == Some("--resume") {
  323. return parse_resume_args(&rest[1..]);
  324. }
  325. if let Some(action) = parse_single_word_command_alias(&rest, &model, permission_mode) {
  326. return action;
  327. }
  328. match rest[0].as_str() {
  329. "dump-manifests" => Ok(CliAction::DumpManifests),
  330. "bootstrap-plan" => Ok(CliAction::BootstrapPlan),
  331. "agents" => Ok(CliAction::Agents {
  332. args: join_optional_args(&rest[1..]),
  333. }),
  334. "mcp" => Ok(CliAction::Mcp {
  335. args: join_optional_args(&rest[1..]),
  336. }),
  337. "skills" => Ok(CliAction::Skills {
  338. args: join_optional_args(&rest[1..]),
  339. }),
  340. "system-prompt" => parse_system_prompt_args(&rest[1..]),
  341. "login" => Ok(CliAction::Login),
  342. "logout" => Ok(CliAction::Logout),
  343. "init" => Ok(CliAction::Init),
  344. "prompt" => {
  345. let prompt = rest[1..].join(" ");
  346. if prompt.trim().is_empty() {
  347. return Err("prompt subcommand requires a prompt string".to_string());
  348. }
  349. Ok(CliAction::Prompt {
  350. prompt,
  351. model,
  352. output_format,
  353. allowed_tools,
  354. permission_mode,
  355. })
  356. }
  357. other if other.starts_with('/') => parse_direct_slash_cli_action(&rest),
  358. _other => Ok(CliAction::Prompt {
  359. prompt: rest.join(" "),
  360. model,
  361. output_format,
  362. allowed_tools,
  363. permission_mode,
  364. }),
  365. }
  366. }
  367. fn parse_single_word_command_alias(
  368. rest: &[String],
  369. model: &str,
  370. permission_mode: PermissionMode,
  371. ) -> Option<Result<CliAction, String>> {
  372. if rest.len() != 1 {
  373. return None;
  374. }
  375. match rest[0].as_str() {
  376. "help" => Some(Ok(CliAction::Help)),
  377. "version" => Some(Ok(CliAction::Version)),
  378. "status" => Some(Ok(CliAction::Status {
  379. model: model.to_string(),
  380. permission_mode,
  381. })),
  382. "sandbox" => Some(Ok(CliAction::Sandbox)),
  383. other => bare_slash_command_guidance(other).map(Err),
  384. }
  385. }
  386. fn bare_slash_command_guidance(command_name: &str) -> Option<String> {
  387. if matches!(
  388. command_name,
  389. "dump-manifests"
  390. | "bootstrap-plan"
  391. | "agents"
  392. | "mcp"
  393. | "skills"
  394. | "system-prompt"
  395. | "login"
  396. | "logout"
  397. | "init"
  398. | "prompt"
  399. ) {
  400. return None;
  401. }
  402. let slash_command = slash_command_specs()
  403. .iter()
  404. .find(|spec| spec.name == command_name)?;
  405. let guidance = if slash_command.resume_supported {
  406. format!(
  407. "`claw {command_name}` is a slash command. Use `claw --resume SESSION.jsonl /{command_name}` or start `claw` and run `/{command_name}`."
  408. )
  409. } else {
  410. format!(
  411. "`claw {command_name}` is a slash command. Start `claw` and run `/{command_name}` inside the REPL."
  412. )
  413. };
  414. Some(guidance)
  415. }
  416. fn join_optional_args(args: &[String]) -> Option<String> {
  417. let joined = args.join(" ");
  418. let trimmed = joined.trim();
  419. (!trimmed.is_empty()).then(|| trimmed.to_string())
  420. }
  421. fn parse_direct_slash_cli_action(rest: &[String]) -> Result<CliAction, String> {
  422. let raw = rest.join(" ");
  423. match SlashCommand::parse(&raw) {
  424. Ok(Some(SlashCommand::Help)) => Ok(CliAction::Help),
  425. Ok(Some(SlashCommand::Agents { args })) => Ok(CliAction::Agents { args }),
  426. Ok(Some(SlashCommand::Mcp { action, target })) => Ok(CliAction::Mcp {
  427. args: match (action, target) {
  428. (None, None) => None,
  429. (Some(action), None) => Some(action),
  430. (Some(action), Some(target)) => Some(format!("{action} {target}")),
  431. (None, Some(target)) => Some(target),
  432. },
  433. }),
  434. Ok(Some(SlashCommand::Skills { args })) => Ok(CliAction::Skills { args }),
  435. Ok(Some(SlashCommand::Unknown(name))) => Err(format_unknown_direct_slash_command(&name)),
  436. Ok(Some(command)) => Err({
  437. let _ = command;
  438. format!(
  439. "slash command {command_name} is interactive-only. Start `claw` and run it there, or use `claw --resume SESSION.jsonl {command_name}` / `claw --resume {latest} {command_name}` when the command is marked [resume] in /help.",
  440. command_name = rest[0],
  441. latest = LATEST_SESSION_REFERENCE,
  442. )
  443. }),
  444. Ok(None) => Err(format!("unknown subcommand: {}", rest[0])),
  445. Err(error) => Err(error.to_string()),
  446. }
  447. }
  448. fn format_unknown_option(option: &str) -> String {
  449. let mut message = format!("unknown option: {option}");
  450. if let Some(suggestion) = suggest_closest_term(option, CLI_OPTION_SUGGESTIONS) {
  451. message.push_str("\nDid you mean ");
  452. message.push_str(suggestion);
  453. message.push('?');
  454. }
  455. message.push_str("\nRun `claw --help` for usage.");
  456. message
  457. }
  458. fn format_unknown_direct_slash_command(name: &str) -> String {
  459. let mut message = format!("unknown slash command outside the REPL: /{name}");
  460. if let Some(suggestions) = render_suggestion_line("Did you mean", &suggest_slash_commands(name))
  461. {
  462. message.push('\n');
  463. message.push_str(&suggestions);
  464. }
  465. message.push_str("\nRun `claw --help` for CLI usage, or start `claw` and use /help.");
  466. message
  467. }
  468. fn format_unknown_slash_command(name: &str) -> String {
  469. let mut message = format!("Unknown slash command: /{name}");
  470. if let Some(suggestions) = render_suggestion_line("Did you mean", &suggest_slash_commands(name))
  471. {
  472. message.push('\n');
  473. message.push_str(&suggestions);
  474. }
  475. message.push_str("\n Help /help lists available slash commands");
  476. message
  477. }
  478. fn render_suggestion_line(label: &str, suggestions: &[String]) -> Option<String> {
  479. (!suggestions.is_empty()).then(|| format!(" {label:<16} {}", suggestions.join(", "),))
  480. }
  481. fn suggest_slash_commands(input: &str) -> Vec<String> {
  482. let mut candidates = slash_command_specs()
  483. .iter()
  484. .flat_map(|spec| {
  485. std::iter::once(spec.name)
  486. .chain(spec.aliases.iter().copied())
  487. .map(|name| format!("/{name}"))
  488. .collect::<Vec<_>>()
  489. })
  490. .collect::<Vec<_>>();
  491. candidates.sort();
  492. candidates.dedup();
  493. let candidate_refs = candidates.iter().map(String::as_str).collect::<Vec<_>>();
  494. ranked_suggestions(input.trim_start_matches('/'), &candidate_refs)
  495. .into_iter()
  496. .map(str::to_string)
  497. .collect()
  498. }
  499. fn suggest_closest_term<'a>(input: &str, candidates: &'a [&'a str]) -> Option<&'a str> {
  500. ranked_suggestions(input, candidates).into_iter().next()
  501. }
  502. fn ranked_suggestions<'a>(input: &str, candidates: &'a [&'a str]) -> Vec<&'a str> {
  503. let normalized_input = input.trim_start_matches('/').to_ascii_lowercase();
  504. let mut ranked = candidates
  505. .iter()
  506. .filter_map(|candidate| {
  507. let normalized_candidate = candidate.trim_start_matches('/').to_ascii_lowercase();
  508. let distance = levenshtein_distance(&normalized_input, &normalized_candidate);
  509. let prefix_bonus = usize::from(
  510. !(normalized_candidate.starts_with(&normalized_input)
  511. || normalized_input.starts_with(&normalized_candidate)),
  512. );
  513. let score = distance + prefix_bonus;
  514. (score <= 4).then_some((score, *candidate))
  515. })
  516. .collect::<Vec<_>>();
  517. ranked.sort_by(|left, right| left.cmp(right).then_with(|| left.1.cmp(right.1)));
  518. ranked
  519. .into_iter()
  520. .map(|(_, candidate)| candidate)
  521. .take(3)
  522. .collect()
  523. }
  524. fn levenshtein_distance(left: &str, right: &str) -> usize {
  525. if left.is_empty() {
  526. return right.chars().count();
  527. }
  528. if right.is_empty() {
  529. return left.chars().count();
  530. }
  531. let right_chars = right.chars().collect::<Vec<_>>();
  532. let mut previous = (0..=right_chars.len()).collect::<Vec<_>>();
  533. let mut current = vec![0; right_chars.len() + 1];
  534. for (left_index, left_char) in left.chars().enumerate() {
  535. current[0] = left_index + 1;
  536. for (right_index, right_char) in right_chars.iter().enumerate() {
  537. let substitution_cost = usize::from(left_char != *right_char);
  538. current[right_index + 1] = (previous[right_index + 1] + 1)
  539. .min(current[right_index] + 1)
  540. .min(previous[right_index] + substitution_cost);
  541. }
  542. previous.clone_from(&current);
  543. }
  544. previous[right_chars.len()]
  545. }
  546. fn resolve_model_alias(model: &str) -> &str {
  547. match model {
  548. "opus" => "claude-opus-4-6",
  549. "sonnet" => "claude-sonnet-4-6",
  550. "haiku" => "claude-haiku-4-5-20251213",
  551. _ => model,
  552. }
  553. }
  554. fn normalize_allowed_tools(values: &[String]) -> Result<Option<AllowedToolSet>, String> {
  555. current_tool_registry()?.normalize_allowed_tools(values)
  556. }
  557. fn current_tool_registry() -> Result<GlobalToolRegistry, String> {
  558. let cwd = env::current_dir().map_err(|error| error.to_string())?;
  559. let loader = ConfigLoader::default_for(&cwd);
  560. let runtime_config = loader.load().map_err(|error| error.to_string())?;
  561. let state = build_runtime_plugin_state_with_loader(&cwd, &loader, &runtime_config)
  562. .map_err(|error| error.to_string())?;
  563. let registry = state.tool_registry.clone();
  564. if let Some(mcp_state) = state.mcp_state {
  565. mcp_state
  566. .lock()
  567. .unwrap_or_else(std::sync::PoisonError::into_inner)
  568. .shutdown()
  569. .map_err(|error| error.to_string())?;
  570. }
  571. Ok(registry)
  572. }
  573. fn parse_permission_mode_arg(value: &str) -> Result<PermissionMode, String> {
  574. normalize_permission_mode(value)
  575. .ok_or_else(|| {
  576. format!(
  577. "unsupported permission mode '{value}'. Use read-only, workspace-write, or danger-full-access."
  578. )
  579. })
  580. .map(permission_mode_from_label)
  581. }
  582. fn permission_mode_from_label(mode: &str) -> PermissionMode {
  583. match mode {
  584. "read-only" => PermissionMode::ReadOnly,
  585. "workspace-write" => PermissionMode::WorkspaceWrite,
  586. "danger-full-access" => PermissionMode::DangerFullAccess,
  587. other => panic!("unsupported permission mode label: {other}"),
  588. }
  589. }
  590. fn permission_mode_from_resolved(mode: ResolvedPermissionMode) -> PermissionMode {
  591. match mode {
  592. ResolvedPermissionMode::ReadOnly => PermissionMode::ReadOnly,
  593. ResolvedPermissionMode::WorkspaceWrite => PermissionMode::WorkspaceWrite,
  594. ResolvedPermissionMode::DangerFullAccess => PermissionMode::DangerFullAccess,
  595. }
  596. }
  597. fn default_permission_mode() -> PermissionMode {
  598. env::var("RUSTY_CLAUDE_PERMISSION_MODE")
  599. .ok()
  600. .as_deref()
  601. .and_then(normalize_permission_mode)
  602. .map(permission_mode_from_label)
  603. .or_else(config_permission_mode_for_current_dir)
  604. .unwrap_or(PermissionMode::DangerFullAccess)
  605. }
  606. fn config_permission_mode_for_current_dir() -> Option<PermissionMode> {
  607. let cwd = env::current_dir().ok()?;
  608. let loader = ConfigLoader::default_for(&cwd);
  609. loader
  610. .load()
  611. .ok()?
  612. .permission_mode()
  613. .map(permission_mode_from_resolved)
  614. }
  615. fn filter_tool_specs(
  616. tool_registry: &GlobalToolRegistry,
  617. allowed_tools: Option<&AllowedToolSet>,
  618. ) -> Vec<ToolDefinition> {
  619. tool_registry.definitions(allowed_tools)
  620. }
  621. fn parse_system_prompt_args(args: &[String]) -> Result<CliAction, String> {
  622. let mut cwd = env::current_dir().map_err(|error| error.to_string())?;
  623. let mut date = DEFAULT_DATE.to_string();
  624. let mut index = 0;
  625. while index < args.len() {
  626. match args[index].as_str() {
  627. "--cwd" => {
  628. let value = args
  629. .get(index + 1)
  630. .ok_or_else(|| "missing value for --cwd".to_string())?;
  631. cwd = PathBuf::from(value);
  632. index += 2;
  633. }
  634. "--date" => {
  635. let value = args
  636. .get(index + 1)
  637. .ok_or_else(|| "missing value for --date".to_string())?;
  638. date.clone_from(value);
  639. index += 2;
  640. }
  641. other => return Err(format!("unknown system-prompt option: {other}")),
  642. }
  643. }
  644. Ok(CliAction::PrintSystemPrompt { cwd, date })
  645. }
  646. fn parse_resume_args(args: &[String]) -> Result<CliAction, String> {
  647. let (session_path, command_tokens): (PathBuf, &[String]) = match args.first() {
  648. None => (PathBuf::from(LATEST_SESSION_REFERENCE), &[]),
  649. Some(first) if looks_like_slash_command_token(first) => {
  650. (PathBuf::from(LATEST_SESSION_REFERENCE), args)
  651. }
  652. Some(first) => (PathBuf::from(first), &args[1..]),
  653. };
  654. let mut commands = Vec::new();
  655. let mut current_command = String::new();
  656. for token in command_tokens {
  657. if token.trim_start().starts_with('/') {
  658. if resume_command_can_absorb_token(&current_command, token) {
  659. current_command.push(' ');
  660. current_command.push_str(token);
  661. continue;
  662. }
  663. if !current_command.is_empty() {
  664. commands.push(current_command);
  665. }
  666. current_command = String::from(token.as_str());
  667. continue;
  668. }
  669. if current_command.is_empty() {
  670. return Err("--resume trailing arguments must be slash commands".to_string());
  671. }
  672. current_command.push(' ');
  673. current_command.push_str(token);
  674. }
  675. if !current_command.is_empty() {
  676. commands.push(current_command);
  677. }
  678. Ok(CliAction::ResumeSession {
  679. session_path,
  680. commands,
  681. })
  682. }
  683. fn resume_command_can_absorb_token(current_command: &str, token: &str) -> bool {
  684. matches!(
  685. SlashCommand::parse(current_command),
  686. Ok(Some(SlashCommand::Export { path: None }))
  687. ) && !looks_like_slash_command_token(token)
  688. }
  689. fn looks_like_slash_command_token(token: &str) -> bool {
  690. let trimmed = token.trim_start();
  691. let Some(name) = trimmed.strip_prefix('/').and_then(|value| {
  692. value
  693. .split_whitespace()
  694. .next()
  695. .map(str::trim)
  696. .filter(|value| !value.is_empty())
  697. }) else {
  698. return false;
  699. };
  700. slash_command_specs()
  701. .iter()
  702. .any(|spec| spec.name == name || spec.aliases.contains(&name))
  703. }
  704. fn dump_manifests() {
  705. let workspace_dir = PathBuf::from(env!("CARGO_MANIFEST_DIR")).join("../..");
  706. let paths = UpstreamPaths::from_workspace_dir(&workspace_dir);
  707. match extract_manifest(&paths) {
  708. Ok(manifest) => {
  709. println!("commands: {}", manifest.commands.entries().len());
  710. println!("tools: {}", manifest.tools.entries().len());
  711. println!("bootstrap phases: {}", manifest.bootstrap.phases().len());
  712. }
  713. Err(error) => {
  714. eprintln!("failed to extract manifests: {error}");
  715. std::process::exit(1);
  716. }
  717. }
  718. }
  719. fn print_bootstrap_plan() {
  720. for phase in runtime::BootstrapPlan::claude_code_default().phases() {
  721. println!("- {phase:?}");
  722. }
  723. }
  724. fn default_oauth_config() -> OAuthConfig {
  725. OAuthConfig {
  726. client_id: String::from("9d1c250a-e61b-44d9-88ed-5944d1962f5e"),
  727. authorize_url: String::from("https://platform.claude.com/oauth/authorize"),
  728. token_url: String::from("https://platform.claude.com/v1/oauth/token"),
  729. callback_port: None,
  730. manual_redirect_url: None,
  731. scopes: vec![
  732. String::from("user:profile"),
  733. String::from("user:inference"),
  734. String::from("user:sessions:claude_code"),
  735. ],
  736. }
  737. }
  738. fn run_login() -> Result<(), Box<dyn std::error::Error>> {
  739. let cwd = env::current_dir()?;
  740. let config = ConfigLoader::default_for(&cwd).load()?;
  741. let default_oauth = default_oauth_config();
  742. let oauth = config.oauth().unwrap_or(&default_oauth);
  743. let callback_port = oauth.callback_port.unwrap_or(DEFAULT_OAUTH_CALLBACK_PORT);
  744. let redirect_uri = runtime::loopback_redirect_uri(callback_port);
  745. let pkce = generate_pkce_pair()?;
  746. let state = generate_state()?;
  747. let authorize_url =
  748. OAuthAuthorizationRequest::from_config(oauth, redirect_uri.clone(), state.clone(), &pkce)
  749. .build_url();
  750. println!("Starting Claude OAuth login...");
  751. println!("Listening for callback on {redirect_uri}");
  752. if let Err(error) = open_browser(&authorize_url) {
  753. eprintln!("warning: failed to open browser automatically: {error}");
  754. println!("Open this URL manually:\n{authorize_url}");
  755. }
  756. let callback = wait_for_oauth_callback(callback_port)?;
  757. if let Some(error) = callback.error {
  758. let description = callback
  759. .error_description
  760. .unwrap_or_else(|| "authorization failed".to_string());
  761. return Err(io::Error::other(format!("{error}: {description}")).into());
  762. }
  763. let code = callback.code.ok_or_else(|| {
  764. io::Error::new(io::ErrorKind::InvalidData, "callback did not include code")
  765. })?;
  766. let returned_state = callback.state.ok_or_else(|| {
  767. io::Error::new(io::ErrorKind::InvalidData, "callback did not include state")
  768. })?;
  769. if returned_state != state {
  770. return Err(io::Error::new(io::ErrorKind::InvalidData, "oauth state mismatch").into());
  771. }
  772. let client = AnthropicClient::from_auth(AuthSource::None).with_base_url(api::read_base_url());
  773. let exchange_request =
  774. OAuthTokenExchangeRequest::from_config(oauth, code, state, pkce.verifier, redirect_uri);
  775. let runtime = tokio::runtime::Runtime::new()?;
  776. let token_set = runtime.block_on(client.exchange_oauth_code(oauth, &exchange_request))?;
  777. save_oauth_credentials(&runtime::OAuthTokenSet {
  778. access_token: token_set.access_token,
  779. refresh_token: token_set.refresh_token,
  780. expires_at: token_set.expires_at,
  781. scopes: token_set.scopes,
  782. })?;
  783. println!("Claude OAuth login complete.");
  784. Ok(())
  785. }
  786. fn run_logout() -> Result<(), Box<dyn std::error::Error>> {
  787. clear_oauth_credentials()?;
  788. println!("Claude OAuth credentials cleared.");
  789. Ok(())
  790. }
  791. fn open_browser(url: &str) -> io::Result<()> {
  792. let commands = if cfg!(target_os = "macos") {
  793. vec![("open", vec![url])]
  794. } else if cfg!(target_os = "windows") {
  795. vec![("cmd", vec!["/C", "start", "", url])]
  796. } else {
  797. vec![("xdg-open", vec![url])]
  798. };
  799. for (program, args) in commands {
  800. match Command::new(program).args(args).spawn() {
  801. Ok(_) => return Ok(()),
  802. Err(error) if error.kind() == io::ErrorKind::NotFound => {}
  803. Err(error) => return Err(error),
  804. }
  805. }
  806. Err(io::Error::new(
  807. io::ErrorKind::NotFound,
  808. "no supported browser opener command found",
  809. ))
  810. }
  811. fn wait_for_oauth_callback(
  812. port: u16,
  813. ) -> Result<runtime::OAuthCallbackParams, Box<dyn std::error::Error>> {
  814. let listener = TcpListener::bind(("127.0.0.1", port))?;
  815. let (mut stream, _) = listener.accept()?;
  816. let mut buffer = [0_u8; 4096];
  817. let bytes_read = stream.read(&mut buffer)?;
  818. let request = String::from_utf8_lossy(&buffer[..bytes_read]);
  819. let request_line = request.lines().next().ok_or_else(|| {
  820. io::Error::new(io::ErrorKind::InvalidData, "missing callback request line")
  821. })?;
  822. let target = request_line.split_whitespace().nth(1).ok_or_else(|| {
  823. io::Error::new(
  824. io::ErrorKind::InvalidData,
  825. "missing callback request target",
  826. )
  827. })?;
  828. let callback = parse_oauth_callback_request_target(target)
  829. .map_err(|error| io::Error::new(io::ErrorKind::InvalidData, error))?;
  830. let body = if callback.error.is_some() {
  831. "Claude OAuth login failed. You can close this window."
  832. } else {
  833. "Claude OAuth login succeeded. You can close this window."
  834. };
  835. let response = format!(
  836. "HTTP/1.1 200 OK\r\ncontent-type: text/plain; charset=utf-8\r\ncontent-length: {}\r\nconnection: close\r\n\r\n{}",
  837. body.len(),
  838. body
  839. );
  840. stream.write_all(response.as_bytes())?;
  841. Ok(callback)
  842. }
  843. fn print_system_prompt(cwd: PathBuf, date: String) {
  844. match load_system_prompt(cwd, date, env::consts::OS, "unknown") {
  845. Ok(sections) => println!("{}", sections.join("\n\n")),
  846. Err(error) => {
  847. eprintln!("failed to build system prompt: {error}");
  848. std::process::exit(1);
  849. }
  850. }
  851. }
  852. fn print_version() {
  853. println!("{}", render_version_report());
  854. }
  855. fn resume_session(session_path: &Path, commands: &[String]) {
  856. let resolved_path = if session_path.exists() {
  857. session_path.to_path_buf()
  858. } else {
  859. match resolve_session_reference(&session_path.display().to_string()) {
  860. Ok(handle) => handle.path,
  861. Err(error) => {
  862. eprintln!("failed to restore session: {error}");
  863. std::process::exit(1);
  864. }
  865. }
  866. };
  867. let session = match Session::load_from_path(&resolved_path) {
  868. Ok(session) => session,
  869. Err(error) => {
  870. eprintln!("failed to restore session: {error}");
  871. std::process::exit(1);
  872. }
  873. };
  874. if commands.is_empty() {
  875. println!(
  876. "Restored session from {} ({} messages).",
  877. resolved_path.display(),
  878. session.messages.len()
  879. );
  880. return;
  881. }
  882. let mut session = session;
  883. for raw_command in commands {
  884. let command = match SlashCommand::parse(raw_command) {
  885. Ok(Some(command)) => command,
  886. Ok(None) => {
  887. eprintln!("unsupported resumed command: {raw_command}");
  888. std::process::exit(2);
  889. }
  890. Err(error) => {
  891. eprintln!("{error}");
  892. std::process::exit(2);
  893. }
  894. };
  895. match run_resume_command(&resolved_path, &session, &command) {
  896. Ok(ResumeCommandOutcome {
  897. session: next_session,
  898. message,
  899. }) => {
  900. session = next_session;
  901. if let Some(message) = message {
  902. println!("{message}");
  903. }
  904. }
  905. Err(error) => {
  906. eprintln!("{error}");
  907. std::process::exit(2);
  908. }
  909. }
  910. }
  911. }
  912. #[derive(Debug, Clone)]
  913. struct ResumeCommandOutcome {
  914. session: Session,
  915. message: Option<String>,
  916. }
  917. #[derive(Debug, Clone)]
  918. struct StatusContext {
  919. cwd: PathBuf,
  920. session_path: Option<PathBuf>,
  921. loaded_config_files: usize,
  922. discovered_config_files: usize,
  923. memory_file_count: usize,
  924. project_root: Option<PathBuf>,
  925. git_branch: Option<String>,
  926. git_summary: GitWorkspaceSummary,
  927. sandbox_status: runtime::SandboxStatus,
  928. }
  929. #[derive(Debug, Clone, Copy)]
  930. struct StatusUsage {
  931. message_count: usize,
  932. turns: u32,
  933. latest: TokenUsage,
  934. cumulative: TokenUsage,
  935. estimated_tokens: usize,
  936. }
  937. #[allow(clippy::struct_field_names)]
  938. #[derive(Debug, Clone, Copy, Default, PartialEq, Eq)]
  939. struct GitWorkspaceSummary {
  940. changed_files: usize,
  941. staged_files: usize,
  942. unstaged_files: usize,
  943. untracked_files: usize,
  944. conflicted_files: usize,
  945. }
  946. impl GitWorkspaceSummary {
  947. fn is_clean(self) -> bool {
  948. self.changed_files == 0
  949. }
  950. fn headline(self) -> String {
  951. if self.is_clean() {
  952. "clean".to_string()
  953. } else {
  954. let mut details = Vec::new();
  955. if self.staged_files > 0 {
  956. details.push(format!("{} staged", self.staged_files));
  957. }
  958. if self.unstaged_files > 0 {
  959. details.push(format!("{} unstaged", self.unstaged_files));
  960. }
  961. if self.untracked_files > 0 {
  962. details.push(format!("{} untracked", self.untracked_files));
  963. }
  964. if self.conflicted_files > 0 {
  965. details.push(format!("{} conflicted", self.conflicted_files));
  966. }
  967. format!(
  968. "dirty · {} files · {}",
  969. self.changed_files,
  970. details.join(", ")
  971. )
  972. }
  973. }
  974. }
  975. #[cfg(test)]
  976. fn format_unknown_slash_command_message(name: &str) -> String {
  977. let suggestions = suggest_slash_commands(name);
  978. if suggestions.is_empty() {
  979. format!("unknown slash command: /{name}. Use /help to list available commands.")
  980. } else {
  981. format!(
  982. "unknown slash command: /{name}. Did you mean {}? Use /help to list available commands.",
  983. suggestions.join(", ")
  984. )
  985. }
  986. }
  987. fn format_model_report(model: &str, message_count: usize, turns: u32) -> String {
  988. format!(
  989. "Model
  990. Current model {model}
  991. Session messages {message_count}
  992. Session turns {turns}
  993. Usage
  994. Inspect current model with /model
  995. Switch models with /model <name>"
  996. )
  997. }
  998. fn format_model_switch_report(previous: &str, next: &str, message_count: usize) -> String {
  999. format!(
  1000. "Model updated
  1001. Previous {previous}
  1002. Current {next}
  1003. Preserved msgs {message_count}"
  1004. )
  1005. }
  1006. fn format_permissions_report(mode: &str) -> String {
  1007. let modes = [
  1008. ("read-only", "Read/search tools only", mode == "read-only"),
  1009. (
  1010. "workspace-write",
  1011. "Edit files inside the workspace",
  1012. mode == "workspace-write",
  1013. ),
  1014. (
  1015. "danger-full-access",
  1016. "Unrestricted tool access",
  1017. mode == "danger-full-access",
  1018. ),
  1019. ]
  1020. .into_iter()
  1021. .map(|(name, description, is_current)| {
  1022. let marker = if is_current {
  1023. "● current"
  1024. } else {
  1025. "○ available"
  1026. };
  1027. format!(" {name:<18} {marker:<11} {description}")
  1028. })
  1029. .collect::<Vec<_>>()
  1030. .join(
  1031. "
  1032. ",
  1033. );
  1034. format!(
  1035. "Permissions
  1036. Active mode {mode}
  1037. Mode status live session default
  1038. Modes
  1039. {modes}
  1040. Usage
  1041. Inspect current mode with /permissions
  1042. Switch modes with /permissions <mode>"
  1043. )
  1044. }
  1045. fn format_permissions_switch_report(previous: &str, next: &str) -> String {
  1046. format!(
  1047. "Permissions updated
  1048. Result mode switched
  1049. Previous mode {previous}
  1050. Active mode {next}
  1051. Applies to subsequent tool calls
  1052. Usage /permissions to inspect current mode"
  1053. )
  1054. }
  1055. fn format_cost_report(usage: TokenUsage) -> String {
  1056. format!(
  1057. "Cost
  1058. Input tokens {}
  1059. Output tokens {}
  1060. Cache create {}
  1061. Cache read {}
  1062. Total tokens {}",
  1063. usage.input_tokens,
  1064. usage.output_tokens,
  1065. usage.cache_creation_input_tokens,
  1066. usage.cache_read_input_tokens,
  1067. usage.total_tokens(),
  1068. )
  1069. }
  1070. fn format_resume_report(session_path: &str, message_count: usize, turns: u32) -> String {
  1071. format!(
  1072. "Session resumed
  1073. Session file {session_path}
  1074. Messages {message_count}
  1075. Turns {turns}"
  1076. )
  1077. }
  1078. fn render_resume_usage() -> String {
  1079. format!(
  1080. "Resume
  1081. Usage /resume <session-path|session-id|{LATEST_SESSION_REFERENCE}>
  1082. Auto-save .claw/sessions/<session-id>.{PRIMARY_SESSION_EXTENSION}
  1083. Tip use /session list to inspect saved sessions"
  1084. )
  1085. }
  1086. fn format_compact_report(removed: usize, resulting_messages: usize, skipped: bool) -> String {
  1087. if skipped {
  1088. format!(
  1089. "Compact
  1090. Result skipped
  1091. Reason session below compaction threshold
  1092. Messages kept {resulting_messages}"
  1093. )
  1094. } else {
  1095. format!(
  1096. "Compact
  1097. Result compacted
  1098. Messages removed {removed}
  1099. Messages kept {resulting_messages}"
  1100. )
  1101. }
  1102. }
  1103. fn format_auto_compaction_notice(removed: usize) -> String {
  1104. format!("[auto-compacted: removed {removed} messages]")
  1105. }
  1106. fn parse_git_status_metadata(status: Option<&str>) -> (Option<PathBuf>, Option<String>) {
  1107. parse_git_status_metadata_for(
  1108. &env::current_dir().unwrap_or_else(|_| PathBuf::from(".")),
  1109. status,
  1110. )
  1111. }
  1112. fn parse_git_status_branch(status: Option<&str>) -> Option<String> {
  1113. let status = status?;
  1114. let first_line = status.lines().next()?;
  1115. let line = first_line.strip_prefix("## ")?;
  1116. if line.starts_with("HEAD") {
  1117. return Some("detached HEAD".to_string());
  1118. }
  1119. let branch = line.split(['.', ' ']).next().unwrap_or_default().trim();
  1120. if branch.is_empty() {
  1121. None
  1122. } else {
  1123. Some(branch.to_string())
  1124. }
  1125. }
  1126. fn parse_git_workspace_summary(status: Option<&str>) -> GitWorkspaceSummary {
  1127. let mut summary = GitWorkspaceSummary::default();
  1128. let Some(status) = status else {
  1129. return summary;
  1130. };
  1131. for line in status.lines() {
  1132. if line.starts_with("## ") || line.trim().is_empty() {
  1133. continue;
  1134. }
  1135. summary.changed_files += 1;
  1136. let mut chars = line.chars();
  1137. let index_status = chars.next().unwrap_or(' ');
  1138. let worktree_status = chars.next().unwrap_or(' ');
  1139. if index_status == '?' && worktree_status == '?' {
  1140. summary.untracked_files += 1;
  1141. continue;
  1142. }
  1143. if index_status != ' ' {
  1144. summary.staged_files += 1;
  1145. }
  1146. if worktree_status != ' ' {
  1147. summary.unstaged_files += 1;
  1148. }
  1149. if (matches!(index_status, 'U' | 'A') && matches!(worktree_status, 'U' | 'A'))
  1150. || index_status == 'U'
  1151. || worktree_status == 'U'
  1152. {
  1153. summary.conflicted_files += 1;
  1154. }
  1155. }
  1156. summary
  1157. }
  1158. fn resolve_git_branch_for(cwd: &Path) -> Option<String> {
  1159. let branch = run_git_capture_in(cwd, &["branch", "--show-current"])?;
  1160. let branch = branch.trim();
  1161. if !branch.is_empty() {
  1162. return Some(branch.to_string());
  1163. }
  1164. let fallback = run_git_capture_in(cwd, &["rev-parse", "--abbrev-ref", "HEAD"])?;
  1165. let fallback = fallback.trim();
  1166. if fallback.is_empty() {
  1167. None
  1168. } else if fallback == "HEAD" {
  1169. Some("detached HEAD".to_string())
  1170. } else {
  1171. Some(fallback.to_string())
  1172. }
  1173. }
  1174. fn run_git_capture_in(cwd: &Path, args: &[&str]) -> Option<String> {
  1175. let output = std::process::Command::new("git")
  1176. .args(args)
  1177. .current_dir(cwd)
  1178. .output()
  1179. .ok()?;
  1180. if !output.status.success() {
  1181. return None;
  1182. }
  1183. String::from_utf8(output.stdout).ok()
  1184. }
  1185. fn find_git_root_in(cwd: &Path) -> Result<PathBuf, Box<dyn std::error::Error>> {
  1186. let output = std::process::Command::new("git")
  1187. .args(["rev-parse", "--show-toplevel"])
  1188. .current_dir(cwd)
  1189. .output()?;
  1190. if !output.status.success() {
  1191. return Err("not a git repository".into());
  1192. }
  1193. let path = String::from_utf8(output.stdout)?.trim().to_string();
  1194. if path.is_empty() {
  1195. return Err("empty git root".into());
  1196. }
  1197. Ok(PathBuf::from(path))
  1198. }
  1199. fn parse_git_status_metadata_for(
  1200. cwd: &Path,
  1201. status: Option<&str>,
  1202. ) -> (Option<PathBuf>, Option<String>) {
  1203. let branch = resolve_git_branch_for(cwd).or_else(|| parse_git_status_branch(status));
  1204. let project_root = find_git_root_in(cwd).ok();
  1205. (project_root, branch)
  1206. }
  1207. #[allow(clippy::too_many_lines)]
  1208. fn run_resume_command(
  1209. session_path: &Path,
  1210. session: &Session,
  1211. command: &SlashCommand,
  1212. ) -> Result<ResumeCommandOutcome, Box<dyn std::error::Error>> {
  1213. match command {
  1214. SlashCommand::Help => Ok(ResumeCommandOutcome {
  1215. session: session.clone(),
  1216. message: Some(render_repl_help()),
  1217. }),
  1218. SlashCommand::Compact => {
  1219. let result = runtime::compact_session(
  1220. session,
  1221. CompactionConfig {
  1222. max_estimated_tokens: 0,
  1223. ..CompactionConfig::default()
  1224. },
  1225. );
  1226. let removed = result.removed_message_count;
  1227. let kept = result.compacted_session.messages.len();
  1228. let skipped = removed == 0;
  1229. result.compacted_session.save_to_path(session_path)?;
  1230. Ok(ResumeCommandOutcome {
  1231. session: result.compacted_session,
  1232. message: Some(format_compact_report(removed, kept, skipped)),
  1233. })
  1234. }
  1235. SlashCommand::Clear { confirm } => {
  1236. if !confirm {
  1237. return Ok(ResumeCommandOutcome {
  1238. session: session.clone(),
  1239. message: Some(
  1240. "clear: confirmation required; rerun with /clear --confirm".to_string(),
  1241. ),
  1242. });
  1243. }
  1244. let backup_path = write_session_clear_backup(session, session_path)?;
  1245. let previous_session_id = session.session_id.clone();
  1246. let cleared = Session::new();
  1247. let new_session_id = cleared.session_id.clone();
  1248. cleared.save_to_path(session_path)?;
  1249. Ok(ResumeCommandOutcome {
  1250. session: cleared,
  1251. message: Some(format!(
  1252. "Session cleared\n Mode resumed session reset\n Previous session {previous_session_id}\n Backup {}\n Resume previous claw --resume {}\n New session {new_session_id}\n Session file {}",
  1253. backup_path.display(),
  1254. backup_path.display(),
  1255. session_path.display()
  1256. )),
  1257. })
  1258. }
  1259. SlashCommand::Status => {
  1260. let tracker = UsageTracker::from_session(session);
  1261. let usage = tracker.cumulative_usage();
  1262. Ok(ResumeCommandOutcome {
  1263. session: session.clone(),
  1264. message: Some(format_status_report(
  1265. "restored-session",
  1266. StatusUsage {
  1267. message_count: session.messages.len(),
  1268. turns: tracker.turns(),
  1269. latest: tracker.current_turn_usage(),
  1270. cumulative: usage,
  1271. estimated_tokens: 0,
  1272. },
  1273. default_permission_mode().as_str(),
  1274. &status_context(Some(session_path))?,
  1275. )),
  1276. })
  1277. }
  1278. SlashCommand::Sandbox => {
  1279. let cwd = env::current_dir()?;
  1280. let loader = ConfigLoader::default_for(&cwd);
  1281. let runtime_config = loader.load()?;
  1282. Ok(ResumeCommandOutcome {
  1283. session: session.clone(),
  1284. message: Some(format_sandbox_report(&resolve_sandbox_status(
  1285. runtime_config.sandbox(),
  1286. &cwd,
  1287. ))),
  1288. })
  1289. }
  1290. SlashCommand::Cost => {
  1291. let usage = UsageTracker::from_session(session).cumulative_usage();
  1292. Ok(ResumeCommandOutcome {
  1293. session: session.clone(),
  1294. message: Some(format_cost_report(usage)),
  1295. })
  1296. }
  1297. SlashCommand::Config { section } => Ok(ResumeCommandOutcome {
  1298. session: session.clone(),
  1299. message: Some(render_config_report(section.as_deref())?),
  1300. }),
  1301. SlashCommand::Mcp { action, target } => {
  1302. let cwd = env::current_dir()?;
  1303. let args = match (action.as_deref(), target.as_deref()) {
  1304. (None, None) => None,
  1305. (Some(action), None) => Some(action.to_string()),
  1306. (Some(action), Some(target)) => Some(format!("{action} {target}")),
  1307. (None, Some(target)) => Some(target.to_string()),
  1308. };
  1309. Ok(ResumeCommandOutcome {
  1310. session: session.clone(),
  1311. message: Some(handle_mcp_slash_command(args.as_deref(), &cwd)?),
  1312. })
  1313. }
  1314. SlashCommand::Memory => Ok(ResumeCommandOutcome {
  1315. session: session.clone(),
  1316. message: Some(render_memory_report()?),
  1317. }),
  1318. SlashCommand::Init => Ok(ResumeCommandOutcome {
  1319. session: session.clone(),
  1320. message: Some(init_claude_md()?),
  1321. }),
  1322. SlashCommand::Diff => Ok(ResumeCommandOutcome {
  1323. session: session.clone(),
  1324. message: Some(render_diff_report_for(
  1325. session_path.parent().unwrap_or_else(|| Path::new(".")),
  1326. )?),
  1327. }),
  1328. SlashCommand::Version => Ok(ResumeCommandOutcome {
  1329. session: session.clone(),
  1330. message: Some(render_version_report()),
  1331. }),
  1332. SlashCommand::Export { path } => {
  1333. let export_path = resolve_export_path(path.as_deref(), session)?;
  1334. fs::write(&export_path, render_export_text(session))?;
  1335. Ok(ResumeCommandOutcome {
  1336. session: session.clone(),
  1337. message: Some(format!(
  1338. "Export\n Result wrote transcript\n File {}\n Messages {}",
  1339. export_path.display(),
  1340. session.messages.len(),
  1341. )),
  1342. })
  1343. }
  1344. SlashCommand::Agents { args } => {
  1345. let cwd = env::current_dir()?;
  1346. Ok(ResumeCommandOutcome {
  1347. session: session.clone(),
  1348. message: Some(handle_agents_slash_command(args.as_deref(), &cwd)?),
  1349. })
  1350. }
  1351. SlashCommand::Skills { args } => {
  1352. let cwd = env::current_dir()?;
  1353. Ok(ResumeCommandOutcome {
  1354. session: session.clone(),
  1355. message: Some(handle_skills_slash_command(args.as_deref(), &cwd)?),
  1356. })
  1357. }
  1358. SlashCommand::Unknown(name) => Err(format_unknown_slash_command(name).into()),
  1359. SlashCommand::Bughunter { .. }
  1360. | SlashCommand::Commit { .. }
  1361. | SlashCommand::Pr { .. }
  1362. | SlashCommand::Issue { .. }
  1363. | SlashCommand::Ultraplan { .. }
  1364. | SlashCommand::Teleport { .. }
  1365. | SlashCommand::DebugToolCall { .. }
  1366. | SlashCommand::Resume { .. }
  1367. | SlashCommand::Model { .. }
  1368. | SlashCommand::Permissions { .. }
  1369. | SlashCommand::Session { .. }
  1370. | SlashCommand::Plugins { .. }
  1371. | SlashCommand::Doctor
  1372. | SlashCommand::Login
  1373. | SlashCommand::Logout
  1374. | SlashCommand::Vim
  1375. | SlashCommand::Upgrade
  1376. | SlashCommand::Stats
  1377. | SlashCommand::Share
  1378. | SlashCommand::Feedback
  1379. | SlashCommand::Files
  1380. | SlashCommand::Fast
  1381. | SlashCommand::Exit
  1382. | SlashCommand::Summary
  1383. | SlashCommand::Desktop
  1384. | SlashCommand::Brief
  1385. | SlashCommand::Advisor
  1386. | SlashCommand::Stickers
  1387. | SlashCommand::Insights
  1388. | SlashCommand::Thinkback
  1389. | SlashCommand::ReleaseNotes
  1390. | SlashCommand::SecurityReview
  1391. | SlashCommand::Keybindings
  1392. | SlashCommand::PrivacySettings
  1393. | SlashCommand::Plan { .. }
  1394. | SlashCommand::Review { .. }
  1395. | SlashCommand::Tasks { .. }
  1396. | SlashCommand::Theme { .. }
  1397. | SlashCommand::Voice { .. }
  1398. | SlashCommand::Usage { .. }
  1399. | SlashCommand::Rename { .. }
  1400. | SlashCommand::Copy { .. }
  1401. | SlashCommand::Hooks { .. }
  1402. | SlashCommand::Context { .. }
  1403. | SlashCommand::Color { .. }
  1404. | SlashCommand::Effort { .. }
  1405. | SlashCommand::Branch { .. }
  1406. | SlashCommand::Rewind { .. }
  1407. | SlashCommand::Ide { .. }
  1408. | SlashCommand::Tag { .. }
  1409. | SlashCommand::OutputStyle { .. }
  1410. | SlashCommand::AddDir { .. } => Err("unsupported resumed slash command".into()),
  1411. }
  1412. }
  1413. fn run_repl(
  1414. model: String,
  1415. allowed_tools: Option<AllowedToolSet>,
  1416. permission_mode: PermissionMode,
  1417. ) -> Result<(), Box<dyn std::error::Error>> {
  1418. let mut cli = LiveCli::new(model, true, allowed_tools, permission_mode)?;
  1419. let mut editor =
  1420. input::LineEditor::new("> ", cli.repl_completion_candidates().unwrap_or_default());
  1421. println!("{}", cli.startup_banner());
  1422. loop {
  1423. editor.set_completions(cli.repl_completion_candidates().unwrap_or_default());
  1424. match editor.read_line()? {
  1425. input::ReadOutcome::Submit(input) => {
  1426. let trimmed = input.trim().to_string();
  1427. if trimmed.is_empty() {
  1428. continue;
  1429. }
  1430. if matches!(trimmed.as_str(), "/exit" | "/quit") {
  1431. cli.persist_session()?;
  1432. break;
  1433. }
  1434. match SlashCommand::parse(&trimmed) {
  1435. Ok(Some(command)) => {
  1436. if cli.handle_repl_command(command)? {
  1437. cli.persist_session()?;
  1438. }
  1439. continue;
  1440. }
  1441. Ok(None) => {}
  1442. Err(error) => {
  1443. eprintln!("{error}");
  1444. continue;
  1445. }
  1446. }
  1447. editor.push_history(input);
  1448. cli.run_turn(&trimmed)?;
  1449. }
  1450. input::ReadOutcome::Cancel => {}
  1451. input::ReadOutcome::Exit => {
  1452. cli.persist_session()?;
  1453. break;
  1454. }
  1455. }
  1456. }
  1457. Ok(())
  1458. }
  1459. #[derive(Debug, Clone)]
  1460. struct SessionHandle {
  1461. id: String,
  1462. path: PathBuf,
  1463. }
  1464. #[derive(Debug, Clone)]
  1465. struct ManagedSessionSummary {
  1466. id: String,
  1467. path: PathBuf,
  1468. modified_epoch_millis: u128,
  1469. message_count: usize,
  1470. parent_session_id: Option<String>,
  1471. branch_name: Option<String>,
  1472. }
  1473. struct LiveCli {
  1474. model: String,
  1475. allowed_tools: Option<AllowedToolSet>,
  1476. permission_mode: PermissionMode,
  1477. system_prompt: Vec<String>,
  1478. runtime: BuiltRuntime,
  1479. session: SessionHandle,
  1480. }
  1481. struct RuntimePluginState {
  1482. feature_config: runtime::RuntimeFeatureConfig,
  1483. tool_registry: GlobalToolRegistry,
  1484. plugin_registry: PluginRegistry,
  1485. mcp_state: Option<Arc<Mutex<RuntimeMcpState>>>,
  1486. }
  1487. struct RuntimeMcpState {
  1488. runtime: tokio::runtime::Runtime,
  1489. manager: McpServerManager,
  1490. pending_servers: Vec<String>,
  1491. degraded_report: Option<runtime::McpDegradedReport>,
  1492. }
  1493. struct BuiltRuntime {
  1494. runtime: Option<ConversationRuntime<AnthropicRuntimeClient, CliToolExecutor>>,
  1495. plugin_registry: PluginRegistry,
  1496. plugins_active: bool,
  1497. mcp_state: Option<Arc<Mutex<RuntimeMcpState>>>,
  1498. mcp_active: bool,
  1499. }
  1500. impl BuiltRuntime {
  1501. fn new(
  1502. runtime: ConversationRuntime<AnthropicRuntimeClient, CliToolExecutor>,
  1503. plugin_registry: PluginRegistry,
  1504. mcp_state: Option<Arc<Mutex<RuntimeMcpState>>>,
  1505. ) -> Self {
  1506. Self {
  1507. runtime: Some(runtime),
  1508. plugin_registry,
  1509. plugins_active: true,
  1510. mcp_state,
  1511. mcp_active: true,
  1512. }
  1513. }
  1514. fn with_hook_abort_signal(mut self, hook_abort_signal: runtime::HookAbortSignal) -> Self {
  1515. let runtime = self
  1516. .runtime
  1517. .take()
  1518. .expect("runtime should exist before installing hook abort signal");
  1519. self.runtime = Some(runtime.with_hook_abort_signal(hook_abort_signal));
  1520. self
  1521. }
  1522. fn shutdown_plugins(&mut self) -> Result<(), Box<dyn std::error::Error>> {
  1523. if self.plugins_active {
  1524. self.plugin_registry.shutdown()?;
  1525. self.plugins_active = false;
  1526. }
  1527. Ok(())
  1528. }
  1529. fn shutdown_mcp(&mut self) -> Result<(), Box<dyn std::error::Error>> {
  1530. if self.mcp_active {
  1531. if let Some(mcp_state) = &self.mcp_state {
  1532. mcp_state
  1533. .lock()
  1534. .unwrap_or_else(std::sync::PoisonError::into_inner)
  1535. .shutdown()?;
  1536. }
  1537. self.mcp_active = false;
  1538. }
  1539. Ok(())
  1540. }
  1541. }
  1542. impl Deref for BuiltRuntime {
  1543. type Target = ConversationRuntime<AnthropicRuntimeClient, CliToolExecutor>;
  1544. fn deref(&self) -> &Self::Target {
  1545. self.runtime
  1546. .as_ref()
  1547. .expect("runtime should exist while built runtime is alive")
  1548. }
  1549. }
  1550. impl DerefMut for BuiltRuntime {
  1551. fn deref_mut(&mut self) -> &mut Self::Target {
  1552. self.runtime
  1553. .as_mut()
  1554. .expect("runtime should exist while built runtime is alive")
  1555. }
  1556. }
  1557. impl Drop for BuiltRuntime {
  1558. fn drop(&mut self) {
  1559. let _ = self.shutdown_mcp();
  1560. let _ = self.shutdown_plugins();
  1561. }
  1562. }
  1563. #[derive(Debug, Deserialize)]
  1564. struct ToolSearchRequest {
  1565. query: String,
  1566. max_results: Option<usize>,
  1567. }
  1568. #[derive(Debug, Deserialize)]
  1569. struct McpToolRequest {
  1570. #[serde(rename = "qualifiedName")]
  1571. qualified_name: Option<String>,
  1572. tool: Option<String>,
  1573. arguments: Option<serde_json::Value>,
  1574. }
  1575. #[derive(Debug, Deserialize)]
  1576. struct ListMcpResourcesRequest {
  1577. server: Option<String>,
  1578. }
  1579. #[derive(Debug, Deserialize)]
  1580. struct ReadMcpResourceRequest {
  1581. server: String,
  1582. uri: String,
  1583. }
  1584. impl RuntimeMcpState {
  1585. fn new(
  1586. runtime_config: &runtime::RuntimeConfig,
  1587. ) -> Result<Option<(Self, runtime::McpToolDiscoveryReport)>, Box<dyn std::error::Error>> {
  1588. let mut manager = McpServerManager::from_runtime_config(runtime_config);
  1589. if manager.server_names().is_empty() && manager.unsupported_servers().is_empty() {
  1590. return Ok(None);
  1591. }
  1592. let runtime = tokio::runtime::Runtime::new()?;
  1593. let discovery = runtime.block_on(manager.discover_tools_best_effort());
  1594. let pending_servers = discovery
  1595. .failed_servers
  1596. .iter()
  1597. .map(|failure| failure.server_name.clone())
  1598. .chain(
  1599. discovery
  1600. .unsupported_servers
  1601. .iter()
  1602. .map(|server| server.server_name.clone()),
  1603. )
  1604. .collect::<BTreeSet<_>>()
  1605. .into_iter()
  1606. .collect::<Vec<_>>();
  1607. let available_tools = discovery
  1608. .tools
  1609. .iter()
  1610. .map(|tool| tool.qualified_name.clone())
  1611. .collect::<Vec<_>>();
  1612. let failed_server_names = pending_servers.iter().cloned().collect::<BTreeSet<_>>();
  1613. let working_servers = manager
  1614. .server_names()
  1615. .into_iter()
  1616. .filter(|server_name| !failed_server_names.contains(server_name))
  1617. .collect::<Vec<_>>();
  1618. let failed_servers = discovery
  1619. .failed_servers
  1620. .iter()
  1621. .map(|failure| runtime::McpFailedServer {
  1622. server_name: failure.server_name.clone(),
  1623. phase: runtime::McpLifecyclePhase::ToolDiscovery,
  1624. error: runtime::McpErrorSurface::new(
  1625. runtime::McpLifecyclePhase::ToolDiscovery,
  1626. Some(failure.server_name.clone()),
  1627. failure.error.clone(),
  1628. std::collections::BTreeMap::new(),
  1629. true,
  1630. ),
  1631. })
  1632. .chain(discovery.unsupported_servers.iter().map(|server| {
  1633. runtime::McpFailedServer {
  1634. server_name: server.server_name.clone(),
  1635. phase: runtime::McpLifecyclePhase::ServerRegistration,
  1636. error: runtime::McpErrorSurface::new(
  1637. runtime::McpLifecyclePhase::ServerRegistration,
  1638. Some(server.server_name.clone()),
  1639. server.reason.clone(),
  1640. std::collections::BTreeMap::from([(
  1641. "transport".to_string(),
  1642. format!("{:?}", server.transport).to_ascii_lowercase(),
  1643. )]),
  1644. false,
  1645. ),
  1646. }
  1647. }))
  1648. .collect::<Vec<_>>();
  1649. let degraded_report = (!failed_servers.is_empty()).then(|| {
  1650. runtime::McpDegradedReport::new(
  1651. working_servers,
  1652. failed_servers,
  1653. available_tools.clone(),
  1654. available_tools,
  1655. )
  1656. });
  1657. Ok(Some((
  1658. Self {
  1659. runtime,
  1660. manager,
  1661. pending_servers,
  1662. degraded_report,
  1663. },
  1664. discovery,
  1665. )))
  1666. }
  1667. fn shutdown(&mut self) -> Result<(), Box<dyn std::error::Error>> {
  1668. self.runtime.block_on(self.manager.shutdown())?;
  1669. Ok(())
  1670. }
  1671. fn pending_servers(&self) -> Option<Vec<String>> {
  1672. (!self.pending_servers.is_empty()).then(|| self.pending_servers.clone())
  1673. }
  1674. fn degraded_report(&self) -> Option<runtime::McpDegradedReport> {
  1675. self.degraded_report.clone()
  1676. }
  1677. fn server_names(&self) -> Vec<String> {
  1678. self.manager.server_names()
  1679. }
  1680. fn call_tool(
  1681. &mut self,
  1682. qualified_tool_name: &str,
  1683. arguments: Option<serde_json::Value>,
  1684. ) -> Result<String, ToolError> {
  1685. let response = self
  1686. .runtime
  1687. .block_on(self.manager.call_tool(qualified_tool_name, arguments))
  1688. .map_err(|error| ToolError::new(error.to_string()))?;
  1689. if let Some(error) = response.error {
  1690. return Err(ToolError::new(format!(
  1691. "MCP tool `{qualified_tool_name}` returned JSON-RPC error: {} ({})",
  1692. error.message, error.code
  1693. )));
  1694. }
  1695. let result = response.result.ok_or_else(|| {
  1696. ToolError::new(format!(
  1697. "MCP tool `{qualified_tool_name}` returned no result payload"
  1698. ))
  1699. })?;
  1700. serde_json::to_string_pretty(&result).map_err(|error| ToolError::new(error.to_string()))
  1701. }
  1702. fn list_resources_for_server(&mut self, server_name: &str) -> Result<String, ToolError> {
  1703. let result = self
  1704. .runtime
  1705. .block_on(self.manager.list_resources(server_name))
  1706. .map_err(|error| ToolError::new(error.to_string()))?;
  1707. serde_json::to_string_pretty(&json!({
  1708. "server": server_name,
  1709. "resources": result.resources,
  1710. }))
  1711. .map_err(|error| ToolError::new(error.to_string()))
  1712. }
  1713. fn list_resources_for_all_servers(&mut self) -> Result<String, ToolError> {
  1714. let mut resources = Vec::new();
  1715. let mut failures = Vec::new();
  1716. for server_name in self.server_names() {
  1717. match self
  1718. .runtime
  1719. .block_on(self.manager.list_resources(&server_name))
  1720. {
  1721. Ok(result) => resources.push(json!({
  1722. "server": server_name,
  1723. "resources": result.resources,
  1724. })),
  1725. Err(error) => failures.push(json!({
  1726. "server": server_name,
  1727. "error": error.to_string(),
  1728. })),
  1729. }
  1730. }
  1731. if resources.is_empty() && !failures.is_empty() {
  1732. let message = failures
  1733. .iter()
  1734. .filter_map(|failure| failure.get("error").and_then(serde_json::Value::as_str))
  1735. .collect::<Vec<_>>()
  1736. .join("; ");
  1737. return Err(ToolError::new(message));
  1738. }
  1739. serde_json::to_string_pretty(&json!({
  1740. "resources": resources,
  1741. "failures": failures,
  1742. }))
  1743. .map_err(|error| ToolError::new(error.to_string()))
  1744. }
  1745. fn read_resource(&mut self, server_name: &str, uri: &str) -> Result<String, ToolError> {
  1746. let result = self
  1747. .runtime
  1748. .block_on(self.manager.read_resource(server_name, uri))
  1749. .map_err(|error| ToolError::new(error.to_string()))?;
  1750. serde_json::to_string_pretty(&json!({
  1751. "server": server_name,
  1752. "contents": result.contents,
  1753. }))
  1754. .map_err(|error| ToolError::new(error.to_string()))
  1755. }
  1756. }
  1757. fn build_runtime_mcp_state(
  1758. runtime_config: &runtime::RuntimeConfig,
  1759. ) -> Result<
  1760. (
  1761. Option<Arc<Mutex<RuntimeMcpState>>>,
  1762. Vec<RuntimeToolDefinition>,
  1763. ),
  1764. Box<dyn std::error::Error>,
  1765. > {
  1766. let Some((mcp_state, discovery)) = RuntimeMcpState::new(runtime_config)? else {
  1767. return Ok((None, Vec::new()));
  1768. };
  1769. let mut runtime_tools = discovery
  1770. .tools
  1771. .iter()
  1772. .map(mcp_runtime_tool_definition)
  1773. .collect::<Vec<_>>();
  1774. if !mcp_state.server_names().is_empty() {
  1775. runtime_tools.extend(mcp_wrapper_tool_definitions());
  1776. }
  1777. Ok((Some(Arc::new(Mutex::new(mcp_state))), runtime_tools))
  1778. }
  1779. fn mcp_runtime_tool_definition(tool: &runtime::ManagedMcpTool) -> RuntimeToolDefinition {
  1780. RuntimeToolDefinition {
  1781. name: tool.qualified_name.clone(),
  1782. description: Some(
  1783. tool.tool
  1784. .description
  1785. .clone()
  1786. .unwrap_or_else(|| format!("Invoke MCP tool `{}`.", tool.qualified_name)),
  1787. ),
  1788. input_schema: tool
  1789. .tool
  1790. .input_schema
  1791. .clone()
  1792. .unwrap_or_else(|| json!({ "type": "object", "additionalProperties": true })),
  1793. required_permission: permission_mode_for_mcp_tool(&tool.tool),
  1794. }
  1795. }
  1796. fn mcp_wrapper_tool_definitions() -> Vec<RuntimeToolDefinition> {
  1797. vec![
  1798. RuntimeToolDefinition {
  1799. name: "MCPTool".to_string(),
  1800. description: Some(
  1801. "Call a configured MCP tool by its qualified name and JSON arguments.".to_string(),
  1802. ),
  1803. input_schema: json!({
  1804. "type": "object",
  1805. "properties": {
  1806. "qualifiedName": { "type": "string" },
  1807. "arguments": {}
  1808. },
  1809. "required": ["qualifiedName"],
  1810. "additionalProperties": false
  1811. }),
  1812. required_permission: PermissionMode::DangerFullAccess,
  1813. },
  1814. RuntimeToolDefinition {
  1815. name: "ListMcpResourcesTool".to_string(),
  1816. description: Some(
  1817. "List MCP resources from one configured server or from every connected server."
  1818. .to_string(),
  1819. ),
  1820. input_schema: json!({
  1821. "type": "object",
  1822. "properties": {
  1823. "server": { "type": "string" }
  1824. },
  1825. "additionalProperties": false
  1826. }),
  1827. required_permission: PermissionMode::ReadOnly,
  1828. },
  1829. RuntimeToolDefinition {
  1830. name: "ReadMcpResourceTool".to_string(),
  1831. description: Some("Read a specific MCP resource from a configured server.".to_string()),
  1832. input_schema: json!({
  1833. "type": "object",
  1834. "properties": {
  1835. "server": { "type": "string" },
  1836. "uri": { "type": "string" }
  1837. },
  1838. "required": ["server", "uri"],
  1839. "additionalProperties": false
  1840. }),
  1841. required_permission: PermissionMode::ReadOnly,
  1842. },
  1843. ]
  1844. }
  1845. fn permission_mode_for_mcp_tool(tool: &McpTool) -> PermissionMode {
  1846. let read_only = mcp_annotation_flag(tool, "readOnlyHint");
  1847. let destructive = mcp_annotation_flag(tool, "destructiveHint");
  1848. let open_world = mcp_annotation_flag(tool, "openWorldHint");
  1849. if read_only && !destructive && !open_world {
  1850. PermissionMode::ReadOnly
  1851. } else if destructive || open_world {
  1852. PermissionMode::DangerFullAccess
  1853. } else {
  1854. PermissionMode::WorkspaceWrite
  1855. }
  1856. }
  1857. fn mcp_annotation_flag(tool: &McpTool, key: &str) -> bool {
  1858. tool.annotations
  1859. .as_ref()
  1860. .and_then(|annotations| annotations.get(key))
  1861. .and_then(serde_json::Value::as_bool)
  1862. .unwrap_or(false)
  1863. }
  1864. struct HookAbortMonitor {
  1865. stop_tx: Option<Sender<()>>,
  1866. join_handle: Option<JoinHandle<()>>,
  1867. }
  1868. impl HookAbortMonitor {
  1869. fn spawn(abort_signal: runtime::HookAbortSignal) -> Self {
  1870. Self::spawn_with_waiter(abort_signal, move |stop_rx, abort_signal| {
  1871. let Ok(runtime) = tokio::runtime::Builder::new_current_thread()
  1872. .enable_all()
  1873. .build()
  1874. else {
  1875. return;
  1876. };
  1877. runtime.block_on(async move {
  1878. let wait_for_stop = tokio::task::spawn_blocking(move || {
  1879. let _ = stop_rx.recv();
  1880. });
  1881. tokio::select! {
  1882. result = tokio::signal::ctrl_c() => {
  1883. if result.is_ok() {
  1884. abort_signal.abort();
  1885. }
  1886. }
  1887. _ = wait_for_stop => {}
  1888. }
  1889. });
  1890. })
  1891. }
  1892. fn spawn_with_waiter<F>(abort_signal: runtime::HookAbortSignal, wait_for_interrupt: F) -> Self
  1893. where
  1894. F: FnOnce(Receiver<()>, runtime::HookAbortSignal) + Send + 'static,
  1895. {
  1896. let (stop_tx, stop_rx) = mpsc::channel();
  1897. let join_handle = thread::spawn(move || wait_for_interrupt(stop_rx, abort_signal));
  1898. Self {
  1899. stop_tx: Some(stop_tx),
  1900. join_handle: Some(join_handle),
  1901. }
  1902. }
  1903. fn stop(mut self) {
  1904. if let Some(stop_tx) = self.stop_tx.take() {
  1905. let _ = stop_tx.send(());
  1906. }
  1907. if let Some(join_handle) = self.join_handle.take() {
  1908. let _ = join_handle.join();
  1909. }
  1910. }
  1911. }
  1912. impl LiveCli {
  1913. fn new(
  1914. model: String,
  1915. enable_tools: bool,
  1916. allowed_tools: Option<AllowedToolSet>,
  1917. permission_mode: PermissionMode,
  1918. ) -> Result<Self, Box<dyn std::error::Error>> {
  1919. let system_prompt = build_system_prompt()?;
  1920. let session_state = Session::new();
  1921. let session = create_managed_session_handle(&session_state.session_id)?;
  1922. let runtime = build_runtime(
  1923. session_state.with_persistence_path(session.path.clone()),
  1924. &session.id,
  1925. model.clone(),
  1926. system_prompt.clone(),
  1927. enable_tools,
  1928. true,
  1929. allowed_tools.clone(),
  1930. permission_mode,
  1931. None,
  1932. )?;
  1933. let cli = Self {
  1934. model,
  1935. allowed_tools,
  1936. permission_mode,
  1937. system_prompt,
  1938. runtime,
  1939. session,
  1940. };
  1941. cli.persist_session()?;
  1942. Ok(cli)
  1943. }
  1944. fn startup_banner(&self) -> String {
  1945. let cwd = env::current_dir().map_or_else(
  1946. |_| "<unknown>".to_string(),
  1947. |path| path.display().to_string(),
  1948. );
  1949. let status = status_context(None).ok();
  1950. let git_branch = status
  1951. .as_ref()
  1952. .and_then(|context| context.git_branch.as_deref())
  1953. .unwrap_or("unknown");
  1954. let workspace = status.as_ref().map_or_else(
  1955. || "unknown".to_string(),
  1956. |context| context.git_summary.headline(),
  1957. );
  1958. let session_path = self.session.path.strip_prefix(Path::new(&cwd)).map_or_else(
  1959. |_| self.session.path.display().to_string(),
  1960. |path| path.display().to_string(),
  1961. );
  1962. format!(
  1963. "\x1b[38;5;196m\
  1964. ██████╗██╗ █████╗ ██╗ ██╗\n\
  1965. ██╔════╝██║ ██╔══██╗██║ ██║\n\
  1966. ██║ ██║ ███████║██║ █╗ ██║\n\
  1967. ██║ ██║ ██╔══██║██║███╗██║\n\
  1968. ╚██████╗███████╗██║ ██║╚███╔███╔╝\n\
  1969. ╚═════╝╚══════╝╚═╝ ╚═╝ ╚══╝╚══╝\x1b[0m \x1b[38;5;208mCode\x1b[0m 🦞\n\n\
  1970. \x1b[2mModel\x1b[0m {}\n\
  1971. \x1b[2mPermissions\x1b[0m {}\n\
  1972. \x1b[2mBranch\x1b[0m {}\n\
  1973. \x1b[2mWorkspace\x1b[0m {}\n\
  1974. \x1b[2mDirectory\x1b[0m {}\n\
  1975. \x1b[2mSession\x1b[0m {}\n\
  1976. \x1b[2mAuto-save\x1b[0m {}\n\n\
  1977. Type \x1b[1m/help\x1b[0m for commands · \x1b[1m/status\x1b[0m for live context · \x1b[2m/resume latest\x1b[0m jumps back to the newest session · \x1b[1m/diff\x1b[0m then \x1b[1m/commit\x1b[0m to ship · \x1b[2mTab\x1b[0m for workflow completions · \x1b[2mShift+Enter\x1b[0m for newline",
  1978. self.model,
  1979. self.permission_mode.as_str(),
  1980. git_branch,
  1981. workspace,
  1982. cwd,
  1983. self.session.id,
  1984. session_path,
  1985. )
  1986. }
  1987. fn repl_completion_candidates(&self) -> Result<Vec<String>, Box<dyn std::error::Error>> {
  1988. Ok(slash_command_completion_candidates_with_sessions(
  1989. &self.model,
  1990. Some(&self.session.id),
  1991. list_managed_sessions()?
  1992. .into_iter()
  1993. .map(|session| session.id)
  1994. .collect(),
  1995. ))
  1996. }
  1997. fn prepare_turn_runtime(
  1998. &self,
  1999. emit_output: bool,
  2000. ) -> Result<(BuiltRuntime, HookAbortMonitor), Box<dyn std::error::Error>> {
  2001. let hook_abort_signal = runtime::HookAbortSignal::new();
  2002. let runtime = build_runtime(
  2003. self.runtime.session().clone(),
  2004. &self.session.id,
  2005. self.model.clone(),
  2006. self.system_prompt.clone(),
  2007. true,
  2008. emit_output,
  2009. self.allowed_tools.clone(),
  2010. self.permission_mode,
  2011. None,
  2012. )?
  2013. .with_hook_abort_signal(hook_abort_signal.clone());
  2014. let hook_abort_monitor = HookAbortMonitor::spawn(hook_abort_signal);
  2015. Ok((runtime, hook_abort_monitor))
  2016. }
  2017. fn replace_runtime(&mut self, runtime: BuiltRuntime) -> Result<(), Box<dyn std::error::Error>> {
  2018. self.runtime.shutdown_plugins()?;
  2019. self.runtime = runtime;
  2020. Ok(())
  2021. }
  2022. fn run_turn(&mut self, input: &str) -> Result<(), Box<dyn std::error::Error>> {
  2023. let (mut runtime, hook_abort_monitor) = self.prepare_turn_runtime(true)?;
  2024. let mut spinner = Spinner::new();
  2025. let mut stdout = io::stdout();
  2026. spinner.tick(
  2027. "🦀 Thinking...",
  2028. TerminalRenderer::new().color_theme(),
  2029. &mut stdout,
  2030. )?;
  2031. let mut permission_prompter = CliPermissionPrompter::new(self.permission_mode);
  2032. let result = runtime.run_turn(input, Some(&mut permission_prompter));
  2033. hook_abort_monitor.stop();
  2034. match result {
  2035. Ok(summary) => {
  2036. self.replace_runtime(runtime)?;
  2037. spinner.finish(
  2038. "✨ Done",
  2039. TerminalRenderer::new().color_theme(),
  2040. &mut stdout,
  2041. )?;
  2042. println!();
  2043. if let Some(event) = summary.auto_compaction {
  2044. println!(
  2045. "{}",
  2046. format_auto_compaction_notice(event.removed_message_count)
  2047. );
  2048. }
  2049. self.persist_session()?;
  2050. Ok(())
  2051. }
  2052. Err(error) => {
  2053. runtime.shutdown_plugins()?;
  2054. spinner.fail(
  2055. "❌ Request failed",
  2056. TerminalRenderer::new().color_theme(),
  2057. &mut stdout,
  2058. )?;
  2059. Err(Box::new(error))
  2060. }
  2061. }
  2062. }
  2063. fn run_turn_with_output(
  2064. &mut self,
  2065. input: &str,
  2066. output_format: CliOutputFormat,
  2067. ) -> Result<(), Box<dyn std::error::Error>> {
  2068. match output_format {
  2069. CliOutputFormat::Text => self.run_turn(input),
  2070. CliOutputFormat::Json => self.run_prompt_json(input),
  2071. }
  2072. }
  2073. fn run_prompt_json(&mut self, input: &str) -> Result<(), Box<dyn std::error::Error>> {
  2074. let (mut runtime, hook_abort_monitor) = self.prepare_turn_runtime(false)?;
  2075. let mut permission_prompter = CliPermissionPrompter::new(self.permission_mode);
  2076. let result = runtime.run_turn(input, Some(&mut permission_prompter));
  2077. hook_abort_monitor.stop();
  2078. let summary = result?;
  2079. self.replace_runtime(runtime)?;
  2080. self.persist_session()?;
  2081. println!(
  2082. "{}",
  2083. json!({
  2084. "message": final_assistant_text(&summary),
  2085. "model": self.model,
  2086. "iterations": summary.iterations,
  2087. "auto_compaction": summary.auto_compaction.map(|event| json!({
  2088. "removed_messages": event.removed_message_count,
  2089. "notice": format_auto_compaction_notice(event.removed_message_count),
  2090. })),
  2091. "tool_uses": collect_tool_uses(&summary),
  2092. "tool_results": collect_tool_results(&summary),
  2093. "prompt_cache_events": collect_prompt_cache_events(&summary),
  2094. "usage": {
  2095. "input_tokens": summary.usage.input_tokens,
  2096. "output_tokens": summary.usage.output_tokens,
  2097. "cache_creation_input_tokens": summary.usage.cache_creation_input_tokens,
  2098. "cache_read_input_tokens": summary.usage.cache_read_input_tokens,
  2099. },
  2100. "estimated_cost": format_usd(
  2101. summary.usage.estimate_cost_usd_with_pricing(
  2102. pricing_for_model(&self.model)
  2103. .unwrap_or_else(runtime::ModelPricing::default_sonnet_tier)
  2104. ).total_cost_usd()
  2105. )
  2106. })
  2107. );
  2108. Ok(())
  2109. }
  2110. #[allow(clippy::too_many_lines)]
  2111. fn handle_repl_command(
  2112. &mut self,
  2113. command: SlashCommand,
  2114. ) -> Result<bool, Box<dyn std::error::Error>> {
  2115. Ok(match command {
  2116. SlashCommand::Help => {
  2117. println!("{}", render_repl_help());
  2118. false
  2119. }
  2120. SlashCommand::Status => {
  2121. self.print_status();
  2122. false
  2123. }
  2124. SlashCommand::Bughunter { scope } => {
  2125. self.run_bughunter(scope.as_deref())?;
  2126. false
  2127. }
  2128. SlashCommand::Commit => {
  2129. self.run_commit(None)?;
  2130. false
  2131. }
  2132. SlashCommand::Pr { context } => {
  2133. self.run_pr(context.as_deref())?;
  2134. false
  2135. }
  2136. SlashCommand::Issue { context } => {
  2137. self.run_issue(context.as_deref())?;
  2138. false
  2139. }
  2140. SlashCommand::Ultraplan { task } => {
  2141. self.run_ultraplan(task.as_deref())?;
  2142. false
  2143. }
  2144. SlashCommand::Teleport { target } => {
  2145. Self::run_teleport(target.as_deref())?;
  2146. false
  2147. }
  2148. SlashCommand::DebugToolCall => {
  2149. self.run_debug_tool_call(None)?;
  2150. false
  2151. }
  2152. SlashCommand::Sandbox => {
  2153. Self::print_sandbox_status();
  2154. false
  2155. }
  2156. SlashCommand::Compact => {
  2157. self.compact()?;
  2158. false
  2159. }
  2160. SlashCommand::Model { model } => self.set_model(model)?,
  2161. SlashCommand::Permissions { mode } => self.set_permissions(mode)?,
  2162. SlashCommand::Clear { confirm } => self.clear_session(confirm)?,
  2163. SlashCommand::Cost => {
  2164. self.print_cost();
  2165. false
  2166. }
  2167. SlashCommand::Resume { session_path } => self.resume_session(session_path)?,
  2168. SlashCommand::Config { section } => {
  2169. Self::print_config(section.as_deref())?;
  2170. false
  2171. }
  2172. SlashCommand::Mcp { action, target } => {
  2173. let args = match (action.as_deref(), target.as_deref()) {
  2174. (None, None) => None,
  2175. (Some(action), None) => Some(action.to_string()),
  2176. (Some(action), Some(target)) => Some(format!("{action} {target}")),
  2177. (None, Some(target)) => Some(target.to_string()),
  2178. };
  2179. Self::print_mcp(args.as_deref())?;
  2180. false
  2181. }
  2182. SlashCommand::Memory => {
  2183. Self::print_memory()?;
  2184. false
  2185. }
  2186. SlashCommand::Init => {
  2187. run_init()?;
  2188. false
  2189. }
  2190. SlashCommand::Diff => {
  2191. Self::print_diff()?;
  2192. false
  2193. }
  2194. SlashCommand::Version => {
  2195. Self::print_version();
  2196. false
  2197. }
  2198. SlashCommand::Export { path } => {
  2199. self.export_session(path.as_deref())?;
  2200. false
  2201. }
  2202. SlashCommand::Session { action, target } => {
  2203. self.handle_session_command(action.as_deref(), target.as_deref())?
  2204. }
  2205. SlashCommand::Plugins { action, target } => {
  2206. self.handle_plugins_command(action.as_deref(), target.as_deref())?
  2207. }
  2208. SlashCommand::Agents { args } => {
  2209. Self::print_agents(args.as_deref())?;
  2210. false
  2211. }
  2212. SlashCommand::Skills { args } => {
  2213. Self::print_skills(args.as_deref())?;
  2214. false
  2215. }
  2216. SlashCommand::Doctor
  2217. | SlashCommand::Login
  2218. | SlashCommand::Logout
  2219. | SlashCommand::Vim
  2220. | SlashCommand::Upgrade
  2221. | SlashCommand::Stats
  2222. | SlashCommand::Share
  2223. | SlashCommand::Feedback
  2224. | SlashCommand::Files
  2225. | SlashCommand::Fast
  2226. | SlashCommand::Exit
  2227. | SlashCommand::Summary
  2228. | SlashCommand::Desktop
  2229. | SlashCommand::Brief
  2230. | SlashCommand::Advisor
  2231. | SlashCommand::Stickers
  2232. | SlashCommand::Insights
  2233. | SlashCommand::Thinkback
  2234. | SlashCommand::ReleaseNotes
  2235. | SlashCommand::SecurityReview
  2236. | SlashCommand::Keybindings
  2237. | SlashCommand::PrivacySettings
  2238. | SlashCommand::Plan { .. }
  2239. | SlashCommand::Review { .. }
  2240. | SlashCommand::Tasks { .. }
  2241. | SlashCommand::Theme { .. }
  2242. | SlashCommand::Voice { .. }
  2243. | SlashCommand::Usage { .. }
  2244. | SlashCommand::Rename { .. }
  2245. | SlashCommand::Copy { .. }
  2246. | SlashCommand::Hooks { .. }
  2247. | SlashCommand::Context { .. }
  2248. | SlashCommand::Color { .. }
  2249. | SlashCommand::Effort { .. }
  2250. | SlashCommand::Branch { .. }
  2251. | SlashCommand::Rewind { .. }
  2252. | SlashCommand::Ide { .. }
  2253. | SlashCommand::Tag { .. }
  2254. | SlashCommand::OutputStyle { .. }
  2255. | SlashCommand::AddDir { .. } => {
  2256. eprintln!("Command registered but not yet implemented.");
  2257. false
  2258. }
  2259. SlashCommand::Unknown(name) => {
  2260. eprintln!("{}", format_unknown_slash_command(&name));
  2261. false
  2262. }
  2263. })
  2264. }
  2265. fn persist_session(&self) -> Result<(), Box<dyn std::error::Error>> {
  2266. self.runtime.session().save_to_path(&self.session.path)?;
  2267. Ok(())
  2268. }
  2269. fn print_status(&self) {
  2270. let cumulative = self.runtime.usage().cumulative_usage();
  2271. let latest = self.runtime.usage().current_turn_usage();
  2272. println!(
  2273. "{}",
  2274. format_status_report(
  2275. &self.model,
  2276. StatusUsage {
  2277. message_count: self.runtime.session().messages.len(),
  2278. turns: self.runtime.usage().turns(),
  2279. latest,
  2280. cumulative,
  2281. estimated_tokens: self.runtime.estimated_tokens(),
  2282. },
  2283. self.permission_mode.as_str(),
  2284. &status_context(Some(&self.session.path)).expect("status context should load"),
  2285. )
  2286. );
  2287. }
  2288. fn print_sandbox_status() {
  2289. let cwd = env::current_dir().expect("current dir");
  2290. let loader = ConfigLoader::default_for(&cwd);
  2291. let runtime_config = loader
  2292. .load()
  2293. .unwrap_or_else(|_| runtime::RuntimeConfig::empty());
  2294. println!(
  2295. "{}",
  2296. format_sandbox_report(&resolve_sandbox_status(runtime_config.sandbox(), &cwd))
  2297. );
  2298. }
  2299. fn set_model(&mut self, model: Option<String>) -> Result<bool, Box<dyn std::error::Error>> {
  2300. let Some(model) = model else {
  2301. println!(
  2302. "{}",
  2303. format_model_report(
  2304. &self.model,
  2305. self.runtime.session().messages.len(),
  2306. self.runtime.usage().turns(),
  2307. )
  2308. );
  2309. return Ok(false);
  2310. };
  2311. let model = resolve_model_alias(&model).to_string();
  2312. if model == self.model {
  2313. println!(
  2314. "{}",
  2315. format_model_report(
  2316. &self.model,
  2317. self.runtime.session().messages.len(),
  2318. self.runtime.usage().turns(),
  2319. )
  2320. );
  2321. return Ok(false);
  2322. }
  2323. let previous = self.model.clone();
  2324. let session = self.runtime.session().clone();
  2325. let message_count = session.messages.len();
  2326. let runtime = build_runtime(
  2327. session,
  2328. &self.session.id,
  2329. model.clone(),
  2330. self.system_prompt.clone(),
  2331. true,
  2332. true,
  2333. self.allowed_tools.clone(),
  2334. self.permission_mode,
  2335. None,
  2336. )?;
  2337. self.replace_runtime(runtime)?;
  2338. self.model.clone_from(&model);
  2339. println!(
  2340. "{}",
  2341. format_model_switch_report(&previous, &model, message_count)
  2342. );
  2343. Ok(true)
  2344. }
  2345. fn set_permissions(
  2346. &mut self,
  2347. mode: Option<String>,
  2348. ) -> Result<bool, Box<dyn std::error::Error>> {
  2349. let Some(mode) = mode else {
  2350. println!(
  2351. "{}",
  2352. format_permissions_report(self.permission_mode.as_str())
  2353. );
  2354. return Ok(false);
  2355. };
  2356. let normalized = normalize_permission_mode(&mode).ok_or_else(|| {
  2357. format!(
  2358. "unsupported permission mode '{mode}'. Use read-only, workspace-write, or danger-full-access."
  2359. )
  2360. })?;
  2361. if normalized == self.permission_mode.as_str() {
  2362. println!("{}", format_permissions_report(normalized));
  2363. return Ok(false);
  2364. }
  2365. let previous = self.permission_mode.as_str().to_string();
  2366. let session = self.runtime.session().clone();
  2367. self.permission_mode = permission_mode_from_label(normalized);
  2368. let runtime = build_runtime(
  2369. session,
  2370. &self.session.id,
  2371. self.model.clone(),
  2372. self.system_prompt.clone(),
  2373. true,
  2374. true,
  2375. self.allowed_tools.clone(),
  2376. self.permission_mode,
  2377. None,
  2378. )?;
  2379. self.replace_runtime(runtime)?;
  2380. println!(
  2381. "{}",
  2382. format_permissions_switch_report(&previous, normalized)
  2383. );
  2384. Ok(true)
  2385. }
  2386. fn clear_session(&mut self, confirm: bool) -> Result<bool, Box<dyn std::error::Error>> {
  2387. if !confirm {
  2388. println!(
  2389. "clear: confirmation required; run /clear --confirm to start a fresh session."
  2390. );
  2391. return Ok(false);
  2392. }
  2393. let previous_session = self.session.clone();
  2394. let session_state = Session::new();
  2395. self.session = create_managed_session_handle(&session_state.session_id)?;
  2396. let runtime = build_runtime(
  2397. session_state.with_persistence_path(self.session.path.clone()),
  2398. &self.session.id,
  2399. self.model.clone(),
  2400. self.system_prompt.clone(),
  2401. true,
  2402. true,
  2403. self.allowed_tools.clone(),
  2404. self.permission_mode,
  2405. None,
  2406. )?;
  2407. self.replace_runtime(runtime)?;
  2408. println!(
  2409. "Session cleared\n Mode fresh session\n Previous session {}\n Resume previous /resume {}\n Preserved model {}\n Permission mode {}\n New session {}\n Session file {}",
  2410. previous_session.id,
  2411. previous_session.id,
  2412. self.model,
  2413. self.permission_mode.as_str(),
  2414. self.session.id,
  2415. self.session.path.display(),
  2416. );
  2417. Ok(true)
  2418. }
  2419. fn print_cost(&self) {
  2420. let cumulative = self.runtime.usage().cumulative_usage();
  2421. println!("{}", format_cost_report(cumulative));
  2422. }
  2423. fn resume_session(
  2424. &mut self,
  2425. session_path: Option<String>,
  2426. ) -> Result<bool, Box<dyn std::error::Error>> {
  2427. let Some(session_ref) = session_path else {
  2428. println!("{}", render_resume_usage());
  2429. return Ok(false);
  2430. };
  2431. let handle = resolve_session_reference(&session_ref)?;
  2432. let session = Session::load_from_path(&handle.path)?;
  2433. let message_count = session.messages.len();
  2434. let session_id = session.session_id.clone();
  2435. let runtime = build_runtime(
  2436. session,
  2437. &handle.id,
  2438. self.model.clone(),
  2439. self.system_prompt.clone(),
  2440. true,
  2441. true,
  2442. self.allowed_tools.clone(),
  2443. self.permission_mode,
  2444. None,
  2445. )?;
  2446. self.replace_runtime(runtime)?;
  2447. self.session = SessionHandle {
  2448. id: session_id,
  2449. path: handle.path,
  2450. };
  2451. println!(
  2452. "{}",
  2453. format_resume_report(
  2454. &self.session.path.display().to_string(),
  2455. message_count,
  2456. self.runtime.usage().turns(),
  2457. )
  2458. );
  2459. Ok(true)
  2460. }
  2461. fn print_config(section: Option<&str>) -> Result<(), Box<dyn std::error::Error>> {
  2462. println!("{}", render_config_report(section)?);
  2463. Ok(())
  2464. }
  2465. fn print_memory() -> Result<(), Box<dyn std::error::Error>> {
  2466. println!("{}", render_memory_report()?);
  2467. Ok(())
  2468. }
  2469. fn print_agents(args: Option<&str>) -> Result<(), Box<dyn std::error::Error>> {
  2470. let cwd = env::current_dir()?;
  2471. println!("{}", handle_agents_slash_command(args, &cwd)?);
  2472. Ok(())
  2473. }
  2474. fn print_mcp(args: Option<&str>) -> Result<(), Box<dyn std::error::Error>> {
  2475. let cwd = env::current_dir()?;
  2476. println!("{}", handle_mcp_slash_command(args, &cwd)?);
  2477. Ok(())
  2478. }
  2479. fn print_skills(args: Option<&str>) -> Result<(), Box<dyn std::error::Error>> {
  2480. let cwd = env::current_dir()?;
  2481. println!("{}", handle_skills_slash_command(args, &cwd)?);
  2482. Ok(())
  2483. }
  2484. fn print_diff() -> Result<(), Box<dyn std::error::Error>> {
  2485. println!("{}", render_diff_report()?);
  2486. Ok(())
  2487. }
  2488. fn print_version() {
  2489. println!("{}", render_version_report());
  2490. }
  2491. fn export_session(
  2492. &self,
  2493. requested_path: Option<&str>,
  2494. ) -> Result<(), Box<dyn std::error::Error>> {
  2495. let export_path = resolve_export_path(requested_path, self.runtime.session())?;
  2496. fs::write(&export_path, render_export_text(self.runtime.session()))?;
  2497. println!(
  2498. "Export\n Result wrote transcript\n File {}\n Messages {}",
  2499. export_path.display(),
  2500. self.runtime.session().messages.len(),
  2501. );
  2502. Ok(())
  2503. }
  2504. fn handle_session_command(
  2505. &mut self,
  2506. action: Option<&str>,
  2507. target: Option<&str>,
  2508. ) -> Result<bool, Box<dyn std::error::Error>> {
  2509. match action {
  2510. None | Some("list") => {
  2511. println!("{}", render_session_list(&self.session.id)?);
  2512. Ok(false)
  2513. }
  2514. Some("switch") => {
  2515. let Some(target) = target else {
  2516. println!("Usage: /session switch <session-id>");
  2517. return Ok(false);
  2518. };
  2519. let handle = resolve_session_reference(target)?;
  2520. let session = Session::load_from_path(&handle.path)?;
  2521. let message_count = session.messages.len();
  2522. let session_id = session.session_id.clone();
  2523. let runtime = build_runtime(
  2524. session,
  2525. &handle.id,
  2526. self.model.clone(),
  2527. self.system_prompt.clone(),
  2528. true,
  2529. true,
  2530. self.allowed_tools.clone(),
  2531. self.permission_mode,
  2532. None,
  2533. )?;
  2534. self.replace_runtime(runtime)?;
  2535. self.session = SessionHandle {
  2536. id: session_id,
  2537. path: handle.path,
  2538. };
  2539. println!(
  2540. "Session switched\n Active session {}\n File {}\n Messages {}",
  2541. self.session.id,
  2542. self.session.path.display(),
  2543. message_count,
  2544. );
  2545. Ok(true)
  2546. }
  2547. Some("fork") => {
  2548. let forked = self.runtime.fork_session(target.map(ToOwned::to_owned));
  2549. let parent_session_id = self.session.id.clone();
  2550. let handle = create_managed_session_handle(&forked.session_id)?;
  2551. let branch_name = forked
  2552. .fork
  2553. .as_ref()
  2554. .and_then(|fork| fork.branch_name.clone());
  2555. let forked = forked.with_persistence_path(handle.path.clone());
  2556. let message_count = forked.messages.len();
  2557. forked.save_to_path(&handle.path)?;
  2558. let runtime = build_runtime(
  2559. forked,
  2560. &handle.id,
  2561. self.model.clone(),
  2562. self.system_prompt.clone(),
  2563. true,
  2564. true,
  2565. self.allowed_tools.clone(),
  2566. self.permission_mode,
  2567. None,
  2568. )?;
  2569. self.replace_runtime(runtime)?;
  2570. self.session = handle;
  2571. println!(
  2572. "Session forked\n Parent session {}\n Active session {}\n Branch {}\n File {}\n Messages {}",
  2573. parent_session_id,
  2574. self.session.id,
  2575. branch_name.as_deref().unwrap_or("(unnamed)"),
  2576. self.session.path.display(),
  2577. message_count,
  2578. );
  2579. Ok(true)
  2580. }
  2581. Some(other) => {
  2582. println!(
  2583. "Unknown /session action '{other}'. Use /session list, /session switch <session-id>, or /session fork [branch-name]."
  2584. );
  2585. Ok(false)
  2586. }
  2587. }
  2588. }
  2589. fn handle_plugins_command(
  2590. &mut self,
  2591. action: Option<&str>,
  2592. target: Option<&str>,
  2593. ) -> Result<bool, Box<dyn std::error::Error>> {
  2594. let cwd = env::current_dir()?;
  2595. let loader = ConfigLoader::default_for(&cwd);
  2596. let runtime_config = loader.load()?;
  2597. let mut manager = build_plugin_manager(&cwd, &loader, &runtime_config);
  2598. let result = handle_plugins_slash_command(action, target, &mut manager)?;
  2599. println!("{}", result.message);
  2600. if result.reload_runtime {
  2601. self.reload_runtime_features()?;
  2602. }
  2603. Ok(false)
  2604. }
  2605. fn reload_runtime_features(&mut self) -> Result<(), Box<dyn std::error::Error>> {
  2606. let runtime = build_runtime(
  2607. self.runtime.session().clone(),
  2608. &self.session.id,
  2609. self.model.clone(),
  2610. self.system_prompt.clone(),
  2611. true,
  2612. true,
  2613. self.allowed_tools.clone(),
  2614. self.permission_mode,
  2615. None,
  2616. )?;
  2617. self.replace_runtime(runtime)?;
  2618. self.persist_session()
  2619. }
  2620. fn compact(&mut self) -> Result<(), Box<dyn std::error::Error>> {
  2621. let result = self.runtime.compact(CompactionConfig::default());
  2622. let removed = result.removed_message_count;
  2623. let kept = result.compacted_session.messages.len();
  2624. let skipped = removed == 0;
  2625. let runtime = build_runtime(
  2626. result.compacted_session,
  2627. &self.session.id,
  2628. self.model.clone(),
  2629. self.system_prompt.clone(),
  2630. true,
  2631. true,
  2632. self.allowed_tools.clone(),
  2633. self.permission_mode,
  2634. None,
  2635. )?;
  2636. self.replace_runtime(runtime)?;
  2637. self.persist_session()?;
  2638. println!("{}", format_compact_report(removed, kept, skipped));
  2639. Ok(())
  2640. }
  2641. fn run_internal_prompt_text_with_progress(
  2642. &self,
  2643. prompt: &str,
  2644. enable_tools: bool,
  2645. progress: Option<InternalPromptProgressReporter>,
  2646. ) -> Result<String, Box<dyn std::error::Error>> {
  2647. let session = self.runtime.session().clone();
  2648. let mut runtime = build_runtime(
  2649. session,
  2650. &self.session.id,
  2651. self.model.clone(),
  2652. self.system_prompt.clone(),
  2653. enable_tools,
  2654. false,
  2655. self.allowed_tools.clone(),
  2656. self.permission_mode,
  2657. progress,
  2658. )?;
  2659. let mut permission_prompter = CliPermissionPrompter::new(self.permission_mode);
  2660. let summary = runtime.run_turn(prompt, Some(&mut permission_prompter))?;
  2661. let text = final_assistant_text(&summary).trim().to_string();
  2662. runtime.shutdown_plugins()?;
  2663. Ok(text)
  2664. }
  2665. fn run_internal_prompt_text(
  2666. &self,
  2667. prompt: &str,
  2668. enable_tools: bool,
  2669. ) -> Result<String, Box<dyn std::error::Error>> {
  2670. self.run_internal_prompt_text_with_progress(prompt, enable_tools, None)
  2671. }
  2672. fn run_bughunter(&self, scope: Option<&str>) -> Result<(), Box<dyn std::error::Error>> {
  2673. println!("{}", format_bughunter_report(scope));
  2674. Ok(())
  2675. }
  2676. fn run_ultraplan(&self, task: Option<&str>) -> Result<(), Box<dyn std::error::Error>> {
  2677. println!("{}", format_ultraplan_report(task));
  2678. Ok(())
  2679. }
  2680. fn run_teleport(target: Option<&str>) -> Result<(), Box<dyn std::error::Error>> {
  2681. let Some(target) = target.map(str::trim).filter(|value| !value.is_empty()) else {
  2682. println!("Usage: /teleport <symbol-or-path>");
  2683. return Ok(());
  2684. };
  2685. println!("{}", render_teleport_report(target)?);
  2686. Ok(())
  2687. }
  2688. fn run_debug_tool_call(&self, args: Option<&str>) -> Result<(), Box<dyn std::error::Error>> {
  2689. validate_no_args("/debug-tool-call", args)?;
  2690. println!("{}", render_last_tool_debug_report(self.runtime.session())?);
  2691. Ok(())
  2692. }
  2693. fn run_commit(&mut self, args: Option<&str>) -> Result<(), Box<dyn std::error::Error>> {
  2694. validate_no_args("/commit", args)?;
  2695. let status = git_output(&["status", "--short", "--branch"])?;
  2696. let summary = parse_git_workspace_summary(Some(&status));
  2697. let branch = parse_git_status_branch(Some(&status));
  2698. if summary.is_clean() {
  2699. println!("{}", format_commit_skipped_report());
  2700. return Ok(());
  2701. }
  2702. println!(
  2703. "{}",
  2704. format_commit_preflight_report(branch.as_deref(), summary)
  2705. );
  2706. Ok(())
  2707. }
  2708. fn run_pr(&self, context: Option<&str>) -> Result<(), Box<dyn std::error::Error>> {
  2709. let branch =
  2710. resolve_git_branch_for(&env::current_dir()?).unwrap_or_else(|| "unknown".to_string());
  2711. println!("{}", format_pr_report(&branch, context));
  2712. Ok(())
  2713. }
  2714. fn run_issue(&self, context: Option<&str>) -> Result<(), Box<dyn std::error::Error>> {
  2715. println!("{}", format_issue_report(context));
  2716. Ok(())
  2717. }
  2718. }
  2719. fn sessions_dir() -> Result<PathBuf, Box<dyn std::error::Error>> {
  2720. let cwd = env::current_dir()?;
  2721. let path = cwd.join(".claw").join("sessions");
  2722. fs::create_dir_all(&path)?;
  2723. Ok(path)
  2724. }
  2725. fn create_managed_session_handle(
  2726. session_id: &str,
  2727. ) -> Result<SessionHandle, Box<dyn std::error::Error>> {
  2728. let id = session_id.to_string();
  2729. let path = sessions_dir()?.join(format!("{id}.{PRIMARY_SESSION_EXTENSION}"));
  2730. Ok(SessionHandle { id, path })
  2731. }
  2732. fn resolve_session_reference(reference: &str) -> Result<SessionHandle, Box<dyn std::error::Error>> {
  2733. if SESSION_REFERENCE_ALIASES
  2734. .iter()
  2735. .any(|alias| reference.eq_ignore_ascii_case(alias))
  2736. {
  2737. let latest = latest_managed_session()?;
  2738. return Ok(SessionHandle {
  2739. id: latest.id,
  2740. path: latest.path,
  2741. });
  2742. }
  2743. let direct = PathBuf::from(reference);
  2744. let looks_like_path = direct.extension().is_some() || direct.components().count() > 1;
  2745. let path = if direct.exists() {
  2746. direct
  2747. } else if looks_like_path {
  2748. return Err(format_missing_session_reference(reference).into());
  2749. } else {
  2750. resolve_managed_session_path(reference)?
  2751. };
  2752. let id = path
  2753. .file_name()
  2754. .and_then(|value| value.to_str())
  2755. .and_then(|name| {
  2756. name.strip_suffix(&format!(".{PRIMARY_SESSION_EXTENSION}"))
  2757. .or_else(|| name.strip_suffix(&format!(".{LEGACY_SESSION_EXTENSION}")))
  2758. })
  2759. .unwrap_or(reference)
  2760. .to_string();
  2761. Ok(SessionHandle { id, path })
  2762. }
  2763. fn resolve_managed_session_path(session_id: &str) -> Result<PathBuf, Box<dyn std::error::Error>> {
  2764. let directory = sessions_dir()?;
  2765. for extension in [PRIMARY_SESSION_EXTENSION, LEGACY_SESSION_EXTENSION] {
  2766. let path = directory.join(format!("{session_id}.{extension}"));
  2767. if path.exists() {
  2768. return Ok(path);
  2769. }
  2770. }
  2771. Err(format_missing_session_reference(session_id).into())
  2772. }
  2773. fn is_managed_session_file(path: &Path) -> bool {
  2774. path.extension()
  2775. .and_then(|ext| ext.to_str())
  2776. .is_some_and(|extension| {
  2777. extension == PRIMARY_SESSION_EXTENSION || extension == LEGACY_SESSION_EXTENSION
  2778. })
  2779. }
  2780. fn list_managed_sessions() -> Result<Vec<ManagedSessionSummary>, Box<dyn std::error::Error>> {
  2781. let mut sessions = Vec::new();
  2782. for entry in fs::read_dir(sessions_dir()?)? {
  2783. let entry = entry?;
  2784. let path = entry.path();
  2785. if !is_managed_session_file(&path) {
  2786. continue;
  2787. }
  2788. let metadata = entry.metadata()?;
  2789. let modified_epoch_millis = metadata
  2790. .modified()
  2791. .ok()
  2792. .and_then(|time| time.duration_since(UNIX_EPOCH).ok())
  2793. .map(|duration| duration.as_millis())
  2794. .unwrap_or_default();
  2795. let (id, message_count, parent_session_id, branch_name) =
  2796. match Session::load_from_path(&path) {
  2797. Ok(session) => {
  2798. let parent_session_id = session
  2799. .fork
  2800. .as_ref()
  2801. .map(|fork| fork.parent_session_id.clone());
  2802. let branch_name = session
  2803. .fork
  2804. .as_ref()
  2805. .and_then(|fork| fork.branch_name.clone());
  2806. (
  2807. session.session_id,
  2808. session.messages.len(),
  2809. parent_session_id,
  2810. branch_name,
  2811. )
  2812. }
  2813. Err(_) => (
  2814. path.file_stem()
  2815. .and_then(|value| value.to_str())
  2816. .unwrap_or("unknown")
  2817. .to_string(),
  2818. 0,
  2819. None,
  2820. None,
  2821. ),
  2822. };
  2823. sessions.push(ManagedSessionSummary {
  2824. id,
  2825. path,
  2826. modified_epoch_millis,
  2827. message_count,
  2828. parent_session_id,
  2829. branch_name,
  2830. });
  2831. }
  2832. sessions.sort_by(|left, right| {
  2833. right
  2834. .modified_epoch_millis
  2835. .cmp(&left.modified_epoch_millis)
  2836. .then_with(|| right.id.cmp(&left.id))
  2837. });
  2838. Ok(sessions)
  2839. }
  2840. fn latest_managed_session() -> Result<ManagedSessionSummary, Box<dyn std::error::Error>> {
  2841. list_managed_sessions()?
  2842. .into_iter()
  2843. .next()
  2844. .ok_or_else(|| format_no_managed_sessions().into())
  2845. }
  2846. fn format_missing_session_reference(reference: &str) -> String {
  2847. format!(
  2848. "session not found: {reference}\nHint: managed sessions live in .claw/sessions/. Try `{LATEST_SESSION_REFERENCE}` for the most recent session or `/session list` in the REPL."
  2849. )
  2850. }
  2851. fn format_no_managed_sessions() -> String {
  2852. format!(
  2853. "no managed sessions found in .claw/sessions/\nStart `claw` to create a session, then rerun with `--resume {LATEST_SESSION_REFERENCE}`."
  2854. )
  2855. }
  2856. fn render_session_list(active_session_id: &str) -> Result<String, Box<dyn std::error::Error>> {
  2857. let sessions = list_managed_sessions()?;
  2858. let mut lines = vec![
  2859. "Sessions".to_string(),
  2860. format!(" Directory {}", sessions_dir()?.display()),
  2861. ];
  2862. if sessions.is_empty() {
  2863. lines.push(" No managed sessions saved yet.".to_string());
  2864. return Ok(lines.join("\n"));
  2865. }
  2866. for session in sessions {
  2867. let marker = if session.id == active_session_id {
  2868. "● current"
  2869. } else {
  2870. "○ saved"
  2871. };
  2872. let lineage = match (
  2873. session.branch_name.as_deref(),
  2874. session.parent_session_id.as_deref(),
  2875. ) {
  2876. (Some(branch_name), Some(parent_session_id)) => {
  2877. format!(" branch={branch_name} from={parent_session_id}")
  2878. }
  2879. (None, Some(parent_session_id)) => format!(" from={parent_session_id}"),
  2880. (Some(branch_name), None) => format!(" branch={branch_name}"),
  2881. (None, None) => String::new(),
  2882. };
  2883. lines.push(format!(
  2884. " {id:<20} {marker:<10} msgs={msgs:<4} modified={modified}{lineage} path={path}",
  2885. id = session.id,
  2886. msgs = session.message_count,
  2887. modified = format_session_modified_age(session.modified_epoch_millis),
  2888. lineage = lineage,
  2889. path = session.path.display(),
  2890. ));
  2891. }
  2892. Ok(lines.join("\n"))
  2893. }
  2894. fn format_session_modified_age(modified_epoch_millis: u128) -> String {
  2895. let now = std::time::SystemTime::now()
  2896. .duration_since(UNIX_EPOCH)
  2897. .ok()
  2898. .map_or(modified_epoch_millis, |duration| duration.as_millis());
  2899. let delta_seconds = now
  2900. .saturating_sub(modified_epoch_millis)
  2901. .checked_div(1_000)
  2902. .unwrap_or_default();
  2903. match delta_seconds {
  2904. 0..=4 => "just-now".to_string(),
  2905. 5..=59 => format!("{delta_seconds}s-ago"),
  2906. 60..=3_599 => format!("{}m-ago", delta_seconds / 60),
  2907. 3_600..=86_399 => format!("{}h-ago", delta_seconds / 3_600),
  2908. _ => format!("{}d-ago", delta_seconds / 86_400),
  2909. }
  2910. }
  2911. fn write_session_clear_backup(
  2912. session: &Session,
  2913. session_path: &Path,
  2914. ) -> Result<PathBuf, Box<dyn std::error::Error>> {
  2915. let backup_path = session_clear_backup_path(session_path);
  2916. session.save_to_path(&backup_path)?;
  2917. Ok(backup_path)
  2918. }
  2919. fn session_clear_backup_path(session_path: &Path) -> PathBuf {
  2920. let timestamp = std::time::SystemTime::now()
  2921. .duration_since(UNIX_EPOCH)
  2922. .ok()
  2923. .map_or(0, |duration| duration.as_millis());
  2924. let file_name = session_path
  2925. .file_name()
  2926. .and_then(|value| value.to_str())
  2927. .unwrap_or("session.jsonl");
  2928. session_path.with_file_name(format!("{file_name}.before-clear-{timestamp}.bak"))
  2929. }
  2930. fn render_repl_help() -> String {
  2931. [
  2932. "REPL".to_string(),
  2933. " /exit Quit the REPL".to_string(),
  2934. " /quit Quit the REPL".to_string(),
  2935. " Up/Down Navigate prompt history".to_string(),
  2936. " Tab Complete commands, modes, and recent sessions".to_string(),
  2937. " Ctrl-C Clear input (or exit on empty prompt)".to_string(),
  2938. " Shift+Enter/Ctrl+J Insert a newline".to_string(),
  2939. " Auto-save .claw/sessions/<session-id>.jsonl".to_string(),
  2940. " Resume latest /resume latest".to_string(),
  2941. " Browse sessions /session list".to_string(),
  2942. String::new(),
  2943. render_slash_command_help(),
  2944. ]
  2945. .join(
  2946. "
  2947. ",
  2948. )
  2949. }
  2950. fn print_status_snapshot(
  2951. model: &str,
  2952. permission_mode: PermissionMode,
  2953. ) -> Result<(), Box<dyn std::error::Error>> {
  2954. println!(
  2955. "{}",
  2956. format_status_report(
  2957. model,
  2958. StatusUsage {
  2959. message_count: 0,
  2960. turns: 0,
  2961. latest: TokenUsage::default(),
  2962. cumulative: TokenUsage::default(),
  2963. estimated_tokens: 0,
  2964. },
  2965. permission_mode.as_str(),
  2966. &status_context(None)?,
  2967. )
  2968. );
  2969. Ok(())
  2970. }
  2971. fn status_context(
  2972. session_path: Option<&Path>,
  2973. ) -> Result<StatusContext, Box<dyn std::error::Error>> {
  2974. let cwd = env::current_dir()?;
  2975. let loader = ConfigLoader::default_for(&cwd);
  2976. let discovered_config_files = loader.discover().len();
  2977. let runtime_config = loader.load()?;
  2978. let project_context = ProjectContext::discover_with_git(&cwd, DEFAULT_DATE)?;
  2979. let (project_root, git_branch) =
  2980. parse_git_status_metadata(project_context.git_status.as_deref());
  2981. let git_summary = parse_git_workspace_summary(project_context.git_status.as_deref());
  2982. let sandbox_status = resolve_sandbox_status(runtime_config.sandbox(), &cwd);
  2983. Ok(StatusContext {
  2984. cwd,
  2985. session_path: session_path.map(Path::to_path_buf),
  2986. loaded_config_files: runtime_config.loaded_entries().len(),
  2987. discovered_config_files,
  2988. memory_file_count: project_context.instruction_files.len(),
  2989. project_root,
  2990. git_branch,
  2991. git_summary,
  2992. sandbox_status,
  2993. })
  2994. }
  2995. fn format_status_report(
  2996. model: &str,
  2997. usage: StatusUsage,
  2998. permission_mode: &str,
  2999. context: &StatusContext,
  3000. ) -> String {
  3001. [
  3002. format!(
  3003. "Status
  3004. Model {model}
  3005. Permission mode {permission_mode}
  3006. Messages {}
  3007. Turns {}
  3008. Estimated tokens {}",
  3009. usage.message_count, usage.turns, usage.estimated_tokens,
  3010. ),
  3011. format!(
  3012. "Usage
  3013. Latest total {}
  3014. Cumulative input {}
  3015. Cumulative output {}
  3016. Cumulative total {}",
  3017. usage.latest.total_tokens(),
  3018. usage.cumulative.input_tokens,
  3019. usage.cumulative.output_tokens,
  3020. usage.cumulative.total_tokens(),
  3021. ),
  3022. format!(
  3023. "Workspace
  3024. Cwd {}
  3025. Project root {}
  3026. Git branch {}
  3027. Git state {}
  3028. Changed files {}
  3029. Staged {}
  3030. Unstaged {}
  3031. Untracked {}
  3032. Session {}
  3033. Config files loaded {}/{}
  3034. Memory files {}
  3035. Suggested flow /status → /diff → /commit",
  3036. context.cwd.display(),
  3037. context
  3038. .project_root
  3039. .as_ref()
  3040. .map_or_else(|| "unknown".to_string(), |path| path.display().to_string()),
  3041. context.git_branch.as_deref().unwrap_or("unknown"),
  3042. context.git_summary.headline(),
  3043. context.git_summary.changed_files,
  3044. context.git_summary.staged_files,
  3045. context.git_summary.unstaged_files,
  3046. context.git_summary.untracked_files,
  3047. context.session_path.as_ref().map_or_else(
  3048. || "live-repl".to_string(),
  3049. |path| path.display().to_string()
  3050. ),
  3051. context.loaded_config_files,
  3052. context.discovered_config_files,
  3053. context.memory_file_count,
  3054. ),
  3055. format_sandbox_report(&context.sandbox_status),
  3056. ]
  3057. .join(
  3058. "
  3059. ",
  3060. )
  3061. }
  3062. fn format_sandbox_report(status: &runtime::SandboxStatus) -> String {
  3063. format!(
  3064. "Sandbox
  3065. Enabled {}
  3066. Active {}
  3067. Supported {}
  3068. In container {}
  3069. Requested ns {}
  3070. Active ns {}
  3071. Requested net {}
  3072. Active net {}
  3073. Filesystem mode {}
  3074. Filesystem active {}
  3075. Allowed mounts {}
  3076. Markers {}
  3077. Fallback reason {}",
  3078. status.enabled,
  3079. status.active,
  3080. status.supported,
  3081. status.in_container,
  3082. status.requested.namespace_restrictions,
  3083. status.namespace_active,
  3084. status.requested.network_isolation,
  3085. status.network_active,
  3086. status.filesystem_mode.as_str(),
  3087. status.filesystem_active,
  3088. if status.allowed_mounts.is_empty() {
  3089. "<none>".to_string()
  3090. } else {
  3091. status.allowed_mounts.join(", ")
  3092. },
  3093. if status.container_markers.is_empty() {
  3094. "<none>".to_string()
  3095. } else {
  3096. status.container_markers.join(", ")
  3097. },
  3098. status
  3099. .fallback_reason
  3100. .clone()
  3101. .unwrap_or_else(|| "<none>".to_string()),
  3102. )
  3103. }
  3104. fn format_commit_preflight_report(branch: Option<&str>, summary: GitWorkspaceSummary) -> String {
  3105. format!(
  3106. "Commit
  3107. Result ready
  3108. Branch {}
  3109. Workspace {}
  3110. Changed files {}
  3111. Action create a git commit from the current workspace changes",
  3112. branch.unwrap_or("unknown"),
  3113. summary.headline(),
  3114. summary.changed_files,
  3115. )
  3116. }
  3117. fn format_commit_skipped_report() -> String {
  3118. "Commit
  3119. Result skipped
  3120. Reason no workspace changes
  3121. Action create a git commit from the current workspace changes
  3122. Next /status to inspect context · /diff to inspect repo changes"
  3123. .to_string()
  3124. }
  3125. fn print_sandbox_status_snapshot() -> Result<(), Box<dyn std::error::Error>> {
  3126. let cwd = env::current_dir()?;
  3127. let loader = ConfigLoader::default_for(&cwd);
  3128. let runtime_config = loader
  3129. .load()
  3130. .unwrap_or_else(|_| runtime::RuntimeConfig::empty());
  3131. println!(
  3132. "{}",
  3133. format_sandbox_report(&resolve_sandbox_status(runtime_config.sandbox(), &cwd))
  3134. );
  3135. Ok(())
  3136. }
  3137. fn render_config_report(section: Option<&str>) -> Result<String, Box<dyn std::error::Error>> {
  3138. let cwd = env::current_dir()?;
  3139. let loader = ConfigLoader::default_for(&cwd);
  3140. let discovered = loader.discover();
  3141. let runtime_config = loader.load()?;
  3142. let mut lines = vec![
  3143. format!(
  3144. "Config
  3145. Working directory {}
  3146. Loaded files {}
  3147. Merged keys {}",
  3148. cwd.display(),
  3149. runtime_config.loaded_entries().len(),
  3150. runtime_config.merged().len()
  3151. ),
  3152. "Discovered files".to_string(),
  3153. ];
  3154. for entry in discovered {
  3155. let source = match entry.source {
  3156. ConfigSource::User => "user",
  3157. ConfigSource::Project => "project",
  3158. ConfigSource::Local => "local",
  3159. };
  3160. let status = if runtime_config
  3161. .loaded_entries()
  3162. .iter()
  3163. .any(|loaded_entry| loaded_entry.path == entry.path)
  3164. {
  3165. "loaded"
  3166. } else {
  3167. "missing"
  3168. };
  3169. lines.push(format!(
  3170. " {source:<7} {status:<7} {}",
  3171. entry.path.display()
  3172. ));
  3173. }
  3174. if let Some(section) = section {
  3175. lines.push(format!("Merged section: {section}"));
  3176. let value = match section {
  3177. "env" => runtime_config.get("env"),
  3178. "hooks" => runtime_config.get("hooks"),
  3179. "model" => runtime_config.get("model"),
  3180. "plugins" => runtime_config
  3181. .get("plugins")
  3182. .or_else(|| runtime_config.get("enabledPlugins")),
  3183. other => {
  3184. lines.push(format!(
  3185. " Unsupported config section '{other}'. Use env, hooks, model, or plugins."
  3186. ));
  3187. return Ok(lines.join(
  3188. "
  3189. ",
  3190. ));
  3191. }
  3192. };
  3193. lines.push(format!(
  3194. " {}",
  3195. match value {
  3196. Some(value) => value.render(),
  3197. None => "<unset>".to_string(),
  3198. }
  3199. ));
  3200. return Ok(lines.join(
  3201. "
  3202. ",
  3203. ));
  3204. }
  3205. lines.push("Merged JSON".to_string());
  3206. lines.push(format!(" {}", runtime_config.as_json().render()));
  3207. Ok(lines.join(
  3208. "
  3209. ",
  3210. ))
  3211. }
  3212. fn render_memory_report() -> Result<String, Box<dyn std::error::Error>> {
  3213. let cwd = env::current_dir()?;
  3214. let project_context = ProjectContext::discover(&cwd, DEFAULT_DATE)?;
  3215. let mut lines = vec![format!(
  3216. "Memory
  3217. Working directory {}
  3218. Instruction files {}",
  3219. cwd.display(),
  3220. project_context.instruction_files.len()
  3221. )];
  3222. if project_context.instruction_files.is_empty() {
  3223. lines.push("Discovered files".to_string());
  3224. lines.push(
  3225. " No CLAUDE instruction files discovered in the current directory ancestry."
  3226. .to_string(),
  3227. );
  3228. } else {
  3229. lines.push("Discovered files".to_string());
  3230. for (index, file) in project_context.instruction_files.iter().enumerate() {
  3231. let preview = file.content.lines().next().unwrap_or("").trim();
  3232. let preview = if preview.is_empty() {
  3233. "<empty>"
  3234. } else {
  3235. preview
  3236. };
  3237. lines.push(format!(" {}. {}", index + 1, file.path.display(),));
  3238. lines.push(format!(
  3239. " lines={} preview={}",
  3240. file.content.lines().count(),
  3241. preview
  3242. ));
  3243. }
  3244. }
  3245. Ok(lines.join(
  3246. "
  3247. ",
  3248. ))
  3249. }
  3250. fn init_claude_md() -> Result<String, Box<dyn std::error::Error>> {
  3251. let cwd = env::current_dir()?;
  3252. Ok(initialize_repo(&cwd)?.render())
  3253. }
  3254. fn run_init() -> Result<(), Box<dyn std::error::Error>> {
  3255. println!("{}", init_claude_md()?);
  3256. Ok(())
  3257. }
  3258. fn normalize_permission_mode(mode: &str) -> Option<&'static str> {
  3259. match mode.trim() {
  3260. "read-only" => Some("read-only"),
  3261. "workspace-write" => Some("workspace-write"),
  3262. "danger-full-access" => Some("danger-full-access"),
  3263. _ => None,
  3264. }
  3265. }
  3266. fn render_diff_report() -> Result<String, Box<dyn std::error::Error>> {
  3267. render_diff_report_for(&env::current_dir()?)
  3268. }
  3269. fn render_diff_report_for(cwd: &Path) -> Result<String, Box<dyn std::error::Error>> {
  3270. let staged = run_git_diff_command_in(cwd, &["diff", "--cached"])?;
  3271. let unstaged = run_git_diff_command_in(cwd, &["diff"])?;
  3272. if staged.trim().is_empty() && unstaged.trim().is_empty() {
  3273. return Ok(
  3274. "Diff\n Result clean working tree\n Detail no current changes"
  3275. .to_string(),
  3276. );
  3277. }
  3278. let mut sections = Vec::new();
  3279. if !staged.trim().is_empty() {
  3280. sections.push(format!("Staged changes:\n{}", staged.trim_end()));
  3281. }
  3282. if !unstaged.trim().is_empty() {
  3283. sections.push(format!("Unstaged changes:\n{}", unstaged.trim_end()));
  3284. }
  3285. Ok(format!("Diff\n\n{}", sections.join("\n\n")))
  3286. }
  3287. fn run_git_diff_command_in(
  3288. cwd: &Path,
  3289. args: &[&str],
  3290. ) -> Result<String, Box<dyn std::error::Error>> {
  3291. let output = std::process::Command::new("git")
  3292. .args(args)
  3293. .current_dir(cwd)
  3294. .output()?;
  3295. if !output.status.success() {
  3296. let stderr = String::from_utf8_lossy(&output.stderr).trim().to_string();
  3297. return Err(format!("git {} failed: {stderr}", args.join(" ")).into());
  3298. }
  3299. Ok(String::from_utf8(output.stdout)?)
  3300. }
  3301. fn render_teleport_report(target: &str) -> Result<String, Box<dyn std::error::Error>> {
  3302. let cwd = env::current_dir()?;
  3303. let file_list = Command::new("rg")
  3304. .args(["--files"])
  3305. .current_dir(&cwd)
  3306. .output()?;
  3307. let file_matches = if file_list.status.success() {
  3308. String::from_utf8(file_list.stdout)?
  3309. .lines()
  3310. .filter(|line| line.contains(target))
  3311. .take(10)
  3312. .map(ToOwned::to_owned)
  3313. .collect::<Vec<_>>()
  3314. } else {
  3315. Vec::new()
  3316. };
  3317. let content_output = Command::new("rg")
  3318. .args(["-n", "-S", "--color", "never", target, "."])
  3319. .current_dir(&cwd)
  3320. .output()?;
  3321. let mut lines = vec![
  3322. "Teleport".to_string(),
  3323. format!(" Target {target}"),
  3324. " Action search workspace files and content for the target".to_string(),
  3325. ];
  3326. if !file_matches.is_empty() {
  3327. lines.push(String::new());
  3328. lines.push("File matches".to_string());
  3329. lines.extend(file_matches.into_iter().map(|path| format!(" {path}")));
  3330. }
  3331. if content_output.status.success() {
  3332. let matches = String::from_utf8(content_output.stdout)?;
  3333. if !matches.trim().is_empty() {
  3334. lines.push(String::new());
  3335. lines.push("Content matches".to_string());
  3336. lines.push(truncate_for_prompt(&matches, 4_000));
  3337. }
  3338. }
  3339. if lines.len() == 1 {
  3340. lines.push(" Result no matches found".to_string());
  3341. }
  3342. Ok(lines.join("\n"))
  3343. }
  3344. fn render_last_tool_debug_report(session: &Session) -> Result<String, Box<dyn std::error::Error>> {
  3345. let last_tool_use = session
  3346. .messages
  3347. .iter()
  3348. .rev()
  3349. .find_map(|message| {
  3350. message.blocks.iter().rev().find_map(|block| match block {
  3351. ContentBlock::ToolUse { id, name, input } => {
  3352. Some((id.clone(), name.clone(), input.clone()))
  3353. }
  3354. _ => None,
  3355. })
  3356. })
  3357. .ok_or_else(|| "no prior tool call found in session".to_string())?;
  3358. let tool_result = session.messages.iter().rev().find_map(|message| {
  3359. message.blocks.iter().rev().find_map(|block| match block {
  3360. ContentBlock::ToolResult {
  3361. tool_use_id,
  3362. tool_name,
  3363. output,
  3364. is_error,
  3365. } if tool_use_id == &last_tool_use.0 => {
  3366. Some((tool_name.clone(), output.clone(), *is_error))
  3367. }
  3368. _ => None,
  3369. })
  3370. });
  3371. let mut lines = vec![
  3372. "Debug tool call".to_string(),
  3373. " Action inspect the last recorded tool call and its result".to_string(),
  3374. format!(" Tool id {}", last_tool_use.0),
  3375. format!(" Tool name {}", last_tool_use.1),
  3376. " Input".to_string(),
  3377. indent_block(&last_tool_use.2, 4),
  3378. ];
  3379. match tool_result {
  3380. Some((tool_name, output, is_error)) => {
  3381. lines.push(" Result".to_string());
  3382. lines.push(format!(" name {tool_name}"));
  3383. lines.push(format!(
  3384. " status {}",
  3385. if is_error { "error" } else { "ok" }
  3386. ));
  3387. lines.push(indent_block(&output, 4));
  3388. }
  3389. None => lines.push(" Result missing tool result".to_string()),
  3390. }
  3391. Ok(lines.join("\n"))
  3392. }
  3393. fn indent_block(value: &str, spaces: usize) -> String {
  3394. let indent = " ".repeat(spaces);
  3395. value
  3396. .lines()
  3397. .map(|line| format!("{indent}{line}"))
  3398. .collect::<Vec<_>>()
  3399. .join("\n")
  3400. }
  3401. fn validate_no_args(
  3402. command_name: &str,
  3403. args: Option<&str>,
  3404. ) -> Result<(), Box<dyn std::error::Error>> {
  3405. if let Some(args) = args.map(str::trim).filter(|value| !value.is_empty()) {
  3406. return Err(format!(
  3407. "{command_name} does not accept arguments. Received: {args}\nUsage: {command_name}"
  3408. )
  3409. .into());
  3410. }
  3411. Ok(())
  3412. }
  3413. fn format_bughunter_report(scope: Option<&str>) -> String {
  3414. format!(
  3415. "Bughunter
  3416. Scope {}
  3417. Action inspect the selected code for likely bugs and correctness issues
  3418. Output findings should include file paths, severity, and suggested fixes",
  3419. scope.unwrap_or("the current repository")
  3420. )
  3421. }
  3422. fn format_ultraplan_report(task: Option<&str>) -> String {
  3423. format!(
  3424. "Ultraplan
  3425. Task {}
  3426. Action break work into a multi-step execution plan
  3427. Output plan should cover goals, risks, sequencing, verification, and rollback",
  3428. task.unwrap_or("the current repo work")
  3429. )
  3430. }
  3431. fn format_pr_report(branch: &str, context: Option<&str>) -> String {
  3432. format!(
  3433. "PR
  3434. Branch {branch}
  3435. Context {}
  3436. Action draft or create a pull request for the current branch
  3437. Output title and markdown body suitable for GitHub",
  3438. context.unwrap_or("none")
  3439. )
  3440. }
  3441. fn format_issue_report(context: Option<&str>) -> String {
  3442. format!(
  3443. "Issue
  3444. Context {}
  3445. Action draft or create a GitHub issue from the current context
  3446. Output title and markdown body suitable for GitHub",
  3447. context.unwrap_or("none")
  3448. )
  3449. }
  3450. fn git_output(args: &[&str]) -> Result<String, Box<dyn std::error::Error>> {
  3451. let output = Command::new("git")
  3452. .args(args)
  3453. .current_dir(env::current_dir()?)
  3454. .output()?;
  3455. if !output.status.success() {
  3456. let stderr = String::from_utf8_lossy(&output.stderr).trim().to_string();
  3457. return Err(format!("git {} failed: {stderr}", args.join(" ")).into());
  3458. }
  3459. Ok(String::from_utf8(output.stdout)?)
  3460. }
  3461. fn git_status_ok(args: &[&str]) -> Result<(), Box<dyn std::error::Error>> {
  3462. let output = Command::new("git")
  3463. .args(args)
  3464. .current_dir(env::current_dir()?)
  3465. .output()?;
  3466. if !output.status.success() {
  3467. let stderr = String::from_utf8_lossy(&output.stderr).trim().to_string();
  3468. return Err(format!("git {} failed: {stderr}", args.join(" ")).into());
  3469. }
  3470. Ok(())
  3471. }
  3472. fn command_exists(name: &str) -> bool {
  3473. Command::new("which")
  3474. .arg(name)
  3475. .output()
  3476. .map(|output| output.status.success())
  3477. .unwrap_or(false)
  3478. }
  3479. fn write_temp_text_file(
  3480. filename: &str,
  3481. contents: &str,
  3482. ) -> Result<PathBuf, Box<dyn std::error::Error>> {
  3483. let path = env::temp_dir().join(filename);
  3484. fs::write(&path, contents)?;
  3485. Ok(path)
  3486. }
  3487. fn recent_user_context(session: &Session, limit: usize) -> String {
  3488. let requests = session
  3489. .messages
  3490. .iter()
  3491. .filter(|message| message.role == MessageRole::User)
  3492. .filter_map(|message| {
  3493. message.blocks.iter().find_map(|block| match block {
  3494. ContentBlock::Text { text } => Some(text.trim().to_string()),
  3495. _ => None,
  3496. })
  3497. })
  3498. .rev()
  3499. .take(limit)
  3500. .collect::<Vec<_>>();
  3501. if requests.is_empty() {
  3502. "<no prior user messages>".to_string()
  3503. } else {
  3504. requests
  3505. .into_iter()
  3506. .rev()
  3507. .enumerate()
  3508. .map(|(index, text)| format!("{}. {}", index + 1, text))
  3509. .collect::<Vec<_>>()
  3510. .join("\n")
  3511. }
  3512. }
  3513. fn truncate_for_prompt(value: &str, limit: usize) -> String {
  3514. if value.chars().count() <= limit {
  3515. value.trim().to_string()
  3516. } else {
  3517. let truncated = value.chars().take(limit).collect::<String>();
  3518. format!("{}\n…[truncated]", truncated.trim_end())
  3519. }
  3520. }
  3521. fn sanitize_generated_message(value: &str) -> String {
  3522. value.trim().trim_matches('`').trim().replace("\r\n", "\n")
  3523. }
  3524. fn parse_titled_body(value: &str) -> Option<(String, String)> {
  3525. let normalized = sanitize_generated_message(value);
  3526. let title = normalized
  3527. .lines()
  3528. .find_map(|line| line.strip_prefix("TITLE:").map(str::trim))?;
  3529. let body_start = normalized.find("BODY:")?;
  3530. let body = normalized[body_start + "BODY:".len()..].trim();
  3531. Some((title.to_string(), body.to_string()))
  3532. }
  3533. fn render_version_report() -> String {
  3534. let git_sha = GIT_SHA.unwrap_or("unknown");
  3535. let target = BUILD_TARGET.unwrap_or("unknown");
  3536. format!(
  3537. "Claw Code\n Version {VERSION}\n Git SHA {git_sha}\n Target {target}\n Build date {DEFAULT_DATE}"
  3538. )
  3539. }
  3540. fn render_export_text(session: &Session) -> String {
  3541. let mut lines = vec!["# Conversation Export".to_string(), String::new()];
  3542. for (index, message) in session.messages.iter().enumerate() {
  3543. let role = match message.role {
  3544. MessageRole::System => "system",
  3545. MessageRole::User => "user",
  3546. MessageRole::Assistant => "assistant",
  3547. MessageRole::Tool => "tool",
  3548. };
  3549. lines.push(format!("## {}. {role}", index + 1));
  3550. for block in &message.blocks {
  3551. match block {
  3552. ContentBlock::Text { text } => lines.push(text.clone()),
  3553. ContentBlock::ToolUse { id, name, input } => {
  3554. lines.push(format!("[tool_use id={id} name={name}] {input}"));
  3555. }
  3556. ContentBlock::ToolResult {
  3557. tool_use_id,
  3558. tool_name,
  3559. output,
  3560. is_error,
  3561. } => {
  3562. lines.push(format!(
  3563. "[tool_result id={tool_use_id} name={tool_name} error={is_error}] {output}"
  3564. ));
  3565. }
  3566. }
  3567. }
  3568. lines.push(String::new());
  3569. }
  3570. lines.join("\n")
  3571. }
  3572. fn default_export_filename(session: &Session) -> String {
  3573. let stem = session
  3574. .messages
  3575. .iter()
  3576. .find_map(|message| match message.role {
  3577. MessageRole::User => message.blocks.iter().find_map(|block| match block {
  3578. ContentBlock::Text { text } => Some(text.as_str()),
  3579. _ => None,
  3580. }),
  3581. _ => None,
  3582. })
  3583. .map_or("conversation", |text| {
  3584. text.lines().next().unwrap_or("conversation")
  3585. })
  3586. .chars()
  3587. .map(|ch| {
  3588. if ch.is_ascii_alphanumeric() {
  3589. ch.to_ascii_lowercase()
  3590. } else {
  3591. '-'
  3592. }
  3593. })
  3594. .collect::<String>()
  3595. .split('-')
  3596. .filter(|part| !part.is_empty())
  3597. .take(8)
  3598. .collect::<Vec<_>>()
  3599. .join("-");
  3600. let fallback = if stem.is_empty() {
  3601. "conversation"
  3602. } else {
  3603. &stem
  3604. };
  3605. format!("{fallback}.txt")
  3606. }
  3607. fn resolve_export_path(
  3608. requested_path: Option<&str>,
  3609. session: &Session,
  3610. ) -> Result<PathBuf, Box<dyn std::error::Error>> {
  3611. let cwd = env::current_dir()?;
  3612. let file_name =
  3613. requested_path.map_or_else(|| default_export_filename(session), ToOwned::to_owned);
  3614. let final_name = if Path::new(&file_name)
  3615. .extension()
  3616. .is_some_and(|ext| ext.eq_ignore_ascii_case("txt"))
  3617. {
  3618. file_name
  3619. } else {
  3620. format!("{file_name}.txt")
  3621. };
  3622. Ok(cwd.join(final_name))
  3623. }
  3624. fn build_system_prompt() -> Result<Vec<String>, Box<dyn std::error::Error>> {
  3625. Ok(load_system_prompt(
  3626. env::current_dir()?,
  3627. DEFAULT_DATE,
  3628. env::consts::OS,
  3629. "unknown",
  3630. )?)
  3631. }
  3632. fn build_runtime_plugin_state() -> Result<RuntimePluginState, Box<dyn std::error::Error>> {
  3633. let cwd = env::current_dir()?;
  3634. let loader = ConfigLoader::default_for(&cwd);
  3635. let runtime_config = loader.load()?;
  3636. build_runtime_plugin_state_with_loader(&cwd, &loader, &runtime_config)
  3637. }
  3638. fn build_runtime_plugin_state_with_loader(
  3639. cwd: &Path,
  3640. loader: &ConfigLoader,
  3641. runtime_config: &runtime::RuntimeConfig,
  3642. ) -> Result<RuntimePluginState, Box<dyn std::error::Error>> {
  3643. let plugin_manager = build_plugin_manager(cwd, loader, runtime_config);
  3644. let plugin_registry = plugin_manager.plugin_registry()?;
  3645. let plugin_hook_config =
  3646. runtime_hook_config_from_plugin_hooks(plugin_registry.aggregated_hooks()?);
  3647. let feature_config = runtime_config
  3648. .feature_config()
  3649. .clone()
  3650. .with_hooks(runtime_config.hooks().merged(&plugin_hook_config));
  3651. let (mcp_state, runtime_tools) = build_runtime_mcp_state(runtime_config)?;
  3652. let tool_registry = GlobalToolRegistry::with_plugin_tools(plugin_registry.aggregated_tools()?)?
  3653. .with_runtime_tools(runtime_tools)?;
  3654. Ok(RuntimePluginState {
  3655. feature_config,
  3656. tool_registry,
  3657. plugin_registry,
  3658. mcp_state,
  3659. })
  3660. }
  3661. fn build_plugin_manager(
  3662. cwd: &Path,
  3663. loader: &ConfigLoader,
  3664. runtime_config: &runtime::RuntimeConfig,
  3665. ) -> PluginManager {
  3666. let plugin_settings = runtime_config.plugins();
  3667. let mut plugin_config = PluginManagerConfig::new(loader.config_home().to_path_buf());
  3668. plugin_config.enabled_plugins = plugin_settings.enabled_plugins().clone();
  3669. plugin_config.external_dirs = plugin_settings
  3670. .external_directories()
  3671. .iter()
  3672. .map(|path| resolve_plugin_path(cwd, loader.config_home(), path))
  3673. .collect();
  3674. plugin_config.install_root = plugin_settings
  3675. .install_root()
  3676. .map(|path| resolve_plugin_path(cwd, loader.config_home(), path));
  3677. plugin_config.registry_path = plugin_settings
  3678. .registry_path()
  3679. .map(|path| resolve_plugin_path(cwd, loader.config_home(), path));
  3680. plugin_config.bundled_root = plugin_settings
  3681. .bundled_root()
  3682. .map(|path| resolve_plugin_path(cwd, loader.config_home(), path));
  3683. PluginManager::new(plugin_config)
  3684. }
  3685. fn resolve_plugin_path(cwd: &Path, config_home: &Path, value: &str) -> PathBuf {
  3686. let path = PathBuf::from(value);
  3687. if path.is_absolute() {
  3688. path
  3689. } else if value.starts_with('.') {
  3690. cwd.join(path)
  3691. } else {
  3692. config_home.join(path)
  3693. }
  3694. }
  3695. fn runtime_hook_config_from_plugin_hooks(hooks: PluginHooks) -> runtime::RuntimeHookConfig {
  3696. runtime::RuntimeHookConfig::new(
  3697. hooks.pre_tool_use,
  3698. hooks.post_tool_use,
  3699. hooks.post_tool_use_failure,
  3700. )
  3701. }
  3702. #[derive(Debug, Clone, PartialEq, Eq)]
  3703. struct InternalPromptProgressState {
  3704. command_label: &'static str,
  3705. task_label: String,
  3706. step: usize,
  3707. phase: String,
  3708. detail: Option<String>,
  3709. saw_final_text: bool,
  3710. }
  3711. #[derive(Debug, Clone, Copy, PartialEq, Eq)]
  3712. enum InternalPromptProgressEvent {
  3713. Started,
  3714. Update,
  3715. Heartbeat,
  3716. Complete,
  3717. Failed,
  3718. }
  3719. #[derive(Debug)]
  3720. struct InternalPromptProgressShared {
  3721. state: Mutex<InternalPromptProgressState>,
  3722. output_lock: Mutex<()>,
  3723. started_at: Instant,
  3724. }
  3725. #[derive(Debug, Clone)]
  3726. struct InternalPromptProgressReporter {
  3727. shared: Arc<InternalPromptProgressShared>,
  3728. }
  3729. #[derive(Debug)]
  3730. struct InternalPromptProgressRun {
  3731. reporter: InternalPromptProgressReporter,
  3732. heartbeat_stop: Option<mpsc::Sender<()>>,
  3733. heartbeat_handle: Option<thread::JoinHandle<()>>,
  3734. }
  3735. impl InternalPromptProgressReporter {
  3736. fn ultraplan(task: &str) -> Self {
  3737. Self {
  3738. shared: Arc::new(InternalPromptProgressShared {
  3739. state: Mutex::new(InternalPromptProgressState {
  3740. command_label: "Ultraplan",
  3741. task_label: task.to_string(),
  3742. step: 0,
  3743. phase: "planning started".to_string(),
  3744. detail: Some(format!("task: {task}")),
  3745. saw_final_text: false,
  3746. }),
  3747. output_lock: Mutex::new(()),
  3748. started_at: Instant::now(),
  3749. }),
  3750. }
  3751. }
  3752. fn emit(&self, event: InternalPromptProgressEvent, error: Option<&str>) {
  3753. let snapshot = self.snapshot();
  3754. let line = format_internal_prompt_progress_line(event, &snapshot, self.elapsed(), error);
  3755. self.write_line(&line);
  3756. }
  3757. fn mark_model_phase(&self) {
  3758. let snapshot = {
  3759. let mut state = self
  3760. .shared
  3761. .state
  3762. .lock()
  3763. .expect("internal prompt progress state poisoned");
  3764. state.step += 1;
  3765. state.phase = if state.step == 1 {
  3766. "analyzing request".to_string()
  3767. } else {
  3768. "reviewing findings".to_string()
  3769. };
  3770. state.detail = Some(format!("task: {}", state.task_label));
  3771. state.clone()
  3772. };
  3773. self.write_line(&format_internal_prompt_progress_line(
  3774. InternalPromptProgressEvent::Update,
  3775. &snapshot,
  3776. self.elapsed(),
  3777. None,
  3778. ));
  3779. }
  3780. fn mark_tool_phase(&self, name: &str, input: &str) {
  3781. let detail = describe_tool_progress(name, input);
  3782. let snapshot = {
  3783. let mut state = self
  3784. .shared
  3785. .state
  3786. .lock()
  3787. .expect("internal prompt progress state poisoned");
  3788. state.step += 1;
  3789. state.phase = format!("running {name}");
  3790. state.detail = Some(detail);
  3791. state.clone()
  3792. };
  3793. self.write_line(&format_internal_prompt_progress_line(
  3794. InternalPromptProgressEvent::Update,
  3795. &snapshot,
  3796. self.elapsed(),
  3797. None,
  3798. ));
  3799. }
  3800. fn mark_text_phase(&self, text: &str) {
  3801. let trimmed = text.trim();
  3802. if trimmed.is_empty() {
  3803. return;
  3804. }
  3805. let detail = truncate_for_summary(first_visible_line(trimmed), 120);
  3806. let snapshot = {
  3807. let mut state = self
  3808. .shared
  3809. .state
  3810. .lock()
  3811. .expect("internal prompt progress state poisoned");
  3812. if state.saw_final_text {
  3813. return;
  3814. }
  3815. state.saw_final_text = true;
  3816. state.step += 1;
  3817. state.phase = "drafting final plan".to_string();
  3818. state.detail = (!detail.is_empty()).then_some(detail);
  3819. state.clone()
  3820. };
  3821. self.write_line(&format_internal_prompt_progress_line(
  3822. InternalPromptProgressEvent::Update,
  3823. &snapshot,
  3824. self.elapsed(),
  3825. None,
  3826. ));
  3827. }
  3828. fn emit_heartbeat(&self) {
  3829. let snapshot = self.snapshot();
  3830. self.write_line(&format_internal_prompt_progress_line(
  3831. InternalPromptProgressEvent::Heartbeat,
  3832. &snapshot,
  3833. self.elapsed(),
  3834. None,
  3835. ));
  3836. }
  3837. fn snapshot(&self) -> InternalPromptProgressState {
  3838. self.shared
  3839. .state
  3840. .lock()
  3841. .expect("internal prompt progress state poisoned")
  3842. .clone()
  3843. }
  3844. fn elapsed(&self) -> Duration {
  3845. self.shared.started_at.elapsed()
  3846. }
  3847. fn write_line(&self, line: &str) {
  3848. let _guard = self
  3849. .shared
  3850. .output_lock
  3851. .lock()
  3852. .expect("internal prompt progress output lock poisoned");
  3853. let mut stdout = io::stdout();
  3854. let _ = writeln!(stdout, "{line}");
  3855. let _ = stdout.flush();
  3856. }
  3857. }
  3858. impl InternalPromptProgressRun {
  3859. fn start_ultraplan(task: &str) -> Self {
  3860. let reporter = InternalPromptProgressReporter::ultraplan(task);
  3861. reporter.emit(InternalPromptProgressEvent::Started, None);
  3862. let (heartbeat_stop, heartbeat_rx) = mpsc::channel();
  3863. let heartbeat_reporter = reporter.clone();
  3864. let heartbeat_handle = thread::spawn(move || loop {
  3865. match heartbeat_rx.recv_timeout(INTERNAL_PROGRESS_HEARTBEAT_INTERVAL) {
  3866. Ok(()) | Err(RecvTimeoutError::Disconnected) => break,
  3867. Err(RecvTimeoutError::Timeout) => heartbeat_reporter.emit_heartbeat(),
  3868. }
  3869. });
  3870. Self {
  3871. reporter,
  3872. heartbeat_stop: Some(heartbeat_stop),
  3873. heartbeat_handle: Some(heartbeat_handle),
  3874. }
  3875. }
  3876. fn reporter(&self) -> InternalPromptProgressReporter {
  3877. self.reporter.clone()
  3878. }
  3879. fn finish_success(&mut self) {
  3880. self.stop_heartbeat();
  3881. self.reporter
  3882. .emit(InternalPromptProgressEvent::Complete, None);
  3883. }
  3884. fn finish_failure(&mut self, error: &str) {
  3885. self.stop_heartbeat();
  3886. self.reporter
  3887. .emit(InternalPromptProgressEvent::Failed, Some(error));
  3888. }
  3889. fn stop_heartbeat(&mut self) {
  3890. if let Some(sender) = self.heartbeat_stop.take() {
  3891. let _ = sender.send(());
  3892. }
  3893. if let Some(handle) = self.heartbeat_handle.take() {
  3894. let _ = handle.join();
  3895. }
  3896. }
  3897. }
  3898. impl Drop for InternalPromptProgressRun {
  3899. fn drop(&mut self) {
  3900. self.stop_heartbeat();
  3901. }
  3902. }
  3903. fn format_internal_prompt_progress_line(
  3904. event: InternalPromptProgressEvent,
  3905. snapshot: &InternalPromptProgressState,
  3906. elapsed: Duration,
  3907. error: Option<&str>,
  3908. ) -> String {
  3909. let elapsed_seconds = elapsed.as_secs();
  3910. let step_label = if snapshot.step == 0 {
  3911. "current step pending".to_string()
  3912. } else {
  3913. format!("current step {}", snapshot.step)
  3914. };
  3915. let mut status_bits = vec![step_label, format!("phase {}", snapshot.phase)];
  3916. if let Some(detail) = snapshot
  3917. .detail
  3918. .as_deref()
  3919. .filter(|detail| !detail.is_empty())
  3920. {
  3921. status_bits.push(detail.to_string());
  3922. }
  3923. let status = status_bits.join(" · ");
  3924. match event {
  3925. InternalPromptProgressEvent::Started => {
  3926. format!(
  3927. "🧭 {} status · planning started · {status}",
  3928. snapshot.command_label
  3929. )
  3930. }
  3931. InternalPromptProgressEvent::Update => {
  3932. format!("… {} status · {status}", snapshot.command_label)
  3933. }
  3934. InternalPromptProgressEvent::Heartbeat => format!(
  3935. "… {} heartbeat · {elapsed_seconds}s elapsed · {status}",
  3936. snapshot.command_label
  3937. ),
  3938. InternalPromptProgressEvent::Complete => format!(
  3939. "✔ {} status · completed · {elapsed_seconds}s elapsed · {} steps total",
  3940. snapshot.command_label, snapshot.step
  3941. ),
  3942. InternalPromptProgressEvent::Failed => format!(
  3943. "✘ {} status · failed · {elapsed_seconds}s elapsed · {}",
  3944. snapshot.command_label,
  3945. error.unwrap_or("unknown error")
  3946. ),
  3947. }
  3948. }
  3949. fn describe_tool_progress(name: &str, input: &str) -> String {
  3950. let parsed: serde_json::Value =
  3951. serde_json::from_str(input).unwrap_or(serde_json::Value::String(input.to_string()));
  3952. match name {
  3953. "bash" | "Bash" => {
  3954. let command = parsed
  3955. .get("command")
  3956. .and_then(|value| value.as_str())
  3957. .unwrap_or_default();
  3958. if command.is_empty() {
  3959. "running shell command".to_string()
  3960. } else {
  3961. format!("command {}", truncate_for_summary(command.trim(), 100))
  3962. }
  3963. }
  3964. "read_file" | "Read" => format!("reading {}", extract_tool_path(&parsed)),
  3965. "write_file" | "Write" => format!("writing {}", extract_tool_path(&parsed)),
  3966. "edit_file" | "Edit" => format!("editing {}", extract_tool_path(&parsed)),
  3967. "glob_search" | "Glob" => {
  3968. let pattern = parsed
  3969. .get("pattern")
  3970. .and_then(|value| value.as_str())
  3971. .unwrap_or("?");
  3972. let scope = parsed
  3973. .get("path")
  3974. .and_then(|value| value.as_str())
  3975. .unwrap_or(".");
  3976. format!("glob `{pattern}` in {scope}")
  3977. }
  3978. "grep_search" | "Grep" => {
  3979. let pattern = parsed
  3980. .get("pattern")
  3981. .and_then(|value| value.as_str())
  3982. .unwrap_or("?");
  3983. let scope = parsed
  3984. .get("path")
  3985. .and_then(|value| value.as_str())
  3986. .unwrap_or(".");
  3987. format!("grep `{pattern}` in {scope}")
  3988. }
  3989. "web_search" | "WebSearch" => parsed
  3990. .get("query")
  3991. .and_then(|value| value.as_str())
  3992. .map_or_else(
  3993. || "running web search".to_string(),
  3994. |query| format!("query {}", truncate_for_summary(query, 100)),
  3995. ),
  3996. _ => {
  3997. let summary = summarize_tool_payload(input);
  3998. if summary.is_empty() {
  3999. format!("running {name}")
  4000. } else {
  4001. format!("{name}: {summary}")
  4002. }
  4003. }
  4004. }
  4005. }
  4006. #[allow(clippy::needless_pass_by_value)]
  4007. #[allow(clippy::too_many_arguments)]
  4008. fn build_runtime(
  4009. session: Session,
  4010. session_id: &str,
  4011. model: String,
  4012. system_prompt: Vec<String>,
  4013. enable_tools: bool,
  4014. emit_output: bool,
  4015. allowed_tools: Option<AllowedToolSet>,
  4016. permission_mode: PermissionMode,
  4017. progress_reporter: Option<InternalPromptProgressReporter>,
  4018. ) -> Result<BuiltRuntime, Box<dyn std::error::Error>> {
  4019. let runtime_plugin_state = build_runtime_plugin_state()?;
  4020. build_runtime_with_plugin_state(
  4021. session,
  4022. session_id,
  4023. model,
  4024. system_prompt,
  4025. enable_tools,
  4026. emit_output,
  4027. allowed_tools,
  4028. permission_mode,
  4029. progress_reporter,
  4030. runtime_plugin_state,
  4031. )
  4032. }
  4033. #[allow(clippy::needless_pass_by_value)]
  4034. #[allow(clippy::too_many_arguments)]
  4035. fn build_runtime_with_plugin_state(
  4036. session: Session,
  4037. session_id: &str,
  4038. model: String,
  4039. system_prompt: Vec<String>,
  4040. enable_tools: bool,
  4041. emit_output: bool,
  4042. allowed_tools: Option<AllowedToolSet>,
  4043. permission_mode: PermissionMode,
  4044. progress_reporter: Option<InternalPromptProgressReporter>,
  4045. runtime_plugin_state: RuntimePluginState,
  4046. ) -> Result<BuiltRuntime, Box<dyn std::error::Error>> {
  4047. let RuntimePluginState {
  4048. feature_config,
  4049. tool_registry,
  4050. plugin_registry,
  4051. mcp_state,
  4052. } = runtime_plugin_state;
  4053. plugin_registry.initialize()?;
  4054. let policy = permission_policy(permission_mode, &feature_config, &tool_registry)
  4055. .map_err(std::io::Error::other)?;
  4056. let mut runtime = ConversationRuntime::new_with_features(
  4057. session,
  4058. AnthropicRuntimeClient::new(
  4059. session_id,
  4060. model,
  4061. enable_tools,
  4062. emit_output,
  4063. allowed_tools.clone(),
  4064. tool_registry.clone(),
  4065. progress_reporter,
  4066. )?,
  4067. CliToolExecutor::new(
  4068. allowed_tools.clone(),
  4069. emit_output,
  4070. tool_registry.clone(),
  4071. mcp_state.clone(),
  4072. ),
  4073. policy,
  4074. system_prompt,
  4075. &feature_config,
  4076. );
  4077. if emit_output {
  4078. runtime = runtime.with_hook_progress_reporter(Box::new(CliHookProgressReporter));
  4079. }
  4080. Ok(BuiltRuntime::new(runtime, plugin_registry, mcp_state))
  4081. }
  4082. struct CliHookProgressReporter;
  4083. impl runtime::HookProgressReporter for CliHookProgressReporter {
  4084. fn on_event(&mut self, event: &runtime::HookProgressEvent) {
  4085. match event {
  4086. runtime::HookProgressEvent::Started {
  4087. event,
  4088. tool_name,
  4089. command,
  4090. } => eprintln!(
  4091. "[hook {event_name}] {tool_name}: {command}",
  4092. event_name = event.as_str()
  4093. ),
  4094. runtime::HookProgressEvent::Completed {
  4095. event,
  4096. tool_name,
  4097. command,
  4098. } => eprintln!(
  4099. "[hook done {event_name}] {tool_name}: {command}",
  4100. event_name = event.as_str()
  4101. ),
  4102. runtime::HookProgressEvent::Cancelled {
  4103. event,
  4104. tool_name,
  4105. command,
  4106. } => eprintln!(
  4107. "[hook cancelled {event_name}] {tool_name}: {command}",
  4108. event_name = event.as_str()
  4109. ),
  4110. }
  4111. }
  4112. }
  4113. struct CliPermissionPrompter {
  4114. current_mode: PermissionMode,
  4115. }
  4116. impl CliPermissionPrompter {
  4117. fn new(current_mode: PermissionMode) -> Self {
  4118. Self { current_mode }
  4119. }
  4120. }
  4121. impl runtime::PermissionPrompter for CliPermissionPrompter {
  4122. fn decide(
  4123. &mut self,
  4124. request: &runtime::PermissionRequest,
  4125. ) -> runtime::PermissionPromptDecision {
  4126. println!();
  4127. println!("Permission approval required");
  4128. println!(" Tool {}", request.tool_name);
  4129. println!(" Current mode {}", self.current_mode.as_str());
  4130. println!(" Required mode {}", request.required_mode.as_str());
  4131. if let Some(reason) = &request.reason {
  4132. println!(" Reason {reason}");
  4133. }
  4134. println!(" Input {}", request.input);
  4135. print!("Approve this tool call? [y/N]: ");
  4136. let _ = io::stdout().flush();
  4137. let mut response = String::new();
  4138. match io::stdin().read_line(&mut response) {
  4139. Ok(_) => {
  4140. let normalized = response.trim().to_ascii_lowercase();
  4141. if matches!(normalized.as_str(), "y" | "yes") {
  4142. runtime::PermissionPromptDecision::Allow
  4143. } else {
  4144. runtime::PermissionPromptDecision::Deny {
  4145. reason: format!(
  4146. "tool '{}' denied by user approval prompt",
  4147. request.tool_name
  4148. ),
  4149. }
  4150. }
  4151. }
  4152. Err(error) => runtime::PermissionPromptDecision::Deny {
  4153. reason: format!("permission approval failed: {error}"),
  4154. },
  4155. }
  4156. }
  4157. }
  4158. struct AnthropicRuntimeClient {
  4159. runtime: tokio::runtime::Runtime,
  4160. client: AnthropicClient,
  4161. model: String,
  4162. enable_tools: bool,
  4163. emit_output: bool,
  4164. allowed_tools: Option<AllowedToolSet>,
  4165. tool_registry: GlobalToolRegistry,
  4166. progress_reporter: Option<InternalPromptProgressReporter>,
  4167. }
  4168. impl AnthropicRuntimeClient {
  4169. fn new(
  4170. session_id: &str,
  4171. model: String,
  4172. enable_tools: bool,
  4173. emit_output: bool,
  4174. allowed_tools: Option<AllowedToolSet>,
  4175. tool_registry: GlobalToolRegistry,
  4176. progress_reporter: Option<InternalPromptProgressReporter>,
  4177. ) -> Result<Self, Box<dyn std::error::Error>> {
  4178. Ok(Self {
  4179. runtime: tokio::runtime::Runtime::new()?,
  4180. client: AnthropicClient::from_auth(resolve_cli_auth_source()?)
  4181. .with_base_url(api::read_base_url())
  4182. .with_prompt_cache(PromptCache::new(session_id)),
  4183. model,
  4184. enable_tools,
  4185. emit_output,
  4186. allowed_tools,
  4187. tool_registry,
  4188. progress_reporter,
  4189. })
  4190. }
  4191. }
  4192. fn resolve_cli_auth_source() -> Result<AuthSource, Box<dyn std::error::Error>> {
  4193. Ok(resolve_startup_auth_source(|| {
  4194. let cwd = env::current_dir().map_err(api::ApiError::from)?;
  4195. let config = ConfigLoader::default_for(&cwd).load().map_err(|error| {
  4196. api::ApiError::Auth(format!("failed to load runtime OAuth config: {error}"))
  4197. })?;
  4198. Ok(config.oauth().cloned())
  4199. })?)
  4200. }
  4201. impl ApiClient for AnthropicRuntimeClient {
  4202. #[allow(clippy::too_many_lines)]
  4203. fn stream(&mut self, request: ApiRequest) -> Result<Vec<AssistantEvent>, RuntimeError> {
  4204. if let Some(progress_reporter) = &self.progress_reporter {
  4205. progress_reporter.mark_model_phase();
  4206. }
  4207. let message_request = MessageRequest {
  4208. model: self.model.clone(),
  4209. max_tokens: max_tokens_for_model(&self.model),
  4210. messages: convert_messages(&request.messages),
  4211. system: (!request.system_prompt.is_empty()).then(|| request.system_prompt.join("\n\n")),
  4212. tools: self
  4213. .enable_tools
  4214. .then(|| filter_tool_specs(&self.tool_registry, self.allowed_tools.as_ref())),
  4215. tool_choice: self.enable_tools.then_some(ToolChoice::Auto),
  4216. stream: true,
  4217. };
  4218. self.runtime.block_on(async {
  4219. let mut stream = self
  4220. .client
  4221. .stream_message(&message_request)
  4222. .await
  4223. .map_err(|error| RuntimeError::new(error.to_string()))?;
  4224. let mut stdout = io::stdout();
  4225. let mut sink = io::sink();
  4226. let out: &mut dyn Write = if self.emit_output {
  4227. &mut stdout
  4228. } else {
  4229. &mut sink
  4230. };
  4231. let renderer = TerminalRenderer::new();
  4232. let mut markdown_stream = MarkdownStreamState::default();
  4233. let mut events = Vec::new();
  4234. let mut pending_tool: Option<(String, String, String)> = None;
  4235. let mut saw_stop = false;
  4236. while let Some(event) = stream
  4237. .next_event()
  4238. .await
  4239. .map_err(|error| RuntimeError::new(error.to_string()))?
  4240. {
  4241. match event {
  4242. ApiStreamEvent::MessageStart(start) => {
  4243. for block in start.message.content {
  4244. push_output_block(block, out, &mut events, &mut pending_tool, true)?;
  4245. }
  4246. }
  4247. ApiStreamEvent::ContentBlockStart(start) => {
  4248. push_output_block(
  4249. start.content_block,
  4250. out,
  4251. &mut events,
  4252. &mut pending_tool,
  4253. true,
  4254. )?;
  4255. }
  4256. ApiStreamEvent::ContentBlockDelta(delta) => match delta.delta {
  4257. ContentBlockDelta::TextDelta { text } => {
  4258. if !text.is_empty() {
  4259. if let Some(progress_reporter) = &self.progress_reporter {
  4260. progress_reporter.mark_text_phase(&text);
  4261. }
  4262. if let Some(rendered) = markdown_stream.push(&renderer, &text) {
  4263. write!(out, "{rendered}")
  4264. .and_then(|()| out.flush())
  4265. .map_err(|error| RuntimeError::new(error.to_string()))?;
  4266. }
  4267. events.push(AssistantEvent::TextDelta(text));
  4268. }
  4269. }
  4270. ContentBlockDelta::InputJsonDelta { partial_json } => {
  4271. if let Some((_, _, input)) = &mut pending_tool {
  4272. input.push_str(&partial_json);
  4273. }
  4274. }
  4275. ContentBlockDelta::ThinkingDelta { .. }
  4276. | ContentBlockDelta::SignatureDelta { .. } => {}
  4277. },
  4278. ApiStreamEvent::ContentBlockStop(_) => {
  4279. if let Some(rendered) = markdown_stream.flush(&renderer) {
  4280. write!(out, "{rendered}")
  4281. .and_then(|()| out.flush())
  4282. .map_err(|error| RuntimeError::new(error.to_string()))?;
  4283. }
  4284. if let Some((id, name, input)) = pending_tool.take() {
  4285. if let Some(progress_reporter) = &self.progress_reporter {
  4286. progress_reporter.mark_tool_phase(&name, &input);
  4287. }
  4288. // Display tool call now that input is fully accumulated
  4289. writeln!(out, "\n{}", format_tool_call_start(&name, &input))
  4290. .and_then(|()| out.flush())
  4291. .map_err(|error| RuntimeError::new(error.to_string()))?;
  4292. events.push(AssistantEvent::ToolUse { id, name, input });
  4293. }
  4294. }
  4295. ApiStreamEvent::MessageDelta(delta) => {
  4296. events.push(AssistantEvent::Usage(delta.usage.token_usage()));
  4297. }
  4298. ApiStreamEvent::MessageStop(_) => {
  4299. saw_stop = true;
  4300. if let Some(rendered) = markdown_stream.flush(&renderer) {
  4301. write!(out, "{rendered}")
  4302. .and_then(|()| out.flush())
  4303. .map_err(|error| RuntimeError::new(error.to_string()))?;
  4304. }
  4305. events.push(AssistantEvent::MessageStop);
  4306. }
  4307. }
  4308. }
  4309. push_prompt_cache_record(&self.client, &mut events);
  4310. if !saw_stop
  4311. && events.iter().any(|event| {
  4312. matches!(event, AssistantEvent::TextDelta(text) if !text.is_empty())
  4313. || matches!(event, AssistantEvent::ToolUse { .. })
  4314. })
  4315. {
  4316. events.push(AssistantEvent::MessageStop);
  4317. }
  4318. if events
  4319. .iter()
  4320. .any(|event| matches!(event, AssistantEvent::MessageStop))
  4321. {
  4322. return Ok(events);
  4323. }
  4324. let response = self
  4325. .client
  4326. .send_message(&MessageRequest {
  4327. stream: false,
  4328. ..message_request.clone()
  4329. })
  4330. .await
  4331. .map_err(|error| RuntimeError::new(error.to_string()))?;
  4332. let mut events = response_to_events(response, out)?;
  4333. push_prompt_cache_record(&self.client, &mut events);
  4334. Ok(events)
  4335. })
  4336. }
  4337. }
  4338. fn final_assistant_text(summary: &runtime::TurnSummary) -> String {
  4339. summary
  4340. .assistant_messages
  4341. .last()
  4342. .map(|message| {
  4343. message
  4344. .blocks
  4345. .iter()
  4346. .filter_map(|block| match block {
  4347. ContentBlock::Text { text } => Some(text.as_str()),
  4348. _ => None,
  4349. })
  4350. .collect::<Vec<_>>()
  4351. .join("")
  4352. })
  4353. .unwrap_or_default()
  4354. }
  4355. fn collect_tool_uses(summary: &runtime::TurnSummary) -> Vec<serde_json::Value> {
  4356. summary
  4357. .assistant_messages
  4358. .iter()
  4359. .flat_map(|message| message.blocks.iter())
  4360. .filter_map(|block| match block {
  4361. ContentBlock::ToolUse { id, name, input } => Some(json!({
  4362. "id": id,
  4363. "name": name,
  4364. "input": input,
  4365. })),
  4366. _ => None,
  4367. })
  4368. .collect()
  4369. }
  4370. fn collect_tool_results(summary: &runtime::TurnSummary) -> Vec<serde_json::Value> {
  4371. summary
  4372. .tool_results
  4373. .iter()
  4374. .flat_map(|message| message.blocks.iter())
  4375. .filter_map(|block| match block {
  4376. ContentBlock::ToolResult {
  4377. tool_use_id,
  4378. tool_name,
  4379. output,
  4380. is_error,
  4381. } => Some(json!({
  4382. "tool_use_id": tool_use_id,
  4383. "tool_name": tool_name,
  4384. "output": output,
  4385. "is_error": is_error,
  4386. })),
  4387. _ => None,
  4388. })
  4389. .collect()
  4390. }
  4391. fn collect_prompt_cache_events(summary: &runtime::TurnSummary) -> Vec<serde_json::Value> {
  4392. summary
  4393. .prompt_cache_events
  4394. .iter()
  4395. .map(|event| {
  4396. json!({
  4397. "unexpected": event.unexpected,
  4398. "reason": event.reason,
  4399. "previous_cache_read_input_tokens": event.previous_cache_read_input_tokens,
  4400. "current_cache_read_input_tokens": event.current_cache_read_input_tokens,
  4401. "token_drop": event.token_drop,
  4402. })
  4403. })
  4404. .collect()
  4405. }
  4406. fn slash_command_completion_candidates_with_sessions(
  4407. model: &str,
  4408. active_session_id: Option<&str>,
  4409. recent_session_ids: Vec<String>,
  4410. ) -> Vec<String> {
  4411. let mut completions = BTreeSet::new();
  4412. for spec in slash_command_specs() {
  4413. completions.insert(format!("/{}", spec.name));
  4414. for alias in spec.aliases {
  4415. completions.insert(format!("/{alias}"));
  4416. }
  4417. }
  4418. for candidate in [
  4419. "/bughunter ",
  4420. "/clear --confirm",
  4421. "/config ",
  4422. "/config env",
  4423. "/config hooks",
  4424. "/config model",
  4425. "/config plugins",
  4426. "/mcp ",
  4427. "/mcp list",
  4428. "/mcp show ",
  4429. "/export ",
  4430. "/issue ",
  4431. "/model ",
  4432. "/model opus",
  4433. "/model sonnet",
  4434. "/model haiku",
  4435. "/permissions ",
  4436. "/permissions read-only",
  4437. "/permissions workspace-write",
  4438. "/permissions danger-full-access",
  4439. "/plugin list",
  4440. "/plugin install ",
  4441. "/plugin enable ",
  4442. "/plugin disable ",
  4443. "/plugin uninstall ",
  4444. "/plugin update ",
  4445. "/plugins list",
  4446. "/pr ",
  4447. "/resume ",
  4448. "/session list",
  4449. "/session switch ",
  4450. "/session fork ",
  4451. "/teleport ",
  4452. "/ultraplan ",
  4453. "/agents help",
  4454. "/mcp help",
  4455. "/skills help",
  4456. ] {
  4457. completions.insert(candidate.to_string());
  4458. }
  4459. if !model.trim().is_empty() {
  4460. completions.insert(format!("/model {}", resolve_model_alias(model)));
  4461. completions.insert(format!("/model {model}"));
  4462. }
  4463. if let Some(active_session_id) = active_session_id.filter(|value| !value.trim().is_empty()) {
  4464. completions.insert(format!("/resume {active_session_id}"));
  4465. completions.insert(format!("/session switch {active_session_id}"));
  4466. }
  4467. for session_id in recent_session_ids
  4468. .into_iter()
  4469. .filter(|value| !value.trim().is_empty())
  4470. .take(10)
  4471. {
  4472. completions.insert(format!("/resume {session_id}"));
  4473. completions.insert(format!("/session switch {session_id}"));
  4474. }
  4475. completions.into_iter().collect()
  4476. }
  4477. fn format_tool_call_start(name: &str, input: &str) -> String {
  4478. let parsed: serde_json::Value =
  4479. serde_json::from_str(input).unwrap_or(serde_json::Value::String(input.to_string()));
  4480. let detail = match name {
  4481. "bash" | "Bash" => format_bash_call(&parsed),
  4482. "read_file" | "Read" => {
  4483. let path = extract_tool_path(&parsed);
  4484. format!("\x1b[2m📄 Reading {path}…\x1b[0m")
  4485. }
  4486. "write_file" | "Write" => {
  4487. let path = extract_tool_path(&parsed);
  4488. let lines = parsed
  4489. .get("content")
  4490. .and_then(|value| value.as_str())
  4491. .map_or(0, |content| content.lines().count());
  4492. format!("\x1b[1;32m✏️ Writing {path}\x1b[0m \x1b[2m({lines} lines)\x1b[0m")
  4493. }
  4494. "edit_file" | "Edit" => {
  4495. let path = extract_tool_path(&parsed);
  4496. let old_value = parsed
  4497. .get("old_string")
  4498. .or_else(|| parsed.get("oldString"))
  4499. .and_then(|value| value.as_str())
  4500. .unwrap_or_default();
  4501. let new_value = parsed
  4502. .get("new_string")
  4503. .or_else(|| parsed.get("newString"))
  4504. .and_then(|value| value.as_str())
  4505. .unwrap_or_default();
  4506. format!(
  4507. "\x1b[1;33m📝 Editing {path}\x1b[0m{}",
  4508. format_patch_preview(old_value, new_value)
  4509. .map(|preview| format!("\n{preview}"))
  4510. .unwrap_or_default()
  4511. )
  4512. }
  4513. "glob_search" | "Glob" => format_search_start("🔎 Glob", &parsed),
  4514. "grep_search" | "Grep" => format_search_start("🔎 Grep", &parsed),
  4515. "web_search" | "WebSearch" => parsed
  4516. .get("query")
  4517. .and_then(|value| value.as_str())
  4518. .unwrap_or("?")
  4519. .to_string(),
  4520. _ => summarize_tool_payload(input),
  4521. };
  4522. let border = "─".repeat(name.len() + 8);
  4523. format!(
  4524. "\x1b[38;5;245m╭─ \x1b[1;36m{name}\x1b[0;38;5;245m ─╮\x1b[0m\n\x1b[38;5;245m│\x1b[0m {detail}\n\x1b[38;5;245m╰{border}╯\x1b[0m"
  4525. )
  4526. }
  4527. fn format_tool_result(name: &str, output: &str, is_error: bool) -> String {
  4528. let icon = if is_error {
  4529. "\x1b[1;31m✗\x1b[0m"
  4530. } else {
  4531. "\x1b[1;32m✓\x1b[0m"
  4532. };
  4533. if is_error {
  4534. let summary = truncate_for_summary(output.trim(), 160);
  4535. return if summary.is_empty() {
  4536. format!("{icon} \x1b[38;5;245m{name}\x1b[0m")
  4537. } else {
  4538. format!("{icon} \x1b[38;5;245m{name}\x1b[0m\n\x1b[38;5;203m{summary}\x1b[0m")
  4539. };
  4540. }
  4541. let parsed: serde_json::Value =
  4542. serde_json::from_str(output).unwrap_or(serde_json::Value::String(output.to_string()));
  4543. match name {
  4544. "bash" | "Bash" => format_bash_result(icon, &parsed),
  4545. "read_file" | "Read" => format_read_result(icon, &parsed),
  4546. "write_file" | "Write" => format_write_result(icon, &parsed),
  4547. "edit_file" | "Edit" => format_edit_result(icon, &parsed),
  4548. "glob_search" | "Glob" => format_glob_result(icon, &parsed),
  4549. "grep_search" | "Grep" => format_grep_result(icon, &parsed),
  4550. _ => format_generic_tool_result(icon, name, &parsed),
  4551. }
  4552. }
  4553. const DISPLAY_TRUNCATION_NOTICE: &str =
  4554. "\x1b[2m… output truncated for display; full result preserved in session.\x1b[0m";
  4555. const READ_DISPLAY_MAX_LINES: usize = 80;
  4556. const READ_DISPLAY_MAX_CHARS: usize = 6_000;
  4557. const TOOL_OUTPUT_DISPLAY_MAX_LINES: usize = 60;
  4558. const TOOL_OUTPUT_DISPLAY_MAX_CHARS: usize = 4_000;
  4559. fn extract_tool_path(parsed: &serde_json::Value) -> String {
  4560. parsed
  4561. .get("file_path")
  4562. .or_else(|| parsed.get("filePath"))
  4563. .or_else(|| parsed.get("path"))
  4564. .and_then(|value| value.as_str())
  4565. .unwrap_or("?")
  4566. .to_string()
  4567. }
  4568. fn format_search_start(label: &str, parsed: &serde_json::Value) -> String {
  4569. let pattern = parsed
  4570. .get("pattern")
  4571. .and_then(|value| value.as_str())
  4572. .unwrap_or("?");
  4573. let scope = parsed
  4574. .get("path")
  4575. .and_then(|value| value.as_str())
  4576. .unwrap_or(".");
  4577. format!("{label} {pattern}\n\x1b[2min {scope}\x1b[0m")
  4578. }
  4579. fn format_patch_preview(old_value: &str, new_value: &str) -> Option<String> {
  4580. if old_value.is_empty() && new_value.is_empty() {
  4581. return None;
  4582. }
  4583. Some(format!(
  4584. "\x1b[38;5;203m- {}\x1b[0m\n\x1b[38;5;70m+ {}\x1b[0m",
  4585. truncate_for_summary(first_visible_line(old_value), 72),
  4586. truncate_for_summary(first_visible_line(new_value), 72)
  4587. ))
  4588. }
  4589. fn format_bash_call(parsed: &serde_json::Value) -> String {
  4590. let command = parsed
  4591. .get("command")
  4592. .and_then(|value| value.as_str())
  4593. .unwrap_or_default();
  4594. if command.is_empty() {
  4595. String::new()
  4596. } else {
  4597. format!(
  4598. "\x1b[48;5;236;38;5;255m $ {} \x1b[0m",
  4599. truncate_for_summary(command, 160)
  4600. )
  4601. }
  4602. }
  4603. fn first_visible_line(text: &str) -> &str {
  4604. text.lines()
  4605. .find(|line| !line.trim().is_empty())
  4606. .unwrap_or(text)
  4607. }
  4608. fn format_bash_result(icon: &str, parsed: &serde_json::Value) -> String {
  4609. use std::fmt::Write as _;
  4610. let mut lines = vec![format!("{icon} \x1b[38;5;245mbash\x1b[0m")];
  4611. if let Some(task_id) = parsed
  4612. .get("backgroundTaskId")
  4613. .and_then(|value| value.as_str())
  4614. {
  4615. write!(&mut lines[0], " backgrounded ({task_id})").expect("write to string");
  4616. } else if let Some(status) = parsed
  4617. .get("returnCodeInterpretation")
  4618. .and_then(|value| value.as_str())
  4619. .filter(|status| !status.is_empty())
  4620. {
  4621. write!(&mut lines[0], " {status}").expect("write to string");
  4622. }
  4623. if let Some(stdout) = parsed.get("stdout").and_then(|value| value.as_str()) {
  4624. if !stdout.trim().is_empty() {
  4625. lines.push(truncate_output_for_display(
  4626. stdout,
  4627. TOOL_OUTPUT_DISPLAY_MAX_LINES,
  4628. TOOL_OUTPUT_DISPLAY_MAX_CHARS,
  4629. ));
  4630. }
  4631. }
  4632. if let Some(stderr) = parsed.get("stderr").and_then(|value| value.as_str()) {
  4633. if !stderr.trim().is_empty() {
  4634. lines.push(format!(
  4635. "\x1b[38;5;203m{}\x1b[0m",
  4636. truncate_output_for_display(
  4637. stderr,
  4638. TOOL_OUTPUT_DISPLAY_MAX_LINES,
  4639. TOOL_OUTPUT_DISPLAY_MAX_CHARS,
  4640. )
  4641. ));
  4642. }
  4643. }
  4644. lines.join("\n\n")
  4645. }
  4646. fn format_read_result(icon: &str, parsed: &serde_json::Value) -> String {
  4647. let file = parsed.get("file").unwrap_or(parsed);
  4648. let path = extract_tool_path(file);
  4649. let start_line = file
  4650. .get("startLine")
  4651. .and_then(serde_json::Value::as_u64)
  4652. .unwrap_or(1);
  4653. let num_lines = file
  4654. .get("numLines")
  4655. .and_then(serde_json::Value::as_u64)
  4656. .unwrap_or(0);
  4657. let total_lines = file
  4658. .get("totalLines")
  4659. .and_then(serde_json::Value::as_u64)
  4660. .unwrap_or(num_lines);
  4661. let content = file
  4662. .get("content")
  4663. .and_then(|value| value.as_str())
  4664. .unwrap_or_default();
  4665. let end_line = start_line.saturating_add(num_lines.saturating_sub(1));
  4666. format!(
  4667. "{icon} \x1b[2m📄 Read {path} (lines {}-{} of {})\x1b[0m\n{}",
  4668. start_line,
  4669. end_line.max(start_line),
  4670. total_lines,
  4671. truncate_output_for_display(content, READ_DISPLAY_MAX_LINES, READ_DISPLAY_MAX_CHARS)
  4672. )
  4673. }
  4674. fn format_write_result(icon: &str, parsed: &serde_json::Value) -> String {
  4675. let path = extract_tool_path(parsed);
  4676. let kind = parsed
  4677. .get("type")
  4678. .and_then(|value| value.as_str())
  4679. .unwrap_or("write");
  4680. let line_count = parsed
  4681. .get("content")
  4682. .and_then(|value| value.as_str())
  4683. .map_or(0, |content| content.lines().count());
  4684. format!(
  4685. "{icon} \x1b[1;32m✏️ {} {path}\x1b[0m \x1b[2m({line_count} lines)\x1b[0m",
  4686. if kind == "create" { "Wrote" } else { "Updated" },
  4687. )
  4688. }
  4689. fn format_structured_patch_preview(parsed: &serde_json::Value) -> Option<String> {
  4690. let hunks = parsed.get("structuredPatch")?.as_array()?;
  4691. let mut preview = Vec::new();
  4692. for hunk in hunks.iter().take(2) {
  4693. let lines = hunk.get("lines")?.as_array()?;
  4694. for line in lines.iter().filter_map(|value| value.as_str()).take(6) {
  4695. match line.chars().next() {
  4696. Some('+') => preview.push(format!("\x1b[38;5;70m{line}\x1b[0m")),
  4697. Some('-') => preview.push(format!("\x1b[38;5;203m{line}\x1b[0m")),
  4698. _ => preview.push(line.to_string()),
  4699. }
  4700. }
  4701. }
  4702. if preview.is_empty() {
  4703. None
  4704. } else {
  4705. Some(preview.join("\n"))
  4706. }
  4707. }
  4708. fn format_edit_result(icon: &str, parsed: &serde_json::Value) -> String {
  4709. let path = extract_tool_path(parsed);
  4710. let suffix = if parsed
  4711. .get("replaceAll")
  4712. .and_then(serde_json::Value::as_bool)
  4713. .unwrap_or(false)
  4714. {
  4715. " (replace all)"
  4716. } else {
  4717. ""
  4718. };
  4719. let preview = format_structured_patch_preview(parsed).or_else(|| {
  4720. let old_value = parsed
  4721. .get("oldString")
  4722. .and_then(|value| value.as_str())
  4723. .unwrap_or_default();
  4724. let new_value = parsed
  4725. .get("newString")
  4726. .and_then(|value| value.as_str())
  4727. .unwrap_or_default();
  4728. format_patch_preview(old_value, new_value)
  4729. });
  4730. match preview {
  4731. Some(preview) => format!("{icon} \x1b[1;33m📝 Edited {path}{suffix}\x1b[0m\n{preview}"),
  4732. None => format!("{icon} \x1b[1;33m📝 Edited {path}{suffix}\x1b[0m"),
  4733. }
  4734. }
  4735. fn format_glob_result(icon: &str, parsed: &serde_json::Value) -> String {
  4736. let num_files = parsed
  4737. .get("numFiles")
  4738. .and_then(serde_json::Value::as_u64)
  4739. .unwrap_or(0);
  4740. let filenames = parsed
  4741. .get("filenames")
  4742. .and_then(|value| value.as_array())
  4743. .map(|files| {
  4744. files
  4745. .iter()
  4746. .filter_map(|value| value.as_str())
  4747. .take(8)
  4748. .collect::<Vec<_>>()
  4749. .join("\n")
  4750. })
  4751. .unwrap_or_default();
  4752. if filenames.is_empty() {
  4753. format!("{icon} \x1b[38;5;245mglob_search\x1b[0m matched {num_files} files")
  4754. } else {
  4755. format!("{icon} \x1b[38;5;245mglob_search\x1b[0m matched {num_files} files\n{filenames}")
  4756. }
  4757. }
  4758. fn format_grep_result(icon: &str, parsed: &serde_json::Value) -> String {
  4759. let num_matches = parsed
  4760. .get("numMatches")
  4761. .and_then(serde_json::Value::as_u64)
  4762. .unwrap_or(0);
  4763. let num_files = parsed
  4764. .get("numFiles")
  4765. .and_then(serde_json::Value::as_u64)
  4766. .unwrap_or(0);
  4767. let content = parsed
  4768. .get("content")
  4769. .and_then(|value| value.as_str())
  4770. .unwrap_or_default();
  4771. let filenames = parsed
  4772. .get("filenames")
  4773. .and_then(|value| value.as_array())
  4774. .map(|files| {
  4775. files
  4776. .iter()
  4777. .filter_map(|value| value.as_str())
  4778. .take(8)
  4779. .collect::<Vec<_>>()
  4780. .join("\n")
  4781. })
  4782. .unwrap_or_default();
  4783. let summary = format!(
  4784. "{icon} \x1b[38;5;245mgrep_search\x1b[0m {num_matches} matches across {num_files} files"
  4785. );
  4786. if !content.trim().is_empty() {
  4787. format!(
  4788. "{summary}\n{}",
  4789. truncate_output_for_display(
  4790. content,
  4791. TOOL_OUTPUT_DISPLAY_MAX_LINES,
  4792. TOOL_OUTPUT_DISPLAY_MAX_CHARS,
  4793. )
  4794. )
  4795. } else if !filenames.is_empty() {
  4796. format!("{summary}\n{filenames}")
  4797. } else {
  4798. summary
  4799. }
  4800. }
  4801. fn format_generic_tool_result(icon: &str, name: &str, parsed: &serde_json::Value) -> String {
  4802. let rendered_output = match parsed {
  4803. serde_json::Value::String(text) => text.clone(),
  4804. serde_json::Value::Null => String::new(),
  4805. serde_json::Value::Object(_) | serde_json::Value::Array(_) => {
  4806. serde_json::to_string_pretty(parsed).unwrap_or_else(|_| parsed.to_string())
  4807. }
  4808. _ => parsed.to_string(),
  4809. };
  4810. let preview = truncate_output_for_display(
  4811. &rendered_output,
  4812. TOOL_OUTPUT_DISPLAY_MAX_LINES,
  4813. TOOL_OUTPUT_DISPLAY_MAX_CHARS,
  4814. );
  4815. if preview.is_empty() {
  4816. format!("{icon} \x1b[38;5;245m{name}\x1b[0m")
  4817. } else if preview.contains('\n') {
  4818. format!("{icon} \x1b[38;5;245m{name}\x1b[0m\n{preview}")
  4819. } else {
  4820. format!("{icon} \x1b[38;5;245m{name}:\x1b[0m {preview}")
  4821. }
  4822. }
  4823. fn summarize_tool_payload(payload: &str) -> String {
  4824. let compact = match serde_json::from_str::<serde_json::Value>(payload) {
  4825. Ok(value) => value.to_string(),
  4826. Err(_) => payload.trim().to_string(),
  4827. };
  4828. truncate_for_summary(&compact, 96)
  4829. }
  4830. fn truncate_for_summary(value: &str, limit: usize) -> String {
  4831. let mut chars = value.chars();
  4832. let truncated = chars.by_ref().take(limit).collect::<String>();
  4833. if chars.next().is_some() {
  4834. format!("{truncated}…")
  4835. } else {
  4836. truncated
  4837. }
  4838. }
  4839. fn truncate_output_for_display(content: &str, max_lines: usize, max_chars: usize) -> String {
  4840. let original = content.trim_end_matches('\n');
  4841. if original.is_empty() {
  4842. return String::new();
  4843. }
  4844. let mut preview_lines = Vec::new();
  4845. let mut used_chars = 0usize;
  4846. let mut truncated = false;
  4847. for (index, line) in original.lines().enumerate() {
  4848. if index >= max_lines {
  4849. truncated = true;
  4850. break;
  4851. }
  4852. let newline_cost = usize::from(!preview_lines.is_empty());
  4853. let available = max_chars.saturating_sub(used_chars + newline_cost);
  4854. if available == 0 {
  4855. truncated = true;
  4856. break;
  4857. }
  4858. let line_chars = line.chars().count();
  4859. if line_chars > available {
  4860. preview_lines.push(line.chars().take(available).collect::<String>());
  4861. truncated = true;
  4862. break;
  4863. }
  4864. preview_lines.push(line.to_string());
  4865. used_chars += newline_cost + line_chars;
  4866. }
  4867. let mut preview = preview_lines.join("\n");
  4868. if truncated {
  4869. if !preview.is_empty() {
  4870. preview.push('\n');
  4871. }
  4872. preview.push_str(DISPLAY_TRUNCATION_NOTICE);
  4873. }
  4874. preview
  4875. }
  4876. fn push_output_block(
  4877. block: OutputContentBlock,
  4878. out: &mut (impl Write + ?Sized),
  4879. events: &mut Vec<AssistantEvent>,
  4880. pending_tool: &mut Option<(String, String, String)>,
  4881. streaming_tool_input: bool,
  4882. ) -> Result<(), RuntimeError> {
  4883. match block {
  4884. OutputContentBlock::Text { text } => {
  4885. if !text.is_empty() {
  4886. let rendered = TerminalRenderer::new().markdown_to_ansi(&text);
  4887. write!(out, "{rendered}")
  4888. .and_then(|()| out.flush())
  4889. .map_err(|error| RuntimeError::new(error.to_string()))?;
  4890. events.push(AssistantEvent::TextDelta(text));
  4891. }
  4892. }
  4893. OutputContentBlock::ToolUse { id, name, input } => {
  4894. // During streaming, the initial content_block_start has an empty input ({}).
  4895. // The real input arrives via input_json_delta events. In
  4896. // non-streaming responses, preserve a legitimate empty object.
  4897. let initial_input = if streaming_tool_input
  4898. && input.is_object()
  4899. && input.as_object().is_some_and(serde_json::Map::is_empty)
  4900. {
  4901. String::new()
  4902. } else {
  4903. input.to_string()
  4904. };
  4905. *pending_tool = Some((id, name, initial_input));
  4906. }
  4907. OutputContentBlock::Thinking { .. } | OutputContentBlock::RedactedThinking { .. } => {}
  4908. }
  4909. Ok(())
  4910. }
  4911. fn response_to_events(
  4912. response: MessageResponse,
  4913. out: &mut (impl Write + ?Sized),
  4914. ) -> Result<Vec<AssistantEvent>, RuntimeError> {
  4915. let mut events = Vec::new();
  4916. let mut pending_tool = None;
  4917. for block in response.content {
  4918. push_output_block(block, out, &mut events, &mut pending_tool, false)?;
  4919. if let Some((id, name, input)) = pending_tool.take() {
  4920. events.push(AssistantEvent::ToolUse { id, name, input });
  4921. }
  4922. }
  4923. events.push(AssistantEvent::Usage(response.usage.token_usage()));
  4924. events.push(AssistantEvent::MessageStop);
  4925. Ok(events)
  4926. }
  4927. fn push_prompt_cache_record(client: &AnthropicClient, events: &mut Vec<AssistantEvent>) {
  4928. if let Some(record) = client.take_last_prompt_cache_record() {
  4929. if let Some(event) = prompt_cache_record_to_runtime_event(record) {
  4930. events.push(AssistantEvent::PromptCache(event));
  4931. }
  4932. }
  4933. }
  4934. fn prompt_cache_record_to_runtime_event(
  4935. record: api::PromptCacheRecord,
  4936. ) -> Option<PromptCacheEvent> {
  4937. let cache_break = record.cache_break?;
  4938. Some(PromptCacheEvent {
  4939. unexpected: cache_break.unexpected,
  4940. reason: cache_break.reason,
  4941. previous_cache_read_input_tokens: cache_break.previous_cache_read_input_tokens,
  4942. current_cache_read_input_tokens: cache_break.current_cache_read_input_tokens,
  4943. token_drop: cache_break.token_drop,
  4944. })
  4945. }
  4946. struct CliToolExecutor {
  4947. renderer: TerminalRenderer,
  4948. emit_output: bool,
  4949. allowed_tools: Option<AllowedToolSet>,
  4950. tool_registry: GlobalToolRegistry,
  4951. mcp_state: Option<Arc<Mutex<RuntimeMcpState>>>,
  4952. }
  4953. impl CliToolExecutor {
  4954. fn new(
  4955. allowed_tools: Option<AllowedToolSet>,
  4956. emit_output: bool,
  4957. tool_registry: GlobalToolRegistry,
  4958. mcp_state: Option<Arc<Mutex<RuntimeMcpState>>>,
  4959. ) -> Self {
  4960. Self {
  4961. renderer: TerminalRenderer::new(),
  4962. emit_output,
  4963. allowed_tools,
  4964. tool_registry,
  4965. mcp_state,
  4966. }
  4967. }
  4968. fn execute_search_tool(&self, value: serde_json::Value) -> Result<String, ToolError> {
  4969. let input: ToolSearchRequest = serde_json::from_value(value)
  4970. .map_err(|error| ToolError::new(format!("invalid tool input JSON: {error}")))?;
  4971. let (pending_mcp_servers, mcp_degraded) = self
  4972. .mcp_state
  4973. .as_ref()
  4974. .map(|state| {
  4975. let state = state
  4976. .lock()
  4977. .unwrap_or_else(std::sync::PoisonError::into_inner);
  4978. (state.pending_servers(), state.degraded_report())
  4979. })
  4980. .unwrap_or((None, None));
  4981. serde_json::to_string_pretty(&self.tool_registry.search(
  4982. &input.query,
  4983. input.max_results.unwrap_or(5),
  4984. pending_mcp_servers,
  4985. mcp_degraded,
  4986. ))
  4987. .map_err(|error| ToolError::new(error.to_string()))
  4988. }
  4989. fn execute_runtime_tool(
  4990. &self,
  4991. tool_name: &str,
  4992. value: serde_json::Value,
  4993. ) -> Result<String, ToolError> {
  4994. let Some(mcp_state) = &self.mcp_state else {
  4995. return Err(ToolError::new(format!(
  4996. "runtime tool `{tool_name}` is unavailable without configured MCP servers"
  4997. )));
  4998. };
  4999. let mut mcp_state = mcp_state
  5000. .lock()
  5001. .unwrap_or_else(std::sync::PoisonError::into_inner);
  5002. match tool_name {
  5003. "MCPTool" => {
  5004. let input: McpToolRequest = serde_json::from_value(value)
  5005. .map_err(|error| ToolError::new(format!("invalid tool input JSON: {error}")))?;
  5006. let qualified_name = input
  5007. .qualified_name
  5008. .or(input.tool)
  5009. .ok_or_else(|| ToolError::new("missing required field `qualifiedName`"))?;
  5010. mcp_state.call_tool(&qualified_name, input.arguments)
  5011. }
  5012. "ListMcpResourcesTool" => {
  5013. let input: ListMcpResourcesRequest = serde_json::from_value(value)
  5014. .map_err(|error| ToolError::new(format!("invalid tool input JSON: {error}")))?;
  5015. match input.server {
  5016. Some(server_name) => mcp_state.list_resources_for_server(&server_name),
  5017. None => mcp_state.list_resources_for_all_servers(),
  5018. }
  5019. }
  5020. "ReadMcpResourceTool" => {
  5021. let input: ReadMcpResourceRequest = serde_json::from_value(value)
  5022. .map_err(|error| ToolError::new(format!("invalid tool input JSON: {error}")))?;
  5023. mcp_state.read_resource(&input.server, &input.uri)
  5024. }
  5025. _ => mcp_state.call_tool(tool_name, Some(value)),
  5026. }
  5027. }
  5028. }
  5029. impl ToolExecutor for CliToolExecutor {
  5030. fn execute(&mut self, tool_name: &str, input: &str) -> Result<String, ToolError> {
  5031. if self
  5032. .allowed_tools
  5033. .as_ref()
  5034. .is_some_and(|allowed| !allowed.contains(tool_name))
  5035. {
  5036. return Err(ToolError::new(format!(
  5037. "tool `{tool_name}` is not enabled by the current --allowedTools setting"
  5038. )));
  5039. }
  5040. let value = serde_json::from_str(input)
  5041. .map_err(|error| ToolError::new(format!("invalid tool input JSON: {error}")))?;
  5042. let result = if tool_name == "ToolSearch" {
  5043. self.execute_search_tool(value)
  5044. } else if self.tool_registry.has_runtime_tool(tool_name) {
  5045. self.execute_runtime_tool(tool_name, value)
  5046. } else {
  5047. self.tool_registry
  5048. .execute(tool_name, &value)
  5049. .map_err(ToolError::new)
  5050. };
  5051. match result {
  5052. Ok(output) => {
  5053. if self.emit_output {
  5054. let markdown = format_tool_result(tool_name, &output, false);
  5055. self.renderer
  5056. .stream_markdown(&markdown, &mut io::stdout())
  5057. .map_err(|error| ToolError::new(error.to_string()))?;
  5058. }
  5059. Ok(output)
  5060. }
  5061. Err(error) => {
  5062. if self.emit_output {
  5063. let markdown = format_tool_result(tool_name, &error.to_string(), true);
  5064. self.renderer
  5065. .stream_markdown(&markdown, &mut io::stdout())
  5066. .map_err(|stream_error| ToolError::new(stream_error.to_string()))?;
  5067. }
  5068. Err(error)
  5069. }
  5070. }
  5071. }
  5072. }
  5073. fn permission_policy(
  5074. mode: PermissionMode,
  5075. feature_config: &runtime::RuntimeFeatureConfig,
  5076. tool_registry: &GlobalToolRegistry,
  5077. ) -> Result<PermissionPolicy, String> {
  5078. Ok(tool_registry.permission_specs(None)?.into_iter().fold(
  5079. PermissionPolicy::new(mode).with_permission_rules(feature_config.permission_rules()),
  5080. |policy, (name, required_permission)| {
  5081. policy.with_tool_requirement(name, required_permission)
  5082. },
  5083. ))
  5084. }
  5085. fn convert_messages(messages: &[ConversationMessage]) -> Vec<InputMessage> {
  5086. messages
  5087. .iter()
  5088. .filter_map(|message| {
  5089. let role = match message.role {
  5090. MessageRole::System | MessageRole::User | MessageRole::Tool => "user",
  5091. MessageRole::Assistant => "assistant",
  5092. };
  5093. let content = message
  5094. .blocks
  5095. .iter()
  5096. .map(|block| match block {
  5097. ContentBlock::Text { text } => InputContentBlock::Text { text: text.clone() },
  5098. ContentBlock::ToolUse { id, name, input } => InputContentBlock::ToolUse {
  5099. id: id.clone(),
  5100. name: name.clone(),
  5101. input: serde_json::from_str(input)
  5102. .unwrap_or_else(|_| serde_json::json!({ "raw": input })),
  5103. },
  5104. ContentBlock::ToolResult {
  5105. tool_use_id,
  5106. output,
  5107. is_error,
  5108. ..
  5109. } => InputContentBlock::ToolResult {
  5110. tool_use_id: tool_use_id.clone(),
  5111. content: vec![ToolResultContentBlock::Text {
  5112. text: output.clone(),
  5113. }],
  5114. is_error: *is_error,
  5115. },
  5116. })
  5117. .collect::<Vec<_>>();
  5118. (!content.is_empty()).then(|| InputMessage {
  5119. role: role.to_string(),
  5120. content,
  5121. })
  5122. })
  5123. .collect()
  5124. }
  5125. #[allow(clippy::too_many_lines)]
  5126. fn print_help_to(out: &mut impl Write) -> io::Result<()> {
  5127. writeln!(out, "claw v{VERSION}")?;
  5128. writeln!(out)?;
  5129. writeln!(out, "Usage:")?;
  5130. writeln!(
  5131. out,
  5132. " claw [--model MODEL] [--allowedTools TOOL[,TOOL...]]"
  5133. )?;
  5134. writeln!(out, " Start the interactive REPL")?;
  5135. writeln!(
  5136. out,
  5137. " claw [--model MODEL] [--output-format text|json] prompt TEXT"
  5138. )?;
  5139. writeln!(out, " Send one prompt and exit")?;
  5140. writeln!(
  5141. out,
  5142. " claw [--model MODEL] [--output-format text|json] TEXT"
  5143. )?;
  5144. writeln!(out, " Shorthand non-interactive prompt mode")?;
  5145. writeln!(
  5146. out,
  5147. " claw --resume [SESSION.jsonl|session-id|latest] [/status] [/compact] [...]"
  5148. )?;
  5149. writeln!(
  5150. out,
  5151. " Inspect or maintain a saved session without entering the REPL"
  5152. )?;
  5153. writeln!(out, " claw help")?;
  5154. writeln!(out, " Alias for --help")?;
  5155. writeln!(out, " claw version")?;
  5156. writeln!(out, " Alias for --version")?;
  5157. writeln!(out, " claw status")?;
  5158. writeln!(
  5159. out,
  5160. " Show the current local workspace status snapshot"
  5161. )?;
  5162. writeln!(out, " claw sandbox")?;
  5163. writeln!(out, " Show the current sandbox isolation snapshot")?;
  5164. writeln!(out, " claw dump-manifests")?;
  5165. writeln!(out, " claw bootstrap-plan")?;
  5166. writeln!(out, " claw agents")?;
  5167. writeln!(out, " claw mcp")?;
  5168. writeln!(out, " claw skills")?;
  5169. writeln!(out, " claw system-prompt [--cwd PATH] [--date YYYY-MM-DD]")?;
  5170. writeln!(out, " claw login")?;
  5171. writeln!(out, " claw logout")?;
  5172. writeln!(out, " claw init")?;
  5173. writeln!(out)?;
  5174. writeln!(out, "Flags:")?;
  5175. writeln!(
  5176. out,
  5177. " --model MODEL Override the active model"
  5178. )?;
  5179. writeln!(
  5180. out,
  5181. " --output-format FORMAT Non-interactive output format: text or json"
  5182. )?;
  5183. writeln!(
  5184. out,
  5185. " --permission-mode MODE Set read-only, workspace-write, or danger-full-access"
  5186. )?;
  5187. writeln!(
  5188. out,
  5189. " --dangerously-skip-permissions Skip all permission checks"
  5190. )?;
  5191. writeln!(out, " --allowedTools TOOLS Restrict enabled tools (repeatable; comma-separated aliases supported)")?;
  5192. writeln!(
  5193. out,
  5194. " --version, -V Print version and build information locally"
  5195. )?;
  5196. writeln!(out)?;
  5197. writeln!(out, "Interactive slash commands:")?;
  5198. writeln!(out, "{}", render_slash_command_help())?;
  5199. writeln!(out)?;
  5200. let resume_commands = resume_supported_slash_commands()
  5201. .into_iter()
  5202. .map(|spec| match spec.argument_hint {
  5203. Some(argument_hint) => format!("/{} {}", spec.name, argument_hint),
  5204. None => format!("/{}", spec.name),
  5205. })
  5206. .collect::<Vec<_>>()
  5207. .join(", ");
  5208. writeln!(out, "Resume-safe commands: {resume_commands}")?;
  5209. writeln!(out)?;
  5210. writeln!(out, "Session shortcuts:")?;
  5211. writeln!(
  5212. out,
  5213. " REPL turns auto-save to .claw/sessions/<session-id>.{PRIMARY_SESSION_EXTENSION}"
  5214. )?;
  5215. writeln!(
  5216. out,
  5217. " Use `{LATEST_SESSION_REFERENCE}` with --resume, /resume, or /session switch to target the newest saved session"
  5218. )?;
  5219. writeln!(
  5220. out,
  5221. " Use /session list in the REPL to browse managed sessions"
  5222. )?;
  5223. writeln!(out, "Examples:")?;
  5224. writeln!(out, " claw --model claude-opus \"summarize this repo\"")?;
  5225. writeln!(
  5226. out,
  5227. " claw --output-format json prompt \"explain src/main.rs\""
  5228. )?;
  5229. writeln!(
  5230. out,
  5231. " claw --allowedTools read,glob \"summarize Cargo.toml\""
  5232. )?;
  5233. writeln!(out, " claw --resume {LATEST_SESSION_REFERENCE}")?;
  5234. writeln!(
  5235. out,
  5236. " claw --resume {LATEST_SESSION_REFERENCE} /status /diff /export notes.txt"
  5237. )?;
  5238. writeln!(out, " claw agents")?;
  5239. writeln!(out, " claw mcp show my-server")?;
  5240. writeln!(out, " claw /skills")?;
  5241. writeln!(out, " claw login")?;
  5242. writeln!(out, " claw init")?;
  5243. Ok(())
  5244. }
  5245. fn print_help() {
  5246. let _ = print_help_to(&mut io::stdout());
  5247. }
  5248. #[cfg(test)]
  5249. mod tests {
  5250. use super::{
  5251. build_runtime_plugin_state_with_loader, build_runtime_with_plugin_state,
  5252. create_managed_session_handle, describe_tool_progress, filter_tool_specs,
  5253. format_bughunter_report, format_commit_preflight_report, format_commit_skipped_report,
  5254. format_compact_report, format_cost_report, format_internal_prompt_progress_line,
  5255. format_issue_report, format_model_report, format_model_switch_report,
  5256. format_permissions_report, format_permissions_switch_report, format_pr_report,
  5257. format_resume_report, format_status_report, format_tool_call_start, format_tool_result,
  5258. format_ultraplan_report, format_unknown_slash_command,
  5259. format_unknown_slash_command_message, normalize_permission_mode, parse_args,
  5260. parse_git_status_branch, parse_git_status_metadata_for, parse_git_workspace_summary,
  5261. permission_policy, print_help_to, push_output_block, render_config_report,
  5262. render_diff_report, render_diff_report_for, render_memory_report, render_repl_help,
  5263. render_resume_usage, resolve_model_alias, resolve_session_reference, response_to_events,
  5264. resume_supported_slash_commands, run_resume_command,
  5265. slash_command_completion_candidates_with_sessions, status_context, validate_no_args,
  5266. write_mcp_server_fixture, CliAction, CliOutputFormat, CliToolExecutor, GitWorkspaceSummary,
  5267. InternalPromptProgressEvent, InternalPromptProgressState, LiveCli, SlashCommand,
  5268. StatusUsage, DEFAULT_MODEL,
  5269. };
  5270. use api::{MessageResponse, OutputContentBlock, Usage};
  5271. use plugins::{
  5272. PluginManager, PluginManagerConfig, PluginTool, PluginToolDefinition, PluginToolPermission,
  5273. };
  5274. use runtime::{
  5275. AssistantEvent, ConfigLoader, ContentBlock, ConversationMessage, MessageRole,
  5276. PermissionMode, Session, ToolExecutor,
  5277. };
  5278. use serde_json::json;
  5279. use std::fs;
  5280. use std::path::{Path, PathBuf};
  5281. use std::process::Command;
  5282. use std::sync::{Mutex, MutexGuard, OnceLock};
  5283. use std::time::{Duration, SystemTime, UNIX_EPOCH};
  5284. use tools::GlobalToolRegistry;
  5285. fn registry_with_plugin_tool() -> GlobalToolRegistry {
  5286. GlobalToolRegistry::with_plugin_tools(vec![PluginTool::new(
  5287. "plugin-demo@external",
  5288. "plugin-demo",
  5289. PluginToolDefinition {
  5290. name: "plugin_echo".to_string(),
  5291. description: Some("Echo plugin payload".to_string()),
  5292. input_schema: json!({
  5293. "type": "object",
  5294. "properties": {
  5295. "message": { "type": "string" }
  5296. },
  5297. "required": ["message"],
  5298. "additionalProperties": false
  5299. }),
  5300. },
  5301. "echo".to_string(),
  5302. Vec::new(),
  5303. PluginToolPermission::WorkspaceWrite,
  5304. None,
  5305. )])
  5306. .expect("plugin tool registry should build")
  5307. }
  5308. fn temp_dir() -> PathBuf {
  5309. let nanos = SystemTime::now()
  5310. .duration_since(UNIX_EPOCH)
  5311. .expect("time should be after epoch")
  5312. .as_nanos();
  5313. std::env::temp_dir().join(format!("rusty-claude-cli-{nanos}"))
  5314. }
  5315. fn git(args: &[&str], cwd: &Path) {
  5316. let status = Command::new("git")
  5317. .args(args)
  5318. .current_dir(cwd)
  5319. .status()
  5320. .expect("git command should run");
  5321. assert!(
  5322. status.success(),
  5323. "git command failed: git {}",
  5324. args.join(" ")
  5325. );
  5326. }
  5327. fn env_lock() -> MutexGuard<'static, ()> {
  5328. static LOCK: OnceLock<Mutex<()>> = OnceLock::new();
  5329. LOCK.get_or_init(|| Mutex::new(()))
  5330. .lock()
  5331. .unwrap_or_else(std::sync::PoisonError::into_inner)
  5332. }
  5333. fn with_current_dir<T>(cwd: &Path, f: impl FnOnce() -> T) -> T {
  5334. let previous = std::env::current_dir().expect("cwd should load");
  5335. std::env::set_current_dir(cwd).expect("cwd should change");
  5336. let result = f();
  5337. std::env::set_current_dir(previous).expect("cwd should restore");
  5338. result
  5339. }
  5340. fn write_plugin_fixture(root: &Path, name: &str, include_hooks: bool, include_lifecycle: bool) {
  5341. fs::create_dir_all(root.join(".claude-plugin")).expect("manifest dir");
  5342. if include_hooks {
  5343. fs::create_dir_all(root.join("hooks")).expect("hooks dir");
  5344. fs::write(
  5345. root.join("hooks").join("pre.sh"),
  5346. "#!/bin/sh\nprintf 'plugin pre hook'\n",
  5347. )
  5348. .expect("write hook");
  5349. }
  5350. if include_lifecycle {
  5351. fs::create_dir_all(root.join("lifecycle")).expect("lifecycle dir");
  5352. fs::write(
  5353. root.join("lifecycle").join("init.sh"),
  5354. "#!/bin/sh\nprintf 'init\\n' >> lifecycle.log\n",
  5355. )
  5356. .expect("write init lifecycle");
  5357. fs::write(
  5358. root.join("lifecycle").join("shutdown.sh"),
  5359. "#!/bin/sh\nprintf 'shutdown\\n' >> lifecycle.log\n",
  5360. )
  5361. .expect("write shutdown lifecycle");
  5362. }
  5363. let hooks = if include_hooks {
  5364. ",\n \"hooks\": {\n \"PreToolUse\": [\"./hooks/pre.sh\"]\n }"
  5365. } else {
  5366. ""
  5367. };
  5368. let lifecycle = if include_lifecycle {
  5369. ",\n \"lifecycle\": {\n \"Init\": [\"./lifecycle/init.sh\"],\n \"Shutdown\": [\"./lifecycle/shutdown.sh\"]\n }"
  5370. } else {
  5371. ""
  5372. };
  5373. fs::write(
  5374. root.join(".claude-plugin").join("plugin.json"),
  5375. format!(
  5376. "{{\n \"name\": \"{name}\",\n \"version\": \"1.0.0\",\n \"description\": \"runtime plugin fixture\"{hooks}{lifecycle}\n}}"
  5377. ),
  5378. )
  5379. .expect("write plugin manifest");
  5380. }
  5381. #[test]
  5382. fn defaults_to_repl_when_no_args() {
  5383. let _guard = env_lock();
  5384. std::env::remove_var("RUSTY_CLAUDE_PERMISSION_MODE");
  5385. assert_eq!(
  5386. parse_args(&[]).expect("args should parse"),
  5387. CliAction::Repl {
  5388. model: DEFAULT_MODEL.to_string(),
  5389. allowed_tools: None,
  5390. permission_mode: PermissionMode::DangerFullAccess,
  5391. }
  5392. );
  5393. }
  5394. #[test]
  5395. fn default_permission_mode_uses_project_config_when_env_is_unset() {
  5396. let _guard = env_lock();
  5397. let root = temp_dir();
  5398. let cwd = root.join("project");
  5399. let config_home = root.join("config-home");
  5400. std::fs::create_dir_all(cwd.join(".claw")).expect("project config dir should exist");
  5401. std::fs::create_dir_all(&config_home).expect("config home should exist");
  5402. std::fs::write(
  5403. cwd.join(".claw").join("settings.json"),
  5404. r#"{"permissionMode":"acceptEdits"}"#,
  5405. )
  5406. .expect("project config should write");
  5407. let original_config_home = std::env::var("CLAW_CONFIG_HOME").ok();
  5408. let original_permission_mode = std::env::var("RUSTY_CLAUDE_PERMISSION_MODE").ok();
  5409. std::env::set_var("CLAW_CONFIG_HOME", &config_home);
  5410. std::env::remove_var("RUSTY_CLAUDE_PERMISSION_MODE");
  5411. let resolved = with_current_dir(&cwd, super::default_permission_mode);
  5412. match original_config_home {
  5413. Some(value) => std::env::set_var("CLAW_CONFIG_HOME", value),
  5414. None => std::env::remove_var("CLAW_CONFIG_HOME"),
  5415. }
  5416. match original_permission_mode {
  5417. Some(value) => std::env::set_var("RUSTY_CLAUDE_PERMISSION_MODE", value),
  5418. None => std::env::remove_var("RUSTY_CLAUDE_PERMISSION_MODE"),
  5419. }
  5420. std::fs::remove_dir_all(root).expect("temp config root should clean up");
  5421. assert_eq!(resolved, PermissionMode::WorkspaceWrite);
  5422. }
  5423. #[test]
  5424. fn env_permission_mode_overrides_project_config_default() {
  5425. let _guard = env_lock();
  5426. let root = temp_dir();
  5427. let cwd = root.join("project");
  5428. let config_home = root.join("config-home");
  5429. std::fs::create_dir_all(cwd.join(".claw")).expect("project config dir should exist");
  5430. std::fs::create_dir_all(&config_home).expect("config home should exist");
  5431. std::fs::write(
  5432. cwd.join(".claw").join("settings.json"),
  5433. r#"{"permissionMode":"acceptEdits"}"#,
  5434. )
  5435. .expect("project config should write");
  5436. let original_config_home = std::env::var("CLAW_CONFIG_HOME").ok();
  5437. let original_permission_mode = std::env::var("RUSTY_CLAUDE_PERMISSION_MODE").ok();
  5438. std::env::set_var("CLAW_CONFIG_HOME", &config_home);
  5439. std::env::set_var("RUSTY_CLAUDE_PERMISSION_MODE", "read-only");
  5440. let resolved = with_current_dir(&cwd, super::default_permission_mode);
  5441. match original_config_home {
  5442. Some(value) => std::env::set_var("CLAW_CONFIG_HOME", value),
  5443. None => std::env::remove_var("CLAW_CONFIG_HOME"),
  5444. }
  5445. match original_permission_mode {
  5446. Some(value) => std::env::set_var("RUSTY_CLAUDE_PERMISSION_MODE", value),
  5447. None => std::env::remove_var("RUSTY_CLAUDE_PERMISSION_MODE"),
  5448. }
  5449. std::fs::remove_dir_all(root).expect("temp config root should clean up");
  5450. assert_eq!(resolved, PermissionMode::ReadOnly);
  5451. }
  5452. #[test]
  5453. fn parses_prompt_subcommand() {
  5454. let _guard = env_lock();
  5455. std::env::remove_var("RUSTY_CLAUDE_PERMISSION_MODE");
  5456. let args = vec![
  5457. "prompt".to_string(),
  5458. "hello".to_string(),
  5459. "world".to_string(),
  5460. ];
  5461. assert_eq!(
  5462. parse_args(&args).expect("args should parse"),
  5463. CliAction::Prompt {
  5464. prompt: "hello world".to_string(),
  5465. model: DEFAULT_MODEL.to_string(),
  5466. output_format: CliOutputFormat::Text,
  5467. allowed_tools: None,
  5468. permission_mode: PermissionMode::DangerFullAccess,
  5469. }
  5470. );
  5471. }
  5472. #[test]
  5473. fn parses_bare_prompt_and_json_output_flag() {
  5474. let _guard = env_lock();
  5475. std::env::remove_var("RUSTY_CLAUDE_PERMISSION_MODE");
  5476. let args = vec![
  5477. "--output-format=json".to_string(),
  5478. "--model".to_string(),
  5479. "claude-opus".to_string(),
  5480. "explain".to_string(),
  5481. "this".to_string(),
  5482. ];
  5483. assert_eq!(
  5484. parse_args(&args).expect("args should parse"),
  5485. CliAction::Prompt {
  5486. prompt: "explain this".to_string(),
  5487. model: "claude-opus".to_string(),
  5488. output_format: CliOutputFormat::Json,
  5489. allowed_tools: None,
  5490. permission_mode: PermissionMode::DangerFullAccess,
  5491. }
  5492. );
  5493. }
  5494. #[test]
  5495. fn resolves_model_aliases_in_args() {
  5496. let _guard = env_lock();
  5497. std::env::remove_var("RUSTY_CLAUDE_PERMISSION_MODE");
  5498. let args = vec![
  5499. "--model".to_string(),
  5500. "opus".to_string(),
  5501. "explain".to_string(),
  5502. "this".to_string(),
  5503. ];
  5504. assert_eq!(
  5505. parse_args(&args).expect("args should parse"),
  5506. CliAction::Prompt {
  5507. prompt: "explain this".to_string(),
  5508. model: "claude-opus-4-6".to_string(),
  5509. output_format: CliOutputFormat::Text,
  5510. allowed_tools: None,
  5511. permission_mode: PermissionMode::DangerFullAccess,
  5512. }
  5513. );
  5514. }
  5515. #[test]
  5516. fn resolves_known_model_aliases() {
  5517. assert_eq!(resolve_model_alias("opus"), "claude-opus-4-6");
  5518. assert_eq!(resolve_model_alias("sonnet"), "claude-sonnet-4-6");
  5519. assert_eq!(resolve_model_alias("haiku"), "claude-haiku-4-5-20251213");
  5520. assert_eq!(resolve_model_alias("claude-opus"), "claude-opus");
  5521. }
  5522. #[test]
  5523. fn parses_version_flags_without_initializing_prompt_mode() {
  5524. assert_eq!(
  5525. parse_args(&["--version".to_string()]).expect("args should parse"),
  5526. CliAction::Version
  5527. );
  5528. assert_eq!(
  5529. parse_args(&["-V".to_string()]).expect("args should parse"),
  5530. CliAction::Version
  5531. );
  5532. }
  5533. #[test]
  5534. fn parses_permission_mode_flag() {
  5535. let args = vec!["--permission-mode=read-only".to_string()];
  5536. assert_eq!(
  5537. parse_args(&args).expect("args should parse"),
  5538. CliAction::Repl {
  5539. model: DEFAULT_MODEL.to_string(),
  5540. allowed_tools: None,
  5541. permission_mode: PermissionMode::ReadOnly,
  5542. }
  5543. );
  5544. }
  5545. #[test]
  5546. fn parses_allowed_tools_flags_with_aliases_and_lists() {
  5547. let _guard = env_lock();
  5548. std::env::remove_var("RUSTY_CLAUDE_PERMISSION_MODE");
  5549. let args = vec![
  5550. "--allowedTools".to_string(),
  5551. "read,glob".to_string(),
  5552. "--allowed-tools=write_file".to_string(),
  5553. ];
  5554. assert_eq!(
  5555. parse_args(&args).expect("args should parse"),
  5556. CliAction::Repl {
  5557. model: DEFAULT_MODEL.to_string(),
  5558. allowed_tools: Some(
  5559. ["glob_search", "read_file", "write_file"]
  5560. .into_iter()
  5561. .map(str::to_string)
  5562. .collect()
  5563. ),
  5564. permission_mode: PermissionMode::DangerFullAccess,
  5565. }
  5566. );
  5567. }
  5568. #[test]
  5569. fn rejects_unknown_allowed_tools() {
  5570. let error = parse_args(&["--allowedTools".to_string(), "teleport".to_string()])
  5571. .expect_err("tool should be rejected");
  5572. assert!(error.contains("unsupported tool in --allowedTools: teleport"));
  5573. }
  5574. #[test]
  5575. fn parses_system_prompt_options() {
  5576. let args = vec![
  5577. "system-prompt".to_string(),
  5578. "--cwd".to_string(),
  5579. "/tmp/project".to_string(),
  5580. "--date".to_string(),
  5581. "2026-04-01".to_string(),
  5582. ];
  5583. assert_eq!(
  5584. parse_args(&args).expect("args should parse"),
  5585. CliAction::PrintSystemPrompt {
  5586. cwd: PathBuf::from("/tmp/project"),
  5587. date: "2026-04-01".to_string(),
  5588. }
  5589. );
  5590. }
  5591. #[test]
  5592. fn parses_login_and_logout_subcommands() {
  5593. assert_eq!(
  5594. parse_args(&["login".to_string()]).expect("login should parse"),
  5595. CliAction::Login
  5596. );
  5597. assert_eq!(
  5598. parse_args(&["logout".to_string()]).expect("logout should parse"),
  5599. CliAction::Logout
  5600. );
  5601. assert_eq!(
  5602. parse_args(&["init".to_string()]).expect("init should parse"),
  5603. CliAction::Init
  5604. );
  5605. assert_eq!(
  5606. parse_args(&["agents".to_string()]).expect("agents should parse"),
  5607. CliAction::Agents { args: None }
  5608. );
  5609. assert_eq!(
  5610. parse_args(&["mcp".to_string()]).expect("mcp should parse"),
  5611. CliAction::Mcp { args: None }
  5612. );
  5613. assert_eq!(
  5614. parse_args(&["skills".to_string()]).expect("skills should parse"),
  5615. CliAction::Skills { args: None }
  5616. );
  5617. assert_eq!(
  5618. parse_args(&["agents".to_string(), "--help".to_string()])
  5619. .expect("agents help should parse"),
  5620. CliAction::Agents {
  5621. args: Some("--help".to_string())
  5622. }
  5623. );
  5624. }
  5625. #[test]
  5626. fn parses_single_word_command_aliases_without_falling_back_to_prompt_mode() {
  5627. let _guard = env_lock();
  5628. std::env::remove_var("RUSTY_CLAUDE_PERMISSION_MODE");
  5629. assert_eq!(
  5630. parse_args(&["help".to_string()]).expect("help should parse"),
  5631. CliAction::Help
  5632. );
  5633. assert_eq!(
  5634. parse_args(&["version".to_string()]).expect("version should parse"),
  5635. CliAction::Version
  5636. );
  5637. assert_eq!(
  5638. parse_args(&["status".to_string()]).expect("status should parse"),
  5639. CliAction::Status {
  5640. model: DEFAULT_MODEL.to_string(),
  5641. permission_mode: PermissionMode::DangerFullAccess,
  5642. }
  5643. );
  5644. assert_eq!(
  5645. parse_args(&["sandbox".to_string()]).expect("sandbox should parse"),
  5646. CliAction::Sandbox
  5647. );
  5648. }
  5649. #[test]
  5650. fn single_word_slash_command_names_return_guidance_instead_of_hitting_prompt_mode() {
  5651. let error = parse_args(&["cost".to_string()]).expect_err("cost should return guidance");
  5652. assert!(error.contains("slash command"));
  5653. assert!(error.contains("/cost"));
  5654. }
  5655. #[test]
  5656. fn multi_word_prompt_still_uses_shorthand_prompt_mode() {
  5657. let _guard = env_lock();
  5658. std::env::remove_var("RUSTY_CLAUDE_PERMISSION_MODE");
  5659. assert_eq!(
  5660. parse_args(&["help".to_string(), "me".to_string(), "debug".to_string()])
  5661. .expect("prompt shorthand should still work"),
  5662. CliAction::Prompt {
  5663. prompt: "help me debug".to_string(),
  5664. model: DEFAULT_MODEL.to_string(),
  5665. output_format: CliOutputFormat::Text,
  5666. allowed_tools: None,
  5667. permission_mode: PermissionMode::DangerFullAccess,
  5668. }
  5669. );
  5670. }
  5671. #[test]
  5672. fn parses_direct_agents_mcp_and_skills_slash_commands() {
  5673. assert_eq!(
  5674. parse_args(&["/agents".to_string()]).expect("/agents should parse"),
  5675. CliAction::Agents { args: None }
  5676. );
  5677. assert_eq!(
  5678. parse_args(&["/mcp".to_string(), "show".to_string(), "demo".to_string()])
  5679. .expect("/mcp show demo should parse"),
  5680. CliAction::Mcp {
  5681. args: Some("show demo".to_string())
  5682. }
  5683. );
  5684. assert_eq!(
  5685. parse_args(&["/skills".to_string()]).expect("/skills should parse"),
  5686. CliAction::Skills { args: None }
  5687. );
  5688. assert_eq!(
  5689. parse_args(&["/skills".to_string(), "help".to_string()])
  5690. .expect("/skills help should parse"),
  5691. CliAction::Skills {
  5692. args: Some("help".to_string())
  5693. }
  5694. );
  5695. assert_eq!(
  5696. parse_args(&[
  5697. "/skills".to_string(),
  5698. "install".to_string(),
  5699. "./fixtures/help-skill".to_string(),
  5700. ])
  5701. .expect("/skills install should parse"),
  5702. CliAction::Skills {
  5703. args: Some("install ./fixtures/help-skill".to_string())
  5704. }
  5705. );
  5706. let error = parse_args(&["/status".to_string()])
  5707. .expect_err("/status should remain REPL-only when invoked directly");
  5708. assert!(error.contains("interactive-only"));
  5709. assert!(error.contains("claw --resume SESSION.jsonl /status"));
  5710. }
  5711. #[test]
  5712. fn direct_slash_commands_surface_shared_validation_errors() {
  5713. let compact_error = parse_args(&["/compact".to_string(), "now".to_string()])
  5714. .expect_err("invalid /compact shape should be rejected");
  5715. assert!(compact_error.contains("Unexpected arguments for /compact."));
  5716. assert!(compact_error.contains("Usage /compact"));
  5717. let plugins_error = parse_args(&[
  5718. "/plugins".to_string(),
  5719. "list".to_string(),
  5720. "extra".to_string(),
  5721. ])
  5722. .expect_err("invalid /plugins list shape should be rejected");
  5723. assert!(plugins_error.contains("Usage: /plugin list"));
  5724. assert!(plugins_error.contains("Aliases /plugins, /marketplace"));
  5725. }
  5726. #[test]
  5727. fn formats_unknown_slash_command_with_suggestions() {
  5728. let report = format_unknown_slash_command_message("statsu");
  5729. assert!(report.contains("unknown slash command: /statsu"));
  5730. assert!(report.contains("Did you mean"));
  5731. assert!(report.contains("Use /help"));
  5732. }
  5733. #[test]
  5734. fn parses_resume_flag_with_slash_command() {
  5735. let args = vec![
  5736. "--resume".to_string(),
  5737. "session.jsonl".to_string(),
  5738. "/compact".to_string(),
  5739. ];
  5740. assert_eq!(
  5741. parse_args(&args).expect("args should parse"),
  5742. CliAction::ResumeSession {
  5743. session_path: PathBuf::from("session.jsonl"),
  5744. commands: vec!["/compact".to_string()],
  5745. }
  5746. );
  5747. }
  5748. #[test]
  5749. fn parses_resume_flag_without_path_as_latest_session() {
  5750. assert_eq!(
  5751. parse_args(&["--resume".to_string()]).expect("args should parse"),
  5752. CliAction::ResumeSession {
  5753. session_path: PathBuf::from("latest"),
  5754. commands: vec![],
  5755. }
  5756. );
  5757. assert_eq!(
  5758. parse_args(&["--resume".to_string(), "/status".to_string()])
  5759. .expect("resume shortcut should parse"),
  5760. CliAction::ResumeSession {
  5761. session_path: PathBuf::from("latest"),
  5762. commands: vec!["/status".to_string()],
  5763. }
  5764. );
  5765. }
  5766. #[test]
  5767. fn parses_resume_flag_with_multiple_slash_commands() {
  5768. let args = vec![
  5769. "--resume".to_string(),
  5770. "session.jsonl".to_string(),
  5771. "/status".to_string(),
  5772. "/compact".to_string(),
  5773. "/cost".to_string(),
  5774. ];
  5775. assert_eq!(
  5776. parse_args(&args).expect("args should parse"),
  5777. CliAction::ResumeSession {
  5778. session_path: PathBuf::from("session.jsonl"),
  5779. commands: vec![
  5780. "/status".to_string(),
  5781. "/compact".to_string(),
  5782. "/cost".to_string(),
  5783. ],
  5784. }
  5785. );
  5786. }
  5787. #[test]
  5788. fn rejects_unknown_options_with_helpful_guidance() {
  5789. let error = parse_args(&["--resum".to_string()]).expect_err("unknown option should fail");
  5790. assert!(error.contains("unknown option: --resum"));
  5791. assert!(error.contains("Did you mean --resume?"));
  5792. assert!(error.contains("claw --help"));
  5793. }
  5794. #[test]
  5795. fn parses_resume_flag_with_slash_command_arguments() {
  5796. let args = vec![
  5797. "--resume".to_string(),
  5798. "session.jsonl".to_string(),
  5799. "/export".to_string(),
  5800. "notes.txt".to_string(),
  5801. "/clear".to_string(),
  5802. "--confirm".to_string(),
  5803. ];
  5804. assert_eq!(
  5805. parse_args(&args).expect("args should parse"),
  5806. CliAction::ResumeSession {
  5807. session_path: PathBuf::from("session.jsonl"),
  5808. commands: vec![
  5809. "/export notes.txt".to_string(),
  5810. "/clear --confirm".to_string(),
  5811. ],
  5812. }
  5813. );
  5814. }
  5815. #[test]
  5816. fn parses_resume_flag_with_absolute_export_path() {
  5817. let args = vec![
  5818. "--resume".to_string(),
  5819. "session.jsonl".to_string(),
  5820. "/export".to_string(),
  5821. "/tmp/notes.txt".to_string(),
  5822. "/status".to_string(),
  5823. ];
  5824. assert_eq!(
  5825. parse_args(&args).expect("args should parse"),
  5826. CliAction::ResumeSession {
  5827. session_path: PathBuf::from("session.jsonl"),
  5828. commands: vec!["/export /tmp/notes.txt".to_string(), "/status".to_string()],
  5829. }
  5830. );
  5831. }
  5832. #[test]
  5833. fn filtered_tool_specs_respect_allowlist() {
  5834. let allowed = ["read_file", "grep_search"]
  5835. .into_iter()
  5836. .map(str::to_string)
  5837. .collect();
  5838. let filtered = filter_tool_specs(&GlobalToolRegistry::builtin(), Some(&allowed));
  5839. let names = filtered
  5840. .into_iter()
  5841. .map(|spec| spec.name)
  5842. .collect::<Vec<_>>();
  5843. assert_eq!(names, vec!["read_file", "grep_search"]);
  5844. }
  5845. #[test]
  5846. fn filtered_tool_specs_include_plugin_tools() {
  5847. let filtered = filter_tool_specs(&registry_with_plugin_tool(), None);
  5848. let names = filtered
  5849. .into_iter()
  5850. .map(|definition| definition.name)
  5851. .collect::<Vec<_>>();
  5852. assert!(names.contains(&"bash".to_string()));
  5853. assert!(names.contains(&"plugin_echo".to_string()));
  5854. }
  5855. #[test]
  5856. fn permission_policy_uses_plugin_tool_permissions() {
  5857. let feature_config = runtime::RuntimeFeatureConfig::default();
  5858. let policy = permission_policy(
  5859. PermissionMode::ReadOnly,
  5860. &feature_config,
  5861. &registry_with_plugin_tool(),
  5862. )
  5863. .expect("permission policy should build");
  5864. let required = policy.required_mode_for("plugin_echo");
  5865. assert_eq!(required, PermissionMode::WorkspaceWrite);
  5866. }
  5867. #[test]
  5868. fn shared_help_uses_resume_annotation_copy() {
  5869. let help = commands::render_slash_command_help();
  5870. assert!(help.contains("Slash commands"));
  5871. assert!(help.contains("works with --resume SESSION.jsonl"));
  5872. }
  5873. #[test]
  5874. fn repl_help_includes_shared_commands_and_exit() {
  5875. let help = render_repl_help();
  5876. assert!(help.contains("REPL"));
  5877. assert!(help.contains("/help"));
  5878. assert!(help.contains("Complete commands, modes, and recent sessions"));
  5879. assert!(help.contains("/status"));
  5880. assert!(help.contains("/sandbox"));
  5881. assert!(help.contains("/model [model]"));
  5882. assert!(help.contains("/permissions [read-only|workspace-write|danger-full-access]"));
  5883. assert!(help.contains("/clear [--confirm]"));
  5884. assert!(help.contains("/cost"));
  5885. assert!(help.contains("/resume <session-path>"));
  5886. assert!(help.contains("/config [env|hooks|model|plugins]"));
  5887. assert!(help.contains("/mcp [list|show <server>|help]"));
  5888. assert!(help.contains("/memory"));
  5889. assert!(help.contains("/init"));
  5890. assert!(help.contains("/diff"));
  5891. assert!(help.contains("/version"));
  5892. assert!(help.contains("/export [file]"));
  5893. assert!(help.contains("/session [list|switch <session-id>|fork [branch-name]]"));
  5894. assert!(help.contains(
  5895. "/plugin [list|install <path>|enable <name>|disable <name>|uninstall <id>|update <id>]"
  5896. ));
  5897. assert!(help.contains("aliases: /plugins, /marketplace"));
  5898. assert!(help.contains("/agents"));
  5899. assert!(help.contains("/skills"));
  5900. assert!(help.contains("/exit"));
  5901. assert!(help.contains("Auto-save .claw/sessions/<session-id>.jsonl"));
  5902. assert!(help.contains("Resume latest /resume latest"));
  5903. }
  5904. #[test]
  5905. fn completion_candidates_include_workflow_shortcuts_and_dynamic_sessions() {
  5906. let completions = slash_command_completion_candidates_with_sessions(
  5907. "sonnet",
  5908. Some("session-current"),
  5909. vec!["session-old".to_string()],
  5910. );
  5911. assert!(completions.contains(&"/model claude-sonnet-4-6".to_string()));
  5912. assert!(completions.contains(&"/permissions workspace-write".to_string()));
  5913. assert!(completions.contains(&"/session list".to_string()));
  5914. assert!(completions.contains(&"/session switch session-current".to_string()));
  5915. assert!(completions.contains(&"/resume session-old".to_string()));
  5916. assert!(completions.contains(&"/mcp list".to_string()));
  5917. assert!(completions.contains(&"/ultraplan ".to_string()));
  5918. }
  5919. #[test]
  5920. fn startup_banner_mentions_workflow_completions() {
  5921. let _guard = env_lock();
  5922. // Inject dummy credentials so LiveCli can construct without real Anthropic key
  5923. std::env::set_var("ANTHROPIC_API_KEY", "test-dummy-key-for-banner-test");
  5924. let root = temp_dir();
  5925. fs::create_dir_all(&root).expect("root dir");
  5926. let banner = with_current_dir(&root, || {
  5927. LiveCli::new(
  5928. "claude-sonnet-4-6".to_string(),
  5929. true,
  5930. None,
  5931. PermissionMode::DangerFullAccess,
  5932. )
  5933. .expect("cli should initialize")
  5934. .startup_banner()
  5935. });
  5936. assert!(banner.contains("Tab"));
  5937. assert!(banner.contains("workflow completions"));
  5938. fs::remove_dir_all(root).expect("cleanup temp dir");
  5939. std::env::remove_var("ANTHROPIC_API_KEY");
  5940. }
  5941. #[test]
  5942. fn resume_supported_command_list_matches_expected_surface() {
  5943. let names = resume_supported_slash_commands()
  5944. .into_iter()
  5945. .map(|spec| spec.name)
  5946. .collect::<Vec<_>>();
  5947. // Now with 135+ slash commands, verify minimum resume support
  5948. assert!(
  5949. names.len() >= 39,
  5950. "expected at least 39 resume-supported commands, got {}",
  5951. names.len()
  5952. );
  5953. // Verify key resume commands still exist
  5954. assert!(names.contains(&"help"));
  5955. assert!(names.contains(&"status"));
  5956. assert!(names.contains(&"compact"));
  5957. }
  5958. #[test]
  5959. fn resume_report_uses_sectioned_layout() {
  5960. let report = format_resume_report("session.jsonl", 14, 6);
  5961. assert!(report.contains("Session resumed"));
  5962. assert!(report.contains("Session file session.jsonl"));
  5963. assert!(report.contains("Messages 14"));
  5964. assert!(report.contains("Turns 6"));
  5965. }
  5966. #[test]
  5967. fn compact_report_uses_structured_output() {
  5968. let compacted = format_compact_report(8, 5, false);
  5969. assert!(compacted.contains("Compact"));
  5970. assert!(compacted.contains("Result compacted"));
  5971. assert!(compacted.contains("Messages removed 8"));
  5972. let skipped = format_compact_report(0, 3, true);
  5973. assert!(skipped.contains("Result skipped"));
  5974. }
  5975. #[test]
  5976. fn cost_report_uses_sectioned_layout() {
  5977. let report = format_cost_report(runtime::TokenUsage {
  5978. input_tokens: 20,
  5979. output_tokens: 8,
  5980. cache_creation_input_tokens: 3,
  5981. cache_read_input_tokens: 1,
  5982. });
  5983. assert!(report.contains("Cost"));
  5984. assert!(report.contains("Input tokens 20"));
  5985. assert!(report.contains("Output tokens 8"));
  5986. assert!(report.contains("Cache create 3"));
  5987. assert!(report.contains("Cache read 1"));
  5988. assert!(report.contains("Total tokens 32"));
  5989. }
  5990. #[test]
  5991. fn permissions_report_uses_sectioned_layout() {
  5992. let report = format_permissions_report("workspace-write");
  5993. assert!(report.contains("Permissions"));
  5994. assert!(report.contains("Active mode workspace-write"));
  5995. assert!(report.contains("Modes"));
  5996. assert!(report.contains("read-only ○ available Read/search tools only"));
  5997. assert!(report.contains("workspace-write ● current Edit files inside the workspace"));
  5998. assert!(report.contains("danger-full-access ○ available Unrestricted tool access"));
  5999. }
  6000. #[test]
  6001. fn permissions_switch_report_is_structured() {
  6002. let report = format_permissions_switch_report("read-only", "workspace-write");
  6003. assert!(report.contains("Permissions updated"));
  6004. assert!(report.contains("Result mode switched"));
  6005. assert!(report.contains("Previous mode read-only"));
  6006. assert!(report.contains("Active mode workspace-write"));
  6007. assert!(report.contains("Applies to subsequent tool calls"));
  6008. }
  6009. #[test]
  6010. fn init_help_mentions_direct_subcommand() {
  6011. let mut help = Vec::new();
  6012. print_help_to(&mut help).expect("help should render");
  6013. let help = String::from_utf8(help).expect("help should be utf8");
  6014. assert!(help.contains("claw help"));
  6015. assert!(help.contains("claw version"));
  6016. assert!(help.contains("claw status"));
  6017. assert!(help.contains("claw sandbox"));
  6018. assert!(help.contains("claw init"));
  6019. assert!(help.contains("claw agents"));
  6020. assert!(help.contains("claw mcp"));
  6021. assert!(help.contains("claw skills"));
  6022. assert!(help.contains("claw /skills"));
  6023. }
  6024. #[test]
  6025. fn model_report_uses_sectioned_layout() {
  6026. let report = format_model_report("claude-sonnet", 12, 4);
  6027. assert!(report.contains("Model"));
  6028. assert!(report.contains("Current model claude-sonnet"));
  6029. assert!(report.contains("Session messages 12"));
  6030. assert!(report.contains("Switch models with /model <name>"));
  6031. }
  6032. #[test]
  6033. fn model_switch_report_preserves_context_summary() {
  6034. let report = format_model_switch_report("claude-sonnet", "claude-opus", 9);
  6035. assert!(report.contains("Model updated"));
  6036. assert!(report.contains("Previous claude-sonnet"));
  6037. assert!(report.contains("Current claude-opus"));
  6038. assert!(report.contains("Preserved msgs 9"));
  6039. }
  6040. #[test]
  6041. fn status_line_reports_model_and_token_totals() {
  6042. let status = format_status_report(
  6043. "claude-sonnet",
  6044. StatusUsage {
  6045. message_count: 7,
  6046. turns: 3,
  6047. latest: runtime::TokenUsage {
  6048. input_tokens: 5,
  6049. output_tokens: 4,
  6050. cache_creation_input_tokens: 1,
  6051. cache_read_input_tokens: 0,
  6052. },
  6053. cumulative: runtime::TokenUsage {
  6054. input_tokens: 20,
  6055. output_tokens: 8,
  6056. cache_creation_input_tokens: 2,
  6057. cache_read_input_tokens: 1,
  6058. },
  6059. estimated_tokens: 128,
  6060. },
  6061. "workspace-write",
  6062. &super::StatusContext {
  6063. cwd: PathBuf::from("/tmp/project"),
  6064. session_path: Some(PathBuf::from("session.jsonl")),
  6065. loaded_config_files: 2,
  6066. discovered_config_files: 3,
  6067. memory_file_count: 4,
  6068. project_root: Some(PathBuf::from("/tmp")),
  6069. git_branch: Some("main".to_string()),
  6070. git_summary: GitWorkspaceSummary {
  6071. changed_files: 3,
  6072. staged_files: 1,
  6073. unstaged_files: 1,
  6074. untracked_files: 1,
  6075. conflicted_files: 0,
  6076. },
  6077. sandbox_status: runtime::SandboxStatus::default(),
  6078. },
  6079. );
  6080. assert!(status.contains("Status"));
  6081. assert!(status.contains("Model claude-sonnet"));
  6082. assert!(status.contains("Permission mode workspace-write"));
  6083. assert!(status.contains("Messages 7"));
  6084. assert!(status.contains("Latest total 10"));
  6085. assert!(status.contains("Cumulative total 31"));
  6086. assert!(status.contains("Cwd /tmp/project"));
  6087. assert!(status.contains("Project root /tmp"));
  6088. assert!(status.contains("Git branch main"));
  6089. assert!(
  6090. status.contains("Git state dirty · 3 files · 1 staged, 1 unstaged, 1 untracked")
  6091. );
  6092. assert!(status.contains("Changed files 3"));
  6093. assert!(status.contains("Staged 1"));
  6094. assert!(status.contains("Unstaged 1"));
  6095. assert!(status.contains("Untracked 1"));
  6096. assert!(status.contains("Session session.jsonl"));
  6097. assert!(status.contains("Config files loaded 2/3"));
  6098. assert!(status.contains("Memory files 4"));
  6099. assert!(status.contains("Suggested flow /status → /diff → /commit"));
  6100. }
  6101. #[test]
  6102. fn commit_reports_surface_workspace_context() {
  6103. let summary = GitWorkspaceSummary {
  6104. changed_files: 2,
  6105. staged_files: 1,
  6106. unstaged_files: 1,
  6107. untracked_files: 0,
  6108. conflicted_files: 0,
  6109. };
  6110. let preflight = format_commit_preflight_report(Some("feature/ux"), summary);
  6111. assert!(preflight.contains("Result ready"));
  6112. assert!(preflight.contains("Branch feature/ux"));
  6113. assert!(preflight.contains("Workspace dirty · 2 files · 1 staged, 1 unstaged"));
  6114. assert!(preflight
  6115. .contains("Action create a git commit from the current workspace changes"));
  6116. }
  6117. #[test]
  6118. fn commit_skipped_report_points_to_next_steps() {
  6119. let report = format_commit_skipped_report();
  6120. assert!(report.contains("Reason no workspace changes"));
  6121. assert!(report
  6122. .contains("Action create a git commit from the current workspace changes"));
  6123. assert!(report.contains("/status to inspect context"));
  6124. assert!(report.contains("/diff to inspect repo changes"));
  6125. }
  6126. #[test]
  6127. fn runtime_slash_reports_describe_command_behavior() {
  6128. let bughunter = format_bughunter_report(Some("runtime"));
  6129. assert!(bughunter.contains("Scope runtime"));
  6130. assert!(bughunter.contains("inspect the selected code for likely bugs"));
  6131. let ultraplan = format_ultraplan_report(Some("ship the release"));
  6132. assert!(ultraplan.contains("Task ship the release"));
  6133. assert!(ultraplan.contains("break work into a multi-step execution plan"));
  6134. let pr = format_pr_report("feature/ux", Some("ready for review"));
  6135. assert!(pr.contains("Branch feature/ux"));
  6136. assert!(pr.contains("draft or create a pull request"));
  6137. let issue = format_issue_report(Some("flaky test"));
  6138. assert!(issue.contains("Context flaky test"));
  6139. assert!(issue.contains("draft or create a GitHub issue"));
  6140. }
  6141. #[test]
  6142. fn no_arg_commands_reject_unexpected_arguments() {
  6143. assert!(validate_no_args("/commit", None).is_ok());
  6144. let error = validate_no_args("/commit", Some("now"))
  6145. .expect_err("unexpected arguments should fail")
  6146. .to_string();
  6147. assert!(error.contains("/commit does not accept arguments"));
  6148. assert!(error.contains("Received: now"));
  6149. }
  6150. #[test]
  6151. fn config_report_supports_section_views() {
  6152. let report = render_config_report(Some("env")).expect("config report should render");
  6153. assert!(report.contains("Merged section: env"));
  6154. let plugins_report =
  6155. render_config_report(Some("plugins")).expect("plugins config report should render");
  6156. assert!(plugins_report.contains("Merged section: plugins"));
  6157. }
  6158. #[test]
  6159. fn memory_report_uses_sectioned_layout() {
  6160. let report = render_memory_report().expect("memory report should render");
  6161. assert!(report.contains("Memory"));
  6162. assert!(report.contains("Working directory"));
  6163. assert!(report.contains("Instruction files"));
  6164. assert!(report.contains("Discovered files"));
  6165. }
  6166. #[test]
  6167. fn config_report_uses_sectioned_layout() {
  6168. let report = render_config_report(None).expect("config report should render");
  6169. assert!(report.contains("Config"));
  6170. assert!(report.contains("Discovered files"));
  6171. assert!(report.contains("Merged JSON"));
  6172. }
  6173. #[test]
  6174. fn parses_git_status_metadata() {
  6175. let _guard = env_lock();
  6176. let temp_root = temp_dir();
  6177. fs::create_dir_all(&temp_root).expect("root dir");
  6178. let (project_root, branch) = parse_git_status_metadata_for(
  6179. &temp_root,
  6180. Some(
  6181. "## rcc/cli...origin/rcc/cli
  6182. M src/main.rs",
  6183. ),
  6184. );
  6185. assert_eq!(branch.as_deref(), Some("rcc/cli"));
  6186. assert!(project_root.is_none());
  6187. fs::remove_dir_all(temp_root).expect("cleanup temp dir");
  6188. }
  6189. #[test]
  6190. fn parses_detached_head_from_status_snapshot() {
  6191. let _guard = env_lock();
  6192. assert_eq!(
  6193. parse_git_status_branch(Some(
  6194. "## HEAD (no branch)
  6195. M src/main.rs"
  6196. )),
  6197. Some("detached HEAD".to_string())
  6198. );
  6199. }
  6200. #[test]
  6201. fn parses_git_workspace_summary_counts() {
  6202. let summary = parse_git_workspace_summary(Some(
  6203. "## feature/ux
  6204. M src/main.rs
  6205. M README.md
  6206. ?? notes.md
  6207. UU conflicted.rs",
  6208. ));
  6209. assert_eq!(
  6210. summary,
  6211. GitWorkspaceSummary {
  6212. changed_files: 4,
  6213. staged_files: 2,
  6214. unstaged_files: 2,
  6215. untracked_files: 1,
  6216. conflicted_files: 1,
  6217. }
  6218. );
  6219. assert_eq!(
  6220. summary.headline(),
  6221. "dirty · 4 files · 2 staged, 2 unstaged, 1 untracked, 1 conflicted"
  6222. );
  6223. }
  6224. #[test]
  6225. fn render_diff_report_shows_clean_tree_for_committed_repo() {
  6226. let _guard = env_lock();
  6227. let root = temp_dir();
  6228. fs::create_dir_all(&root).expect("root dir");
  6229. git(&["init", "--quiet"], &root);
  6230. git(&["config", "user.email", "tests@example.com"], &root);
  6231. git(&["config", "user.name", "Rusty Claude Tests"], &root);
  6232. fs::write(root.join("tracked.txt"), "hello\n").expect("write file");
  6233. git(&["add", "tracked.txt"], &root);
  6234. git(&["commit", "-m", "init", "--quiet"], &root);
  6235. let report = render_diff_report_for(&root).expect("diff report should render");
  6236. assert!(report.contains("clean working tree"));
  6237. fs::remove_dir_all(root).expect("cleanup temp dir");
  6238. }
  6239. #[test]
  6240. fn render_diff_report_includes_staged_and_unstaged_sections() {
  6241. let _guard = env_lock();
  6242. let root = temp_dir();
  6243. fs::create_dir_all(&root).expect("root dir");
  6244. git(&["init", "--quiet"], &root);
  6245. git(&["config", "user.email", "tests@example.com"], &root);
  6246. git(&["config", "user.name", "Rusty Claude Tests"], &root);
  6247. fs::write(root.join("tracked.txt"), "hello\n").expect("write file");
  6248. git(&["add", "tracked.txt"], &root);
  6249. git(&["commit", "-m", "init", "--quiet"], &root);
  6250. fs::write(root.join("tracked.txt"), "hello\nstaged\n").expect("update file");
  6251. git(&["add", "tracked.txt"], &root);
  6252. fs::write(root.join("tracked.txt"), "hello\nstaged\nunstaged\n")
  6253. .expect("update file twice");
  6254. let report = render_diff_report_for(&root).expect("diff report should render");
  6255. assert!(report.contains("Staged changes:"));
  6256. assert!(report.contains("Unstaged changes:"));
  6257. assert!(report.contains("tracked.txt"));
  6258. fs::remove_dir_all(root).expect("cleanup temp dir");
  6259. }
  6260. #[test]
  6261. fn render_diff_report_omits_ignored_files() {
  6262. let _guard = env_lock();
  6263. let root = temp_dir();
  6264. fs::create_dir_all(&root).expect("root dir");
  6265. git(&["init", "--quiet"], &root);
  6266. git(&["config", "user.email", "tests@example.com"], &root);
  6267. git(&["config", "user.name", "Rusty Claude Tests"], &root);
  6268. fs::write(root.join(".gitignore"), ".omx/\nignored.txt\n").expect("write gitignore");
  6269. fs::write(root.join("tracked.txt"), "hello\n").expect("write tracked");
  6270. git(&["add", ".gitignore", "tracked.txt"], &root);
  6271. git(&["commit", "-m", "init", "--quiet"], &root);
  6272. fs::create_dir_all(root.join(".omx")).expect("write omx dir");
  6273. fs::write(root.join(".omx").join("state.json"), "{}").expect("write ignored omx");
  6274. fs::write(root.join("ignored.txt"), "secret\n").expect("write ignored file");
  6275. fs::write(root.join("tracked.txt"), "hello\nworld\n").expect("write tracked change");
  6276. let report = render_diff_report_for(&root).expect("diff report should render");
  6277. assert!(report.contains("tracked.txt"));
  6278. assert!(!report.contains("+++ b/ignored.txt"));
  6279. assert!(!report.contains("+++ b/.omx/state.json"));
  6280. fs::remove_dir_all(root).expect("cleanup temp dir");
  6281. }
  6282. #[test]
  6283. fn resume_diff_command_renders_report_for_saved_session() {
  6284. let _guard = env_lock();
  6285. let root = temp_dir();
  6286. fs::create_dir_all(&root).expect("root dir");
  6287. git(&["init", "--quiet"], &root);
  6288. git(&["config", "user.email", "tests@example.com"], &root);
  6289. git(&["config", "user.name", "Rusty Claude Tests"], &root);
  6290. fs::write(root.join("tracked.txt"), "hello\n").expect("write tracked");
  6291. git(&["add", "tracked.txt"], &root);
  6292. git(&["commit", "-m", "init", "--quiet"], &root);
  6293. fs::write(root.join("tracked.txt"), "hello\nworld\n").expect("modify tracked");
  6294. let session_path = root.join("session.json");
  6295. Session::new()
  6296. .save_to_path(&session_path)
  6297. .expect("session should save");
  6298. let session = Session::load_from_path(&session_path).expect("session should load");
  6299. let outcome = with_current_dir(&root, || {
  6300. run_resume_command(&session_path, &session, &SlashCommand::Diff)
  6301. .expect("resume diff should work")
  6302. });
  6303. let message = outcome.message.expect("diff message should exist");
  6304. assert!(message.contains("Unstaged changes:"));
  6305. assert!(message.contains("tracked.txt"));
  6306. fs::remove_dir_all(root).expect("cleanup temp dir");
  6307. }
  6308. #[test]
  6309. fn status_context_reads_real_workspace_metadata() {
  6310. let context = status_context(None).expect("status context should load");
  6311. assert!(context.cwd.is_absolute());
  6312. assert!(context.discovered_config_files >= context.loaded_config_files);
  6313. assert!(context.loaded_config_files <= context.discovered_config_files);
  6314. }
  6315. #[test]
  6316. fn normalizes_supported_permission_modes() {
  6317. assert_eq!(normalize_permission_mode("read-only"), Some("read-only"));
  6318. assert_eq!(
  6319. normalize_permission_mode("workspace-write"),
  6320. Some("workspace-write")
  6321. );
  6322. assert_eq!(
  6323. normalize_permission_mode("danger-full-access"),
  6324. Some("danger-full-access")
  6325. );
  6326. assert_eq!(normalize_permission_mode("unknown"), None);
  6327. }
  6328. #[test]
  6329. fn clear_command_requires_explicit_confirmation_flag() {
  6330. assert_eq!(
  6331. SlashCommand::parse("/clear"),
  6332. Ok(Some(SlashCommand::Clear { confirm: false }))
  6333. );
  6334. assert_eq!(
  6335. SlashCommand::parse("/clear --confirm"),
  6336. Ok(Some(SlashCommand::Clear { confirm: true }))
  6337. );
  6338. }
  6339. #[test]
  6340. fn parses_resume_and_config_slash_commands() {
  6341. assert_eq!(
  6342. SlashCommand::parse("/resume saved-session.jsonl"),
  6343. Ok(Some(SlashCommand::Resume {
  6344. session_path: Some("saved-session.jsonl".to_string())
  6345. }))
  6346. );
  6347. assert_eq!(
  6348. SlashCommand::parse("/clear --confirm"),
  6349. Ok(Some(SlashCommand::Clear { confirm: true }))
  6350. );
  6351. assert_eq!(
  6352. SlashCommand::parse("/config"),
  6353. Ok(Some(SlashCommand::Config { section: None }))
  6354. );
  6355. assert_eq!(
  6356. SlashCommand::parse("/config env"),
  6357. Ok(Some(SlashCommand::Config {
  6358. section: Some("env".to_string())
  6359. }))
  6360. );
  6361. assert_eq!(
  6362. SlashCommand::parse("/memory"),
  6363. Ok(Some(SlashCommand::Memory))
  6364. );
  6365. assert_eq!(SlashCommand::parse("/init"), Ok(Some(SlashCommand::Init)));
  6366. assert_eq!(
  6367. SlashCommand::parse("/session fork incident-review"),
  6368. Ok(Some(SlashCommand::Session {
  6369. action: Some("fork".to_string()),
  6370. target: Some("incident-review".to_string())
  6371. }))
  6372. );
  6373. }
  6374. #[test]
  6375. fn help_mentions_jsonl_resume_examples() {
  6376. let mut help = Vec::new();
  6377. print_help_to(&mut help).expect("help should render");
  6378. let help = String::from_utf8(help).expect("help should be utf8");
  6379. assert!(help.contains("claw --resume [SESSION.jsonl|session-id|latest]"));
  6380. assert!(help.contains("Use `latest` with --resume, /resume, or /session switch"));
  6381. assert!(help.contains("claw --resume latest"));
  6382. assert!(help.contains("claw --resume latest /status /diff /export notes.txt"));
  6383. }
  6384. #[test]
  6385. fn managed_sessions_default_to_jsonl_and_resolve_legacy_json() {
  6386. let _guard = cwd_lock().lock().expect("cwd lock");
  6387. let workspace = temp_workspace("session-resolution");
  6388. std::fs::create_dir_all(&workspace).expect("workspace should create");
  6389. let previous = std::env::current_dir().expect("cwd");
  6390. std::env::set_current_dir(&workspace).expect("switch cwd");
  6391. let handle = create_managed_session_handle("session-alpha").expect("jsonl handle");
  6392. assert!(handle.path.ends_with("session-alpha.jsonl"));
  6393. let legacy_path = workspace.join(".claw/sessions/legacy.json");
  6394. std::fs::create_dir_all(
  6395. legacy_path
  6396. .parent()
  6397. .expect("legacy path should have parent directory"),
  6398. )
  6399. .expect("session dir should exist");
  6400. Session::new()
  6401. .with_persistence_path(legacy_path.clone())
  6402. .save_to_path(&legacy_path)
  6403. .expect("legacy session should save");
  6404. let resolved = resolve_session_reference("legacy").expect("legacy session should resolve");
  6405. assert_eq!(
  6406. resolved
  6407. .path
  6408. .canonicalize()
  6409. .expect("resolved path should exist"),
  6410. legacy_path
  6411. .canonicalize()
  6412. .expect("legacy path should exist")
  6413. );
  6414. std::env::set_current_dir(previous).expect("restore cwd");
  6415. std::fs::remove_dir_all(workspace).expect("workspace should clean up");
  6416. }
  6417. #[test]
  6418. fn latest_session_alias_resolves_most_recent_managed_session() {
  6419. let _guard = cwd_lock().lock().expect("cwd lock");
  6420. let workspace = temp_workspace("latest-session-alias");
  6421. std::fs::create_dir_all(&workspace).expect("workspace should create");
  6422. let previous = std::env::current_dir().expect("cwd");
  6423. std::env::set_current_dir(&workspace).expect("switch cwd");
  6424. let older = create_managed_session_handle("session-older").expect("older handle");
  6425. Session::new()
  6426. .with_persistence_path(older.path.clone())
  6427. .save_to_path(&older.path)
  6428. .expect("older session should save");
  6429. std::thread::sleep(Duration::from_millis(20));
  6430. let newer = create_managed_session_handle("session-newer").expect("newer handle");
  6431. Session::new()
  6432. .with_persistence_path(newer.path.clone())
  6433. .save_to_path(&newer.path)
  6434. .expect("newer session should save");
  6435. let resolved = resolve_session_reference("latest").expect("latest session should resolve");
  6436. assert_eq!(
  6437. resolved
  6438. .path
  6439. .canonicalize()
  6440. .expect("resolved path should exist"),
  6441. newer.path.canonicalize().expect("newer path should exist")
  6442. );
  6443. std::env::set_current_dir(previous).expect("restore cwd");
  6444. std::fs::remove_dir_all(workspace).expect("workspace should clean up");
  6445. }
  6446. #[test]
  6447. fn unknown_slash_command_guidance_suggests_nearby_commands() {
  6448. let message = format_unknown_slash_command("stats");
  6449. assert!(message.contains("Unknown slash command: /stats"));
  6450. assert!(message.contains("/status"));
  6451. assert!(message.contains("/help"));
  6452. }
  6453. #[test]
  6454. fn resume_usage_mentions_latest_shortcut() {
  6455. let usage = render_resume_usage();
  6456. assert!(usage.contains("/resume <session-path|session-id|latest>"));
  6457. assert!(usage.contains(".claw/sessions/<session-id>.jsonl"));
  6458. assert!(usage.contains("/session list"));
  6459. }
  6460. fn cwd_lock() -> &'static Mutex<()> {
  6461. static LOCK: OnceLock<Mutex<()>> = OnceLock::new();
  6462. LOCK.get_or_init(|| Mutex::new(()))
  6463. }
  6464. fn temp_workspace(label: &str) -> PathBuf {
  6465. let nanos = std::time::SystemTime::now()
  6466. .duration_since(std::time::UNIX_EPOCH)
  6467. .expect("system time should be after epoch")
  6468. .as_nanos();
  6469. std::env::temp_dir().join(format!("claw-cli-{label}-{nanos}"))
  6470. }
  6471. #[test]
  6472. fn init_template_mentions_detected_rust_workspace() {
  6473. let _guard = cwd_lock()
  6474. .lock()
  6475. .unwrap_or_else(std::sync::PoisonError::into_inner);
  6476. let workspace_root = PathBuf::from(env!("CARGO_MANIFEST_DIR")).join("../..");
  6477. let rendered = crate::init::render_init_claude_md(&workspace_root);
  6478. assert!(rendered.contains("# CLAUDE.md"));
  6479. assert!(rendered.contains("cargo clippy --workspace --all-targets -- -D warnings"));
  6480. }
  6481. #[test]
  6482. fn converts_tool_roundtrip_messages() {
  6483. let messages = vec![
  6484. ConversationMessage::user_text("hello"),
  6485. ConversationMessage::assistant(vec![ContentBlock::ToolUse {
  6486. id: "tool-1".to_string(),
  6487. name: "bash".to_string(),
  6488. input: "{\"command\":\"pwd\"}".to_string(),
  6489. }]),
  6490. ConversationMessage {
  6491. role: MessageRole::Tool,
  6492. blocks: vec![ContentBlock::ToolResult {
  6493. tool_use_id: "tool-1".to_string(),
  6494. tool_name: "bash".to_string(),
  6495. output: "ok".to_string(),
  6496. is_error: false,
  6497. }],
  6498. usage: None,
  6499. },
  6500. ];
  6501. let converted = super::convert_messages(&messages);
  6502. assert_eq!(converted.len(), 3);
  6503. assert_eq!(converted[1].role, "assistant");
  6504. assert_eq!(converted[2].role, "user");
  6505. }
  6506. #[test]
  6507. fn repl_help_mentions_history_completion_and_multiline() {
  6508. let help = render_repl_help();
  6509. assert!(help.contains("Up/Down"));
  6510. assert!(help.contains("Tab"));
  6511. assert!(help.contains("Shift+Enter/Ctrl+J"));
  6512. }
  6513. #[test]
  6514. fn tool_rendering_helpers_compact_output() {
  6515. let start = format_tool_call_start("read_file", r#"{"path":"src/main.rs"}"#);
  6516. assert!(start.contains("read_file"));
  6517. assert!(start.contains("src/main.rs"));
  6518. let done = format_tool_result(
  6519. "read_file",
  6520. r#"{"file":{"filePath":"src/main.rs","content":"hello","numLines":1,"startLine":1,"totalLines":1}}"#,
  6521. false,
  6522. );
  6523. assert!(done.contains("📄 Read src/main.rs"));
  6524. assert!(done.contains("hello"));
  6525. }
  6526. #[test]
  6527. fn tool_rendering_truncates_large_read_output_for_display_only() {
  6528. let content = (0..200)
  6529. .map(|index| format!("line {index:03}"))
  6530. .collect::<Vec<_>>()
  6531. .join("\n");
  6532. let output = json!({
  6533. "file": {
  6534. "filePath": "src/main.rs",
  6535. "content": content,
  6536. "numLines": 200,
  6537. "startLine": 1,
  6538. "totalLines": 200
  6539. }
  6540. })
  6541. .to_string();
  6542. let rendered = format_tool_result("read_file", &output, false);
  6543. assert!(rendered.contains("line 000"));
  6544. assert!(rendered.contains("line 079"));
  6545. assert!(!rendered.contains("line 199"));
  6546. assert!(rendered.contains("full result preserved in session"));
  6547. assert!(output.contains("line 199"));
  6548. }
  6549. #[test]
  6550. fn tool_rendering_truncates_large_bash_output_for_display_only() {
  6551. let stdout = (0..120)
  6552. .map(|index| format!("stdout {index:03}"))
  6553. .collect::<Vec<_>>()
  6554. .join("\n");
  6555. let output = json!({
  6556. "stdout": stdout,
  6557. "stderr": "",
  6558. "returnCodeInterpretation": "completed successfully"
  6559. })
  6560. .to_string();
  6561. let rendered = format_tool_result("bash", &output, false);
  6562. assert!(rendered.contains("stdout 000"));
  6563. assert!(rendered.contains("stdout 059"));
  6564. assert!(!rendered.contains("stdout 119"));
  6565. assert!(rendered.contains("full result preserved in session"));
  6566. assert!(output.contains("stdout 119"));
  6567. }
  6568. #[test]
  6569. fn tool_rendering_truncates_generic_long_output_for_display_only() {
  6570. let items = (0..120)
  6571. .map(|index| format!("payload {index:03}"))
  6572. .collect::<Vec<_>>();
  6573. let output = json!({
  6574. "summary": "plugin payload",
  6575. "items": items,
  6576. })
  6577. .to_string();
  6578. let rendered = format_tool_result("plugin_echo", &output, false);
  6579. assert!(rendered.contains("plugin_echo"));
  6580. assert!(rendered.contains("payload 000"));
  6581. assert!(rendered.contains("payload 040"));
  6582. assert!(!rendered.contains("payload 080"));
  6583. assert!(!rendered.contains("payload 119"));
  6584. assert!(rendered.contains("full result preserved in session"));
  6585. assert!(output.contains("payload 119"));
  6586. }
  6587. #[test]
  6588. fn tool_rendering_truncates_raw_generic_output_for_display_only() {
  6589. let output = (0..120)
  6590. .map(|index| format!("raw {index:03}"))
  6591. .collect::<Vec<_>>()
  6592. .join("\n");
  6593. let rendered = format_tool_result("plugin_echo", &output, false);
  6594. assert!(rendered.contains("plugin_echo"));
  6595. assert!(rendered.contains("raw 000"));
  6596. assert!(rendered.contains("raw 059"));
  6597. assert!(!rendered.contains("raw 119"));
  6598. assert!(rendered.contains("full result preserved in session"));
  6599. assert!(output.contains("raw 119"));
  6600. }
  6601. #[test]
  6602. fn ultraplan_progress_lines_include_phase_step_and_elapsed_status() {
  6603. let snapshot = InternalPromptProgressState {
  6604. command_label: "Ultraplan",
  6605. task_label: "ship plugin progress".to_string(),
  6606. step: 3,
  6607. phase: "running read_file".to_string(),
  6608. detail: Some("reading rust/crates/rusty-claude-cli/src/main.rs".to_string()),
  6609. saw_final_text: false,
  6610. };
  6611. let started = format_internal_prompt_progress_line(
  6612. InternalPromptProgressEvent::Started,
  6613. &snapshot,
  6614. Duration::from_secs(0),
  6615. None,
  6616. );
  6617. let heartbeat = format_internal_prompt_progress_line(
  6618. InternalPromptProgressEvent::Heartbeat,
  6619. &snapshot,
  6620. Duration::from_secs(9),
  6621. None,
  6622. );
  6623. let completed = format_internal_prompt_progress_line(
  6624. InternalPromptProgressEvent::Complete,
  6625. &snapshot,
  6626. Duration::from_secs(12),
  6627. None,
  6628. );
  6629. let failed = format_internal_prompt_progress_line(
  6630. InternalPromptProgressEvent::Failed,
  6631. &snapshot,
  6632. Duration::from_secs(12),
  6633. Some("network timeout"),
  6634. );
  6635. assert!(started.contains("planning started"));
  6636. assert!(started.contains("current step 3"));
  6637. assert!(heartbeat.contains("heartbeat"));
  6638. assert!(heartbeat.contains("9s elapsed"));
  6639. assert!(heartbeat.contains("phase running read_file"));
  6640. assert!(completed.contains("completed"));
  6641. assert!(completed.contains("3 steps total"));
  6642. assert!(failed.contains("failed"));
  6643. assert!(failed.contains("network timeout"));
  6644. }
  6645. #[test]
  6646. fn describe_tool_progress_summarizes_known_tools() {
  6647. assert_eq!(
  6648. describe_tool_progress("read_file", r#"{"path":"src/main.rs"}"#),
  6649. "reading src/main.rs"
  6650. );
  6651. assert!(
  6652. describe_tool_progress("bash", r#"{"command":"cargo test -p rusty-claude-cli"}"#)
  6653. .contains("cargo test -p rusty-claude-cli")
  6654. );
  6655. assert_eq!(
  6656. describe_tool_progress("grep_search", r#"{"pattern":"ultraplan","path":"rust"}"#),
  6657. "grep `ultraplan` in rust"
  6658. );
  6659. }
  6660. #[test]
  6661. fn push_output_block_renders_markdown_text() {
  6662. let mut out = Vec::new();
  6663. let mut events = Vec::new();
  6664. let mut pending_tool = None;
  6665. push_output_block(
  6666. OutputContentBlock::Text {
  6667. text: "# Heading".to_string(),
  6668. },
  6669. &mut out,
  6670. &mut events,
  6671. &mut pending_tool,
  6672. false,
  6673. )
  6674. .expect("text block should render");
  6675. let rendered = String::from_utf8(out).expect("utf8");
  6676. assert!(rendered.contains("Heading"));
  6677. assert!(rendered.contains('\u{1b}'));
  6678. }
  6679. #[test]
  6680. fn push_output_block_skips_empty_object_prefix_for_tool_streams() {
  6681. let mut out = Vec::new();
  6682. let mut events = Vec::new();
  6683. let mut pending_tool = None;
  6684. push_output_block(
  6685. OutputContentBlock::ToolUse {
  6686. id: "tool-1".to_string(),
  6687. name: "read_file".to_string(),
  6688. input: json!({}),
  6689. },
  6690. &mut out,
  6691. &mut events,
  6692. &mut pending_tool,
  6693. true,
  6694. )
  6695. .expect("tool block should accumulate");
  6696. assert!(events.is_empty());
  6697. assert_eq!(
  6698. pending_tool,
  6699. Some(("tool-1".to_string(), "read_file".to_string(), String::new(),))
  6700. );
  6701. }
  6702. #[test]
  6703. fn response_to_events_preserves_empty_object_json_input_outside_streaming() {
  6704. let mut out = Vec::new();
  6705. let events = response_to_events(
  6706. MessageResponse {
  6707. id: "msg-1".to_string(),
  6708. kind: "message".to_string(),
  6709. model: "claude-opus-4-6".to_string(),
  6710. role: "assistant".to_string(),
  6711. content: vec![OutputContentBlock::ToolUse {
  6712. id: "tool-1".to_string(),
  6713. name: "read_file".to_string(),
  6714. input: json!({}),
  6715. }],
  6716. stop_reason: Some("tool_use".to_string()),
  6717. stop_sequence: None,
  6718. usage: Usage {
  6719. input_tokens: 1,
  6720. output_tokens: 1,
  6721. cache_creation_input_tokens: 0,
  6722. cache_read_input_tokens: 0,
  6723. },
  6724. request_id: None,
  6725. },
  6726. &mut out,
  6727. )
  6728. .expect("response conversion should succeed");
  6729. assert!(matches!(
  6730. &events[0],
  6731. AssistantEvent::ToolUse { name, input, .. }
  6732. if name == "read_file" && input == "{}"
  6733. ));
  6734. }
  6735. #[test]
  6736. fn response_to_events_preserves_non_empty_json_input_outside_streaming() {
  6737. let mut out = Vec::new();
  6738. let events = response_to_events(
  6739. MessageResponse {
  6740. id: "msg-2".to_string(),
  6741. kind: "message".to_string(),
  6742. model: "claude-opus-4-6".to_string(),
  6743. role: "assistant".to_string(),
  6744. content: vec![OutputContentBlock::ToolUse {
  6745. id: "tool-2".to_string(),
  6746. name: "read_file".to_string(),
  6747. input: json!({ "path": "rust/Cargo.toml" }),
  6748. }],
  6749. stop_reason: Some("tool_use".to_string()),
  6750. stop_sequence: None,
  6751. usage: Usage {
  6752. input_tokens: 1,
  6753. output_tokens: 1,
  6754. cache_creation_input_tokens: 0,
  6755. cache_read_input_tokens: 0,
  6756. },
  6757. request_id: None,
  6758. },
  6759. &mut out,
  6760. )
  6761. .expect("response conversion should succeed");
  6762. assert!(matches!(
  6763. &events[0],
  6764. AssistantEvent::ToolUse { name, input, .. }
  6765. if name == "read_file" && input == "{\"path\":\"rust/Cargo.toml\"}"
  6766. ));
  6767. }
  6768. #[test]
  6769. fn response_to_events_ignores_thinking_blocks() {
  6770. let mut out = Vec::new();
  6771. let events = response_to_events(
  6772. MessageResponse {
  6773. id: "msg-3".to_string(),
  6774. kind: "message".to_string(),
  6775. model: "claude-opus-4-6".to_string(),
  6776. role: "assistant".to_string(),
  6777. content: vec![
  6778. OutputContentBlock::Thinking {
  6779. thinking: "step 1".to_string(),
  6780. signature: Some("sig_123".to_string()),
  6781. },
  6782. OutputContentBlock::Text {
  6783. text: "Final answer".to_string(),
  6784. },
  6785. ],
  6786. stop_reason: Some("end_turn".to_string()),
  6787. stop_sequence: None,
  6788. usage: Usage {
  6789. input_tokens: 1,
  6790. output_tokens: 1,
  6791. cache_creation_input_tokens: 0,
  6792. cache_read_input_tokens: 0,
  6793. },
  6794. request_id: None,
  6795. },
  6796. &mut out,
  6797. )
  6798. .expect("response conversion should succeed");
  6799. assert!(matches!(
  6800. &events[0],
  6801. AssistantEvent::TextDelta(text) if text == "Final answer"
  6802. ));
  6803. assert!(!String::from_utf8(out).expect("utf8").contains("step 1"));
  6804. }
  6805. #[test]
  6806. fn build_runtime_plugin_state_merges_plugin_hooks_into_runtime_features() {
  6807. let config_home = temp_dir();
  6808. let workspace = temp_dir();
  6809. let source_root = temp_dir();
  6810. fs::create_dir_all(&config_home).expect("config home");
  6811. fs::create_dir_all(&workspace).expect("workspace");
  6812. fs::create_dir_all(&source_root).expect("source root");
  6813. write_plugin_fixture(&source_root, "hook-runtime-demo", true, false);
  6814. let mut manager = PluginManager::new(PluginManagerConfig::new(&config_home));
  6815. manager
  6816. .install(source_root.to_str().expect("utf8 source path"))
  6817. .expect("plugin install should succeed");
  6818. let loader = ConfigLoader::new(&workspace, &config_home);
  6819. let runtime_config = loader.load().expect("runtime config should load");
  6820. let state = build_runtime_plugin_state_with_loader(&workspace, &loader, &runtime_config)
  6821. .expect("plugin state should load");
  6822. let pre_hooks = state.feature_config.hooks().pre_tool_use();
  6823. assert_eq!(pre_hooks.len(), 1);
  6824. assert!(
  6825. pre_hooks[0].ends_with("hooks/pre.sh"),
  6826. "expected installed plugin hook path, got {pre_hooks:?}"
  6827. );
  6828. let _ = fs::remove_dir_all(config_home);
  6829. let _ = fs::remove_dir_all(workspace);
  6830. let _ = fs::remove_dir_all(source_root);
  6831. }
  6832. #[test]
  6833. fn build_runtime_plugin_state_discovers_mcp_tools_and_surfaces_pending_servers() {
  6834. let config_home = temp_dir();
  6835. let workspace = temp_dir();
  6836. fs::create_dir_all(&config_home).expect("config home");
  6837. fs::create_dir_all(&workspace).expect("workspace");
  6838. let script_path = workspace.join("fixture-mcp.py");
  6839. write_mcp_server_fixture(&script_path);
  6840. fs::write(
  6841. config_home.join("settings.json"),
  6842. format!(
  6843. r#"{{
  6844. "mcpServers": {{
  6845. "alpha": {{
  6846. "command": "python3",
  6847. "args": ["{}"]
  6848. }},
  6849. "broken": {{
  6850. "command": "python3",
  6851. "args": ["-c", "import sys; sys.exit(0)"]
  6852. }}
  6853. }}
  6854. }}"#,
  6855. script_path.to_string_lossy()
  6856. ),
  6857. )
  6858. .expect("write mcp settings");
  6859. let loader = ConfigLoader::new(&workspace, &config_home);
  6860. let runtime_config = loader.load().expect("runtime config should load");
  6861. let state = build_runtime_plugin_state_with_loader(&workspace, &loader, &runtime_config)
  6862. .expect("runtime plugin state should load");
  6863. let allowed = state
  6864. .tool_registry
  6865. .normalize_allowed_tools(&["mcp__alpha__echo".to_string(), "MCPTool".to_string()])
  6866. .expect("mcp tools should be allow-listable")
  6867. .expect("allow-list should exist");
  6868. assert!(allowed.contains("mcp__alpha__echo"));
  6869. assert!(allowed.contains("MCPTool"));
  6870. let mut executor = CliToolExecutor::new(
  6871. None,
  6872. false,
  6873. state.tool_registry.clone(),
  6874. state.mcp_state.clone(),
  6875. );
  6876. let tool_output = executor
  6877. .execute("mcp__alpha__echo", r#"{"text":"hello"}"#)
  6878. .expect("discovered mcp tool should execute");
  6879. let tool_json: serde_json::Value =
  6880. serde_json::from_str(&tool_output).expect("tool output should be json");
  6881. assert_eq!(tool_json["structuredContent"]["echoed"], "hello");
  6882. let wrapped_output = executor
  6883. .execute(
  6884. "MCPTool",
  6885. r#"{"qualifiedName":"mcp__alpha__echo","arguments":{"text":"wrapped"}}"#,
  6886. )
  6887. .expect("generic mcp wrapper should execute");
  6888. let wrapped_json: serde_json::Value =
  6889. serde_json::from_str(&wrapped_output).expect("wrapped output should be json");
  6890. assert_eq!(wrapped_json["structuredContent"]["echoed"], "wrapped");
  6891. let search_output = executor
  6892. .execute("ToolSearch", r#"{"query":"alpha echo","max_results":5}"#)
  6893. .expect("tool search should execute");
  6894. let search_json: serde_json::Value =
  6895. serde_json::from_str(&search_output).expect("search output should be json");
  6896. assert_eq!(search_json["matches"][0], "mcp__alpha__echo");
  6897. assert_eq!(search_json["pending_mcp_servers"][0], "broken");
  6898. assert_eq!(
  6899. search_json["mcp_degraded"]["failed_servers"][0]["server_name"],
  6900. "broken"
  6901. );
  6902. assert_eq!(
  6903. search_json["mcp_degraded"]["failed_servers"][0]["phase"],
  6904. "tool_discovery"
  6905. );
  6906. assert_eq!(
  6907. search_json["mcp_degraded"]["available_tools"][0],
  6908. "mcp__alpha__echo"
  6909. );
  6910. let listed = executor
  6911. .execute("ListMcpResourcesTool", r#"{"server":"alpha"}"#)
  6912. .expect("resources should list");
  6913. let listed_json: serde_json::Value =
  6914. serde_json::from_str(&listed).expect("resource output should be json");
  6915. assert_eq!(listed_json["resources"][0]["uri"], "file://guide.txt");
  6916. let read = executor
  6917. .execute(
  6918. "ReadMcpResourceTool",
  6919. r#"{"server":"alpha","uri":"file://guide.txt"}"#,
  6920. )
  6921. .expect("resource should read");
  6922. let read_json: serde_json::Value =
  6923. serde_json::from_str(&read).expect("resource read output should be json");
  6924. assert_eq!(
  6925. read_json["contents"][0]["text"],
  6926. "contents for file://guide.txt"
  6927. );
  6928. if let Some(mcp_state) = state.mcp_state {
  6929. mcp_state
  6930. .lock()
  6931. .unwrap_or_else(std::sync::PoisonError::into_inner)
  6932. .shutdown()
  6933. .expect("mcp shutdown should succeed");
  6934. }
  6935. let _ = fs::remove_dir_all(config_home);
  6936. let _ = fs::remove_dir_all(workspace);
  6937. }
  6938. #[test]
  6939. fn build_runtime_plugin_state_surfaces_unsupported_mcp_servers_structurally() {
  6940. let config_home = temp_dir();
  6941. let workspace = temp_dir();
  6942. fs::create_dir_all(&config_home).expect("config home");
  6943. fs::create_dir_all(&workspace).expect("workspace");
  6944. fs::write(
  6945. config_home.join("settings.json"),
  6946. r#"{
  6947. "mcpServers": {
  6948. "remote": {
  6949. "url": "https://example.test/mcp"
  6950. }
  6951. }
  6952. }"#,
  6953. )
  6954. .expect("write mcp settings");
  6955. let loader = ConfigLoader::new(&workspace, &config_home);
  6956. let runtime_config = loader.load().expect("runtime config should load");
  6957. let state = build_runtime_plugin_state_with_loader(&workspace, &loader, &runtime_config)
  6958. .expect("runtime plugin state should load");
  6959. let mut executor =
  6960. CliToolExecutor::new(None, false, state.tool_registry.clone(), state.mcp_state.clone());
  6961. let search_output = executor
  6962. .execute("ToolSearch", r#"{"query":"remote","max_results":5}"#)
  6963. .expect("tool search should execute");
  6964. let search_json: serde_json::Value =
  6965. serde_json::from_str(&search_output).expect("search output should be json");
  6966. assert_eq!(search_json["pending_mcp_servers"][0], "remote");
  6967. assert_eq!(
  6968. search_json["mcp_degraded"]["failed_servers"][0]["server_name"],
  6969. "remote"
  6970. );
  6971. assert_eq!(
  6972. search_json["mcp_degraded"]["failed_servers"][0]["phase"],
  6973. "server_registration"
  6974. );
  6975. assert_eq!(
  6976. search_json["mcp_degraded"]["failed_servers"][0]["error"]["context"]["transport"],
  6977. "http"
  6978. );
  6979. let _ = fs::remove_dir_all(config_home);
  6980. let _ = fs::remove_dir_all(workspace);
  6981. }
  6982. #[test]
  6983. fn build_runtime_runs_plugin_lifecycle_init_and_shutdown() {
  6984. let config_home = temp_dir();
  6985. // Inject a dummy API key so runtime construction succeeds without real credentials.
  6986. // This test only exercises plugin lifecycle (init/shutdown), never calls the API.
  6987. std::env::set_var("ANTHROPIC_API_KEY", "test-dummy-key-for-plugin-lifecycle");
  6988. let workspace = temp_dir();
  6989. let source_root = temp_dir();
  6990. fs::create_dir_all(&config_home).expect("config home");
  6991. fs::create_dir_all(&workspace).expect("workspace");
  6992. fs::create_dir_all(&source_root).expect("source root");
  6993. write_plugin_fixture(&source_root, "lifecycle-runtime-demo", false, true);
  6994. let mut manager = PluginManager::new(PluginManagerConfig::new(&config_home));
  6995. let install = manager
  6996. .install(source_root.to_str().expect("utf8 source path"))
  6997. .expect("plugin install should succeed");
  6998. let log_path = install.install_path.join("lifecycle.log");
  6999. let loader = ConfigLoader::new(&workspace, &config_home);
  7000. let runtime_config = loader.load().expect("runtime config should load");
  7001. let runtime_plugin_state =
  7002. build_runtime_plugin_state_with_loader(&workspace, &loader, &runtime_config)
  7003. .expect("plugin state should load");
  7004. let mut runtime = build_runtime_with_plugin_state(
  7005. Session::new(),
  7006. "runtime-plugin-lifecycle",
  7007. DEFAULT_MODEL.to_string(),
  7008. vec!["test system prompt".to_string()],
  7009. true,
  7010. false,
  7011. None,
  7012. PermissionMode::DangerFullAccess,
  7013. None,
  7014. runtime_plugin_state,
  7015. )
  7016. .expect("runtime should build");
  7017. assert_eq!(
  7018. fs::read_to_string(&log_path).expect("init log should exist"),
  7019. "init\n"
  7020. );
  7021. runtime
  7022. .shutdown_plugins()
  7023. .expect("plugin shutdown should succeed");
  7024. assert_eq!(
  7025. fs::read_to_string(&log_path).expect("shutdown log should exist"),
  7026. "init\nshutdown\n"
  7027. );
  7028. let _ = fs::remove_dir_all(config_home);
  7029. let _ = fs::remove_dir_all(workspace);
  7030. let _ = fs::remove_dir_all(source_root);
  7031. std::env::remove_var("ANTHROPIC_API_KEY");
  7032. }
  7033. }
  7034. fn write_mcp_server_fixture(script_path: &Path) {
  7035. let script = [
  7036. "#!/usr/bin/env python3",
  7037. "import json, sys",
  7038. "",
  7039. "def read_message():",
  7040. " header = b''",
  7041. r" while not header.endswith(b'\r\n\r\n'):",
  7042. " chunk = sys.stdin.buffer.read(1)",
  7043. " if not chunk:",
  7044. " return None",
  7045. " header += chunk",
  7046. " length = 0",
  7047. r" for line in header.decode().split('\r\n'):",
  7048. r" if line.lower().startswith('content-length:'):",
  7049. " length = int(line.split(':', 1)[1].strip())",
  7050. " payload = sys.stdin.buffer.read(length)",
  7051. " return json.loads(payload.decode())",
  7052. "",
  7053. "def send_message(message):",
  7054. " payload = json.dumps(message).encode()",
  7055. r" sys.stdout.buffer.write(f'Content-Length: {len(payload)}\r\n\r\n'.encode() + payload)",
  7056. " sys.stdout.buffer.flush()",
  7057. "",
  7058. "while True:",
  7059. " request = read_message()",
  7060. " if request is None:",
  7061. " break",
  7062. " method = request['method']",
  7063. " if method == 'initialize':",
  7064. " send_message({",
  7065. " 'jsonrpc': '2.0',",
  7066. " 'id': request['id'],",
  7067. " 'result': {",
  7068. " 'protocolVersion': request['params']['protocolVersion'],",
  7069. " 'capabilities': {'tools': {}, 'resources': {}},",
  7070. " 'serverInfo': {'name': 'fixture', 'version': '1.0.0'}",
  7071. " }",
  7072. " })",
  7073. " elif method == 'tools/list':",
  7074. " send_message({",
  7075. " 'jsonrpc': '2.0',",
  7076. " 'id': request['id'],",
  7077. " 'result': {",
  7078. " 'tools': [",
  7079. " {",
  7080. " 'name': 'echo',",
  7081. " 'description': 'Echo from MCP fixture',",
  7082. " 'inputSchema': {",
  7083. " 'type': 'object',",
  7084. " 'properties': {'text': {'type': 'string'}},",
  7085. " 'required': ['text'],",
  7086. " 'additionalProperties': False",
  7087. " },",
  7088. " 'annotations': {'readOnlyHint': True}",
  7089. " }",
  7090. " ]",
  7091. " }",
  7092. " })",
  7093. " elif method == 'tools/call':",
  7094. " args = request['params'].get('arguments') or {}",
  7095. " send_message({",
  7096. " 'jsonrpc': '2.0',",
  7097. " 'id': request['id'],",
  7098. " 'result': {",
  7099. " 'content': [{'type': 'text', 'text': f\"echo:{args.get('text', '')}\"}],",
  7100. " 'structuredContent': {'echoed': args.get('text', '')},",
  7101. " 'isError': False",
  7102. " }",
  7103. " })",
  7104. " elif method == 'resources/list':",
  7105. " send_message({",
  7106. " 'jsonrpc': '2.0',",
  7107. " 'id': request['id'],",
  7108. " 'result': {",
  7109. " 'resources': [{'uri': 'file://guide.txt', 'name': 'guide', 'mimeType': 'text/plain'}]",
  7110. " }",
  7111. " })",
  7112. " elif method == 'resources/read':",
  7113. " uri = request['params']['uri']",
  7114. " send_message({",
  7115. " 'jsonrpc': '2.0',",
  7116. " 'id': request['id'],",
  7117. " 'result': {",
  7118. " 'contents': [{'uri': uri, 'mimeType': 'text/plain', 'text': f'contents for {uri}'}]",
  7119. " }",
  7120. " })",
  7121. " else:",
  7122. " send_message({",
  7123. " 'jsonrpc': '2.0',",
  7124. " 'id': request['id'],",
  7125. " 'error': {'code': -32601, 'message': method}",
  7126. " })",
  7127. "",
  7128. ]
  7129. .join("\n");
  7130. fs::write(script_path, script).expect("mcp fixture script should write");
  7131. }
  7132. #[cfg(test)]
  7133. mod sandbox_report_tests {
  7134. use super::{format_sandbox_report, HookAbortMonitor};
  7135. use runtime::HookAbortSignal;
  7136. use std::sync::mpsc;
  7137. use std::time::Duration;
  7138. #[test]
  7139. fn sandbox_report_renders_expected_fields() {
  7140. let report = format_sandbox_report(&runtime::SandboxStatus::default());
  7141. assert!(report.contains("Sandbox"));
  7142. assert!(report.contains("Enabled"));
  7143. assert!(report.contains("Filesystem mode"));
  7144. assert!(report.contains("Fallback reason"));
  7145. }
  7146. #[test]
  7147. fn hook_abort_monitor_stops_without_aborting() {
  7148. let abort_signal = HookAbortSignal::new();
  7149. let (ready_tx, ready_rx) = mpsc::channel();
  7150. let monitor = HookAbortMonitor::spawn_with_waiter(
  7151. abort_signal.clone(),
  7152. move |stop_rx, abort_signal| {
  7153. ready_tx.send(()).expect("ready signal");
  7154. let _ = stop_rx.recv();
  7155. assert!(!abort_signal.is_aborted());
  7156. },
  7157. );
  7158. ready_rx.recv().expect("waiter should be ready");
  7159. monitor.stop();
  7160. assert!(!abort_signal.is_aborted());
  7161. }
  7162. #[test]
  7163. fn hook_abort_monitor_propagates_interrupt() {
  7164. let abort_signal = HookAbortSignal::new();
  7165. let (done_tx, done_rx) = mpsc::channel();
  7166. let monitor = HookAbortMonitor::spawn_with_waiter(
  7167. abort_signal.clone(),
  7168. move |_stop_rx, abort_signal| {
  7169. abort_signal.abort();
  7170. done_tx.send(()).expect("done signal");
  7171. },
  7172. );
  7173. done_rx
  7174. .recv_timeout(Duration::from_secs(1))
  7175. .expect("interrupt should complete");
  7176. monitor.stop();
  7177. assert!(abort_signal.is_aborted());
  7178. }
  7179. }