`gja+UdZddlmZddlZddlZddlZddlZddlmZddl m Z m Z m Z m Z ddlmZej"eZiZded <ej,Zd Zej2d Zdd Zdd ZddZddZddZddZ ddZ! d d dZ" d! d"dZ#dZ$d#dZ%d$dZ&d%dZ'y)&udisk-cleanup plugin — auto-cleanup of ephemeral Hermes session files. Wires three behaviours: 1. ``post_tool_call`` hook — inspects ``write_file`` and ``terminal`` tool results for newly-created paths matching test/temp patterns under ``HERMES_HOME`` and tracks them silently. Zero agent compliance required. 2. ``on_session_end`` hook — when any test files were auto-tracked during the just-finished turn, runs :func:`disk_cleanup.quick` and logs a single line to ``$HERMES_HOME/disk-cleanup/cleanup.log``. 3. ``/disk-cleanup`` slash command — manual ``status``, ``dry-run``, ``quick``, ``deep``, ``track``, ``forget``. Replaces PR #12212's skill-plus-script design: the agent no longer needs to remember to run commands. ) annotationsN)Path)AnyDictOptionalSet) disk_cleanupzDict[str, Set[str]]_recent_test_trackspathz#(?:^|\s)(/[^\s'\"`]+|\~/[^\s'\"`]+)c|xs|xsdS)Ndefault)task_id session_ids P/root/.hermes/venv/lib/python3.12/site-packages/plugins/disk-cleanup/__init__.py _tracker_keyr4s  -j -I-c|dk7ryt||}t5tj|t j t |dddy#1swYyxYw)z=Record that we tracked *path* as *category* during this turn.testN)r_lockr setdefaultsetaddstr)rrr categorykeys r _record_trackr8sV6 w +C B&&sCE266s4yABBBs 7AA"ct||}t5tj|t cdddS#1swYyxYw)z3Pop the set of test paths tracked during this turn.N)rrr popr)rrrs r_drainr!As8 w +C 3"&&sCE2333s ;Ac  t|j}|jsyt j |}|yt j t||d}|rt||||yy#t$rYywxYw)z%Best-effort auto-track. Never raises.NTsilent) r expanduser Exceptionexistsdgguess_categorytrackrr)path_strrrprnewlys r_attempt_trackr.Hs N % % ' 88:  #H HHSVXd 3E gz1h7  sA77 BBcj|jt}t|tr|r|hSt S)N)get_WRITE_FILE_PATH_KEY isinstancerrargsr s r_extract_paths_from_write_filer5Xs+ 88( )Dc*tD6>>rcb|jd}t|tr|r|hStS)Nr )r0r2rrr3s r_extract_paths_from_patchr7]s, 88F Dc*tD6>>rct}|jdxsd}t|trC|rA t j |dD]%}|j ds|j|' t|tr9t|dkr+tj|D]}|j||S#t$rYVwxYw)zBest-effort: pull candidate filesystem paths from a terminal command and its output, then let ``guess_category`` / ``is_safe_path`` filter. commandT)posix)/~i) rr0r2rshlexsplit startswithr ValueErrorlen_TERMINAL_PATH_REGEXfindall)r4resultpathscmdtokmatchs r_extract_paths_from_terminalrJgseE ((9  #C#s {{3d3 #>>*-IIcN # &#3v;#5)11&9 E IIe   L    s+B>B>> C  C c t|tsyt}|dk(r t|}n6|dk(r t |}n%|dk(rt |t|t r|nd}ny|D]}t|||y)z8Auto-track ephemeral files created by recent tool calls.N write_filepatchterminalr:)r2dictrr5r7rJrr.) tool_namer4rErr tool_call_id_ candidatesr+s r_on_post_tool_callrTs{ dD !5JL 3D9 g .t4 j 1$*VUXBY_ab 6x*56rc td|}t5ttj }dddD]!}|s||k7s tj |d#|s|sy t j}|ds|dr;t jd|dd|ddt j|d yy#1swYxYw#t$r }tjd|Yd}~yd}~wwxYw) zBRun quick cleanup if any test files were tracked during this turn.r:Nz%disk-cleanup quick cleanup failed: %sdeleted empty_dirsz"AUTO_QUICK (session_end): deleted=z dirs=z freed=freed) r!rlistr keysr r(quickr&loggerdebug_logfmt_size) r completed interruptedrRdrained_session task_bucketsrsummaryexcs r_on_session_endrfsR,O 8/4467 8/ 3*$  # #C ./ <((* yW\2 01C0DEL)*'"++gg>N2O1P R 388   Manually add a path to tracking forget Stop tracking a path (does not delete) Categories: temp | test | research | download | chrome-profile | cron-output | other All operations are scoped to HERMES_HOME and /tmp/hermes-*. Test files are auto-tracked on write_file / terminal and auto-cleaned at session end. c d|dd|ddtj|dd}|jdr|d t|dd z }|S) Nz[disk-cleanup] Cleaned rVz files + rWz empty dirs, freed rX.errorsz z error(s); see cleanup.log.)r(r_r0rB)rdbases r _fmt_summaryrksr !')"4!5Y < !!4R[[AQ5R4SST V {{8 $s78,-..IJJ Krcn|jj}|r|ddvrtS|d}|dk(r'tjtj S|dk(rtj \}}td|D}td|D}ddt|d tj|d g}|D]}|jd |d d |d!|jdt|d tj|d |D]}|jd |d d |d!|jdtj||zdj|S|dk(rttjS|dk(rtj} tj \} } t| g}| r{td| D} |jdt| dtj| d| D]}|jd|d d |d!|jddj|S|dk(rpt|dkry|d} |d}|tjvrd|dttjStj | |d !r d"| d#|d$Sd%| S|d&k(rGt|dkry'tj"|d}|rd(|d)|dk(rd*nd+d,|dd-Sd.|dSd/|d0tS)1Nr>-h--helphelpstatuszdry-runc3&K|] }|d ywsizeNr.0is r z _handle_slash..s0a& 0c3&K|] }|d ywrrrrts rrwz _handle_slash..s4!F)4rxz"Dry-run preview (nothing deleted):z Auto-delete : z files ()z [rz] r z Needs prompt: z Total potential:  r[deepc3&K|] }|d ywrrrrts rrwz _handle_slash..s7Qqy7rxz item(s) need confirmation (z):z [zL Run `/disk-cleanup forget ` to skip, or delete manually via terminal.r*z,Usage: /disk-cleanup track r zUnknown category 'z '. Allowed: Tr#zTracked z as 'z'.z@Not tracked (already present, missing, or outside HERMES_HOME): forgetz"Usage: /disk-cleanup forget zRemoved z tracking entryiesz for rhzNot found in tracking: zUnknown subcommand: z )stripr? _HELP_TEXTr( format_statusrpdry_runsumrBr_appendjoinrkr[ALLOWED_CATEGORIESsortedr*r)raw_argsargvsubautoprompt auto_size prompt_sizelinesitem quick_summary_auto prompt_itemsrspath_argrns r _handle_slashrs >>  ! ! #D 4766 q'C h ,, izz| f0400 4V44 0s4yk"++i2H1I K  ED LL5j!1 2"T&\NC D E s6{m8BKK 4L3MQ O  ED LL5j!1 2"T&\NC D E #BKK K0G$H#I J yy g~BHHJ'' f}  jjl|m,- 7,77D LLS&'(KK%&b* % G s4 #3"4BtF|nEF G LL) yy g~ t9q=A77 200 0$XJ/"2#8#89:<  88Hht 4hZuXJb9 9Nj   h t9q=7 IId1g qcqAv5'AtAwiq Q /Qy9 "#d:, 77rc|jdt|jdt|jdtdy)Npost_tool_callon_session_endz disk-cleanupz2Track and clean up ephemeral Hermes session files.)handler description) register_hookrTrfregister_commandr)ctxs rregisterr5sA&(:;&8Hr)rrrrreturnr) rrrrr rrrrNone)rrrrrSet[str])r+rrrrrrr)r4Dict[str, Any]rr)r4rrErrr)r:NNr:r:r:)rPrr4zOptional[Dict[str, Any]]rErrrrrrQrrRrrr)r:TF) rrr`boolrarrRrrr)rdrrr)rrrz Optional[str])rr)(__doc__ __future__rloggingrer> threadingpathlibrtypingrrrrr:r r( getLogger__name__r\r __annotations__Lockrr1compilerCrrr!r.r5r7rJrTrfrrkrrrrrrsO(# ++   8 $,.(- !rzz"HI.B3 8 ? ?4%) 66 "6 6 6  6  6 6 68! ! ! !  !  ! P $O8lr