@OOfv'FddlZddlZddlZddlZddlZddlZddlZddlmZddlm Z ddl m Z ddl m Z ddlmZddlmZddlmZej$eZd Zd Zej.d ZGd d ZdZe Gddej6ZGddZy)N)Callable)Sequence) dataclass)Any) exceptions)flow)hooksc`t|d|jjjS)Nname)getattr __class____name__lower)itms W/var/www/premiumrankchecker/venv/lib/python3.12/site-packages/mitmproxy/addonmanager.py _get_namers$ 3 6 6 < < > ??cv|}tj|D]\}}}}|j}||k(sn|xs|S)z Cut off a traceback at the function with the given name. The func_name's frame is excluded. Args: tb: traceback object, as returned by sys.exc_info()[2] func_name: function name Returns: Reduced traceback. ) traceback extract_tbtb_next)tb func_nametb_orig_fnames r cut_tracebackrsKG#..r21eQ ZZ I   =rc#$K dy#tjtjf$rt$rXt j \}}}t |d}t |d}|sJ|sJtjd||||fYywxYww)Ninvoke_addon_sync invoke_addonz Addon error: )exc_info) r AddonHalt OptionsError Exceptionsysr!rloggererror)etypevaluers rsafecallr*,s    *"9"9 :   <<>ub 22 3 2~ . u u E7 #UB'   s&B BBB  B B  Bc ZeZdZdZdZ ddededededeedzd df d Z d ed e d dfd Z y)LoaderzM A loader object is passed to the load() event when addons start up. c||_yN)masterselfr/s r__init__zLoader.__init__Cs  rNr typespecdefaulthelpchoicesreturnct|trJ||jjvr|jjj|}|j |k(xrB|j |k(xr1|j|k(xr |j|k(xr|j|k(}|rytjd|z|jjj|||||y)z Add an option to mitmproxy. Help should be a single paragraph with no linebreaks - it will be reflowed by tools. Information on the data type should be omitted - it will be generated and added by tools as needed. NzOver-riding existing option %s) isinstancestrr/options_optionsr r3r4r5r6r&warning add_option)r1r r3r4r5r6existingsame_signatures rr>zLoader.add_optionFsgs+++ 4;;&& &{{**33D9H %0%%10$$/0MMT)0$$/ ?$FG &&tXwgNrpathfunccP|jjj||y)zAdd a command to mitmproxy. Unless you are generating commands programatically, this API should be avoided. Decorate your function with `@mitmproxy.command.command` instead. N)r/commandsadd)r1rArBs r add_commandzLoader.add_commandes   t,rr.) r __module__ __qualname____doc__r2r:typerrr>rrFrrr,r,>sz)- OOO O  O #% O O>--8--rr,c#zK|D]0}|t|dst|jEd{2y7w)z. Recursively traverse an addon chain. addonsN)hasattrtraverserM)chainas rrOrOns=* 1h ) ) )* *s;;9;ceZdZUdZeed<y)LoadHookz Called when an addon is first loaded. This event receives a Loader object, which contains methods for adding options and commands. This method is where the addon configures itself. loaderN)rrGrHrIr,__annotations__rKrrrSrSxs NrrSceZdZdZdZdZdZdZdZdZ dZ d Z d Z d e jfd Zd e jfd Zd e jfdZd e jfdZd e jfdZd e jfdZy) AddonManagerci|_g|_||_|jjj |j yr.)lookuprPr/r;changedconnect_configure_allr0s rr2zAddonManager.__init__s5   &&t':':;rcL|jtj|yr.)triggerr ConfigureHook)r1updateds rr\zAddonManager._configure_alls U((12rc|jD]&}|j|tj(i|_g|_y)z$ Remove all addons. N)rPrr DoneHookrY)r1rQs rclearzAddonManager.clears< 8A  " "1enn&6 7 8  rc:|jj|dS)z Retrieve an addon by name. Addon names are equal to the .name attribute on the instance, or the lower case class name if that does not exist. N)rYget)r1r s rrezAddonManager.gets {{tT**rc~dddddd}t|gD]p}|jD]*\}}t||stj |d,t |}||j vsZtjd|zt|j}|j|t|t|gD]}t |}||j |<t|gD]'}|jjj|)|jjj!|S) aL Register an addon, call its load event, and then register all its sub-addons. This should be used by addons that dynamically manage addons. If the calling addon is already running, it should follow with running and configure events. Must be called within a current context. zFThe clientconnect event has been removed, use client_connected insteadzLThe clientdisconnect event has been removed, use client_disconnected insteadzYThe serverconnect event has been removed, use server_connect and server_connected insteadzLThe serverdisconnect event has been removed, use server_disconnected insteadzRThe add_log event has been deprecated, use Python's builtin logging module instead) clientconnectclientdisconnect serverconnectserverdisconnectadd_logzM. For more details, see https://docs.mitmproxy.org/dev/addons-api-changelog/.z$An addon called '%s' already exists.)rOitemsrNr&r=rrYrAddonManagerErrorr,r/rrSrDcollect_commandsr;process_deferred)r1addon api_changesrQoldmsgr rTs rregisterzAddonManager.registers> f"nx"nk 5'" A'--/ S1c?NN%lm  Q1A:!>DJ IaL)  * uenn&67?s  B"B"c,t|jSr.)lenrP)r1s r__len__zAddonManager.__len__s4::rcxtj|jDcgc] }t|c}Scc}wr.)pprintpformatrPr:)r1rws r__str__zAddonManager.__str__s'~~tzz:!s1v:;;:s7c4t|}||jvSr.)rrY)r1itemr s r __contains__zAddonManager.__contains__st{{""reventcK|jd}|j|d{t|tjr.|jt j |gd{yy7M7w)z+ Handle a lifecycle event. rN)args trigger_eventr9rFlowr UpdateHook)r1rmessages rhandle_lifecyclezAddonManager.handle_lifecyclesk**,q/  ''' gtyy )$$U%5%5wi%@A A A * ( Bs"(A<A8AA<1A:2A<:A<c#>Kt|tjsJt|gD]n}t ||j d}|st |r||f/t|tjrJtjd|j d|dyw)zK Enumerate all hook callables belonging to the given addon NzAddon handler  (z) not callable) r9r HookrOr r callabletypes ModuleTyperrm)r1rprrQrBs r _iter_hookszAddonManager._iter_hookss%,,,5'" A1ejj$/DD>T'Me&6&67 $66( BqcH s ABABcK|j||D];\}}||j}|tj|s2|d{=y7w)zR Asynchronously invoke an event on an addon and all its children. N)rrinspect isawaitable)r1rprrBress rr zAddonManager.invoke_addonsV ++E59 KE4 %C7#6#6s#;  s.AAA A Ac|j||D]S\}}tj|r&tjd|j d|d||j Uy)zC Invoke an event on an addon and all its children. zAsync handler rz$) cannot be called from sync contextN)rriscoroutinefunctionrrmr r)r1rprrBs rrzAddonManager.invoke_addon_syncsl ++E59 KE4**40 22$UZZL5'9]^ %**,   rcK|jD]0} t5|j||d{ddd2y7#1swYxYw#tj$rYywxYww)zD Asynchronously trigger an event across all addons. N)rPr*r rr"r1rrws rrzAddonManager.trigger_eventsm A Z6++Au5556 666''  sNA+ AAAAAA+AA AA($A+'A((A+c|jD](} t5|j||ddd*y#1swY xYw#tj$rYywxYw)z Trigger an event across all addons. This API is discouraged and may be deprecated in the future. Use `trigger_event()` instead, which provides the same functionality but supports async hooks. N)rPr*rrr"rs rr^zAddonManager.trigger(s^ A Z5**1e45 55''  s$ A9AA AAAN)rrGrHr2r\rcrertrErzr}rrr rrrr rrr^rKrrrWrWs< 3+&P08 <# BEJJ B *uzz  ejj     UZZ rrW) contextlibrloggingrr%rrcollections.abcrr dataclassesrtypingr mitmproxyrrr getLoggerrr&rrcontextmanagerr*r,rOrrSrWrKrrrs  $$!    8 $@(   "----`* uzz qqr