yj3$dZddlmZddlZddlmZmZddlmZm Z m Z m Z m Z dZ dZe fdd ZeGddZGddZdZdZdZdZdZ dddZdS) uCtrl+S prompt stash — pure state machine for the classic CLI composer. Park a half-written prompt, send something else, then bring the draft back. Mirrors Claude Code's ``ctrl + s to stash prompt`` affordance. The state machine lives here (no prompt_toolkit imports) so it can be unit tested directly; ``cli.py`` owns only the keybinding and the rendering. Gesture ------- - Buffer has content → push it onto the stash, clear the composer. - Buffer empty, 1 item → pop it straight back into the composer. - Buffer empty, 2+ items → open the browse panel (↑↓ / Enter / D / Esc). Newest-first ordering: index 0 is always the most recently stashed draft, so the common "undo my last Ctrl+S" case is a single keystroke. Nothing is written to disk. Drafts frequently contain credentials, prompts under NDA, or pasted secrets, and a session-scoped stash keeps that material in memory only. Callers that later want cross-restart persistence must route through ``get_hermes_home()`` rather than hardcoding ``~/.hermes``. ) annotationsN) dataclassfield)AnyListOptionalSequenceTuple<textstrwidthintreturncV|sdS|dddd}|dddd}d|}|dkr#t||kr|d |dz d zS|S) uCollapse a possibly multi-line draft into one preview line. Newlines and tabs become ``⏎``/space so a 40-line draft still renders as a single panel row, and the result is ellipsized to ``width`` display chars. z   u ⏎   Nu…)replacejoinsplitlen)r rflats =/home/ragecks/.hermes/hermes-agent/hermes_cli/prompt_stash.py build_previewr%s r << % % - -dD 9 9D <<g & & . .tS 9 9D 88DJJLL ! !D qyySYY&&KeaiK 5(( KcdeZdZUdZded<eeZded<dZded <d Z ded <ddZ dS) StashEntryz@One parked draft: exact text plus any images that were attached.rr )default_factoryz List[Any]imagesgfloat stashed_atrpreviewrdictcR|jt|j|j|jdS)z?Render in the shape ``HermesCLI._render_stash_panel`` consumes.r r$r&r')r listr$r&r'selfs ras_dictzStashEntry.as_dict>s/I4;''/|    r N)rr() __name__ __module__ __qualname____doc____annotations__rr+r$r&r'r.r rr"r"5syJJ IIId333F3333JG      r r"ceZdZdZeddd&dZd'dZd(d Zed)d Z d*dZ d+dZ d+dZ d,d-dZ d.d/dZd.d0dZd1dZd2dZd(d Zd1d!Zd3d#Zd(d$Zd4d%ZdS)5 PromptStashzSession-scoped stack of parked composer drafts. Pure state: no I/O, no prompt_toolkit, no global clock beyond ``time.monotonic`` (injectable for tests via ``clock``). N) max_itemsclockr7rcg|_tdt||_|p tj|_d|_d|_dS)NrFr) _itemsmaxr _max_itemstime monotonic_clock panel_open panel_cursor)r-r7r8s r__init__zPromptStash.__init__OsC(* aY00-t~ r rc*t|jSN)rr:r,s r__len__zPromptStash.__len__Xs4;r boolc*t|jSrD)rFr:r,s r__bool__zPromptStash.__bool__[sDK   r List[StashEntry]c*t|jS)u=Newest-first list of entries (a copy — mutate via the API).)r+r:r,s ritemszPromptStash.items_sDK   r List[dict]c$d|jDS)z.Entries as plain dicts for the panel renderer.c6g|]}|Sr4)r.).0es r z*PromptStash.panel_rows..fs 111 111r )r:r,s r panel_rowszPromptStash.panel_rowsds11T[1111r rcVt|j}|sdS|jrd|dnd|S)uStatus-bar indicator, or ``""`` when the stash is empty. ``📌 2`` when idle, ``📌 2 ▲`` while the browse panel is open, so the user can always tell a parked draft exists without opening anything. ru📌 u ▲)rr:r@r-ns r indicatorzPromptStash.indicatorhs?    2"&/Bq{q{{Br ctt|j}|sdS|dkrd|jdjSd|dS)z8Composer placeholder text advertising the stashed draft.rrzCtrl+S to restore: rzCtrl+S to browse z stashed drafts)rr:r'rTs rplaceholder_hintzPromptStash.placeholder_hintssO    2 66AQ)?AA A515555r r r$Optional[Sequence[Any]]cVt|}|pds|sdSt|pdt|pg|t |pdpd}|jd||j|jd=d|_ d|_ dS)uPush a draft. Returns False (no-op) for a blank buffer. A buffer that is empty or whitespace-only is not worth parking and must stay a no-op, otherwise Ctrl+S on an empty composer would push a junk entry instead of triggering the restore half of the gesture. Text is stored verbatim — leading/trailing whitespace and newlines are preserved so a restore round-trips byte-for-byte. rFz (images only)r*rNT) rFstripr"r+r?rr:insertr<r@rA)r-r r$ has_imagesentrys rstashzPromptStash.stash~s&\\  !!## J 5 "%%{{}}!$*"--@     1e$$$ K(( )tr rindexOptional[Tuple[str, List[Any]]]c|jr!d|cxkrt|jksndS|j|}|jsd|_||j|_|jt|jfS)z;Remove and return ``(text, images)`` at ``index``, or None.rNF) r:rpopr@ _clamp_cursorrAr r+r$)r-r`r^s rrczPromptStash.pops{ 1#@#@#@#@DK0@0@#@#@#@#@4 &&{ $#DO ..t/@AAz4 ----r Optional[StashEntry]cp|jr!d|cxkrt|jksndS|j|S)z2Return the entry at ``index`` without removing it.rN)r:r)r-r`s rpeekzPromptStash.peeksF{ 1#@#@#@#@DK0@0@#@#@#@#@4{5!!r NonecT|jd|_d|_dSNFr)r:clearr@rAr,s rrkzPromptStash.clears+ r valuec |jsdStdtt|t |jdz S)Nrr)r:r;minrr)r-rls rrdzPromptStash._clamp_cursors@{ 11c#e**c$+&6&6&:;;<<??@@  r )r7r)rr)rrF)rrI)rrL)rrrD)r rr$rYrrF)r)r`rrra)r`rrre)rrh)rlrrr)rsrrr)rra)r/r0r1r2MAX_STASH_ITEMSrBrErHpropertyrKrRrVrXr_rcrgrkrdrprrrurxr{r4r rr6r6Hs ,;$    !!!!!!!X!2222 C C C C66666.....""""" ==== !!!!    r r6noopstashedrestoredrprrr_ buffer_textr$rY+Tuple[str, Optional[Tuple[str, List[Any]]]]c|jr|tdfS|pds|r(|||r t dfSt dfSt|}|dkr t dfS|dkrt| dfS| tdfS)aDecide what one Ctrl+S press does. Returns ``(action, payload)``. ``payload`` carries ``(text, images)`` for :data:`ACTION_RESTORED`, else None. This is the whole decision table in one pure function so the keybinding handler in ``cli.py`` stays a thin adapter. Nrrr) r@rrACTION_CLOSE_PANELr[r_ACTION_STASHED ACTION_NOOPrACTION_RESTOREDrcrpACTION_OPEN_PANEL)r_rr$counts rresolve_ctrl_srs ( !4'' r  ""!f! ;;{F + + (!4' 'D   JJE zzD   zz ! ,,  d ""r )r rrrrrrD)r_r6rrr$rYrr)r2 __future__rr= dataclassesrrtypingrrrr r PREVIEW_WIDTHr|rr"r6rrrrrrr4r rrsD.#""""" ((((((((77777777777777 +8                $QQQQQQQQn  " '+#######r