)jvdZddlZddlmZddlmZmZmZmZddl 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.dS)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|}|}|sdS|D].}d}|D]'}|||}|dkrdS|dz }(/dS)zEReturn True when every query token is a case-insensitive subsequence.TrF)lowersplitfind)r r normalizedtokenstokenposchs ;/home/wildlama/.hermes/hermes-agent/hermes_cli/curses_ui.py_query_matchesrsJ [[]] " "F t    B//"c**CQwwuuu 1HCC   4z-_/. targetindexc|dkrdS||dz }|tvrdS||}||ko/||ko||kS)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!'sr  zzt %!) D ~t -C 4::<<  MC399;;$6 M3#))++;MMrorigrrcd}d}d}g}|D]}|||}|dkrdS|||dz }|dkr||dzkr|dz }n|dkr|t||z dz dz}t||r|dz }|dkr|dz }|}|dz}|r-|ddkr!|dt |dz kr|dz }||kr|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;sS E DKIjj[)) 7744   199q QJEE QYY Sta++ +E c " "  QJE !88 QJEAg Yq\Q&&9R=C NNQz!_filter_indices..s47(DG!4r)keycg|]\}}|Sr9r9).0i_s r z#_filter_indices..s ! ! !$!QA ! ! !r)striplistranger, enumerater5r*sort)r6r qscoredr?r r-s r_filter_indicesrIs  A 'E#e**%%&&& Fe$$&&5UA&&   MM1e* % % % KK44K555 ! !& ! ! !!rc0eZdZUdZdZeed<dZeed<dS) _SearchStatez3Mutable search state shared by curses picker loops.Factiver N) __name__ __module__ __qualname____doc__rLbool__annotations__r strr9rrrKrKs3==FDE3OOOOOrrKfilteredcursorcT|s|dfS||vr|d}|||fS)z?Return ``(cursor, cursor_pos)`` inside the filtered index list.r)r)rUrVs r_reconcile_cursorrXs? qy X! 8>>&)) ))r cursor_posdeltac@|s|S|||zt|zS)zEMove through the filtered index list, wrapping like the legacy menus.)r,)rUrVrYrZs r_move_filtered_cursorr\s,  Z%'3x==8 99r scroll_offset visible_rows total_rowsc td|}||kr|}n|||zkr||z dz}tdt|td||z S)z2Clamp scroll offset so the cursor remains visible.rr)maxr+)r]rYr^r_s r_scroll_for_cursorrbsmq,''LM!!" }|3 3 3"\1A5 q#mSJ,E%F%FGG H HHrr<searchcN|jsdS|dkr't|j}d|_d|_dd|fS||jddfvr|jdd |_d S|d kr d|_d S||jd d fvrdSd|cxkrdkr"nn|xjt |z c_d SdS)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. )FFFFrMTr(Nr%)TFT )TTF )rLrRr KEY_BACKSPACE KEY_ENTERchr) curses_modr<rc had_querys r_handle_active_search_keyrps =#"" byy&&   UI%% z'a000|CRC(    byy    z#R,,,   S3 C    rc tjsdSddl}|tj|jdS#t $rYdSwxYw)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)rus r flush_stdinrysl y!!  F 7#344444      sA )A AAupdownselecttogglecancelnonecFt||S)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 FFLLNN 3 33rcddl}||jtdfvrtS||jtdfvrt S||jddfvrtS|tdkrtS|tdkrtS|d kr* | d | }| d n#| d wxYw|d krtS|td td fvr| }|tdtdfvrtS|tdtdfvrt Sd|cxkrdkr&nn#| }d|cxkrdk!ntStStS)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. rNkjrhri rGre<r%[OABrj?) cursesKEY_UPordNAV_UPKEY_DOWNNAV_DOWNrl NAV_SELECT NAV_TOGGLE NAV_CANCELtimeoutrNAV_NONE)rr<rnxtfinals rrr's MMM v}c#hh'''  vC))) vR((( c#hh c#hh byy  NN2   ,,..C NN2    FNN2     "99  3s88SXX& & &LLNNES3s88,,, S3s88,,,%''''4''''' %''''4''''O Os )CC(rF)reserve_bottom draw_footerextra_color_pairs searchable search_labelsc  tjs| S| o duot k ddl t g  f d} | tdt urdn| 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  drdjddjdr)djdkrdnjd}d}t} rng} | | \}} rt||j ntt}t!||\}} ||||} n#t"$r|||} YnwxYwt%d|| z z } t'||| t)|} r<|j r5|s3 || dd |dz jn#j$rYnwxYwt1t|t3t)||| zD]0\} } || } | | z}||z krn||| | |k|1 |||| r|}|jr{t;||\}}}|r(d}t!t|j |\}}|r%|r!t<|}|t>ur|d<dS%|r)tA||}n<|tCd kr d|_VtA||}ntE|}|tFkrtI|||d}na|tJkrtI|||d}nC|t<tLtNfvr-|t<kr r|s||}|t>ur|d<dS) Nrrr%r'r(T)rcz No matches/)(curs_set has_colors start_coloruse_default_colors init_pair COLOR_GREEN COLOR_YELLOWCOLORS COLOR_WHITErKcleargetmaxyxrIr rCrDrX TypeErrorrarbr,addnstrA_DIMerrorrEr+refreshrrLrpr_KEEPrrrrr\rrr)!rrVr]rclabelsmax_ymax_xrUrY items_startr^draw_i filtered_posr?yr<handledconfirmchangedoutcomeactionrr draw_headerdraw_rowrinitial_cursor item_count on_actionr result_holderr use_searchs! r_drawz_run_curses_menu.._draws OOA     "" ""$$$))+++  F$6;;;  F$7<<<$$$  1 111v7I2$FM!^^F#-S1J QS U  %00 u"1OFFL999eJ//00 &7x%H%H" D"-+feU6"R"R"RKK DDD"-+feU"C"CKKKD #1ek&9N&JKK 2!:|S]]!! &,x{A~uqyRXR^____!<-6-S]]ML>>>*Ku555   3 ,,..C}?5N"C551'#,-M1B / v| L Lf22.FJ#%'+*3)J*G*G#*%#7#77>M!$4$*F$"%$!1&#!>!>C(,  !1&#!>!>*622FV##28VZQSTTFFx''28VZQRSSFF J CCC++ +8+ 'i77Ge+++2 a(kU s$EE-,E-1!G G G ) rrrsrtr,rrwrapperryKeyboardInterruptrx)rrrrrrrrfallback cancel_valuerrrrrrs```````` ` @@@r_run_curses_menur^sMn 9    ^ T 9^c->P>PT^>^Jv  i i i i i i i i i i i i i i i i V u #0#35#@#@}QlR  xzzsAB C-CC)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}|r||dz}|dd|dz ||ddd|dz |jn#|j$rYnwxYwdS)Nrrru: ↑↓ navigate SPACE toggle ENTER confirm ESC cancelr')rA_BOLDr color_pairrrr)rrrrhattrrs r _draw_headerz&curses_checklist.._draw_header+s  ME  "" .**1--- NN1a 5 9 9 9 NN1L 6<     |    D qsA0A88 BBc"ddl}| vrdnd}|rdnd}d|d|d |}|j} |r3|j} |r| |dz} ||d||dz | dS#|j$rYdSwxYw)Nru✓r→z [z] rrA_NORMALrrrrr) rrr? is_cursorrrcheckarrowlineattrchosenr6s r _draw_rowz#curses_checklist.._draw_row;s f#"+/5//E//U1X//  -=D  "" -))!,,,  NN1auqy$ 7 7 7 7 7|    DD s"B B BcBddl} }|rztd|t|z dz }|j}|r||dz}||dz ||||z dz |dSdS#|j$rYdSwxYw)Nrrr')rrar,rrrrr) rrrr status_textsxsattrrrs r _draw_footerz&curses_checklist.._draw_footerJs  #)F++K REC $4$44q899 $$&&2V..q111Euqy"k52:>5QQQQQ  R R|    DD sBB BBc|tkr|htS|tkrt SSN)rsymmetric_difference_updaterrset)rrVrrs r _on_actionz$curses_checklist.._on_actionXsF Z    . .x 8 8 8L Z  v;; rrrrc*tSr)_numbered_fallback)rr6rrrsrr;z"curses_checklist..is+E5(NT]^^r) rrrrrrrrrr)rrr,rR) rr6rrrrrrrrs ````` @rcurses_checklistrs&X ]]F             u:: &-A$-7LL4y//^^^^^^^^#    r)r descriptionrrrc g |r| d fd }fd}fd}tt|||dfdrtnd 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}|r||dz}||d |dz ||dz } D]3}||dz krn'||d||dz |j|dz }4 r||jr d|jd}n rd}nd}||d||dz |j|dz }n#|j $rYnwxYw|dzS)Nrrr Search: +▎ BACKSPACE edit Ctrl+U clear ESC stopu; ↑↓ navigate ENTER/SPACE select / search ESC cancelu1 ↑↓ navigate ENTER/SPACE select ESC cancel) rrrrrrrLr rr) rrrrcrrowrdlinehint desc_linesrrs rrz&curses_radiolist.._draw_headersV  ME  "" .**1--- NN35%!)U ; ; ; 1HC$  %!)##EsAueaiIIIq Qf0V]0`FL``` QZP NN34FL A A A 1HCC|    D QwsCC C%$C%c&ddl}| krdnd}|rdnd}d|d|d |}|j} |r3|j} |r| |dz} ||d||dz | dS#|j$rYdSwxYw) Nru●u○rrz (z) rr) rrr?rrrradiorrrr6rs rrz#curses_radiolist.._draw_rows MMx%.3/5//E//U1X//  -=D  "" -))!,,,  NN1auqy$ 7 7 7 7 7|    DD s$B BBc,|ttfvr|SSr)rr)rrVrs rrz$curses_radiolist.._on_actions j*- - -Mrrc(tSr)_radio_numbered_fallback)rr6rrsrr;z"curses_radiolist..s1%.YYr rrrrrrrrrrr) splitlinesrr,rC) rr6rrrrrrrrs ```` ` @rcurses_radiolistrns.!J. ++-- <       u:: YYYYYYY#%/9d5kkkT    rc ttd|tjttdtjt |D]C\}}||krtdtjnd}td|d|dzdd |Dt ttd |dzd tj}|s|St|dz }d |cxkrt|krnn|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 rYELLOWDIMrEGREENinputrBintr, ValueErrorrEOFError) rr6rrr?r markervalr0s rrrsi %u . ./// %96: F FGGGe$$11545MMz6<000z /6//AE/////0000 GGG E@x!|@@@&*MMNNTTVV O#hhl  c%jj J )8 4s4AD+80D+)D++EECancel) cancel_labelr default_indexrc$ t||gzt| dfd }fd} fd}tt|tdz t|||d fddrtnd 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). NcPddl} |j}|r||dz}|dd|dz |r||jr d|jd}nrd}nd}|dd||dz |jn#|j$rYnwxYwdS) Nrrrrru8 ↑↓ navigate ENTER confirm / search ESC/q cancelu. ↑↓ navigate ENTER confirm ESC/q cancelr') rrrrrrLr rr) rrrrcrrrrrs rrz*curses_single_select.._draw_headers  ME  "" .**1--- NN1a 5 9 9 9 Hf0V]0`FL``` HQG NN1auqy&, ? ? ? ?|    D qsBB B#"B#c ddl}|rdnd}d|d |}|j}|r3|j}|r||dz} ||d||dz |dS#|j$rYdSwxYw)Nrrrrr) rrr?rrrrrr all_itemss rrz'curses_single_select.._draw_row s "+)5))9Q<))  -=D  "" -))!,,,  NN1auqy$ 7 7 7 7 7|    DD sA55 BBcV|tkr |krdn|S|tkrdStSr)rrr)rrV cancel_idxs rrz(curses_single_select.._on_actions9 Z  "Z//44V ; Z  4 rrc&tSr)_numbered_single_fallback)rrrsrr;z&curses_single_select..)s25)ZPPrrr)rCr,rr+) rr6rrrrrrrrs ` ` @@rcurses_single_selectrs U |n,IUJ$      =#i..1*<==y>> PPPPPP)3=d9ooo    rrctd|dt|dD]\}}td|d|t tdt|d}|sdSt |dz }d |cxkrt|kr nn||kr|S||krdSn#t ttf$rYnwxYwdS) z/Text-based numbered fallback for single-select.r rrrz Choice [1-rNr) rrEr r,rBr rrr)rr6rr?r rr0s rrr0s+     eQ''!!5 1     GGG  23u::2223399;; 4#hhl  c%jj S:%5%5J *  4  )8 4     4s4C6C;CCCc \t|}ttd|tjttdtj t |D]A\}}||vrtdtjnd}td|d|dzd d |B|r7||} | r*ttd| tjt ttd tj } | snut| dz } d | cxkrt|krnn| | hn#tttf$r|cYSwxYwtJ|S) z8Text-based toggle fallback for terminals without curses.rz& Toggle by number, Enter to confirm. Tu[✓]z[ ]rrrrrz" Toggle # (or Enter to confirm): r)rrr rr r rEr r rBr r,rrrr) rr6rrrrr?r rrrr0s rrrHs]]F %u . ./// %96: F FGGG!%(( 5 5HAu56&[[U7FL111eF 3v33A333E33 4 4 4 4  ?#)F++K ?e0;00&*==>>>  "BFJOOPPVVXXC c((Q,CC$$$$#e**$$$$$22C5999-x8 " " "! ! ! ! " %( Ms;;E=7AE==FF)r N)rr)/rQrr dataclassesrtypingrrrrhermes_cli.colorsrr rTrRr frozensetrr r!floatr1r5rIrKtuplerXr\rbrpryrrrrrrrrobjectrrrrrrrrr9rrr(s  !!!!!!000000000000++++++++#cd,7##NNSNTNNNN(3s333s3ut|3333lCEDL2"49"S"T#Y""""2  *S *3*5c?****:3i:!$:25:>A::::: I I$' I7: IHK I I I I I##".# 4t ####L    0      4S4444&/#/#////h rrrrrt'+59 XXX X 9X#hX HtO X #c(S12 X XXXXX|[ "&"[[[ [ 9[[ $J [ t [[ [[[[|  9   :G ! GGG G 9GG  G  G 4ZGGGGT  9 4Z :6:     9 #h H  #c(S12   X      r