`gj_dZddlmZddlZddlmZmZmZeegee fZ ejdZ hdZ dZd'd(dZd)d Zd*d Zd+d Zd*d Zd,d Zd-dZd,dZd.dZd/dZd0dZ d1dZd2dZd3dZd4dZd5dZd6dZd7dZ d8dZ!d9dZ" d:dZ#GddZ$d;dZ%dd$d!ddd% d?d&Z(y)@a"Authoritative project -> repo -> lane -> session tree builder. This is the single source of truth for how the desktop sidebar groups sessions into projects, repos, and lanes. It is pure (all git resolution is injected via ``resolve``) so it can be unit-tested with fixtures and reused by the gateway's ``projects.tree`` / ``projects.project_sessions`` RPCs. It deliberately mirrors the desktop's former client-side grouping (the old ``workspace-groups.ts``) so the emitted ids and lane keys stay byte-compatible with the renderer's persisted state (pins, manual ordering, dismissal), which all key off these exact strings: - explicit project id .......... ``p_`` (from projects.db) - auto/discovered project id ... the repo root path - repo node id ................. the repo root path - main branch lane id .......... ``::branch::`` (or ``::branch::``) - kanban bucket lane id ........ ``::kanban`` - linked worktree lane id ...... the worktree path The one correctness upgrade over the client version: linked worktrees are folded under their MAIN repo via a git common-dir probe (injected as ``resolve``), instead of being treated as separate repos (``git rev-parse --show-toplevel`` returns the worktree's own root, which is why the client double-counted them). ) annotationsN)AnyCallableOptionalz,^(.*[/\\]\.worktrees)[/\\]t_[0-9a-f]+[/\\]?$>maintrunkmasterdeveloprc4|d|xsdjS)zGThe one definition of a main-checkout lane id (must match the desktop). ::branch::)strip) repo_rootbranchs K/root/.hermes/venv/lib/python3.12/site-packages/tui_gateway/project_tree.py_branch_lane_idr-s![ FLb#7#7#9": ;;c |dS)N::kanban)rs r_kanban_lane_idr2s[ !!rctjd|xsdjdDcgc]}|s| c}Scc}w)Nz[/\\]r z/\)resplitrstrip)pathss r _segmentsr;s2xx4:2*=*=e*DE K!A KK Ks;;c|xsdj}ttjd|xs|j dS)Nr z^[A-Za-z]:[/\\])\z//)rboolrmatch startswith)rvalues r_is_windows_pathr%?s= ZR   E +U3 4 V8H8H8VVrcxt|}t|r|Dcgc]}|jc}S|Scc}w)zPath segments suitable for identity comparisons on any host. Windows paths remain case-insensitive even when tests or remote backends run on POSIX. Display paths and emitted IDs keep their original spelling. )rr%casefold)rsegssegments r_comparison_segmentsr*Gs7 T?D7G7Md 37G    3WSWW 3s7c6djt|S)z=Canonical comparison key (separator/trailing-slash agnostic)./)joinr*)rs r _path_keyr.Qs 88(. //rc~dD].}||vs|j|d\}}t|||cSt|S)zCanonicalize only the path portion of a lane id. Branch labels remain byte-preserved; repo/worktree paths follow platform path identity so equivalent Windows spellings do not create duplicate lanes. )r r)rr.) path_or_lanemarkerrootsuffixs r _lane_keyr5VsV -8 \ !'--fa8LD&o&vhvh7 78 \ ""rc*t|}|r|dSdS)Nr )r)rr(s r base_namer8cs T?D48##rc^tj|xsd}|r|jdSdS)zNThe ``/.worktrees`` dir for a ``.../.worktrees/`` path, else None.r r0N)_KANBAN_DIR_REr"group)rms rkanban_worktree_dirr=hs+TZR(A1771:$$rct|t|rttkDrytfdttDS)zLTrue when ``target`` equals ``folder`` or is nested under it (segment-wise).Fc34K|]}||k(ywNr).0ifts r z!_is_path_under..ts3qtqt|3s)r*lenallrange)foldertargetrCrDs @@r_is_path_underrKnsDV$AV$A AQ 3U3q6]3 33rc^tjdd|}tjd||S)N[/\\]+$r z[^/\\]+$)rsub)rnamestrippeds r_with_base_namerQws'vvj"d+H 66+tX ..rc 2|t|xs|||||||dS)N)repo_key repo_label repo_pathlane_key lane_label lane_pathis_main is_kanban)r8)rrVrWrXrYrZs r _placementr[s0 *7i   rcpt|}|syt|}|r:tjddt |d}t |t |d|ddStjd|}|r:t ||jd}t |||jd |ddSt ||||ddS) zDPath-only fallback when there is no git probe and no persisted root.NrMr kanbanFTz^(.+)-wt-(.+)$r0) r8r=rrNrQr[rr"r;)rbase kanban_dirrUr<s r_place_by_heuristicras T?D $T*JFF:r?:r+JK )_Y%?:W\^bcc "D)A#D!''!*5 )T1771:tUEJJ dD$dE ::rc|r||nd}|r|jdr|jdr|d}|d}||k(xst|jd}|r6|xsdjxst}t |t ||||ddSt |} | rt |t|d| ddSt|xs|} t ||| |ddS|r\t |} | rt |t|d| ddS|xsdjxst}t |t ||||ddSt|S)Nr worktree_rootrYr TFr]) getr!rDEFAULT_BRANCH_LABELr[rr=rr8ra) cwdrresolvepersisted_rootinforrcrYbr`labels r_placerlsS"73<D %$((?*C% _- 9,ITXXi5H0I 2$$&>*>AiA)F9VZ\ab b(7 i)CXz[`bfg g-(9M)]E=%QVWW(- non.MxYcejlpq q \r " :&:./.!*LaQ_aeglmm s ##rc|jdxsdj}|r"|r ||}|r|jdr|dS|jdxsdjS)zCThe COMMON repo root a session belongs to (folds linked worktrees).rfr r git_repo_root)rdr)sessionrgrfris r_session_repo_rootrpsb ;;u  # * * ,C ws| DHH[) $ $ KK ( .B 5 5 77rc.t|jdxr|djtv}t|jd}t d|jdxsgDd}|rdnd |rd nd| |djfS) NisMainrkisKanbanc32K|]}t|ywr@ _session_timerArs rrEz!_lane_sort_key..sJM!$Jsessionsdefaultrr0)r!rdlower_TRUNK_BRANCHESmax)r;is_trunkrZactivitys r_lane_sort_keyrsEIIh'(VU7^-A-A-C-VHUYYz*+IJeii .C.IrJTWXH 1 A   g  rc$t|tS)N)key)sortedr)groupss r _sort_lanesrs &n --rc i}|D]&}|j|dgj|(|jD]!}|Dcgc]}|jds|}}t |dkr4|Dcic]}t |t |ddd}}td|jD}d}||ksi} |D]j}|t |} |rdj| | dnd } t|dxs|d} | r| d| n| |d<| j|dd dz| |d<ltd | jDr|dz }||kr$ycc}wcc}w) z>Grow colliding basenames into path-prefixed labels (in place).rkrr^Nr7c32K|]}t|ywr@)rF)rAps rrEz'_disambiguate_labels..s91A9rxr0r,r rc3&K|] }|dk( yw)r0Nr)rAcs rrEz'_disambiguate_labels..s3a163s) setdefaultappendvaluesrdrFidrrr-r8rG) itemsby_labelitembucketgpathedparents max_depthdepthcountsr(prefixr_s r_disambiguate_labelsrs&(H<DM2.55d;<//##5quuV}!55 v;? =CD2a5)AfI.s33DD9(899 y %'F Cr!u~49$vw-0r 6+8qy39xq/t' %+ZZ' A%>%Bqz"  C 36==?33 QJEy 5EsE$E$6"E)cdt|jdxs|jdxsdS)N last_active started_atr)floatrd)ros rrvrv s) ]+Mw{{ lane -> sessions`` subtree for a set of sessions.rfr git_branchrnrVrWrXrYrZ)rrkrrrrsryrSrTrU)r;rSrTrUr;ryTrreverserrrkrr sessionCountrr) rdrrlr5rrsortrvrFr.listrr)ryrghydratelanesrorf placement lane_identityentryreposr;count repo_identityrepo repo_lists r _build_reposrs:E3{{5!'R..0   [[ & ," 3 3 5  [[ ) /R 6 6 8   !)J"78  -( =$J/&|4%k2' 2 )+ 6 " &j1' 5&{3 E$)E-  gz"))'2?3B E&g j=$?E*%& "E* !% "34 yy' <J'|,k* ! D$(E-  Xe$ ^%'&*U\\^$I-$T(^4XT(^,-# rc V|Dchc]6}|jd|jdfD]}|r t|8}}}t|}|xsgD]}|jdxsdj}|s)|r||nd} | xsijdxst j dd|} t| } | r| |vru|j | t| xs| | gdd|j| t|t|k7r t||Scc}}w) uEnsure every declared project folder shows as a repo, even with 0 sessions. A brand-new project (or any project whose sessions haven't loaded yet) has an empty session-derived ``repos`` list. That breaks two things on the desktop: the entered-project view renders blank (it early-returns on no repos), and the optimistic live-session overlay has no lane to drop a freshly-created session into — so a new session in the project only appears after a full tree refresh. Seeding each folder as an empty repo fixes both: the overlay matches a new session's cwd under the folder root, and the drill-in renders a real (if empty) project body. Folders already covered by a session-derived repo (same git root) are left untouched. rrr NrrMrr) rdr.rrrrNrr8addrFr) rfoldersrgrr$seenseededrIrawrir3root_keys r_seed_folder_reposrQs/"  hhtndhhv&67    %  D  %[F-R zz&!'R..0 &ws|D  ,Kz2s0KT?8t+  TIdO,CtT]_qrst   6{c%j V$ M/ s;D%c eZdZdZddZddZy) _FolderIndexu>Maps a normalized folder path → (owning project, depth), so a session is matched to its project by walking its cwd's ancestors (O(path depth) dict lookups) instead of scanning every project × folder per session — the difference between O(sessions × projects) and O(sessions) at power-user scale. c<i|_|D]}|jdxsgD]v}t|jdxsd}|s$dj|}t |}|jj|}| ||dkDsf||f|j|<xy)Nrrr r,r0)_by_pathrdr*r-rF)selfprojectsprojectrIr(rrexistings r__init__z_FolderIndex.__init__s57  :G!++i06B :+FJJv,>,D"EhhtnD ==,,S1#ux{':*15)9DMM#& : :rct|xsd}tt|ddD]5}|jj dj |d|}|s3|cSy)zFOwning project for ``target`` by longest ancestor folder, + its depth.r rr7r,N)Nr7)r*rHrFrrdr-)rrJr(endhits rr"z_FolderIndex.matchs\#FLb1TAr* C--##CHHT$3Z$89C  rN)r list[dict]returnNone)rJstrrztuple[Optional[dict], int])__name__ __module__ __qualname____doc__rr"rrrrrs :rrc*|j|dS)Nr)r")indexrJs r_project_for_pathrs ;;v q !!rc|jdxsdj}|syt||}|r ||k7r||gn|g}d}d}|D]#}|j|\} } | s| |kDs | }| }%|S)Nrfr r7)rdrrpr") rorrgrfr candidatesbestbest_lenrJr"lengths r_project_for_sessionrs ;;u  # * * ,C "7G4I%.93C#y!#JDH F+ v Vh&HD  KrF)coloriconis_autoc |||||| ||||d S)N) rrkrrrisAutor lastActiverpreviewSessionsr) pidrkrr session_countrpreview_sessionsrrrs r _project_noders,%!+  r) preview_limitr is_junk_root is_junk_cwdc~&'|Dcgc]}|jdr|} }|xsd} |xsd} t| } i} g}|D]G}t|| |}|r%| j|dgj |7|j |Ig}d#fd }d$d}g}| D]}| j|dg}|j d|Dt t||||jdxsg|}|j t|d|jd xs|d|jd |jd |jd |t||||| i'd%'fd }|D]}t||}|r| |s |||#|jdxsdj}|r| |rQt||jdxsdj||jdxsdj}|s||d|t}'jD]}|d}|d}t|&t|||}t!&fd|Dd} | =|j#&|j d|D|j t|t%|xs|||| d||||d|xsgD]}!|!jdxsdj}"|"s*|r||"nd}#|#xsijdxs|"}t|}$|$|vs| |s t'| |rs|j#|$|!jdxst%|xs|}%|j t||%|||%|gddgt)|!jdxsdt+|!jd xsdgd t-|Dcgc]}|jd!s|c}||d"Scc}wcc}w)&aBuild the authoritative project tree. ``projects`` are ``projects_db.Project.to_dict()`` shapes (non-archived). ``sessions`` are projected session-row dicts (must carry ``id``, ``cwd``, ``git_branch``, ``git_repo_root``, ``started_at``, ``last_active``). ``discovered_repos`` are ``{"root", "label", "sessions", "last_active"}``. ``is_junk_root`` flags git roots that must never become an AUTO project (the bare home dir, the HERMES_HOME subtree). ``is_junk_cwd`` is the narrower policy for non-git session folders: selected descendants may be intentional workspaces even when their parent tree contains Hermes state. User-created projects are honored regardless. Returns ``{"projects": [...], "scoped_session_ids": [...]}``. When ``hydrate`` is False (overview), lane ``sessions`` arrays are emptied but every count is preserved and each project carries up to ``preview_limit`` ``previewSessions``. When True (drill-in), lanes carry full session rows. archivedcyNFr)_roots rzbuild_tree..rcyrr)_cwds rrzbuild_tree..rrrc@dkrgSt|td}|dS)NrTr)rrv)project_sessionsorderedrs r _previewszbuild_tree.._previewss, A I)}dK~ &&rc*td|DdS)Nc32K|]}t|ywr@rurws rrEz3build_tree.._last_active..s?M!$?rxrzr{)r)rs r _last_activez build_tree.._last_actives?.>?MMrc3JK|]}|jds|dywrNrdrws rrEzbuild_tree..sDad !D'D# #rrO primary_pathrr) rrkrrrrrrrr3ctt|}|syj||gd}|dj|y)N)r3ryry)r.rr)r3rorr by_auto_roots r _add_autozbuild_tree.._add_auto&s<o ((t.LMz!!'*rrfr rrnrSryc3K|]:}t|jdxs|jdxsdk(r|<yw)rrr N)r.rd)rArauto_keys rrEzbuild_tree..IsA TXXd^Etxx/?E2F(R sAANc3JK|]}|jds|dywrrrws rrEzbuild_tree..SsHaAEE$K!D'HrrT)rrkrrrrrrrrkrrrr)rscoped_session_ids)rrrr)rrrr)r3rrodictrr)rdrrrrextendrrrrFrprrlsetrr.nextrr8rintrr)(rrydiscovered_reposrgrrrrractive_projects_junk _junk_cwd folder_index by_projectunownedroowner scoped_idsrrresultr psessionsrrr3rfrrr auto_root auto_sessions repo_noderraw_rootrirrkrrs( ` @@r build_treersc8#+DQ!%% 2CqDOD  10E3 2I0L(*JG$$WlGD   ! !%+r 2 9 9' B NN7 # $J' NF# NN74="5 D9DD" GW 5w{{97M7SQSU\   DMkk&):WT][[0kk'*[[(!)n(3!*9!5   0%'L+6!'73 ;$( {{5!'R..0in   [[ & ," 3 3 5  [[ ) /R 6 6 8   i +W 5'6*UD%%' 6N z* Y']GW= !        H=HH  *7i'7(7!*=!9  # @!&B HHV$*113 $+wx   ,8T? t uT{.? d.S  !Q?!#   6V?quuX!?@j AAaE\@sP5P5P:%P:)r )rrrrrr)rrrr)rrrz list[str])rrrr!)rrrr)r1rrr)rrr Optional[str])rIrrJrrr!)rrrOrrr)rrrVrrWrrXrrYr!rZr!rr)rrrOptional[dict]) rfrrrrgOptional[Resolve]rhrrr)rorrgrrr)r;rrtuple)rrrr)rrrr)rorrr)ryrrgrrr!rr)rrrrrgrrr)rrrJrrr)rorrrrgrrr)rrrkrrrrrrrrrrrrrrrrr!rrr@)rrryrr rrgrrrrr!rOptional[Callable[[str], bool]]rrrr))r __future__rrtypingrrrrrResolvecompiler:r~rerrrr%r*r.r5r8r=rKrQr[rarlrprrrrvrrrrrrrrrrr"s;2# ** C5(4.( ) KL8< "LWX0 #$ % 4/      (;&$B8 .@O@F& & *&5F&&\@">        !   >"& lB 4837lBlBlB!lB lB  lBlB2lB1lB lBr