o i2 @sddlZddlZddlZddlZddlmZddlmZmZm Z ddl Z ddl Z ddl m Z ddlmZddlmZddlmZddlmZmZdd lmZdd lmZdd lmZdd lmZmZm Z m!Z!e j"d dddZ#GdddZ$ee$Z%de&de'fddZ( d,de&dee&de ee&ee&ffddZ)e#j*dddej+e j,ddd d!d"fd#e'fd$dZ-e#j*d%d&dej+ed'd%Z.e#j*d(d)dej+ed*d+Z/dS)-N)asynccontextmanager)AsyncGeneratorOptionalTuple)logger)Columns)Live)Panel)synchronize_api synchronizer)requires_login)console)API)configremove_user_configupdate_user_configuser_config_pathauthz Manage Pipecat Cloud credentialsT)namehelpno_args_is_helpc@s`eZdZddZedeeeeeeeefdffddZ d d e d e deefd d Z dS) _AuthFlowcCsdSN)selfrrR/home/ubuntu/.local/lib/python3.10/site-packages/pipecatcloud/cli/commands/auth.py__init__$sz_AuthFlow.__init__returnNc Cszt4IdHi}td}t||j|ddid4IdH9}|jdkr1td|j| IdH}|d|_ |d|_ |d }| d }|j ||fVWdIdHn 1IdHsbwYWdIdHWdS1IdHsywYWdStyd VYdSw) N login_pathuse_codetrue)paramszFailed to start auth flow: token_flow_id wait_secretweb_urlcode)NNN) aiohttp ClientSessionrconstruct_api_urlrdebugpoststatus Exceptionjsonr#r$get)rsessionurlrespdatar%r&rrrstart's&      *2 z_AuthFlow.startD@@timeoutnetwork_timeoutc st}t4IdH}t||krz|jtdtd|j|jdtj ||dd4IdH\}|j dkr| IdH}|ddkrh|d WdIdHWWdIdHS|dd kr WdIdHWWdIdHd St d IdH WdIdHWq1IdHswYWn tj tjfyYqwt||ks WdIdHdS1IdHswYdS) Napi_hostlogin_status_path)r#r$)total)r!r7r"r,completetokenfailure)asyncioget_event_looptimer'r(r/rr#r$ ClientTimeoutr,r.sleep TimeoutError ClientError)rr7r8 start_timer0r2r3rrrfinish;s@    00z_AuthFlow.finish)r5r6) __name__ __module__ __qualname__rrrrrstrr4floatrHrrrrr#s  rr1rc Csz7dtjvp dtjv}ttjdptjd}|r |s WdSt}t|tjr2|jdvr2WdS||WStj t t fyEYdSw)NWSL_DISTRO_NAME WSL_INTEROPDISPLAYWAYLAND_DISPLAYF)openz x-www-browserzxdg-open) osenvironboolr/ webbrowser isinstanceGenericBrowserr open_new_tabError ImportErrorAttributeError)r1is_wsl has_displaybrowserrrr _open_urlXs r`r= active_orgc sxt4IdH}|jtddd|id4IdH{}|jdkr|IdH}|d}|rZ|D]&}|d|krY|d|dfWdIdHWdIdHSq3|r||d d|d dfWdIdHWdIdHS WdIdHWdIdHd Std |j1IdHswYWdIdHdS1IdHswYdS) Norganization_path AuthorizationzBearer )headersr" organizationsr verboseNamer)NNz)Failed to retrieve account organization: )r'r(r/rr)r,r.r-)r=rar0r2r3reorgrrr_get_account_orgms6     .rhloginz*Login to Pipecat Cloud and get a new token)rrFz --headlessz-hzJSkip opening a browser window for authentication and print the URL instead)rheadlessc s:td}t}|rtd|z|4IdH\}}}|dur8td WdIdHWdS|rTtdddtd|ddtd|d ddntt d |d |d |d ddd|st j dddd}| dkrtdt WdIdHWSt||rtdddtD]}|IdH}|durnq|durtdddt WdIdHWStdddzt||IdH\} } td| d| d| durWntytdddt YWdIdHWSwtdddtdtddnttjd d!d"dd#} tD]}|IdH}|dur9n| td$|d%d&q)|durm| td't WdWdIdHWS| tjd(d!d"z"t||IdH\} } | td| d| d| durWn)ty| td)t YWdWdIdHWSwWdn 1swYtd*td+WdIdHn 1IdHswYWnty} zt| td,WYd} ~ nd} ~ wwtd-t|| dS).NrgzCurrent active org: z[Unable to connect to Pipecat Cloud API. Please check your network connection and try again.z% To authenticate with Pipecat Cloud: T)flushz Visit this URL: z Then enter this code:  zN[bold]To authenticate with Pipecat Cloud:[/bold] Visit this URL: [blue][link=]z1[/link][/blue] Then enter this code: [cyan bold]z [/cyan bold]z$[bold]Authentication Required[/bold]blue)title border_stylez1 Press Enter to open the browser (or 'q' to quit)F)default show_defaultqz)[yellow]Authentication cancelled[/yellow]z)Waiting for authentication to complete...zError: Authentication failedzObtaining account data...zSetting namespace to z ()zError: Account has no associated namespace. Have you completed the onboarding process? Please first sign in via the web dashboard.z)Web authentication finished successfully!zAccount details stored to z4[dim]Waiting for authentication to complete...[/dim]dotsspinner transientz8[dim]Waiting for authentication to complete... (attempt z)[/dim]zAuthentication failedz![dim]Obtaining account data[/dim]z{Account has no associated namespace. Have you completed the onboarding process? Please first sign in via the web dashboard.zRWeb authentication finished successfully! [dim]Account details stored to [magenta]z[/magenta][/dim]z7Unexpected login error occured. Please contact support.z![dim]Storing credentials...[/dim])rr/rrr*r4r errorprintr typerpromptlowerExitr` itertoolscountrHrhr-rrr,updatestopsuccessr) rjra auth_flowr#r%r&responseattemptresult account_nameaccount_name_verboseliveerrrris     #    / 5     U  `&*l  logoutzLogout from Pipecat CloudcsVtjddd tWdn1swYtjddtddddS) NzRemoving user IDrvrwzbUser credentials for Pipecat Cloud removed. Please sign out via dashboard to fully revoke session.z[dim]Please visit:[/dim] dashboard_hostz /sign-out)subtitle)r r,rrrr/rrrrrs whoamiz=Display data about the current user. Also show Daily API key.c std}zttjddddd}tj|dIdH\}}|r,tWdWS| tdtj ||d IdH\}}|rRt tWdWS|d rZ|d s[| tjd ddd}ztj ||d IdH\}}Wn t y~Ynw|td |ddd|d d|d d|r|ddndg}t|WdWdS1swYWdSt ytdYdSw)Nrgz*[dim]Requesting current user data...[/dim]rvrwTry)rz;[dim]Requesting user namespace / organization data...[/dim])rgrr verbose_namez$[dim]Fetching Daily API key...[/dim]zP[bold]User ID[/bold] [bold]Active Organization[/bold] [bold]Daily API Key[/bold]useruserIdrlz [dim](z)[/dim] apiKeyz[dim]N/A[/dim]z2Unable to obtain user data. Please contact support)rr/rr r,rrr~rrorganizations_current print_errororganizations_daily_keyr-rrrr|)rgr user_datar|account daily_api_keymessagerrrwhomai!s\     & -rr)0r@rrSrV contextlibrtypingrrrr'r~logurur rich.columnsr rich.liver rich.panelr pipecatcloud._utils.async_utilsr r pipecatcloud._utils.auth_utilsr !pipecatcloud._utils.console_utilsr pipecatcloud.cli.apirpipecatcloud.cli.configrrrrTyperauth_clirAuthFlowrLrUr`rhcommandcreate_blockingOptionrirrrrrrsf        2