@OOfl)VddlZddlZddlmZddlmZddlmZddlmZddlmZddlm Z ddlm Z dd lm Z dd lm Z dd lm Z dd 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_focusedwindowurwidAttrWrapTextsuper__init__)selfrwidgettitlefocusr __class__s _/var/www/premiumrankchecker/venv/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) WindowStackc ||_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||_|g|_d|_y)Nrrrr r r redit_focus_queryedit_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 CookieEditorSetCookieEditorCookieAttributeEditorRequestMultipartEditorRequestUrlEncodedEditor PathEditorRequestHeaderEditorResponseHeaderEditorwindowsstackr )rrHbases r"rzWindowStack.__init__4so  ))&1 &&v. &&v. $//7  OOF+  v&  &&v. (33F;  *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[rnpop)rargsrds r"ruzWindowStack.popisF||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)getattrrir )rnamervrds r"rnzWindowStack.callusG )!4($9&9 << 'GDLL$ ' 8 8 r#N) r6r7r8rrerir`rqrurnrGr#r"r<r<3s%. , !) 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 StatusBarrrrrrHviewsig_view_refreshconnect view_changed sig_view_addsig_view_removesig_view_updater sig_change focus_changedr sig_focus flow_change flow_changedpop_view_staterur subscribe configurepaner<stacks)rrHr!s r"rzWindow.__init__s",,V4  ennT^^\&R    ))11$2C2CD %%--d.?.?@ ((001B1BC ((001B1BC ))11$2C2CD ))11$2D2DE dnn-##D$5$56&&txx0 %% NN-/GH  "6:6 FJ8WX r#c4|j|jSr^)rrrhs 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^) rr`rHr rrirrr)idxrrrs r"wrappedzWindow.refresh..wrappedsi[[%002F{{""99 C(335;;tVUDII4DE Er#Nvertical) focus_itemr&) dividechars focus_columnr})rHr rrrPile enumeraterColumnsrbodyrwindow_refreshsend)rcrwiss` r"rzWindow.refreshs KK   . . =DI F  = A *_ (1$++(>?1?DIIA (1$++(>?1?!YYA NN1l3 ##%@@s /D =D$ rreturnNc|jjjjrS|j|jjjjjk(r|j yyyr^)rHrr 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. rNrrnrrvrdrs 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)rrerrbs r"rezWindow.set_overlays+ '&&q3F3 r#c|jj||j|j|j y)zA Push a window onto the currently focused stack. N)rrqrrr)rros r"rqz 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)rrurHprompt_for_exitrrrrhs r"ruz Window.popsJ     ! ! # KK ' ' ) LLN        r#c|jj}|jd|j|j|S)zQ Returns: self.stacks, with the focused stack first. r)rcopyinsertrur)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)rr`keyctxrrrts 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)rrirrs 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)rHr rrrtrrrhs 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)rrvrdkr!s r"r(zWindow.mouse_events G  0 0Aw,&&&++}, aM)d1gl d1gt,  aM)d1gl d1gv.r#ct|||}|rL|jjj |j j j|Syr^)rr0rHkeymaphandlerr`r)rr)rr!s r"r0zWindow.keypress,sR G T1 % ;;%%,,T-=-=-?-J-J-L-S-SUVW W r#)rN)r6r7r8rrrrrFlowrrrrerqrurrrrr'r(r0r9r:s@r"r{r{ssY.&&B%%t% $# !  &"XXr#r{ceZdZfdZxZS)Screencrtjrtjdd|}t||y)Nz[])rIS_WINDOWS_OR_WSLresubrwrite)rdatar!s r"rz Screen.write3s,  # #66,D1D  dr#)r6r7r8rr9r:s@r"rr2s r#r)rr mitmproxyrmitmproxy.tools.consolerrrrrr r r r r rrFramerr<r{ raw_displayrrGr#r"rs| ,*,,,.(/+++-%++>J9J9ZoXU[[oXdU   % %r#