kjhUddlZddlZddlmZddlmZddlZddlZddlmZddlm Z ddlm Z ddlm Z ddlm Z dd l mZdd lmZd e j"d e j$fd Zde j$d dfdZd e j"d e j*fdZde j$d efdZd e j"d efdZd e j"d efdZd e j"d efdZd e j"d efdZdd e j"d efdZeeeeeeZ eeee j"geezffe!d<GddZ"y)N)Callable)Sequence)command)ctx) exceptions)flow)http)assemble)strutilsfreturnct|ddstjdt|tj sJ|j j}|jd|S)Nrequestz"Can't export flow with no request.Fstrict) getattrr CommandError isinstancer HTTPFlowrcopydecoder rs `/home/ubuntu/.cache/pipx/cdb5059130eadca/lib/python3.12/site-packages/mitmproxy/addons/export.pycleanup_requestrsW 1i &%%&JKK a '' 'iinnG NN%N NrcN|jjdd|jjdd|jk(r|jjd|jjdd|jk(r|jjdyy)z>Remove some headers that are redundant for curl/httpie export.zcontent-lengthNhostz :authority)headerspopgetr)rs r pop_headersr"sz OO($/62&',,6F#<, <L)=rct|ddstjdt|tj sJ|j j}|jd|S)Nresponsez#Can't export flow with no response.Fr) rrrrr rr$rrr r$s rcleanup_responser&%sW 1j$ '%%&KLL a '' 'zz H OO5O! Orc |jd}|sJ t dDcic]}t |d|dc}dj fd|D}tfd |Drd tj|d Stj|S#t$rtjdwxYwcc}w) NTrz%Request content must be valid unicode z\x02xrc3BK|]}j||ywN)r!).0xescape_control_charss r z.request_content_for_console..7sHa/33Aq9Hsc3&K|]}|v ywr+)r,charr.s rr/z.request_content_for_console..8s 9D4' ' 9sz "$(printf z)") get_text ValueErrorrrrangechrjoinanyshlexquote)rtexti escaped_textr.s @rrequest_content_for_consoler>.sOt, t .as8Cu{{3'8!z -d )rr" server_connpeernameroptionsexport_preserve_original_ip pretty_hostportappendritemslowermethodcontent pretty_urlr7r>)r rargs server_addrresolvekvrs r curl_commandr^?sua G 8D/0}}/E/E!--((+4K //    ; .(()7<<.;-qI K  G%%D%1(1 779) ) KK ' TaS1#;' 'D ( ~~ T./ /D w~~&&KK""#hh8488GT5g>?@@ NrcFt|}t||j}d|j|g}|jj dD]\}}|j |d|djd|D}|jr|dt|zz }|S)Nr TrCrErGc3FK|]}tj|ywr+rIrJs rr/z!httpie_command..ss45;;s#4rLz <<< ) rr"rXrVrrTrSr7rWr>)r rurlrYr\r]cmds rhttpie_commandrcgsa G   C GNNC (D%%D%1!1 qcA3K ! ((4t4 4C w4W=== Jrct|}|jtjdt j |S)NzRequest content missing.)r raw_contentrrr assemble_requestrs r raw_requestrgys;a G"%%&@AA  $ $W --rct|}|jtjdt j |S)NzResponse content missing.)r&rerrr assemble_responser%s r raw_responserjs;"H#%%&ABB  % %h //rc\t|tjxr&|jxr|jjdu}t|tjxr&|j xr|j jdu}|rx|rvt |t|g}t|tjr5|jr)|j|jj|j|S|r t |S|r t|Stjd)zOReturn either the request or response if only one exists, otherwise return bothNz.Can't export flow with no request or response.)rr rrrer$rgrj websocketrS_get_formatted_messagesr7rr)r separatorrequest_presentresponse_presentpartss rrawrrs 1dmm$XXqyy7L7LTX7X 1dmm$ / JJ / JJ " "$ . +Qa1 a 'AKK LL<<> ?~~e$$ 1~ A%%&VWWr)r@httpierrrgrjformatsc peZdZdZejddeefdZejddedejde jjdd fd Z ejd ded ejdd fd Zejdded ejdefdZy )Exportc4|jdtddy)NrPFa[ When exporting a request as an external command, make an effort to connect to the same IP as in the original request. This helps with reproducibility in cases where the behaviour depends on the particular host we are connecting to. Currently this only affects curl exports. ) add_optionbool)selfloaders rloadz Export.loads )    rzexport.formatsr cNtttjS)z@ Return a list of the supported export formats. )listsortedrtkeys)rzs rrtzExport.formatss F7<<>*++rz export.fileformatrpathNc|tvrtjd|zt||} t|d5}t |t r|j |n!|j |jdddddy#1swYyxYw#t$r(}tjt|Yd}~yd}~wwxYw)z( Export a flow to path. No such export format: %swbzutf-8surrogateescapeN) rtrropenrbyteswriteencodeOSErrorloggingerrorstr)rzrrrr]fpes rfilez Export.files  ))*E*NO O FOD ! "dD! CRa'HHQKHHQXXg/@AB  C C C  " MM#a& ! ! "s5 BAB B BBB CCCz export.clipr c|j||} tj|y#tj$r(}t j t |Yd}~yd}~wwxYw)z8 Export a flow to the system clipboard. N) export_str pypercliprPyperclipExceptionrrr)rzrr rWrs rclipz Export.clipsN //&!, " NN7 #++ " MM#a& ! ! "s*A%A  A%exportc|tvrtjd|zt||}tj|dd}tj |dd}|S)z6 Export a flow and return the result. rutf8rbackslashreplace)rtrrr always_bytes always_str)rzrr rWs rrzExport.export_strsb  ))*E*NO O&/!$''9JK%%gv7IJr)__name__ __module__ __qualname__r|rrrrtrFlow mitmproxytypesPathrrrr1rrrvrvs  W__%&,#,', W__]#"3"dii"y7K7K"PT"$" W__]#"3"499""$"W__X   s  rrv)s )#rr9collections.abcrrrmitmproxy.typesrrrrrr mitmproxy.net.http.http1r mitmproxy.utilsr rRequestrr"Responser&rr>r^rcrrgrjrrdictrt__annotations__rvr1rrrsS $$ -$tyyT\\**$* dmm%%#%"%DII%#%PdiiC$.499..0DII0%0X499XX0:>    :c8TYYKu455 6@@r