o ;i|@s8ddlZddlZddlmZddlmZddlZddlm Z ddl m Z ddl m Z ddlmZdd lmZdd lmZdd lmZmZmZmZmZmZdd lmZdd lmZGdddZe eZdddddddee de!de!dee dee f ddZ"dddddde de dee de!de!dee f ddZ#dS)N)AsyncGenerator)Optional)asynccontextmanager)api_pb2)synchronize_api)open_url)run_temporary_http_server)_Client)DEFAULT_SERVER_URL_lookup_workspace_store_user_configconfigconfig_profilesuser_config_path) AuthError) OutputManagerc @sreZdZdefddZe ddeedeedee eeefdffdd Z dd e d e dee j fddZdS) _TokenFlowclientcCs |j|_dSN)stub)selfrrD/home/ubuntu/.local/lib/python3.10/site-packages/modal/token_flow.py__init__s z_TokenFlow.__init__N utm_sourcenext_urlreturnc sfdd}tj}|tjd|gt|4IdH5}tj||t| ddd}j |IdH}|j _ |j _ |j |j|jfVWdIdHdS1IdHsZwYdS) mdmd:hiddencsddi}tjjj|dS)NzAccess-Control-Allow-Origin*)textheaders)aiohttpwebResponse token_flow_id)requestr!rrrslash"sz_TokenFlow.start..slash/N:)rrlocalhost_port)r"r# Application add_routesgetr rTokenFlowCreateRequestintsplitrTokenFlowCreater% wait_secretweb_urlcode)rrrr(appurlreqresprr'rstarts .z_TokenFlow.startD@@timeoutgrpc_extra_timeoutcs@tj|j||jd}|jj|d||dIdH}|js|SdS)r)r%r>r4N)retryr>)rTokenFlowWaitRequestr%r4r TokenFlowWaitr>)rr>r?r9r:rrrfinish3s z_TokenFlow.finish)NN)r<r=)__name__ __module__ __qualname__r rrrstrrtupler;floatrTokenFlowWaitResponserCrrrrrs& rT)profileactivateverifysourcerrKrLrMrNrc stjd|d}t}d}t|4IdH}t|} | ||4IdH}\} } } |d.t| r:| dn| d| d| d| d| rV| d | d Wdn1s`wY|d '} t D]}| IdH}|dur~n | d |d dqoWdn1swYWdIdHn 1IdHswY| d|j}WdIdHn 1IdHswY|dusJ|jr| d|jdt|j|j||||dIdHdS)N server_urlrKz-Waiting for authentication in the web browserzThe web browser should have opened for you to authenticate and get an API token. If it didn't, please copy this URL into your web browser manually: zd[red]Was not able to launch web browser[/red] Please go to this URL manually and complete the flow: z[link=]z[/link] zEnter this code: [yellow]z [/yellow] z%Waiting for token flow to complete...z/Waiting for token flow to complete... (attempt )z8[green]Web authentication finished successfully![/green]z*[green]Token is connected to the [magenta]z[/magenta] workspace.[/green]rKrLrMrO)rr/rr anonymousrr;statusrprint itertoolscountrCupdaterOworkspace_username _set_tokentoken_id token_secret)rKrLrMrNrrOoutputresultr token_flow_r5r6rVattemptrrr _new_tokenEsP   ( (  rdrTr]r^rOc stjd|d}t}|r'|d|dt|||fIdH|d|dur\dtjvr6tjd}n&z t|||IdH}Wnt yX}z |sRd} t | ||d}~ww|j }||d} |durm|t krm|| d<|srt st|nd} | d t| || d Wdn1swY|d td |d dddD} dd| D} d| }| rt| dkrdnd}t| dkrdnd}|d|d|d|ddSdS)NrOrPzVerifying token against [blue]z[/blue]z+[green]Token verified successfully![/green] MODAL_PROFILEzSNo profile name given, but could not authenticate client to look up workspace name.)r]r^z Storing token)rKactive_profilez![green]Token written to [magenta]z[/magenta] in profile [magenta]z[/magenta].[/green]cSs g|] }tj|r |ndqSr)osenvironr/.0varrrr s z_set_token..)MODAL_TOKEN_IDMODAL_TOKEN_SECRETcSsg|]}|dur|qSrrrirrrrlsz / rsareisz[yellow]Warning: The z environment variable z7 set; this will override your new credentials.[/yellow])rr/rrWr rMrgrhr rusernamer rrVr rjoinlen)r]r^rKrLrMrOr_ workspaceexcmsg config_datarfenv_vars env_vars_used env_vars_strroverbrrrr\|sT        r\)$rXrgcollections.abcrtypingr aiohttp.webr"synchronicity.async_wrapr modal_protor_utils.async_utilsr_utils.browser_utilsr_utils.http_utilsr rr rr r r rr exceptionrr_rr TokenFlowrGboolrdr\rrrrsb           - ;