o iF @s8ddlZddlZddlmZddlmZddlmZddlm Z ddl m Z ddl m Z ddlmZdd lmZdd lmZmZmZejd d d dZejddd dZejddd dZeeeeejddde je edddfdefddZejddde je ddZ ejddde je ejdddddfdefd dZ!ejd!d"de je ejdd#d$d%dejdddd&dejd'd(d)d*dfd+eded,e"fd-d.Z#ejd/d0de je ejdddd1dfdefd2d3Z$ejd4d5de je ejdddd1dfdefd6d7Z%ejdd8de je ejdddd9dfdefd:d;Z&ejddfdefd?d@Z'ejdAdBde je ej(dCdDdej(dCdEdejddddFdfdGedHedefdIdJZ)ejdKdLde je ej(ddMdejddddNdfdOedefdPdQZ*dS)RN)logger)box)Table) synchronizer)requires_login)console)PIPECAT_CLI_NAME)API)configupdate_user_configuser_config_path organizationszUser organizationsT)namehelpno_args_is_helpkeyszAPI key management commands propertiesz Organization property managementselectzSelect an organization to use.)rrz--organizationz-o organizationc sRtdtjdddtIdH\}}|rtWdn1s'wYzmd}|sUtj dfdd|Dd  IdH}|sLtd WS|d |d f}n(d}|D] }|d |krc|}qY|sut d |dtd WS|d |df}t d|d t d|d d|d dtdWdStyt dYdSw)Norg9[dim]Retrieve user namespace / organization data...[/dim]dotsspinner)NNz'Select default namespace / organizationcs@g|]}|dd|dd|d|df|dkdqS) verboseNamez (r))rvaluechecked).0r current_orgr[/home/ubuntu/.local/lib/python3.10/site-packages/pipecatcloud/cli/commands/organizations.py 8s zselect..choicesrrz Unable to find namespace [bold]'z*'[/bold] in user's available organizationsrz(Current organization set to [bold green]z [dim](z7)[/dim][/bold green] [dim]Default namespace updated in [/dim]z:Unable to update user credentials. Please contact support.)r getrstatusr r typerExit questionaryr ask_asyncerrorr successr Exception)rorg_listr. selected_orgrmatchorr r"r$sX        listz'List organizations user is a member of.cs.td}tjdddtIdH\}}|r$tWdSWdn1s.wY|r9t|sHtj dtddtdSt d t j d d d }|j d dd|j ddd|D]'}|r~|d|kr~|d|ddd|ddqa||d|dqatj|t|dddS)NrrrrzYNo namespaces associated with user account. Please complete onboarding via the dashboard.dashboard_hostsubtitler&dimTF) border_styler show_edge show_lines OrganizationwhitestyleNamerz [cyan bold]rz [/cyan bold]z (active)[/cyan bold]z results title_extra)r r(rr)r r r*r+lenr.rrSIMPLE add_columnadd_rowr/)r!r1r.tablerrrr"r5_s8   z"List API keys for an organization.z!Organization to list API keys for)rc sL|ptd}tjd|dddt|IdH\}}|r+tWdSt|ddkrHt dt d td WdSt d d d t j d }|d|d|d|d|dD] }|j|dd|d|d|dr~dnd|drdnddqitj|d|dWddS1swYdS)Nr0[dim]Fetching API keys for organization: [bold]''[/bold][/dim]rrpublicrI[bold]No API keys found.[/bold] [dim]Create a new API key with the [bold]0 organizations keys create[/bold] command.[/dim]r&Tr9 show_headerr<r:rrAKeyz Created AtStatusmetadatarkey createdAtrevokedRevokedActiveredr?zAPI keys for organization: rB)r r(rr)r api_keysr*r+rDr.rrrrErFrGr/)rrdatar.rHrSrrr"rsJ        "createz&Create an API key for an organization.z--namez-nz#Human readable name for new API keyz"Organization to create API key forFz --defaultz-dzESet the newly created key as the active / default key in local config api_key_namedefaultcsn|ptd}|stdIdH}|r|dkr$tdtdSd}tj d|ddd !t ||IdH\}}|rKtdWdSWdn1sUwY|r`d |vrjtd tdS|}|sztj d d dIdH}|rt ||d |ddntdtdddtjd}|d|d|d|||d |tj|dddS)Nrz8Enter human readable name for API key e.g. 'Pipecat Key'z%You must enter a name for the API keyr&z([dim]Creating API key with name: [bold]'rJrrrSz5Invalid response from server. Please contact support.zLWould you like to make this key the default key in your local configuration?F)r]default_public_keydefault_public_key_name active_orgadditional_dataz#[dim]Bypassing using key as defaultTr9rNrArPr=z Using as default in local configr7)r r(r,textr-rr.r*r+r)r api_key_createconfirmr printrrrErFrGr/)r\rr]rrZr. make_activerHrrr" create_keysn           rjdeletez&Delete an API key for an organization.zOrganization to get tokens forcs|ptd}tjd|ddd>t|IdH\}}|r+tWdSt|ddkrJt dt d td  WddSWdn1sTwYt j d d d |dDd IdH}|sstd td|d k}|rt d IdHz t|ddddWntyt dtd YSwtjd|dddd#t|d|IdH\}}|rtd WdSWdn1swYtd|dddS)NrrIrJrrrKrrLrMr&zSelect API key to deletecSs*g|]}|dd|d|dfdqS)rRridrSrrrrrSrrr"r#0szdelete_key..r$r`z`This key is currently set as the default in your local config. Are you sure you want to proceed?)rar`rbz3Unable to remove default key from local user configz[dim]Deleting API key with ID z ...[/dim]zAPI key with ID: [bold]'z'[/bold] deleted successfully.)r r(rr)r rYr*r+rDr.rr,rr-rgr r0api_key_deleter/)rrrZr.rSkey_is_defaultrrr" delete_keysl      rqusez8Set default API key for an organization in local config.c sN|ptd}tjd|dddt|IdH\}}|r+tWdSWdn1s5wYt|ddkrRt dt d td dSt j d d d |dDd IdH}|sntd dSzt||d|d ddtd|d dWdSty}zt|tdWYd}~dSd}~ww)NrrIrJrrrKrrLrMr&zSelect API key to usecSs.g|]}|dd|d|ddfdqS)rRrrSrmrrnrrr"r#us zuse_key..r$r_rbzAPI key with name: [bold]'z'[/bold] set as default.zBUnable to set default key in local config. Please contact support.)r r(rr)r rYr*r+rDrhrr,rr-r r/r0rdebugr.)rrrZr.rSerrr"use_keyTsR     ruz*List current organization property values.z#Organization to list properties forcs|ptd}tjd|dddt|IdH\}}|r+tWdSWdn1s5wY|sCtddSt ddd t j d }|j d d d |j ddd | D] \}}||t|q^tj|d|ddS)Nrz2[dim]Fetching properties for organization: [bold]'rJrrz$[dim]No properties configured.[/dim]TFr9rNPropertycyanr?Valuer>zProperties for organization: rB)r r(rr)r rr*r+rhrrrErFitemsrGstrr/)rrrZr.rH prop_name prop_valuerrr"properties_lists2   r}schemaz(Show available properties with metadata.z*Organization to show properties schema forc s|ptd}tjd|dddt|IdH\}}|r+tWdSWdn1s5wY|sCtddSt dddt j d }|j d d d | d |j ddd | d| d| D]J\}}|dd}|dd}|d} |dd} | r| dddd| Dd7} |||dd|durt|nd|durt|nd| qmtj|d|ddS) Nrz9[dim]Fetching properties schema for organization: [bold]'rJrrz#[dim]No properties available.[/dim]Tr9rNrvrwr?Typez Current ValuegreenDefault Description currentValuer^r]availableValues descriptionz [dim]Available: , css|]}t|VqdS)N)rz)rvrrr" sz$properties_schema..r'typez[dim]not set[/dim]z$Properties schema for organization: rB)r r(rr)r properties_schemar*r+rhrrrErFryjoinrGrzr/) rrrZr.rHr{ prop_infocurrentr] availablerrrr"rsR            rsetz Update an organization property..zName of the property to setz Value to setz#Organization to update property for property_namercs|ptd}tjd|d|ddd"t|||iIdH\}}|r1tWdSWdn1s;wY|sLtdtdS|||}t d |d |d dS) Nrz[dim]Updating property [bold]'"'[/bold] for organization: [bold]'rJrrzFailed to update property.r&zProperty [bold cyan]z [/bold cyan] set to [bold green] [/bold green]) r r(rr)r properties_updater*r+r.r/)rrrrrZr. new_valuerrr"properties_sets&    rzdefault-regionz2Get or set the default region for an organization.z/Region to set as default (omit to show current)zOrganization to configureregioncs|ptd}|r]tjd|d|ddd"t|d|iIdH\}}|r3tWdSWdn1s=wY|sNtdtd St d |d|d dStjd |dddt |IdH\}}|rtWdSWdn1swY|rd|vrt d dS|d}|d|dd}|dg}t d|d |rt dd |ddSdS)Nrz&[dim]Setting default region to [bold]'rrJrr defaultRegionz Failed to update default region.r&z"Default region set to [bold green]rz6[dim]Fetching default region for organization: [bold]'z([dim]No default region configured.[/dim]rr]znot setrzDefault region: [bold green]z[dim]Available regions: rr') r r(rr)r rr*r+r.r/rrhr)rrrrZr.proprrrrr"default_regionsN     r)+r,r*logururrichr rich.tablerpipecatcloud._utils.async_utilsrpipecatcloud._utils.auth_utilsr!pipecatcloud._utils.console_utilsrpipecatcloud.clirpipecatcloud.cli.apir pipecatcloud.cli.configr r r Typerorganization_clikeys_cliproperties_cli add_typercommandcreate_blockingOptionrzrr5rboolrjrqrur}rArgumentrrrrrr"s.           " 8 $ 1  R B 6 % 8  "