i!dZddlmZddlZddlZddlmZddlmZddlm Z m Z dd Z ddZ ddZ d dZd dZd dZd!dZd dZd dZd"dZdS)#a `hermes checkpoints` CLI subcommand. Gives users direct visibility and control over the filesystem checkpoint store at ``~/.hermes/checkpoints/``. Actions: hermes checkpoints # same as `status` hermes checkpoints status # total size, project count, breakdown hermes checkpoints list # per-project checkpoint counts + workdir hermes checkpoints prune [opts] # force a sweep (ignores the 24h marker) hermes checkpoints clear [-f] # nuke the entire base (asks first) hermes checkpoints clear-legacy # delete just the legacy-* archives Examples:: hermes checkpoints hermes checkpoints prune --retention-days 3 --max-size-mb 200 hermes checkpoints clear -f None of these require the agent to be running. Safe to call any time. ) annotationsN)datetime)Path)AnyDictnintreturnstrcd}t|pd}|D]?}|dks ||dkr&|dkrt|d|cS|dd|cS|dz}@|ddS) N)BKBMBGBTBrir  z.1fz TB)floatr )runitssizeunits ;/home/piyush/.hermes/hermes-agent/hermes_cli/checkpoints.py _fmt_bytesrs )E a==D $;;$%)++s{{d)),,d,,,,,''''' ' ' '      tsrc tjt|dS#tt f$rYdSwxYw)Nz%Y-%m-%d %H:%M—)r fromtimestamprstrftime TypeError ValueError)rs r_fmt_tsr"+sT%eBii0099:JKKK z "uus36A  A cR tjt|z }n#ttf$rYdSwxYw|dkrdS|dkrt |dS|dkrt |dz dS|dkrt |dz d St |dz d S) Nrrnow<zs agoizm agoiQzh agozd ago)timerr r!r )rages r_fmt_ager(2sikkE"II% z "uu Qwwu Rxxc((!!!! TzzcBh--&&&& U{{cDj//((((#+ % % %%s #&;;argsargparse.Namespacec ddlm}|}|d}td|tdt|dtdt|dtd t|d td |d t |d dd}|rttdddddddddd|dt |dr|jr|jndD]}|dpd}t|dkr d|ddz}|d }|rd!nd"}|d#d} t|d$} td|dd| dd| dd||d%g} | rttd&t| d't | d(dD]2} td| d)d*dt| d+d,3ttd-dS).Nr) store_statusbasezCheckpoint base: zTotal size: total_size_bytesz store/ store_size_bytesz legacy-* legacy_size_byteszProjects: project_countprojectsc0|dpdS)N last_touchrget)ps rzcmd_status..Osquu\**/arT)keyreverse WORKDIRz<60COMMITSz>7z LAST TOUCHz>12z STATElimitworkdirz (unknown)r%u…iexistsliveorphancommitsr4legacy_archiveszLegacy archives (z):c.|ddS)Nmtimerr5)as rr8zcmd_status..csw1B1Brname<40 size_bytes>10z+Clear with: hermes checkpoints clear-legacy) tools.checkpoint_managerr,printrsortedhasattrr>r6lenr() r)r,infor-r2r7wdrAstaterDlastlegacyarchs r cmd_statusrXBs555555 <>>D ?? A ABBB 1rpromptboolc t|d}n&#ttf$rt YdSwxYw|dvS)Nz [y/N]: F)yyes)inputstriplowerEOFErrorKeyboardInterruptrN)riresps r_confirmrtsu((())//117799 ' ( uu < s69AAc ddlm}m}m}|}|ddkr2t |st ddSt d|dt dt|dt d|d t d t| d gt t d |j s td st ddS|}|dr(t dt|dddSt ddS)Nr)CHECKPOINT_BASE clear_allr,r.u4Nothing to clear — checkpoint base does not exist.z/This will delete the ENTIRE checkpoint base at r-z size: z projects: r1z legacy dirs: rEz?All /rollback history for every working directory will be lost.zProceed?Aborted.deletedzCleared. Reclaimed rd.z+Could not clear checkpoint base (see logs).) rMrvrwr,rrArNrrQr6forcert)r)rvrwr,rRrgs r cmd_clearr~sQQQQQQQQQQ <>>D 1$$T/-B-B-I-I-K-K$ DEEEq JDL J JKKK BJt,>'?@@ B BCCC 3D1 3 3444 BC):B ? ?@@ B BCCC GGG KLLL :hz22 jq Y[[F i HJvm/D$E$EHHHIIIq 7888 1rc ddlm}m}|}|dg}|st ddSt d|D}t dt |dt|d|D]2}t d |d d d t|d d 3t t dt dt d|js tdst ddS|}t d|ddt|dddS)Nr) clear_legacyr,rEzNo legacy archives to clear.c3BK|]}|ddVdS)rKrNr5).0rHs r z#cmd_clear_legacy..s0771lA&&777777rzFound z legacy archive(s), total :r;rIrJrKrLzALegacy archives hold pre-v2 per-project shadow repos, moved asidez?during the single-store migration. Delete when you're confidentz)you don't need the old /rollback history.zDelete all legacy archives?rxryzDeleted rzz archive(s), reclaimed rdr{) rMrr,r6rNsumrQrr}rt)r)rr,rRrVtotalrWrgs rcmd_clear_legacyrsCCCCCCCC <>>D XX' , ,F  ,---q 77777 7 7E N3v;; N N*U:K:K N N NOOOMM K4<KKKz$|2D'E'EKKKLLLL GGG MNNN KLLL 5666 :h'DEE jq \^^F cVI& c cz&Q^J_?`?` c c cddd 1rparserargparse.ArgumentParserNonec|t|dd}|dd}|dt d d |t|d d }|dt d |t |dd}|dt dd |dt dd |ddd|t|dd}|dddd|t|d d!}|dddd|td"S)#z8Wire subcommands onto the ``hermes checkpoints`` parser.)funccheckpoints_commandCOMMAND)destmetavarstatusz9Show total size, project count, and per-project breakdown)helpz--limitr?z!Max projects to list (default 20))typedefaultrlistzAlias for 'status')rrprunez0Delete orphan/stale checkpoints and GC the storez--retention-daysz?Drop projects whose last_touch is older than N days (default 7)z --max-size-mbiz`After orphan/stale prune, drop oldest commits per project until total size <= this (default 500)z--keep-orphans store_truez5Skip deleting projects whose workdir no longer exists)actionrclearz9Delete the entire checkpoint base (all /rollback history)z-fz--forcezSkip confirmation promptz clear-legacyz7Delete only the legacy-/ archives from v1 migrationN) set_defaultsrXadd_subparsers add_parser add_argumentr rZrhr~r)rsubsp_statusp_listp_prunep_clearp_legacys r register_clirs\ Z(((  &;Y O OD HH )#rBDDD z*** __ !F  R888 X&&&oo ?G +#q_aaa sCSTTT ),UWWW i(((oo HG y8::: i((( FH $ ,9;;; /00000r)rr r r )rrr r )r)r*r r )rir r rj)rrr r)__doc__ __future__rargparser&rpathlibrtypingrrrr"r(rXrZrhrtr~rrrrrsD*#"""""      & & & & % % % % P     2        4    4/1/1/1/1/1/1r