AHj$0>dZddlmZddlZddlZddlZddlZddlZddlm Z ddl m Z ddl m Z ddlmZmZddlmZmZmZmZmZdd lmZmZmZdd lmZdd lmZdd l m!Z!dd l"m#Z#ddl$m%Z%ddl&m'Z'ddl(m)Z)ejTdk(Z+Gdde,Z-Gdde#Z.y)z The Remote driver uses the following packet structure. 1 byte for packet type. "D" for data, "M" for meta. 4 byte little endian integer for the size of the payload. Arbitrary payload. ) annotationsN)getincrementaldecoder)partial)Path)EventThread)AnyBinaryIOLiteralTextIOcast)eventslogmessages)dump) XTermParser)App)Driver) ByteStream) InputReader)Sizewin32ceZdZdZy) _ExitInputz/Internal exception to force exit of input loop.N)__name__ __module__ __qualname____doc__R/root/tools/cai/cai_env/lib/python3.12/site-packages/textual/drivers/web_driver.pyrr%s9r rc2eZdZdZdddd dfdZeddZddZd d Zd!d Z d"d Z d"d Z d"d Z d"dZ d"dZd"dZd"dZd"dZd"dZd"dZd#dZd$dZd%d&dZddddd d'dZdddd d'dZxZS)( WebDriverz+A headless driver that may be run remotely.FTNdebugmousesizec@|W ttjjdd}ttjjdd}||f}t |||||tj|_ tjj|_ ttj|j|_ t|_t!|j"d|_t'|_i|_y#t$rYwxYw)NCOLUMNSPROWSr$z textual-input)targetname)intosenvironget ValueErrorsuper__init__sys __stdout__stdoutfilenorwrite_writer exit_eventrrun_input_thread _key_threadr _input_reader _deliveries)selfappr%r&r'widthheight __class__s r!r5zWebDriver.__init__,s < %BJJNN9b9:RZZ^^FB78f} ETBnn nn++- bhh 4 '#)(($ )]9; 0  sAD DDcy)NTrrAs r!is_webzWebDriver.is_webJsr c|jd}|jdt|jdd|fzy)zWrite string data to the output device, which may be piped to the parent process (i.e. textual-web/textual-serve). Args: data: Raw data. utf-8sD%s%sbigN)encoder;lento_bytes)rAdata data_bytess r!r:zWebDriver.writeNs9[[)  HJ 8 8E BJOOPr ctj|jdd}|jdt |j dd|fzy)zWrite a dictionary containing some metadata to stdout, which may be piped to the parent process (i.e. textual-web/textual-serve). Args: data: Meta dict. rJignore)errorssM%s%srKrLN)jsondumpsrMr;rNrO)rArP meta_bytess r! write_metazWebDriver.write_metaYsIZZ%,,WX,F  HJ 8 8E BJOOPr cxt|}|jdt|jdd|fzy)zBinary encode a data-structure and write to stdout. Args: data: The data to binary encode and write. sP%s%srKrLN) binary_dumpr;rNrO)rArP packed_bytess r!write_binary_encodedzWebDriver.write_binary_encodedcs6 #4(  HL 1 : :1e DlSSTr cy)NrrGs r!flushzWebDriver.flushls r c\|j}|d|d|d|dy)z!Enable reporting of mouse events.z[?1000h[?1003hz[?1015hz[?1006hNr:rAr:s r!_enable_mouse_supportzWebDriver._enable_mouse_supporto+  m m m mr c&|jdy)zEnable bracketed paste mode.z[?2004hNrarGs r!_enable_bracketed_pastez!WebDriver._enable_bracketed_pastew =!r c&|jdy)zDisable bracketed paste mode.z[?2004lNrarGs r!_disable_bracketed_pastez"WebDriver._disable_bracketed_paste{rgr c\|j}|d|d|d|dy)z"Disable reporting of mouse events.z[?1000lz[?1003lz[?1015lz[?1006lNrarbs r!_disable_mouse_supportz WebDriver._disable_mouse_supportrdr c&|jdy)zNWrites an escape sequence to query the terminal support for the sync protocol.z [?2026$pNrarGs r!#_request_terminal_sync_mode_supportz-WebDriver._request_terminal_sync_mode_supports >"r cXtjd fd }ts7tjtj fD]}j ||jdjdjjdjdj tddntj}tj||}tjjj!| j#j%j'j(j+jj-jj.tj0y) zStart application mode.ctjjjt j y)zCallback to force exit.loopN)asynciorun_coroutine_threadsafe_app _post_messagerExitApp)rqrAsr!do_exitz1WebDriver.start_application_mode..do_exits.  , , ''(8(8(:;$ r s __GANGLION__ z[?1049hz[?25lr`Nr*r,rpreturnNone)rrget_running_loopWINDOWSsignalSIGINTSIGTERMadd_signal_handlerr;r:rc_sizerrResizersrtrurmrfr^r>start call_later post_messageAppBlur)rArw_signalr'eventrqs` @r!start_application_modez WebDriver.start_application_modes4'')  "MM6>>: :''9 : %& =! ""$ ; =!#zz1tB|tTZZ7H dD)(( II # #E * 002 $$&    TYY33V^^5EFr cy)zDisable further input.NrrGs r! disable_inputzWebDriver.disable_inputsr c|jj|jj|j ddiy)z%Stop application mode, restore state.typeexitN)r<setr?closerXrGs r!stop_application_modezWebDriver.stop_application_modes5    " ()r cZ|j}t|j}tdj}|}t } |D]}|r^|j |D]J\}}|dk(r.|j ||D]} |j| 9|j||L|jD]} |j|  |jy#t$rYt$rddl m } t| Y> !l  !$ 00;0@0@0F@, W&#-)/VG_)E< $ 4 4U ;<!MM+w?@$[[]0E((/0 0&        ,       s7 B C)D) D2D4DDDDD*ctj|}|jdi}t|tr|j ||yt jd|y)zPrivate method to dispatch meta. Args: packet_type: Packet type (currently always "M") payload: Meta payload (JSON encoded as bytes). rz;Protocol error: type field value is not a string. Value is N)rUloadsr2 isinstancestron_metarerror)rArr payload_map_types r!rzWebDriver._on_metasP*.G)< + eS ! LL , IIMeYW r c|dk(rQ|d|df|_t|j}|jjt j ||y |dk(r.|jjt j y |dk(r.|jjt jy |dk(r.|jjtjy |dk(r t|dk(r tjd | tt|d }tt|d }|j$}d } ||}|j'dd } tjd|d||j)|}tjd|dt+||j-d||f|s@tj.d||j1||=|j3|d |y y y #t $rtj"d Yy wxYw#t4$rz} |j1||=tj"d|ddd l} tj"t| j9|j;|| |Yd } ~ y d } ~ wwxYw#t $rtj"d|Yy wxYw)zProcess a dictionary containing information received from the controlling process. Args: packet_type: The type of the packet. payload: meta dict. resizerCrDfocusblurquitrdeliver_chunk_requestzDeliver chunk request: keyr'z9Protocol error: deliver_chunk_request missing key or sizeNr.zReading z bytes from zDelivering chunk z of len deliver_chunkzDelivery complete for ) save_pathr.z$Error delivering file chunk for key z. Cancelling delivery.r) exceptionr.z2Protocol error: deliver_chunk_request invalid key )rrrtrrrAppFocusrrrvrrr%r rr/KeyErrorrr@r2readrNr\infor_delivery_completerrr_delivery_failed) rArrrequested_size delivery_key deliveries file_liker.chunkrrs r!rzWebDriver.on_metas ( "!'*GH,=>DJ!4::.N II " "6==#P Q G # II " "6??#4 5 F " II " "6>>#3 4 F " II " "8#3#3#5 6 F ",  3 3 II/y9 : #C8 !%c76?!; ))J26I T&|4 $+;;vt#<TII(8 \NST%NN>:EII 1,1A#e*VW-- e.TU #9,!HI!)&|4// SW/X !94   UV 6! TOO%"<0II>|>NO//%IIc)"6"6"89:)),%d)SS T%  HHXY s>&HJ=+B%H7H43H47 J:A0J55J:=!K! K!c.|jd||dy)zOpen a URL in the default web browser. Args: url: The URL to open. new_tab: Whether to open the URL in a new tab. open_url)rurlnew_tabN)rX)rArrs r!rzWebDriver.open_url)s CGLMr download) open_methodencoding mime_typer.c 4|j|||||||y)N)rrrrrr.) _deliver_file)rAbinaryrrrrrr.s r!deliver_binaryzWebDriver.deliver_binary2s,  %#  r )rrr.c|jd||j|<d|t|j||xsd|xsd|d}|j |t j d|dd|y) z2Deliver a file to the end-user of the application.rdeliver_file_start)rrpathrrrr.zDelivering file rz: N)seekr@rresolverXrr) rArrrrrrr.metas r!rzWebDriver._deliver_fileGs  A)/&) ))+,& B"b#   #DL#32dX>?r )rBzApp[Any]r%boolr&rr'ztuple[int, int] | None)ryr)rPrryrz)rPdict[str, object]ryrz)rPztuple[str | bytes, ...]ryrzrx)rrrbytesryrz)rrrrryrz)T)rrrrryrz)rzBinaryIO | TextIOrrrrrzLiteral['browser', 'download']r str | Nonerrr.rryrz)rrrrr5propertyrHr:rXr\r^rcrfrirkrmrrrr=rrrrr __classcell__)rEs@r!r#r#)sr5 '+ 0 0 0  0 % 0< QQU ""#"GH%* !> @TDN7A# $ !     4       8 $ $@!@ @  @ 4 @@@@ @r r#)/r __future__rrrrUr0r}r6codecsr functoolsrpathlibr threadingrrtypingr r r r r textualrrrtextual._binary_encoderrZtextual._xterm_parserr textual.apprtextual.driverrtextual.drivers._byte_streamrtextual.drivers._input_readerrtextual.geometryrplatformr|rrr#rr r!rsu # (#77))6-!35! ,,' !::y@y@r