o WZ@sdZddlmZddlmZddlmZddlZddlZddlZddlZddl Z ddl Z ddl m Z ddl mZddl mZdd l mZdd l mZdd lmZdd lmZdd lmZddlZddlmZddlmZejrqdndZe jZgdZ Gddde!ej"e#Z$Gddde!ej"e#Z%Gddde!ej"e#Z&Gddde#Z'ddZ(ddZ)ddZ*d d!Z+d"d#Z,dd2d3Z3d4d5Z4d6d7Z5d8d9Z6d:d;Z7dS)?z@Module for common transport utilities, such as request wrapping.)absolute_import)division)unicode_literalsN) exceptions)config)log)metrics) properties) console_attr) console_io) platforms)urllib)ziputf-8)z*https://accounts.google.com/o/oauth2/tokenz*https://www.googleapis.com/oauth2/v3/tokenz*https://www.googleapis.com/oauth2/v4/tokenz#https://oauth2.googleapis.com/tokenz-https://oauth2.googleapis.com/oauth2/v4/tokenc@s8eZdZdZddZeejddZejddZ dS) Requesta Encapsulates parameters for making a general HTTP request. Attributes: uri: URI of the HTTP resource. method: HTTP method to perform, such as GET, POST, DELETE, etc. headers: Additional headers to include in the request. body: Body of the request. cCs||_||_||_||_dS)aInstantiates a Request object. Args: uri: URI of the HTTP resource. method: HTTP method to perform, such as GET, POST, DELETE, etc. headers: Additional headers to include in the request. body: Body of the request. Returns: Request N)urimethodheadersbody)selfrrrrr:/tmp/google-cloud-sdk/lib/googlecloudsdk/core/transport.py__init__>s  zRequest.__init__cOdS)zReturns a Request object. Args: *args: args to be passed into http.request **kwargs: dictionary of kwargs to be passed into http.request Returns: Request Nr)clsargskwargsrrrFromRequestArgsOzRequest.FromRequestArgscCr)zAReturns the args and kwargs to be used when calling http.request.Nr)rrrr ToRequestArgs\rzRequest.ToRequestArgsN) __name__ __module__ __qualname____doc__r classmethodabcabstractmethodrrrrrrr4s   rc@s*eZdZdZddZeejddZdS)ResponsezEncapsulates responses from making a general HTTP request. Attributes: status_code: headers: Headers of the response. body: Body of the response. cCs||_||_||_dS)zInstantiates a Response object. Args: status_code: headers: Headers of the response. body: Body of the response. Returns: Response N) status_coderr)rr(rrrrrrjs  zResponse.__init__cCr)zReturns a Response object. Args: response: raw response from calling http.request. Returns: Response Nr)rresponserrr FromResponseyrzResponse.FromResponseN) r r!r"r#rr$r%r&r*rrrrr'as r'c@sFeZdZdZeZeZej ddZ   d ddZ de dfdd Z dS) RequestWrapperaKClass for wrapping http requests. The general process is that you can define a series of handlers that get executed before and after the original http request you are mapping. All the request handlers are executed in the order provided. Request handlers must return a result that is used when invoking the corresponding response handler. Request handlers don't actually execute the request but rather just modify the request arguments. After all request handlers are executed, the original http request is executed. Finally, all response handlers are executed in order, getting passed both the http response as well as the result from their corresponding request handler. Attributes: request_class: Class used to represent a generic HTTP request. response_class: Class used to represent a generic HTTP request. cCr)z9Decodes the response body according to response_encoding.Nr)rr)response_encodingrrrDecodeResponserzRequestWrapper.DecodeResponseNFc Cs<ttjjj}ttttt dt j tt d|g}tjj j}|r1|ttd|tjj j}|rD|ttd|tjjj} | rW|ttd| tjj jr}tjj j} tjj j} |tt| | su|ndt|dtjvrtjddkr|ttdd |j|||d |S) a'Wraps request with user-agent, and trace reporting. Args: http_client: The original http client to be wrapped. response_encoding: str, the encoding to use to decode the response. streaming_response_body: bool, True indicates that the response body will be a streaming body. redact_request_body_reason: str, the reason why the request body must be redacted if --log-http is used. If None, the body is not redacted. Returns: http, The same http object but with the request method wrapped. z user-agentCookiezX-Goog-Request-ReasonzX-Goog-Allowed-ResourcesNCLOUDSDK_CORE_DRY_RUN1cWsdSNr)rrrrrz1RequestWrapper.WrapWithDefaults..)r,)MakeUserAgentStringr VALUESr command_nameGetHandlerRecordStartTimeReportDurationMaybePrependToHeaderrCLOUDSDK_USER_AGENTAppendToHeadercore trace_tokenappend SetHeaderrequest_reasonresource_policyorg_restriction_headerlog_httpGetBoollog_http_redact_tokenlog_http_show_request_body LogRequest LogResponseosenvironLogRequestDryRun WrapRequest) r http_clientr,streaming_response_bodyredact_request_body_reason gcloud_uahandlers trace_valuerArequest_org_restriction_headers redact_tokenshow_request_bodyrrrWrapWithDefaultss`          zRequestWrapper.WrapWithDefaultscs&|jfdd}||_dS)aXWraps an http client with request modifiers. Args: http_client: The original http client to be wrapped. handlers: [Handler], The handlers to execute before and after the original request. exc_handler: f(e), A function that takes an exception and handles it. It should also throw an exception if you don't want it to be swallowed. exc_type: The type of exception that should be caught and given to the handler. It could be a tuple to catch more than one exception type. response_encoding: str, the encoding to use to decode the response. c sjj|i|}ddt|jD}i|_t|D]\}}t||\}}||j|<qg}D] }||}||q1z|\} } | i| } Wnyi} zd} rd| WYd} ~ dSd} ~ wwdurt | } j | } t |D] \}}|j r| | |q| S)z)Replacement http_client.request() method.cSsi|]\}}||qSrr).0hvrrr szFRequestWrapper.WrapRequest..WrappedRequest..N) request_classrsix iteritemsr _EncodeHeaderrequestr?rr-response_classr*rr))rrhandler_requestrrYrZ modifier_datahandlermodifier_result modified_argsmodified_kwargsr)ehandler_responsedata exc_handlerexc_typerR orig_requestr,rrrWrappedRequests:       z2RequestWrapper.WrapRequest..WrappedRequestNr`)rrNrRrlrmr,rorrkrrMs %zRequestWrapper.WrapRequest)NFN)r r!r"r#rr\r'rar%r&r-rW ExceptionrMrrrrr+s  Wr+c@seZdZdZdddZdS)r7zA holder object for a pair of request and response handlers. Request handlers are invoked before the original http request, response handlers are invoked after. NcCs||_||_dS)aDCreates a new Handler. Args: request: f(request) -> data, A function that gets called before the original http request gets called. It is passed a Request object that encapsulates the parameters of an http request. It returns data to be passed to its corresponding response hander. response: f(response, data), A function that gets called after the original http request. It is passed a Response object that encapsulates the response of an http request as well as whatever the request handler returned as data. N)r`r))rr`r)rrrr6s zHandler.__init__r1)r r!r"r#rrrrrr7/sr7cCs4t|tjr |d}t|tjr|d}||fS)Nr) isinstancer] text_typeencodeheadervaluerrrr_Gs    r_c t\fdd}|S)aPrepends the given value if the existing header does not start with it. Args: header: str, The name of the header to prepend to. value: str, The value to prepend to the existing header value. Returns: A function that can be used in a Handler.request. csb|j}d}t|D]\}}|kr|}||=nq |s+d|}||<dS)z0Maybe prepends a value to a header on a request. N)rr]r^lower startswithstripr`r current_valuehdrrZrurr_MaybePrependToHeader[s  z3MaybePrependToHeader.._MaybePrependToHeaderr_)rvrwrrrurr:Os r:crx)aAppends the given value to the existing value in the http request. Args: header: str, The name of the header to append to. value: str, The value to append to the existing header value. Returns: A function that can be used in a Handler.request. cs\|j}d}t|D]\}}|kr|}||=nq |r(|dn|<dS)z)Appends a value to a header on a request.ryrzN)rr]r^r{r}r~rurr_AppendToHeaderxs z'AppendToHeader.._AppendToHeaderr)rvrwrrrurr<ls  r<crx)zSets the given header value in the http request. Args: header: str, The name of the header to set to. value: str, The new value of the header. Returns: A function that can be used in a Handler.request. cs<|j}t|D]}|kr||=nq|<dS)zSets a header on a request.N)rr]iterkeysr{)r`rrrurr _SetHeaders zSetHeader.._SetHeaderr)rvrwrrrurr@s  r@cfdd}|S)zAdds the given query parameter to the http request. Args: param: str, The name of the parameter. value: str, The value of the parameter. Returns: A function that can be used in a Handler.request. csVtj|j}tj|j}|<t|}tjj|dd|d<tj|}||_dS)z$Sets a query parameter on a request.T)doseqN) r parseurlsplitrparse_qsquerylist urlencode urlunsplit)r` url_parts query_paramsnew_urlparamrwrr_AddQueryParams  z%AddQueryParam.._AddQueryParamr)rrwrrrr AddQueryParams rTcr)a@Logs the contents of the http request. Args: redact_token: bool, True to redact auth tokens. redact_request_body_reason: str, the reason why the request body must be redacted if --log-http is used. If None, the body is not redacted. Returns: A function that can be used in a Handler.request. c s8|j}|j}|j}|jp d}d}d}rt|rd}d}ndur#}tjdtjdtjdj|dtjd j|d tjd t t |D]\}}r^| d vr^d }tjd||qPtjdtjd|durtj|n tjd|tjdtjdt |dS)zLogs a request.NzdContains oauth token. Set log_http_redact_token property to false to print the body of this request.zeContains oauth token. Set log_http_redact_token property to false to print the body of this response.z=======================z==== request start ====z uri: {uri})rzmethod: {method})rz== headers start ==)s authorizationsx-goog-iam-authorization-tokenz--- Token Redacted ---{0}: {1}z== headers end ==z== body start ==Body redacted: {}z== body end ==z==== request end ====) start_timeredact_resp_body_reason)rrrr IsTokenUrirstatusPrintformatsortedr]r^r{time) r`rrrrredact_req_body_reasonrrYrZrPrUrr _LogRequests@         zLogRequest.._LogRequestr)rUrPrrrrrHs .rHcC dd}|S)z^Dry run the http request. Returns: A function that can be used in a Handler.request. cSs t|)zBlocks a dry-run request.)r DryRunErrorrprrrrs z%LogRequestDryRun.._LogRequestr)rrrrrLsrLFcsfdd}|S)zLogs the contents of the http response. Args: streaming_response_body: bool, True indicates that the response body will be a streaming body. Returns: A function that can be used in a Handler.response. cs|d}t|d}tjdtjd|jtjdtt|j D]\}}tjd||q*tjdtjdrNtjd n|d urZtj|j n tjd |tjd tjd |tjdtjdd S)zLogs a response.rrz---- response start ----z status: {0}z-- headers start --rz-- headers end --z-- body start --zNrz-- body end --z6total round trip time (request+response): {0:.3f} secsz---- response end ----z----------------------) rrrrrr(rr]r^rr)r)rjr time_takenrYrZrOrr _LogResponses,      z!LogResponse.._LogResponser)rOrrrrrI s rIcCr)zwRecords the time at which the request was started. Returns: A function that can be used in a Handler.request. cSs~dtiS)z$Records the start time of a request.r)rrprrr_RecordStartTime6s z)RecordStartTime.._RecordStartTimer)rrrrr8/sr8cCr)z}Reports the duration of response to the metrics module. Returns: A function that can be used in a Handler.response. cSs ~t|d}t|dS)z"Records the duration of a request.rN)rr RPCDuration)r)rjdurationrrr_ReportDurationEsz'ReportDuration.._ReportDurationr)rrrrr9>sr9cCst}z|d}Wn tyd}Ynw|r|r|S|jtjjkr;|jtj j kr;tj r;d tj j tj j}nt|j}|rH|d||S)aGet and cache architecture of client machine. For M1 Macs running x86_64 Python using Rosetta, user_platform.architecture (from platform.machine()) returns x86_64. We can use IsActuallyM1ArmArchitecture() to determine the underlying hardware; however, it requires a system call that might take ~5ms. To mitigate this, we will persist this value as an internal property with INSTALLATION scope. Args: user_platform: platforms.Platform.Current() Returns: client machine architecture client_archNz{}_{})rGetConfigStorer6rqoperating_systemr OperatingSystemMACOSX architecture Architecturex86_64PlatformIsActuallyM1ArmArchitecturerarmstrSet) user_platformactive_config_store cached_archarchrrrGetAndCacheArchitectureNs&    rcCstj}t|}djtjdd|ptj j j t ttj j j |j|jr,|jjnd|tjdddt|tttdS)aReturn a user-agent string for this request. Contains 'gcloud' in addition to several other product IDs used for tracing in metrics reporting. Args: cmd_path: str representing the current command for tracing. Returns: str, User Agent string. a5gcloud/{version} command/{cmd} invocation-id/{inv_id} environment/{environment} environment-version/{env_version} client-os/{os} client-os-ver/{os_version} client-pltf-arch/{architecture} interactive/{is_interactive} from-script/{from_script} python/{py_version} term/{term} {gcloud_mcp_metrics} {ua_fragment} _NT)error heuristic)versioncmdinv_id environment env_versionrJ os_versionris_interactive py_version ua_fragment from_scripttermgcloud_mcp_metrics)r rCurrentrrrCLOUD_SDK_VERSIONreplacer r4rr5r6 INVOCATION_IDGetMetricsEnvironmentenvironment_versionr clean_versionr IsInteractiveplatformpython_versionUserAgentFragmentIsRunFromShellScriptr GetConsoleAttrGetTermIdentifierGetValidMCPMetricsString)cmd_pathrrrrrr3us*      r3cCstjjjpdS)Ni,)r r4r= http_timeoutGetIntrrrrGetDefaultTimeoutsrcCs@|tvrdSd}d}t||durdSt||durdSdS)z=Determine if the given URI is for requesting an access token.Tzb(metadata.google.internal|169.254.169.254)/computeMetadata/.*?/instance/service-accounts/.*?/tokenzUiamcredentials.googleapis.com/v.*?/projects/-/serviceAccounts/.*?:generateAccessTokenNF) TOKEN_URISresearch)rmetadata_regexpimpersonate_service_accountrrrrsrcCs$dtjvrtjd}t|r|SdS)zRReturns the valid MCP metrics string if it exists, otherwise returns empty string.GCLOUD_MCP_METRICSr)rJrKValidateMCPMetricsFormat)metrics_stringrrrrs  rcCsd}t||duS)zChecks if a string follows the MCP metrics format. Args: metrics_string: The string to validate. Returns: True if the string matches the format, False otherwise. zZgoog-mcp/agent/[^/]+/agent-version/[^/]+/mcp-server/[^/]+/mcp-version/[^/]+/mcp-tool/[^/]+N)r fullmatch)rpatternrrrrs r)TN)Fr1)8r# __future__rrrr%rJrrruuidgooglecloudsdk.calliopergooglecloudsdk.corerrrr googlecloudsdk.core.consoler r googlecloudsdk.core.utilr r] six.movesr rPY2ENCODINGuuid4hexrrwith_metaclassABCMetaobjectrr'r+r7r_r:r<r@rrHrLrIr8r9rr3rrrrrrrrsZ               -%* = & '2