lKj"[ddlmZmZmZmZddlmZmZmZm Z m Z ddl m Z m Z ddlmZddlmZmZddlmZddlmZddlZddlZej2eZdZGd d Zy) )INTERACTIVE_ROLESSCROLLABLE_ROLESWINDOW_CONTROL_SUBROLESPRUNABLE_ROLES) TreeStateTreeElementNodeScrollElementNodeTextElementNode BoundingBox)ThreadPoolExecutor as_completed)deque)BROWSER_BUNDLE_IDSSYSTEM_UI_BUNDLE_IDS)WindowNceZdZeZdededdfdZdedzde fdZ dde ed e edzd e edzde e e e ee effd Z dd ed edede e e e ee effdZ dd ed edede e e e ee effdZdededefdZ d dede e dededzfdZ d dede e dededzfdZ d!dej4dede e de ede ededzd eddfdZy)"Tree notificationpidreturnNc2tjd||y)a Callback invoked by WatchDog when focus changes (FocusedUIElementChanged, FocusedWindowChanged, MainWindowChanged). Can be used to invalidate caches or trigger fresh tree reads to overcome macOS accessibility tree laziness. z%Focus changed: notification=%s pid=%dN)loggerdebug)selfelementrrs g/Users/ahmed/devFolder/Ultron/claude-voice/.venv/lib/python3.12/site-packages/macos_mcp/tree/service.pyon_focus_changedzTree.on_focus_changeds  !!#"6"67. 889P9PQQsQ -FM   m55 6 ''+;;;!..449;!..55:  '../?@  & &%"3(? '  C+-B/-"7   rr#r$r%c g}g}g}|g}|g}g}|D]tv}|j|df |D]#|vstv}|j|df%|j} |D cic] \} } } | d } } } } i}|D])\} }}| j|j| ||}| ||<+|ryt t |D]_}|j| |j}|r9|\}}}|j||j||j|a|ry|||fScc} } } w#t$r}| jddz| <tjd| || tkrNtfd|Dd}tfd|Dd}| j|j||}||<ntj!dt|d Yd}~+d}~wwxYw) NFTrz(Error processing bundle %s, retry %d: %sc36K|]\}}}|k(s |ywN).0bib_r.s r z-Tree.get_window_wise_nodes..sN HAr1qI~R c36K|]\}}}|k(s |ywr<r=)r>r?rAdor.s rrBz-Tree.get_window_wise_nodes..sN HAq"qI~R rCz;Task failed for bundle %s after %d retries. Exact error: %s)exc_info)rr, _executorsubmit_get_nodes_pooledr listpopresultextend ExceptiongetrrTHREAD_MAX_RETRIESnexterror)rr#r$r%r'r(r) task_inputs is_browserexecutorbidrA__ retry_countsfuture_to_bundle_id desktop_onlyfuturerL element_nodes scroll_nodes info_nodese new_futurer.s @rr4zTree.get_window_wise_nodesNsd 464679  $ "  " *&( #46 #I"&88J    :u= >$1I *&*<< ""Iz4#@A1 >>AL'M:32Q 'M$&-8 )C\__&&ZF+.  ' .9 "&t,?'@A/33F; !#]]_FBH? |Z)00?(// =-44Z@B"J!"24III[(N$!.:.>.>y!.Lq.PL+LLB!$Y/  $I.1CC%)N NPU& (,N NPU( &.__ 22Iz<& ;D+J7 Y%.%) %)s /D1A D88 G7B+G22G7r.rTrZc|tj5|j|||cdddS#1swYyxYw)aRun get_nodes inside an autorelease pool on the worker thread. get_nodes executes on the shared ThreadPoolExecutor's worker threads. PyObjC only installs an NSAutoreleasePool automatically on the main thread, so without draining a pool per task the autoreleased Objective-C objects created during AX traversal accumulate for the lifetime of these long-lived worker threads (a steady memory leak). N)objcautorelease_pool get_nodes)rr.rTrZs rrIzTree._get_nodes_pooleds, " " $>>)ZF% $ $s2;c Ftj|}|sgggfStj|jd|jxs|}g}g}g}|sJ|j x} r|j | |||g||jx} r|j | |||g||jx} r<| jx} rtj| } |j | ||||| |n|j}|Dcgc]$}tj|jds|&}}|rE|D]?}|j}|rtj|nd}|j |||||||An.|s,|jD]}|j |||||||||fScc}w)a Get interactive and scrollable nodes for an app by bundle_id. Tree traversal begins here: starts from each window and recurses via tree_traversal. Args: bundle_id: Application bundle identifier. is_browser: Whether the app is a browser. desktop_only: When True, skip menu bar scanning (used for Finder desktop items when another app owns the menu bar). g?)rT)main_window_bounding_boxrT AXMinimizedN)r*r+SetMessagingTimeoutr0NameMenuBartree_traversal ExtrasMenuBar MainWindowBoundingRectangler from_bounding_rectangleWindows GetAttribute GetChildren)rr.rTrZr6app_namer'r(r)menubarextras_menubar main_windowmain_window_rectrf all_windowswvisible_windowswindow window_rect window_bboxchilds rrdzTree.get_nodess00;r2:  s{{C088(y354679++%w%##%$) $"%!2!22~2##"%$) $.. (; (#.#@#@@@+6+N+N$,(##%$)-E)$++K&&aqyy-@; -F"(":":K'$;;KH! '' )(-1<#-(. ! __.E'' )(-#- (/!"24III?s9)F window_box element_boxc ft|j|j}t|j|j}t|j|j}t|j |j }||kDr||kDrt ||||||z ||z St ddddddS)N)lefttoprightbottomr2r3r)maxrrminrrr )rrrrrrrs riou_bounding_boxzTree.iou_bounding_boxs:??K$4$45*..+//2J$$k&7&78Z&& (:(:; 4A.3ii 8J8J.K.Qr+ % ",,*?*? @ 6C-"2"'2(,,.==*BB4H +#'#8#80,$L&002!((#%1%4%)$/!)  3&Z I&G11(,,.=='2'J'J4'P$++/+@+@02F,(.88:!((#%9%4W=C%)$/!)  2 r root_controlr(r)c Ft|j|fg}|r|j\} } tj| } | d} | d} | d}| ds| t vrH| $t |D]}|j|| fn| jdkDxr| jdkD}| tvxs| dk(}t| d}|xr| d xst| d xs | d xs|xr|}tj| }|r2|j||}|jd k(s|jd k(r|rdtj| }d}| dtj | dtj"j$tj"j&tj"j(g}|j+tj"j$xsX|j+tj"j&xs-|j+tj"j(xsd}|d xs|r t-|nd}i| |d|i}|j/}i}| dk(r|dx}r||d<|dx}r||d<n| dk(s| dk(r3|dx}r||d<|dx}r||d<|dr|d|d<| d ry| d |d <np| dk(r |dx}rd||d<n^| dk(r|rW||d<nQ| dk(r|dx}rE|j1dr4||d<n.| dk(r)|dx}r||d<|dx}r|j1dr||d<|j+dr|d|d<|jt3|||| ||| r|j5||||n|j7||||| t8vrV|rT|r|d nd} d}!| tj| }"|"d }!|jt;|!| |||j/ t |D]}|j|| f|ryy)!u Traverse the accessibility tree iteratively and collect interactive and scrollable nodes. Uses a two-phase attribute fetch strategy to minimise cross-process IPC cost: Phase 1 (every element): 8 cheap attributes — role, visibility, bounds, children. Phase 2 (interactive elements only, ~5-10%): 9 display/metadata attributes. This avoids marshalling string attributes for the ~90% of non-interactive nodes. rrrhiddenNr:AXImagetitle_ui_elementenabledhelp has_popuprrr AXTextField placeholdervalue AXComboBox AXTextAreaexpanded AXRadioButtonselected AXPopUpButtontitlerurl)zfile://zhttp://zhttps://filenamerrr)rrrr1r)rr0rKr*GetEarlyTraversalBatchrreversedr,r2r3rboolr rorGetLateTraversalBatchrrTitler DescriptionrOstrr startswithrrrrr )#rrrr'r(r)rfrTstackrcurrent_is_browserearlyrrr child_element is_visible has_roleshas_title_ui_elementis_interactiver1latetitle_ui_element_textref_rawrrrrrrrr first_child scroll_label child_lates# rrkzTree.tree_traversals$ ,,j9:;*/))+ 'G'--g6E=D=DZ(HX$."8|%-h%7MLL-1C!DE&8a;DKK!OJ!22J 8II#'.@(A#B /uY/(f &(%((   '>>tDL'#44,l   %%*l.A.AQ.F//8)-%+,8 ;;01LL..LL..LL44G BLL$6$67 ";;r||'9'9: ";;r||'?'?@  *W 2GC-.RU5TDT*<>ST%002=(&*=&99{92=/ $W -u-,1)\)T\-A&*=&99{92=/ $W -u-,1)J'/3J/?,[)05k0B-_, $W -u-/4,_,,,A)X%"5k)s)>>*LM.1HUOY&#' #33x3/7,"5k)s)>>*LM.1HUO88L)/3L/AH^,!((#%1%"%)$/!)  &((0+?W,,0+?W''J-5hqk4 ! *!#!9!9+!FJ#-g#6L ''%)%)$/%1+668 "*(!3  m-?@A"4Yr)NN)Fr<)NF)__name__ __module__ __qualname__r rGrintrrrr8rJtuplerr r r4rrIrdr rdictrrr*Controlrkr=rrrrsr"$IQcQQQ% v}% % T/348 FJIFJ 9t+FJ"&cT!1 FJ tO$d+<&=tO?TT U FJRFK G G*. G>B G tO$d+<&=tO?TT U GFK`J`J*.`J>B`J tO$d+<&=tO?TT U`JDP%P4?P P08< $$ 0$ $ #."4 $V8< CC 0C C #."4 CX8< aBjjaBaB 0 aB 01 aB $O4 aB#."4aBaB aBrr)macos_mcp.tree.configrrrrmacos_mcp.tree.viewsrrr r r concurrent.futuresr r collectionsrmacos_mcp.desktop.configrrmacos_mcp.desktop.viewsr macos_mcp.axr*loggingrb getLoggerrrrPrr=rrrsW @M*   8 $JBJBr