kj&*^ddlZddlZddlZddlmZddlmZddlmZddlmZddlm Z ddlm Z ddlm Z dd lm Z dd lm Z dd lmZdd lmZdd lmZddlmZGddej$ZGddZGddej$ZGddej,j.Zy)N)flow)commands)common)eventlog)flowlist)flowview) grideditor)help) keybindings)options)overlay)signals) statusbarc2eZdZfdZfdZfdZxZS) StackWidgetc||_||_|r.tjtj||rdnd}nd}t |||y)Nheadingheading_inactive)header) is_focusedwindowurwidAttrMapTextsuper__init__)selfrwidgettitlefocusr __class__s g/home/ubuntu/.cache/pipx/cdb5059130eadca/lib/python3.12/site-packages/mitmproxy/tools/console/window.pyrzStackWidget.__init__sN ]] 5!9;MFF /c|dk(r+|dk(r&|js|jjt|||||||S)N mouse press)rrswitchr mouse_event)rsizeeventbuttoncolrowr r!s r"r(zStackWidget.mouse_event"sA M !fk$// KK   w"4S%HHr#cnt|||}|j|}|r|jdry|S)Ncursor)rkeypress _command_map startswith)rr)keyretcommandr!s r"r0zStackWidget.keypress'sCgtS)##   w))(3 r#)__name__ __module__ __qualname__rr(r0 __classcell__r!s@r"rrs 0I  r#rc6eZdZdZdZdZdZdZdZdZ y) WindowStackc8||_tdidtj|dt j |dt j|dtj|dtj|dtj|dtj|dt!j"|d t!j$|d t!j&|d t!j(|d t!j*|d t!j,|dt!j.|dt!j0|dt!j2|dt!j4||_|g|_d|_y)Nrrrr r r redit_focus_queryedit_focus_commentedit_focus_cookiesedit_focus_setcookiesedit_focus_setcookie_attrsedit_focus_multipart_formedit_focus_urlencoded_formedit_focus_pathedit_focus_request_headersedit_focus_response_headers)masterdictr FlowListBoxrFlowViewrCommandsr KeyBindingsr Optionsr HelpViewrEventLogr QueryEditor CommentEditor CookieEditorSetCookieEditorCookieAttributeEditorRequestMultipartEditorRequestUrlEncodedEditor PathEditorRequestHeaderEditorResponseHeaderEditorwindowsstackr )rrIbases r"rzWindowStack.__init__5s  ))&1 &&v. &&v. $//7  OOF+  v&  &&v. (33F;  *77?  *66v> #-"<"U "   %  /4(r#c|jst|jdk(ry|jd|jrd|_y|jj y)zM Pop off the stack, return True if we're already at the top. r&Tlayout_poppingN)r lenr]rppop)rargsrfs r"rwzWindowStack.popksF||DJJ1 4 "# <<DL JJNN r#ct|j||i||jrt|j||i|yy)z Call a function on both the top window, and the overlay if there is one. If the widget has a key_responder, we call the function on the responder instead. N)getattrrkr )rnamerxrfs r"rpzWindowStack.callwsG )!4($9&9 << 'GDLL$ ' 8 8 r#N) r6r7r8rrgrkrbrsrwrprHr#r"r<r<4s%0 , !) 9r#r<ceZdZfdZdZdZdZdejddfdZ d Z d Z d Z d Z dd ZdZdZdZdZdZfdZfdZxZS)Windowctj||_t| ddt j |jd||_|j jjj|j|j jjj|j|j jjj|j|j jjj|j|j jjjj|j|j jjjj|j t"jj|j$t"j&j|j(t"j*j|j,|j j.j1|j2ddgd|_t7|dt7|dg|_y)N background)rfooterconsole_layoutconsole_layout_headersrrr)r StatusBarrrrrrIviewsig_view_refreshconnect view_changed sig_view_addsig_view_removesig_view_updater sig_change focus_changedr sig_focus flow_change flow_changedpop_view_staterwr subscribe configurepaner<stacks)rrIr!s r"rzWindow.__init__s",,V4  emmDNNL&Q    ))11$2C2CD %%--d.?.?@ ((001B1BC ((001B1BC ))11$2C2CD ))11$2D2DE dnn-##D$5$56&&txx0 %% NN-/GH  "6:6 FJ8WX r#c4|j|jSr`)rrrjs r" focus_stackzWindow.focus_stacks{{499%%r#c$|jyr`)refresh)rr updateds r"rzWindow.configures  r#c Tjjj}|dk(rd_fd}d}|dk(r |d}n|dk(rNt j t jDcgc] \}}||c}}j}nNt jt jDcgc] \}}||c}}dj}t j|d _ tjjycc}}wcc}}w) z$ Redraw the layout. singlercj|j}jjjr(j|j j }nd}t||j|k(Sr`) rrbrIr rrkrrr)idxrrrs r"wrappedzWindow.refresh..wrappedsi[[%002F{{""99 C(335;;tVUDII4DE Er#Nvertical) focus_itemr&) dividechars focus_columnr)rIr rrrPile enumeraterColumnsrbodyrwindow_refreshsend)rcrwiss` r"rzWindow.refreshs KK   . . =DI F  = A *_ (1$++(>?1?DIIA (1$++(>?1?!YYA MM!\2 ##%@@s /D =D$ rreturnNc|jjjjrS|j|jjjjjk(r|j yyyr`)rIrr ridr)rrs r"rzWindow.flow_changedsY ;;   ! ! & &ww$++**0055888""$9 'r#cH|jD]}|jdy)z Triggered when the focus changes - either when it's modified, or when it changes to a different flow altogether. rNrrprrxrfrs r"rzWindow.focus_changeds#  $A FF? # $r#cH|jD]}|jdy)z; Triggered when the view list has changed. rNrrs r"rzWindow.view_changeds# #A FF> " #r#c f|jj|fi||jy)z@ Set an overlay on the currently focused stack. N)rrgrrds r"rgzWindow.set_overlays+ '&&q3F3 r#c|jj||j|j|j y)zA Push a window onto the currently focused stack. N)rrsrrr)rrqs r"rsz Window.pushs< &   r#c|jjr|jjy|j |j |j y)z Pop a window from the currently focused stack. If there is only one window on the stack, this prompts for exit. N)rrwrIprompt_for_exitrrrrjs r"rwz Window.popsJ     ! ! # KK ' ' ) LLN        r#c|jj}|jd|j|j|S)zQ Returns: self.stacks, with the focused stack first. r)rcopyinsertrwr)rrs r"stacks_sorted_by_focuszWindow.stacks_sorted_by_focuss6 !!# aDII./ r#ct|jD]%}|j}|j|k(s#|cSy)z Returns the active widget with a matching key context, including overlays. If multiple stacks have an active widget with a matching key context, the currently focused stack is preferred. N)rrbkeyctxrrrts r"currentzWindow.current8 ,,. A Axx6! r#ct|jD]%}|j}|j|k(s#|cSy)z Returns the active window with a matching key context, ignoring overlays. If multiple stacks have an active widget with a matching key context, the currently focused stack is preferred. N)rrkrrs r"current_windowzWindow.current_windowrr#c||_yr`)focus_position)rsections r"rzWindow.sig_focuss %r#c|jjjdk(rd|_n*|jdzt |j z|_|j y)z/ Switch between the two panes. rrr&N)rIr rrrvrrrjs r"r'z Window.switchsI ;;   - - 9DIQ#dkk*::DI r#ct||i|}|sw|ddk(r"tjj ddy |ddk(r|ddk(r|j |dd y |ddk(r|dd k(r|j |dd y y y)Nr&z mouse dragz_Hold down fn, shift, alt or ctrl to select text or use the --set console_mouse=false parameter.)messageexpirer%rupdownFT)rr(rstatus_messagerr0)rrxrfkr!s r"r(zWindow.mouse_events G  0 0Aw,&&&++}, aM)d1gl d1gt,  aM)d1gl d1gv.r#ct|||}|rL|jjj |j j j|Syr`)rr0rIkeymaphandlerrbr)rr)rr!s r"r0zWindow.keypress.sR G T1 % ;;%%,,T-=-=-?-J-J-L-S-SUVW W r#rN)r6r7r8rrrrrFlowrrrrgrsrwrrrrr'r(r0r9r:s@r"r}r}ssY.&&B%%t% $# !  &"XXr#r}c*eZdZdfd ZfdZxZS)ScreencVt|tjd|_y)Nr)rrlogging getLoggerlogger)rr!s r"rzScreen.__init__5s  ''0 r#crtjrtjdd|}t||y)Nz[])rIS_WINDOWS_OR_WSLresubrwrite)rdatar!s r"rz Screen.write9s,  # #66,D1D  dr#r)r6r7r8rrr9r:s@r"rr4s1r#r)rrr mitmproxyrmitmproxy.tools.consolerrrrrr r r r r rrFramerr<r} raw_displayrrHr#r"rs ,*,,,.(/+++-%++>K9K9\oXU[[oXd U   % % r#