`gjvFdZddlZddlmZddlmZmZmZmZddl m Z m Z de de de fd Zed Zd e d ede fd Zde de de dedzfdZde de dedzfdZdee de deefdZeGddZdeededeeeffdZdeededededef dZdededededef d Zd!ed"edee e e ffd#ZdDd$Zd%Zd&Zd'Z d(Z!d)Z"d*Z#de fd+Z$d!ede fd,Z%e&Z'd-dd.d.dd/d0Z(ddd1d2e dee d3eed4eedzd5eeeege fdeef d6Z) dEddd.d7d2e dee d3ed4edzd8e dzd9e defd:Z*d2e dee d3ed4edef d;Z+ dEded?e d9e dedzf d@Z,d2e dee dAededzfdBZ- dFd2e dee d3eed4eed5eeeege fdeef dCZ.y)GzShared curses-based UI components for Hermes CLI. Used by `hermes tools` and `hermes skills` for interactive checklists. Provides a curses multi-select with keyboard navigation, plus a text-based numbered fallback for terminals without curses support. N) dataclass)CallableListOptionalSet)Colorscolorlabelqueryreturnc|j}|jj}|sy|D]*}d}|D]!}|j||}|dkry|dz }#,y)zEReturn True when every query token is a case-insensitive subsequence.TrF)lowersplitfind)r r normalizedtokenstokenposchs G/root/.hermes/venv/lib/python3.12/site-packages/hermes_cli/curses_ui.py_query_matchesrsrJ [[] "F   B//"c*CQw 1HC    z-_/. targetindexc|dk(ry||dz }|tvry||}||jk(xr(||jk7xr||jk(S)zTrue if position ``index`` in ``target`` starts a word. Mirrors ``isBoundary`` in the TS scorer: start-of-string, after a separator char, or a lower->upper camelCase transition. rTr)_WORD_BOUNDARYrupper)rrprevcurs r _is_boundaryr!'sc  z %!) D ~ -C 4::<  MC399;$6 M3#))+;MMrorigrrcd}d}d}g}|D]}|j||}|dkry|j||dz }|dk\r||dzk(r|dz }n|dk\r|t||z dz dz}t||r|dz }|dk(r|dz }|}|dz}|r!|ddk(r|dt |dz k(r|dz }||k(r|d z }|t |d zz}|S) a#Score one token against a target. None if the token isn't a subsequence. A faithful port of ``fuzzyScore`` in ui-tui/src/lib/fuzzy.ts and web/src/lib/fuzzy.ts so all three surfaces rank model ids identically: contiguous runs, word-boundary / first-char starts, prefix matches, and exact matches all score higher than scattered subsequence hits. ``lower`` is ``orig`` lowercased; matching is done against ``lower`` while boundary detection uses ``orig`` (so the camelCase rule works), exactly as in the TS scorer. rNrg{Gz?)rappendminr!len) r"rrscorer search_from positionsridxs r _token_scorer1;s E DKIjj[) 7   19q QJE QY Sta+ +E c " QJE !8 QJEAg +0Yq\Q&9R=C NQz!_filter_indices..s47(DG!4r)key)striplistranger, enumerater5r*sort)r6r qscoredir r-_s r_filter_indicesrFs  A E#e*%&& Fe$&5UA&   MM1e* % &  KK4K5 !$!QA !! !s? Bc.eZdZUdZdZeed<dZeed<y) _SearchStatez3Mutable search state shared by curses picker loops.Factiver N) __name__ __module__ __qualname____doc__rIbool__annotations__r strr9rrrHrHs=FDE3OrrHfilteredcursorcF|s|dfS||vr|d}||j|fS)z?Return ``(cursor, cursor_pos)`` inside the filtered index list.r)r)rRrSs r_reconcile_cursorrUs4 qy X! 8>>&) ))r cursor_posdeltac2|s|S|||zt|zS)zEMove through the filtered index list, wrapping like the legacy menus.)r,)rRrSrVrWs r_move_filtered_cursorrYs%  Z%'3x=8 99r scroll_offset visible_rows total_rowsc td|}||kr|}n|||zk\r||z dz}tdt|td||z S)z2Clamp scroll offset so the cursor remains visible.rr)maxr+)rZrVr[r\s r_scroll_for_cursorr_sYq,'LM!" }|3 3"\1A5 q#mSJ,E%FG HHrr<searchc\|jsy|dk(r(t|j}d|_d|_dd|fS||jddfvr|jdd |_y |d k(rd|_y ||jd d fvryd|cxkrdkr"ny|xjt |z c_y y)zHandle a key while the search prompt is active. Returns ``(handled, confirm, changed)``. Active search consumes query editing keys, but leaves navigation keys for the menu loop to handle. )FFFFrJTr(Nr%)TFT )TTF )rIrOr KEY_BACKSPACE KEY_ENTERchr) curses_modr<r` had_querys r_handle_active_search_keyrms ==" by&   UI%% z''a00||CR(   by   z##R,,  S3   C  rc tjjsyddl}|j tj|j y#t $rYywxYw)uFlush any stray bytes from the stdin input buffer. Must be called after ``curses.wrapper()`` (or any terminal-mode library like simple_term_menu) returns, **before** the next ``input()`` / ``getpass.getpass()`` call. ``curses.endwin()`` restores the terminal but does NOT drain the OS input buffer — leftover escape-sequence bytes (from arrow keys, terminal mode-switch responses, or rapid keypresses) remain buffered and silently get consumed by the next ``input()`` call, corrupting user data (e.g. writing ``^[^[`` into .env files). On non-TTY stdin (piped, redirected) or Windows, this is a no-op. Nr)sysstdinisattytermiostcflushTCIFLUSH Exception)rrs r flush_stdinrvsG yy!  7#3#34    sA.A AAupdownselecttogglecancelnonec6t||jS)uVRead one keypress and normalize it to a menu action. Decodes raw arrow-key escape sequences in addition to the translated ``curses.KEY_*`` values. Even with ``keypad(True)`` (which ``curses.wrapper`` sets), some terminals/terminfo entries deliver cursor keys as raw CSI/SS3 byte sequences — ``getch()`` then returns ``27`` (ESC) followed by e.g. ``[`` ``A``. Treating that leading ``27`` as a cancel is what made the setup wizard's provider/model pickers bail to the numbered fallback the moment a user pressed up/down. Returns one of the ``NAV_*`` constants. A lone ESC (no continuation byte within a short window) is the only thing that maps to ``NAV_CANCEL`` via the escape path; ``q`` also cancels. Unknown sequences map to ``NAV_NONE`` so the caller simply ignores them rather than misfiring. )_decode_menu_keygetch)stdscrs r read_menu_keyrs FFLLN 33rcddl}||jtdfvrtS||jtdfvrt S||j ddfvrtS|tdk(rtS|tdk(rtS|d k(r |jd |j}|jd |d k(rtS|td td fvr|j}|tdtdfvrtS|tdtdfvrt Sd|cxkrdkr+ntS|j}d|cxkr dkrtStStStS#|jd wxYw)zNormalize an already-read keypress to a menu action. Split out from ``read_menu_key`` so search-aware loops can peek the raw key (e.g. to catch ``/``) before falling back to nav decoding. rNkjrerf rBrb<r%[OABrg?) cursesKEY_UPordNAV_UPKEY_DOWNNAV_DOWNri NAV_SELECT NAV_TOGGLE NAV_CANCELtimeoutrNAV_NONE)rr<rnxtfinals rr~r~'sa  v}}c#h''  vC)) vR(( c#h c#h by  NN2 ,,.C NN2  "9  3s8SX& &LLNES3s8,, S3s8,,%'4'O %'4'O(O O) NN2 s !E11FrF)reserve_bottom draw_footerextra_color_pairs searchable search_labelsc ` tjjs| S| xr duxrt k( ddl t g  f d} j | tdt urdS| S#t$r| cYSt$r |cYSwxYw)a2 Shared curses single-/multi-select event loop. Owns every piece the three public menus used to duplicate verbatim: the non-TTY guard, ``curses.wrapper`` setup (cursor hide + color pairs), the per-frame ``clear``/``getmaxyx``/``refresh`` cycle, scroll-offset math, row iteration, the ``read_menu_key`` dispatch with ``NAV_UP``/``NAV_DOWN`` cursor wrap, ``flush_stdin``, and the ``KeyboardInterrupt`` / curses- unavailable fallback. Per-menu behavior is supplied as callbacks so the rendered output stays byte-identical to the old hand-rolled loops. Callbacks / params: draw_header(stdscr, max_y, max_x) -> int Draw the title/hint/description rows. Returns the first screen row index where the scrollable item list should start. When search is active it receives the live ``_SearchState`` via the optional ``search`` keyword (drawn by the menu so the hint line can show it). draw_row(stdscr, y, idx, is_cursor, max_x) -> None Draw one item row. ``idx`` is always the ORIGINAL item index, so per-menu rendering is unchanged whether or not a filter is active. on_action(action, cursor) -> value Reducer for SELECT/TOGGLE/CANCEL. Return ``_KEEP`` to continue the loop; return anything else to resolve the menu with that value. (UP/DOWN cursor movement is handled by the driver itself.) reserve_bottom: number of bottom screen rows kept clear of items (1 = leave the final row blank, matching the old loops). draw_footer(stdscr, max_y, max_x) -> None Optional bottom-row painter (e.g. a status bar). Drawn after the item rows; its row budget must be included in ``reserve_bottom``. extra_color_pairs: also init pair 3 (dim gray) for status bars. fallback() -> value Called when curses errors out on a real TTY (curses unavailable). cancel_value: returned on non-TTY stdin, ESC/cancel, or KeyboardInterrupt. searchable: when true, ``/`` opens a type-to-filter prompt over ``search_labels``. Returned values are always ORIGINAL item indices. search_labels: per-item text used for filtering (required when ``searchable`` is true; length must equal ``item_count``). Nrc  jdjrjjj dj dj dj dr.j djdkDrdn jd}d}t} rng} |j|j\}} rt||jntt}t!||\}} ||||} t%d|| z z } t'||| t)|} r1|jr%|s# |j+| dd |dz j,t1t|t3t)||| zD](\} } || } | | z}||z k\rn||| | |k(|* ||||j5 r|j7}|j8rjt;||\}}}|r%d}t!t|j|\}}|r|rt<|}|t>ur|d<y|rtA||}n/|tCd k(r d|_tA||}n tE|}|tFk(rtI|||d}nQ|tJk(rtI|||d}n9|t<tLtNfvr&|t<k(r r|sC||}|t>ur|d<y[#t"$r|||} YwxYw#j.$rYwxYw) Nrrr%r'r(T)r`z No matches/)(curs_set has_colors start_coloruse_default_colors init_pair COLOR_GREEN COLOR_YELLOWCOLORS COLOR_WHITErHcleargetmaxyxrFr r>r?rU TypeErrorr^r_r,addnstrA_DIMerrorr@r+refreshrrIrmr_KEEPr~rrrrYrrr)!rrSrZr`labelsmax_ymax_xrRrV items_startr[draw_i filtered_posrDyr<handledconfirmchangedoutcomeactionrr draw_headerdraw_rowrinitial_cursor item_count on_actionr result_holderr use_searchs! r_drawz_run_curses_menu.._draws OOA   """$))+  F$6$6;  F$7$7<$$$  11v7I7I2$FM!^F#-1J QS  %0 u"$FFLL9eJ/0 &7x%H" D"-feU6"RK #1ek&9N&JK 2!:|S]! &,,x{A~uqyRXR^R^_-6-S]ML?*u5  ,,.C}}5N"C51'#,-M1B / v|| Lf2.FJ#'*3J*G#*%#77>M!$4$*$"$!1&#!>C(,  !1&#!>*62FV#28VZQSTFx'28VZQRSF J CC+ 8 '7Ge++2 a(k!D"-feU"CKD"<<s$# L)"L9L65L69M  M ) rorprqr,rrwrapperrvKeyboardInterruptru)rrrrrrrrfallback cancel_valuerrrrrrs```````` ` @@@r_run_curses_menur^sn 99   ^ T 9^c->PT^>^Jv i i V u #0#35#@}QRlR  zsAB B B-B-,B-)cancel_returns status_fntitleselectedrrc  tt fd} fd} fd} fd}tdt|||rdndr|ndtfd  S) aCurses multi-select checklist. Returns set of selected indices. Args: title: Header line displayed above the checklist. items: Display labels for each row. selected: Indices that start checked (pre-selected). cancel_returns: Returned on ESC/q. Defaults to the original *selected*. status_fn: Optional callback ``f(chosen_indices) -> str`` whose return value is rendered on the bottom row of the terminal. Use this for live aggregate info (e.g. estimated token counts). Ncddl} |j}|jr||jdz}|j dd|dz ||j ddd|dz |j y#|j $rYywxYw)Nrrru: ↑↓ navigate SPACE toggle ENTER confirm ESC cancelr')rA_BOLDr color_pairrrr)rrrrhattrrs r _draw_headerz&curses_checklist.._draw_header+s MME  "**1-- NN1a 5 9 NN1L 6<< ||   sA*A22BBcddl}| vrdnd}|rdnd}d|d|d |}|j} |r0|j} |jr| |j dz} |j |d||dz | y#|j $rYywxYw)Nru✓r→z [z] rrA_NORMALrrrrr) rrrD is_cursorrrcheckarrowlineattrchosenr6s r _draw_rowz#curses_checklist.._draw_row;sf#"5'E7"U1XJ/ ==D  "))!,,  NN1auqy$ 7||   s!A::B  B c"ddl} }|rjtd|t|z dz }|j}|j r||j dz}|j |dz ||||z dz |yy#|j$rYywxYw)Nrrr')rr^r,rrrrr) rrrr status_textsxsattrrrs r _draw_footerz&curses_checklist.._draw_footerJs #F+KEC $44q89 $$&V..q11Euqy"k52:>5Q ||   sA3A<<B Bcr|tk(rj|htS|tk(r t SSN)rsymmetric_difference_updaterrset)rrSrrs r _on_actionz$curses_checklist.._on_actionXs8 Z   . .x 8L Z v; rrrrc"tSr)_numbered_fallback)rr6rrrsrr;z"curses_checklist..is+E5(NT]^r) rrrrrrrrrr)rrr,rO) rr6rrrrrrrrs ````` @rcurses_checklistrsn&X ]F   u: &A$-L4y/^#  r)r descriptionrrrc g |r|j d fd }fd}fd}tt|||dfdrt Sd S) aCurses single-select radio list. Returns the selected index. Args: title: Header line displayed above the list. items: Display labels for each row. selected: Index that starts selected (pre-selected). cancel_returns: Returned on ESC/q. Defaults to the original *selected*. description: Optional multi-line text shown between the title and the item list. Useful for context that should survive the curses screen clear. searchable: When true, ``/`` opens a type-to-filter prompt. The returned value is always the original item index, not a filtered row position. Ncddl}d} |j}|jr||jdz}|j |d |dz ||dz } D]3}||dz k\rn)|j |d||dz |j |dz }5 r||j rd|jd}n rd}nd}|j |d||dz |j|dz }|dzS#|j$rY|dzSwxYw)Nrrr Search: +▎ BACKSPACE edit Ctrl+U clear ESC stopu; ↑↓ navigate ENTER/SPACE select / search ESC cancelu1 ↑↓ navigate ENTER/SPACE select ESC cancel) rrrrrrrIr rr) rrrr`rrowrdlinehint desc_linesrrs rrz&curses_radiolist.._draw_headers! MME  "**1-- NN35%!)U ; 1HC$ %!)#sAueaiIq   f0V]]#FLL>1_`ZP NN34FLL A 1HCQw||  Qw sCC""C87C8c ddl}| k(rdnd}|rdnd}d|d|d |}|j} |r0|j} |jr| |j dz} |j |d||dz | y#|j $rYywxYw) Nru●u○rrz (z) rr) rrrDrrrradiorrrr6rs rrz#curses_radiolist.._draw_rowsMx%35'E7"U1XJ/ ==D  "))!,,  NN1auqy$ 7||   s"A;;B  B c(|ttfvr|SSr)rr)rrSrs rrz$curses_radiolist.._on_actions j*- -Mrrc tSr)_radio_numbered_fallback)rr6rrsrr;z"curses_radiolist..s1%.Yr rrrrrrrrrrr) splitlinesrr,r>) rr6rrrrrrrrs ```` ` @rcurses_radiolistrns{.!J ++- <  u: Y#%/d5k  6:  rc Dttd|tjttdtjt |D]>\}}||k(rtdtj nd}td|d|dzdd |@t ttd |dzd tjj}|s|St|dz }d |cxkrt|kr|S|S|S#tttf$r|cYSwxYw) z1Text-based numbered fallback for radio selection. z& Select by number, Enter to confirm. u(●)u(○) rr>2. z Choice [default ]: r)printr rYELLOWDIMr@GREENinputr=intr, ValueErrorrEOFError) rr6rrrDr markervalr0s rrrs %$ug ./ %96:: FGe$1545Mz6<<0z 6(!AE":Rw/01 G E.x!|nC@&**MNTTVO#hl  c%j J !x )8 4s$!;D#DDDDDCancel) cancel_labelr default_indexrc t||gzt| dfd }fd} fd}tt|tdz t|||d fddrt Sd S) arCurses single-select menu. Returns selected index or None on cancel. Works inside prompt_toolkit because curses.wrapper() restores the terminal safely, unlike simple_term_menu which conflicts with /dev/tty. When ``searchable`` is true, ``/`` opens a type-to-filter prompt; the returned value is always the original item index (or None for cancel). Nc^ddl} |j}|jr||jdz}|j dd|dz |r||j rd|j d}nrd}nd}|j dd||dz |jy#|j$rYywxYw) Nrrrrru8 ↑↓ navigate ENTER confirm / search ESC/q cancelu. ↑↓ navigate ENTER confirm ESC/q cancelr') rrrrrrIr rr) rrrr`rrrrrs rrz*curses_single_select.._draw_headers MME  "**1-- NN1a 5 9f0V]]#FLL>1_`QG NN1auqy&,, ?||   sBBB,+B,cddl}|rdnd}d|d |}|j}|r0|j}|jr||j dz} |j |d||dz |y#|j $rYywxYw)Nrrrrr) rrrDrrrrrr all_itemss rrz'curses_single_select.._draw_row s"5'9Q<.) ==D  "))!,,  NN1auqy$ 7||   sA//BBcH|tk(r |k\rdS|S|tk(rytSr)rrr)rrS cancel_idxs rrz(curses_single_select.._on_actions/ Z "Z/4 ;V ; Z  rrctSr)_numbered_single_fallback)rrrsrr;z&curses_single_select..)s25)ZPrrr)r>r,rr+) rr6rrrrrrrrs ` ` @@rcurses_single_selectrs U |n,IUJ$  =#i.1*<=y> P)3d9o  :>  rrcntd|dt|dD]\}}td|d|t tdt|dj }|syt |dz }d |cxkrt|kr nn||kr|S||k(ry y#t ttf$rYywxYw) z/Text-based numbered fallback for single-select.r rrrz Choice [1-rNr) rr@r r,r=r r rr )rr6rrDr rr0s rrr0s  Dr eQ'!5 1#Rw ! G  l3u:,c2399;#hl  c%j S:%5J *    )8 4   s(B)+BBB43B4c t|}ttd|tjttdtj  t |D]=\}}||vrtdtjnd}td|d|dzdd |?|r0||} | r&ttd| tj t ttd tj j} | s |St| dz } d | cxkrt|krnn|j| ht#tttf$r|cYSwxYw) z8Text-based toggle fallback for terminals without curses.rz& Toggle by number, Enter to confirm. u[✓]z[ ]rrrrrz" Toggle # (or Enter to confirm): r)rrr rrrr@rr r=r r,rr rr ) rr6rrrrrDr r rrr0s rrrHsK]F %$ug ./ %96:: FG !%( 5HAu56&[U7FLL1eF BvhaAbzE73 4 5 #F+Ked;-0&**=>  "BFJJOPVVXC Mc(Q,CC$#e*$22C59 %  -x8 "! ! "s3E7EE0/E0)r N)rr)/rNro dataclassesrtypingrrrrhermes_cli.colorsrr rQrOr frozensetrr r!floatr1r5rFrHtuplerUrYr_rmrvrrrrrrrr~objectrrrrrrrrr9rrr%s  !00+#cd,7#NNSNTN(3s333s3ut|3lCEDL2"49"S"T#Y"2  *S *3*5c?*:3i:!$:25:>A:: I I$' I7: IHK I I##".# 4t #L 0      4S4&/#/#/h rt'+59 X X 9X#hX HtO X #c(S12 X XX|[ "&"[ [ 9[[ $J [ t [[ [|  9   :G ! G G 9GG  G  G 4ZGT  9 4Z :6:    9 #h H  #c(S12   X r