o i::@sddlZddlmZddlmZddlZddlmZddlm Z ddl m Z ddl m Z mZddlmZejd d d d ZGd ddeeZddZdJdededeedefddZdedefddZdJdefddZ dKdeededededef dd Z dJd!ed"ed#ed$eedef d%d&Zejd'd(d)e jeej dd d*ej!dd+d,d-ej dd.d/d0d1d2ej dd3d4d5d1d2ej dd6d7d1d2ej dd8d9d:d;d2ej d,dd?d;d2fd#ed@edAed$edBedCedDefdEdFZ"dGejfdHdIZ#dS)LN)Enum)Optional)Panel) synchronizer)console)DeployConfigParamswith_deploy_config)PIPECAT_CLI_NAMEdockerzDocker build and push utilitiesT)namehelpno_args_is_helpc@seZdZdZdZdS) RegistryType dockerhubcustomN)__name__ __module__ __qualname__ DOCKERHUBCUSTOMrrT/home/ubuntu/.local/lib/python3.10/site-packages/pipecatcloud/cli/commands/docker.pyrsrcCstdtdtdtdtdtjdddrHtd gd }t|d d d r7tddStdtdtddStdtddS)zMSuggest and optionally install binfmt_misc support for cross-platform builds.u> [yellow]⚠️ Cross-platform build support missing[/yellow]zZ[yellow] Building ARM64 images on x86_64 requires QEMU and binfmt_misc support.[/yellow]zS [yellow] This is a one-time setup that persists across Docker restarts.[/yellow]zU [dim] The following command will be run (requires Docker --privileged flag):[/dim]zJ[dim] docker run --privileged --rm tonistiigi/binfmt --install all[/dim]z. Would you like to install binfmt support now?Tdefaultz* [cyan]Installing binfmt support...[/cyan])r runz --privilegedz--rmztonistiigi/binfmtz --installallz'Setting up cross-platform build supportF stream_outputu;✓ binfmt support installed! You can now retry your build.z!Failed to install binfmt support.z9 [yellow]You can install it manually by running:[/yellow]zP[yellow] docker run --privileged --rm tonistiigi/binfmt --install all[/yellow]zB [yellow]Skipping installation. To install manually, run:[/yellow]N)rprinttyperconfirmrun_docker_commandsuccesserror) install_cmdrrr_suggest_and_install_binfmts6       r%stderrstdoutcommand registry_infocCsb|}|}|d|}d|vsd|vrtdSt|r-dd|vr/t|dSdSdS)zJProvide helpful error hints only when Docker's message isn't clear enough. zexec format errorzexec /bin/sh: exec format errorNpush)lowerr%_is_auth_errorjoin_suggest_docker_login)r&r'r(r) stderr_lower stdout_lowercombined_outputrrr_provide_error_hintsGs r3r0returncsgd}tfdd|DS)z-Check if the error is authentication related.) unauthorizedz access denieddeniedc3s|]}|vVqdSNr).0keywordr0rr ]sz!_is_auth_error..)any)r0 auth_keywordsrr:rr-Zsr-cCsNtd|r |ddkr |dr td|dddStddS) z-Suggest the appropriate docker login command.uA [yellow]💡 You need to authenticate with the registry[/yellow]typerurlz[yellow] docker login z [/yellow]z [yellow] docker login[/yellow]N)rrget)r)rrrr/`s r/ descriptionrc Cszotd|d|rSg}tj|tjtjdddd}|jD]}||td|dq| |j dkrPt |j |}d ||_ |j |_d|_|WdStj|dddd }|jrmtd|jdWdStj y} zEtd d |t| d dpt| d dpd} t| ddpd} | rtd| d| rtd| dt| | ||WYd} ~ dSd} ~ wtytdYdSw)z.Run a docker command and handle output/errors.z[dim]z ...[/dim]T)r'r&textbufsizeuniversal_newlinesz[/dim]r)capture_outputrCcheckzDocker command failed: r*r'outputr&z [red]stdout: z[/red]z [red]stderr: NFzEDocker not found. Please ensure Docker is installed and in your PATH.)rr subprocessPopenPIPESTDOUTr'appendrstripwait returncodeCalledProcessErrorr.rIr&rstripr#getattrr3FileNotFoundError) r(rArr) output_linesprocessliner#resulter'r&rrrr!jsT      r! registry_typeusername agent_name registry_urlcCsP|tjkr |d|S|tjkr!|std|d|d|Std|)z1Build the full image name based on registry type./z.registry_url is required for custom registrieszUnsupported registry type: )rrr ValueError)r[r\r]r^rrr_build_image_names  raz build-pushz!Build, tag, and push Docker image)r r )hiddenz4Name of the agent to build image for e.g. 'my-agent'F)r show_defaultz --registryz-rzRegistry type to push tozRegistry Configuration)r rich_help_panelz --usernamez-uzRegistry usernamez--registry-urlz4Custom registry URL (required for custom registries)z --versionz-vzVersion tag for the imagezBuild Configurationz --no-pushz+Build and tag only, do not push to registryz --no-latestzDo not tag as 'latest'registryregistry_usernameversionno_push no_latestc! s |pt}t|di} d} d} d} d} d}|jrY|jd}t|dkr0d} |d} |d}nt|dkrEd } |d}|d} |d}n|j}d |vrU|d d\} } n|} d } |p_| p_|j}|ph| ph| d }|pq| pq| d }|pz|pz| d}|p| pd }|p| dd }|stdt dS|s|stdt dS|stdt dS|t j kr|stdt dS|r|}n*z t t ||||}Wnty}ztt|t dWYd}~Sd}~ww|d |}|d}d|g}|s |d|gdddddg|d}|g}|s ||ttd|dd|d |r2d!nd"|d#d$d%d&d't jd(dd)sPtt Std*t|d+|dd,set dStd-d||std.|d/||d kr|ndd0}tdd1|gd2|d3|d4st dS|stdd1|gd2|d3|d4st dS|g}|s||td5d|ntd6|std7|D] } td8| d9qtd:td;|d<td=td8td>dSdS)?z>Build, tag, and optionally push a Docker image for your agent. docker_configNr_rrrBr:latestrerfr^ auto_latestTz?Agent name is required. Provide via argument or pcc-deploy.tomlzVRegistry type is required when pushing. Use --registry or configure in pcc-deploy.tomlzZRegistry username is required when pushing. Use --username or configure in pcc-deploy.tomlzbRegistry URL is required for custom registries. Use --registry-url or configure in pcc-deploy.tomlz:latestz-tr buildxbuildz--platform=linux/arm64z--load.z'[bold white]Agent:[/bold white] [green]zm[/green] [bold white]Platform:[/bold white] [green]linux/arm64[/green] [bold white]Tags:[/bold white] [green]z, z([/green] [bold white]Push:[/bold white] z [red]No[/red]z [green]Yes (z )[/green]zDocker Build Configurationleftyellow)title title_align border_stylez&Do you want to proceed with the build?rz0 [bold cyan]Building Docker image...[/bold cyan]z Building rzSuccessfully built image(s): z [bold cyan]Pushing to z...[/bold cyan])r>r?r+zPushing F)rr)zSuccessfully pushed: z5 [yellow]Skipping push (--no-push specified)[/yellow]z+ [dim]Your image is now available at:[/dim]z [bold]z[/bold]z> [dim]To deploy this image, update your pcc-deploy.toml:[/dim]z [bold]image = "z"[/bold]z [dim]Then run:[/dim]z deploy[/bold])rrTimagesplitlenr]r@rr#rExitrrrar`strextendrNrrr.r cancelr!r"r )! deploy_configr]rerfr^rgrhripartial_configrjparsed_registryparsed_usernameparsed_agent_nameparsed_versionparsed_registry_url image_parts repo_and_tagfinal_agent_namefinal_registryfinal_usernamefinal_registry_url final_versionfinal_no_latestbase_image_namerZ version_tag latest_tag build_tags build_command tags_displayr) pushed_tagstagrrr build_pushs 3                          rappcCs|jtdddS)z'Add docker command to the main CLI app.Commands)rdN) add_typer docker_cli)rrrrcreate_docker_commandsrr7)TN)$rJenumrtypingrr rich.panelrpipecatcloud._utils.async_utilsr!pipecatcloud._utils.console_utilsr pipecatcloud._utils.deploy_utilsrrpipecatcloud.clir Typerrr|rr%listdictr3boolr-r/r!rar(create_blockingOptionArgumentrrrrrrs       *  ;    &b