o iQ@shddlZddlZddlZddlmZddlmZddlZddlZddl m Z ddl m Z ddl mZddlmZddlmZdd lmZdd lmZdd lmZdd lmZmZdd lmZddlmZddl m!Z!ddl"m#Z#ej$ddddZ%de&fddZ'de(fddZ)e%j*dddej+eej,ddej,dddej-dd d!d"dej-d#d$d%d&dej-dd'd(d)dej-dd*d+d,dfde(de.e(d-e(d.ed/e(d0ee#f d1dZ/e%j*d2d3dej+eej,dd4dej,dd5dej-d#d$d%d&dej-dd'd(d)dfde(d6e(d.ed/e(fd7d2Z0e%j*d8d9dej+eej,dd:dej-dd;d%ded/e(d0ee#fd?d8Z.e%j*d@dAdej+eej,dBdej-d#d$d%d&de-dd'd(fde(d.ed/e(fdCd@Z1e%j*dDdEdej+eej,dFdej,dGdej,ddHdej-ddIdJdKde-dd'd(ej-dd*d+dLdfde(dMe(dNe(dOe2d/e(d0ee#f dPdQZ3dS)RN)Optional)boolean)logger)box)Live)Panel)Table) synchronizer)requires_login)console)get_region_codesvalidate_region)PIPECAT_CLI_NAME)API)config)Regionsecretsz'Secret and image pull secret managementT)namehelpno_args_is_helpcCstd}|D]7\}}|r|stdtdSt|dkr/tdtdS||s@tdtdSq dS)Nz^[a-zA-Z0-9_-]+$zl[red]Error: Secrets must be provided as key-value pairs. Please reference --help for more information.[/red]@zG[red]Error: Secret names must not exceed 64 characters in length.[/red]zc[red]Error: Secret names must contain only alphanumeric characters, underscores, and hyphens.[/red]) recompileitemsr printtyperExitlenmatch)rvalid_name_patternkeyvaluer#U/home/ubuntu/.local/lib/python3.10/site-packages/pipecatcloud/cli/commands/secrets.pyvalidate_secrets&s"    r%rcCstd}t||S)Nz4^[a-zA-Z0-9][a-zA-Z0-9_-]*[a-zA-Z0-9]$|^[a-zA-Z0-9]$)rrboolr)rr r#r#r$validate_secret_name;s r'setz/Create a new secret set for active organization)rrz5Name of the secret set to create e.g. 'my-secret-set')rzJList of secret key-value pairs e.g. 'KEY1=value1 KEY2="value with spaces"'z--filez-fz&Load secrets from a relative file pathFz--skipz-sz-Skip confirmations / force creation or updatez--organizationz-oz$Organization to create secret set inz--regionz-rzRegion for secret set from_file skip_confirm organizationregionc s8t|stdtdS|s|stdtdS|r+|r+tdtdSi}|p3td}|rtj |sKtd|dtdSzt |dt}|D]i} | } | ra| d rbqTd | vr~td |d tdWdWS| d d\} } | } | } | d r| d r| dd} | r| std|tdWdWS| || <qTWdn1swY|std|tdWSWntty} ztd|dt| tdWYd} ~ Sd} ~ ww|D]L} d | vrtdtdS| d d\} } | } | } | d r;| d r;| dd} | rA| sMtdtdS| || <qt|t||rt|IdHstIdH}td|dd|dtdS|s tdtjdddd}|jddd|jd dd|D]\} } t| d!kr| dd!d"n| }|| |qtd#||rtd$|d%nt |IdH\}}|rtStd$|d&d'tt!|d(d)d*t"#d+$IdH}|s tdSd}tj%d,|d-d.d/,tj&||d0IdH\}}|r2tWdS|rnd?}|rd@|dAndB}dC|dD|dE|}|d?kr |dF7}n |dGt*dH|dI7}t+|dS)JNzM[red]Secret set name must only contain characters, numbers and hyphens.[/red]rzl[red]Command requires either passed key-values or relative file path. See --help for more information.[/red]z3[red]Cannot pass key-value pairs with --file optionorgz[red]Error: File 'z' does not exist.[/red]r#=z#[red]Error: Invalid line format in z?. Each line must be a key-value pair using '=' separator.[/red]"z#Error: Empty key or value found in z!Error: No valid secrets found in zError reading file 'z': zZError: Secrets must be provided as key-value pairs using '=' separator. Example: KEY=valuezF[red]Error: Both key and value must be provided for each secret.[/red][red]Invalid region ''. Valid regions are: , [/red]dimTF) border_styler show_header show_edge show_linesKeywhitestylez Value Previewz...z& [bold white]Secret Set:[/bold white] z![bold white]Region:[/bold white]  defaultRegionz# [dim](organization default)[/dim] z'[bold]Secrets to create / modify[/bold]lefttitle title_alignz-Would you like to proceed with these secrets?z7[dim]Checking for existing secret set with name [bold]''[/bold][/dim]dotsspinner)r- secret_setcSsh|]}|dqS) fieldNamer#).0secretr#r#r$ szset..z)The following secret(s) already exist in z will be overwritten: z. Would you like to continue?z2[bold red]Secret set creation cancelled[/bold red][dim] ModifyingCreatingz secret set [bold]')risImagePullSecret secretKey secretValuedataset_namer-r,r,createdmodified in [bold cyan] [/bold cyan]Secret set [bold green]'z'[/bold green] z successfullyzh [bold white]You must re-deploy any agents using this secret set for changes to take effect[/bold white]z [dim]Deploy your agent with z deploy agent-name --secrets [/dim]),r'r rrrrgetospathexistsopenstrip startswithsplitendswitherror Exceptionstrrdebugr%r r joinrrSIMPLE add_columnrradd_rowr propertiesr questionaryconfirm ask_asyncstatus secrets_list intersectionkeyssecrets_upsertrsuccess)rrr)r*r+r, secrets_dictr-fliner!r"erN valid_regionstablepreviewpropsri looks_good existing_setrWexisting_secret_namesoverlapping_secretscreate used_regionaction region_infomessager#r#r$r(CsL"                      "          unsetz+Delete a secret within specified secret setzCName of the secret set to delete a secret from e.g. 'my-secret-set'z-Name of the secret to delete e.g. 'my-secret' secret_keycs"|ptd}|r |stdtdS|s5td|d|dIdH}|s5tdtdStj d|d |d d d 8t j |||d IdH\}}|sgtd|dtWdS|rttWdSWdn1s~wYt d|d|ddS)Nr-zfError: Secret set name and secret name must be provided. Please reference --help for more information.rz0Are you sure you want to unset secret with key 'z ' from set ''?zSecret key unset cancelledz[dim]Deleting secret [bold]'z '[/bold] from secret set [bold]'rGrHrI)rX secret_namer-zKey not found in set ''zSecret [bold green]'zA'[/bold green] deleted successfully from secret set [bold green]'z'[/bold green]) rr`r rirrrrrsrtrursecrets_deleterz)rrr*r+r-rsrWrir#r#r$r%sB      listzList secret sets and set keysz@Name of the secret set to list secrets from e.g. 'my-secret-set'z--setsz?Filter results to show secret sets only (no image pull secrets)zFilter by regionshow_allcs|ptd}d}|r-t|IdHs-tIdH}td|dd|dtdSt d|tj d |d d d  t j|||d IdH\}}|rl|dkr]tdnt tWdS|rrt|s|rtd|d|dn td|dtWdS|rtddddd} | j|dd|D] } | | dqtt| d|dddnfdd |D} | rt| std!|dtWdStdtjdddd"} | jd#dd| jd$ddr#| jd%dd| D]} | d&d'krd(nd)} | | d*| d+| qn| D] } | | d*| d+q%tj| d,|d-WddSWddS1sQwYdS).Nr-zRetrieving secret setsr3r4r5r6rzSecret set name to lookup: rPr_rHrI)r-rKr,iz#Unable to lookup image pull secretsz!No secrets sets with name [bold]'z'[/bold] found in [bold]''[/bold]z4No secrets sets for namespace / organization [bold]'r7FT)r8r9r:r;r=r>rLz![bold]Secret keys for set [green]z[/green][/bold]rCrDcs g|] }s |ddkr|qS)typeimagePullSecretr#rMsrr#r$ s zlist..z2No secret sets in namespace / organization [bold]')r9rr8r:r;zSecret Set NamerTyperrzImage Pull Secretz Secret Setrr,zSecret sets for ) title_extra)rr`r r r rrmrrrrlrur bubble_errorrvri print_errorrrrorprrrnrz)rrr+r,r- status_titlerrWrirr filtered_setsrKset_typer#rr$r`s      )$deletez,Delete a secret set from active organizationz5Name of the secret set to delete e.g. 'my-secret-set'cs|ptd}|s$td|dIdH}|s$tdtdStj d|ddd 8t j ||d IdH\}}|sRt d |tdWdS|r`tdWdSWdn1sjwYt d |d dS)Nr-z,Are you sure you want to delete secret set 'rz.[bold red]Secret deletion cancelled[/bold red]rzDeleting secret set [bold]'rrHrI)rXr-zNo set found with name 'r^z#'[/bold green] deleted successfully)rr`rrrsrtr rrrrursecrets_delete_setrirz)rr*r+r-rsrWrir#r#r$rs.    zimage-pull-secretz4Create an image pull secret for active organization.zTName of the image pull secret to reference in deployment e.g. 'my-image-pull-secret'zEHost address of the image repository e.g. https://index.docker.io/v1/z<s*z$image_pull_secret..zImage pull secret '[bold]zY'[/bold] already exists. Please choose a different name or delete the existing one first.z'[dim]Creating image pull secret [bold]'rGT)rSrUrrVr,r[r\r]zImage pull secret [bold green]'z'[/bold green] for [bold green]z"[/bold green] created successfully.)rr`r rirrrrrtextrtpasswordbase64 b64encodeencodedecoder r rmrrurrvnextstopupdateryrz)rrrrr+r,r-usernamerrrliverWriexisting_secretrr#rr$image_pull_secrets           '*r)4rrartypingr xmlrpc.clientrrrrlogururrichr rich.liver rich.panelr rich.tablerpipecatcloud._utils.async_utilsr pipecatcloud._utils.auth_utilsr !pipecatcloud._utils.console_utilsr pipecatcloud._utils.regionsr r pipecatcloud.clirpipecatcloud.cli.apirpipecatcloud.cli.configrpipecatcloud.constantsrTyper secrets_clidictr%rkr'commandcreate_blockingArgumentOptionrr(rrr&rr#r#r#r$s                  ` 8    a &