@OOfGddlZddlZddlmZddlmZddlmZddlm Z ddlm Z ddlm Z ddlm Z dd l mZdd lmZdd lmZGd d eZdededefdZGddZy)N)Sequence)Path) NamedTuple)ctx) exceptions) flowfilter)http)Headers)strutils) parse_speccLeZdZUejed<eed<eed<defdZy) ModifySpecmatchessubjectreplacement_strreturnc|jjdr4t|jddjj St j |jS)z Process the replacement str. This usually just involves converting it to bytes. However, if it starts with `@`, we interpret the rest as a file path to read from. Raises: - IOError if the file cannot be read. @N)r startswithr expanduser read_bytesr escaped_str_to_bytesselfs _/var/www/premiumrankchecker/venv/lib/python3.12/site-packages/mitmproxy/addons/modifyheaders.pyread_replacementzModifySpec.read_replacements[    * *3 /,,QR01<<>IIK K001E1EF FN) __name__ __module__ __qualname__rTFilter__annotations__bytesstrrrrrrs'    N G% Grroptionsubject_is_regexrc`t|\}}}tj|}|r tj|t|||} |j|S#tj $r}t d|d|dd}~wwxYw#t$r}t d|ddd|dd}~wwxYw)NzInvalid regular expression z ()zInvalid file path: r) r r rrecompileerror ValueErrorrrOSError)r'r( flow_filter subject_str replacementrespecs rparse_modify_specr5$s,6v,>)Kk++K8G N JJw  k7K 8DH  Kxx N:7+Rs!LM M N H.{12.?r!AFGGHs/A B B0BB B-B((B-cTeZdZd dZdZdZdZdZdejd e ddfd Z y) ModifyHeadersrNcg|_yN) replacementsrs r__init__zModifyHeaders.__init__9s .0rcB|jdttgdy)Nmodify_headersa9 Header modify pattern of the form "[/flow-filter]/header-name/[@]header-value", where the separator can be any character. The @ allows to provide a file path that is used to read the header value string. An empty header-value removes existing header-name headers. ) add_optionrr%)rloaders rloadzModifyHeaders.load<s"  SM   rcd|vrOg|_tjjD]*} t |d}|jj|,yy#t $r!}t jd|d||d}~wwxYw)Nr=Fz#Cannot parse modify_headers option z: ) r:roptionsr=r5r.r OptionsErrorappend)rupdatedr'r4r3s r configurezModifyHeaders.configureHs w & "D ++44 /,VU;D !!((. / ' "$11=fXRsKs A A?A::A?c|js|js |jsy|j||jj yr9)responser-liverunrequestheadersrflows rrKzModifyHeaders.requestTs0 ==DJJdii  t||++,rc|js |jsy|j||jjyr9)r-rIrJrHrLrMs rrHzModifyHeaders.responseYs* ::TYY  t}},,-rrNhdrscg}|jD]"}|j|j|$t|jD]'\}}||s |j |j d)t|jD];\}}||s |j }|s |j|j |=y#t$r"}tjd|Yd}~ed}~wwxYw)Nz!Could not read replacement file: ) r:rDr enumeratepoprraddr/loggingwarning)rrNrPrr4ir2r3s rrJzModifyHeaders.run^s%% /D NN4<<- . /!!2!23 -GAtqzt, - !!2!23 ras[ $ "$+GG(cTj(=<=