iXtddlmZddlZddlZddlZddlZddlZddlZddlZddl Z ddl Z ddl m Z mZmZmZmZmZmZGddZGdde jZd;dZdd Zd?d#Zd@d(ZdAd*Z dBdCd/Z dBdCd0ZdDd2Z dEd3Z!dFd4Z"Gd5d6Z#Gd7d8Z$Gd9d:Z%dS)G) annotationsN) async_lib config_libconsterrors server_lib transformstypescPeZdZdZdddd Zd dd ddZd dd ddZddZddZdS)AuthenticatedHTTPClientz HTTP client that: - Is thread-safe - Works in both async and sync contexts - Handles auth (opt in) - Handles branch environments N)request_timeoutenv str | Noner int | datetime.timedelta | None signing_keysigning_key_fallbackct|_tj|_|jj|_||_||_ ||_ t|tr |dz |_ dSt|tjr||_ dSd|_ dS)Ng>@)ThreadAwareAsyncHTTPClient initialize _http_clienthttpxClient_http_client_sync build_requestbuild_httpx_request_env _signing_key_signing_key_fallback isinstanceint_default_timeoutdatetime timedelta total_seconds)selfrr rrs R/root/projects/qq-shell/venv/lib/python3.11/site-packages/inngest/_internal/net.py__init__z AuthenticatedHTTPClient.__init__"s788CCEE!& $(#9#G  '%9" os + + )$3d$:D ! ! ! ); < < )$3$A$A$C$CD ! ! !$(D ! ! !F)authheadersurlstrr*boolr+dict[str, str] | Nonereturn types.MaybeError[httpx.Response]c K|d|it|jdd|pi}|r/t|j|j||j|jd{V}n!t|j|j|d{V}t|tr|S|j dkrtd|j d|j S|S) a Perform an async HTTP GET request. Args: ---- url: Request URL auth: Include the Authorization header. Never set to True if the request is not to an Inngest server headers: Additional request headers GETNr framework server_kindr+rr HTTP error:  ) rcreate_headersrfetch_with_auth_fallbackrrrrfetch_with_thready_safetyr Exception status_codetextr&r,r*r+reqress r'getzAuthenticatedHTTPClient.get=sB"&&     " $ =b '     0!& -%)%? CC2!&C c9 % % J ?c ! !HCOHHchHHII I r)c l|d|it|jdd|pi}|r#t|j||j|j}n|j|}t|tr|S|j dkrtd|j d|j S|S) a  Perform a sync HTTP GET request. Args: ---- url: Request URL auth: Include the Authorization header. Never set to True if the request is not to an Inngest server headers: Additional request headers r3Nr4r7r8r9r:r;) rr<rfetch_with_auth_fallback_syncrrrsendr r?r@rArBs r'get_syncz AuthenticatedHTTPClient.get_syncss"&&     " $ =b '     3/& -%)%? CC(--c22C c9 % % J ?c ! !HCOHHchHHII I r)bodyobjectc PK|d|t|jdd||j}t |j|j||j|jd{V}t|tr|S|j dkrtd|j d|j S|S) a Perform an asynchronous HTTP POST request. Handles authn Args: ---- url: The pathname to the endpoint, including query string body: The body of the request Returns: ------- A httpx.Response object POSTNr4r+jsontimeoutr8r9r:r;) rr<rr"r=rrrrr r?r@rAr&r,rJrCrDs r'postzAuthenticatedHTTPClient.posts&&  "I  )'   -    " )!%!;           c9 % % J ?c ! !HCOHHchHHII I r)c 4|d|t|jdd||j}t |j||j|j}t|tr|S|j dkrtd|j d|j S|S) a Perform a synchronous HTTP POST request. Handles authn Args: ---- url: The pathname to the endpoint, including query string body: The body of the request Returns: ------- A httpx.Response object rMNr4rNr8r9r:r;) rr<rr"rGrrrr r?r@rArQs r' post_syncz!AuthenticatedHTTPClient.post_syncs&&  "I  )'   ,  " )!%!;     c9 % % J ?c ! !HCOHHchHHII I r))rrr rrrrr)r,r-r*r.r+r/r0r1)r,r-rJrKr0r1) __name__ __module__ __qualname____doc__r(rErIrRrTr)r'r r s<@ ))))))>)- 444444t)- 000000d((((T''''''r)r c2eZdZUdZdZded<d dZd dZdS) rz Thin wrapper around httpx.AsyncClient. It keeps track of the thread it was created in, which is critical since asyncio is not thread safe: calling an async method in a different thread will raise an exception N int | None_creation_thread_idr0r.cj|jtdtj}|j|kS)Nz-did not initialize ThreadAwareAsyncHTTPClient)r\r? threading get_ident)r&current_thread_ids r'is_same_threadz)ThreadAwareAsyncHTTPClient.is_same_threads8  # +KLL L%/11'+<<(T(TU](T(T,2J4Z'-/[%./[/[EM/[/[ G 25 $(./8A $.45    $9?@ Nr) public_path request_urlr- serve_origin serve_pathctj|}tj|}tj|}|j}|j}|j}|r@http HTTPStatus FORBIDDEN UNAUTHORIZEDr?r, __cause__)rrrrrrDerrnew_errs r'r=r=asc @j1+>> @ @  ,:@A-            O)4?+GH I I$0N*56JKKMM OJ0>D E2C  EEEEEFFsB C C>"C93C>9C>c|3dtj||jtjjj< ||}|jtj j tj j fvrJ|Hdtj||jtjjj<||}|S#t$r-}td|jd|}||_|cYd}~Sd}~wwxYw)z Send an HTTP request with the given signing key. If the response is a 401 or 403, then try again with the fallback signing key. Returns an error when receiving a non-OK response Nrrr)r rr+rrjrrlrHr@rrrrr?r,r)rrrrrDrrs r'rGrGs @j1+>> @ @  ,:@Akk'"" O)4?+GH I I$0N*56JKKMM OJ0>D E++g&&C EEEEEFFsB C C8 "C3-C83C8r+%dict[str, str] | dict[str, list[str]]ci}|D]?\}}|}t|tr |d||<:|||<@|S)z> Ensure that known headers are in the correct casing. r)itemslowerr list)r+ new_headerskvs r'normalize_headersrscK 1 GGII a   qTKNNKNN r)r,modeserver_lib.ServerKindtypes.MaybeError[str]cd|vr9d|ddvrtdS|ddur|tjjurd|}nd|} t j|}|j dkrtd S| S#t$r }|cYd}~Sd}~wwxYw) N.:zhttp://zhttps://zinvalid URL: no domain or portrFr|z invalid URL) stripr? startswithr ServerKindCLOUDrrrrgeturl)r,rparsedrs r' parse_urlrs #~~#SYYy%9%9%?%? %K%KKK9::: ~~f&& :(. . ."S""CC!C//C&&s++ =B  ]++ +}}  s$48C-C C C CChttpx.ResponsecK|dur|d{VStj}|S|dfdd{VS)zg Safely handles the situation where the async HTTP client is called in a different thread. TNc.Src)rH)rrsr'z+fetch_with_thready_safety..s   ))r))rarHrget_event_looprun_in_executor)rrrloops `` r'r>r>s$&&[[)))))))))  # % %D |(((%% )))))      r)rJbytesunix_msr[c| ttj}tj|}t |t r|t jtj| d|tj }| t| d|}d|d|S)z Sign an HTTP request in the same way an Inngest server would. This is only needed for tests that mimic Inngest server behavior. Nutf-8t=&s=)roundtimer canonicalizer r?hmacnewremove_signing_key_prefixencodehashlibsha256updater- hexdigest)rJrr canonicalizedmacsigs r' sign_requestrs $$+D11M-++ (,[99@@II  C JJs7||""7++,,, --//C " ! !C ! !!r)cr| ttj}tjt j|d|tj}| t|d| }d|d|S)z Sign an HTTP response. Nrrr) rrrrr rrrrrr-r)rJrrrrs r' sign_responsers $$ (,[99@@II   C JJs7||""7++,,, --//C " ! !C ! !!r)types.MaybeError[str | None]cr|tjjkrdSd}d}|tjjj}|,tjdtjjjdStj |}d|vrt|dd}d|vr|dd}|tj dS|&ttjjjdStjt#j|d|t(j}|r5|t/|dtj||stjS|S) Nz7cannot validate signature in production mode without a z headertrszBcannot validate signature in production mode without a signing keyz header is malformedr)rr DEV_SERVERrErj SIGNATURErlrHeaderMissingErrorrrparse_qsr!SigningKeyMissingErrorr?rrr rrrrrr-compare_digestrSigVerificationFailedError) rJr+rr timestamp signature sig_headerrrs r' _validate_sigr,s z$///tIIZ1;ABBJ( sjFZFdFj s s s   &&z22 &==F3KN++I &==s AI, P   #-3 I I I    (,[99@@II   C 3 3y>>((11222  y#--// : :30222 r)ctj|}t|tr|St ||||}t|tr|t ||||}|S)a[ Validate the request signature. Falls back to the fallback signing key if signature validation fails with the primary signing key. Args: ---- body: Request body. headers: Request headers. mode: Server mode. signing_key: Primary signing key. signing_key_fallback: Fallback signing key. rJr+rr)r rr r?r)rJr+rrrrrs r'validate_request_sigr]s*+D11M-++       C #y!!  &:&F,     Jr)c(t||||S)a3 Validate an HTTP response signature in the same way an Inngest server would. This is only needed for tests that mimic Inngest server behavior. Args: ---- body: Request body. headers: Request headers. mode: Server mode. signing_key: Primary signing key. r)rrs r'validate_response_sigrs'&       r)c.eZdZd dZd dZdd Zdd Zd S) ServerTimingnamer-r0Nonec0||_d|_d|_dSrc)_name_start_counter _end_counterr&rs r'r(zServerTiming.__init__s ,0*.r)cD|jtj|_|Src)rr perf_counterrds r' __enter__zServerTiming.__enter__s!   &"&"3"5"5D  r)argsrKcH|jtj|_dSdSrc)rrrr&rs r'__exit__zServerTiming.__exit__s*   $ $ 1 3 3D    % $r)cx|j|jdSt|j|jz dz}|jd|SNr|rz;dur=)rrr!rr&durs r' to_headerzServerTiming.to_headersK   &$*;*C24$t'::dBCC*((3(((r)Nrr-r0r)r0rrrKr0rr0r-)rUrVrWr(rrrrYr)r'rrsd////  4444))))))r)rc:eZdZdZddZddZdd Zdd Zdd Zd S)_AsyncBlockServerTimingzN Special server timing that tracks how long the event loop is blocked rr-r0rc>d|_||_d|_d|_dSNr) _block_durrr _tracker_taskrs r'r(z _AsyncBlockServerTiming.__init__s&!" ,08<r)c|jtj|_|j+t j||_|Src)rrrrasyncio create_task_trackerrds r'rz!_AsyncBlockServerTiming.__enter__sE   &"&"3"5"5D    %!(!4T]]__!E!ED  r)rrKcJ|j|jdSdSrc)rcancelrs r'rz _AsyncBlockServerTiming.__exit__s.   )   % % ' ' ' ' ' * )r)cKtj}tjdd{Vd} tj}|xjt ||z |z dz c_|}tj|d{VT)Nrg?)rrrsleeprmax)r&lastintervalnows r'r z _AsyncBlockServerTiming._trackers ""mA *#%%C OOs3:#8!<< . s!*/ar))keyc6g|]}|SrY)r).0timings r' z+ServerTimings.to_header..s$FFFFV--//FFFr)cg|] }|dk| S)r|rY)r"rs r'r$z+ServerTimings.to_header..s///qBww!wwwr)z, )rrrrrrsortedjoin)r&timingsvaluess r'rzServerTimings.to_headers     M  #  $ L A  //    GFgFFF0/V///yy   r)Nrr)rUrVrWr(rrYr)r'rrs<////$!!!!!!r)r)rrr5rfr6rgr0rh) rvrrwr-rxrryrr0r-) rrrrrrrrrrr0r1) rrrrrrrrr0r1)r+rr0rh)r,r-rrr0r)rrrrrrr0rrc)rJrrr-rr[r0r) rJrr+rhrrrrr0r) rJrr+rhrrrrrrr0r) rJrr+rhrrrr-r0r)& __future__rrr#rrrr^r urllib.parserrinngest._internalrrrrrr r r AsyncClientrr<rr=rGrrr>rrrrrrrrrYr)r'r.s""""""   ]]]]]]]]@!2*<2222j,,,,^""""J((:"""""@"""""0....b****Z6)))))))).))))))))))))))))X,!,!,!,!,!,!,!,!,!,!r)