o d۷i1 @sddlZddlZddlZddlZddlmZddlmZddlZddl m Z ddl m Z ddl mZddlmZmZeeZdZed ZGd d d eZd ed efddZd ed edBfddZd efddZd efddZd efddZdeded dfddZ deded e!eeffddZ"d efd d!Z#d"ed#ed dfd$d%Z$dej%dd&d'd(d)ej%d*d+d,d-d.d/ej%d*d0d1d2d.d/ej%dd3d4d5d)fd6eedBej&d7d)fd"edBd8ed9ed:edBd df d;d<Z'dS)=N)Path) Annotated) APIClient)get_app_config)Identity)get_rich_toolkithandle_http_errorsimz.github/workflows/deploy.ymlc@seZdZdZdS)GitHubSecretErrorz2Raised when setting a GitHub Actions secret fails.N)__name__ __module__ __qualname____doc__rrY/home/ubuntu/vllm_env/lib/python3.10/site-packages/fastapi_cloud_cli/commands/setup_ci.pyr sr originreturncCtd|}|r |dSdS)a2Extract the GitHub host from a git remote URL. Supports both github.com and GitHub Enterprise hosts. Examples: git@github.com:owner/repo.git -> github.com https://github.com/owner/repo.git -> github.com git@enterprise.github.com:owner/repo.git -> enterprise.github.com z(?:git@|https://)([^:/]+)z github.comresearchgrouprmatchrrr_get_github_hosts rcCr)z.Extract 'owner/repo' from a GitHub remote URL.z[:/]([^:/]+/[^/]+?)(?:\.git)?$rNrrrrr_repo_slug_from_origin)s rcCtdduS)z(Check if git is installed and available.gitNshutilwhichrrrr_check_git_installed2r!cCr)z8Check if the GitHub CLI (gh) is installed and available.ghNrrrrr_check_gh_cli_installed7r"r$c Cs\ztjgddddd}|jWStjtfy-tjgddddd}|jYSw)z0Get the remote origin URL of the Git repository.)r#repoview--jsonurl-qz.urlTcapture_outputtextcheck)rconfigz--getzremote.origin.url subprocessrunstdoutstripCalledProcessErrorFileNotFoundErrorresultrrr_get_remote_origin<s" r8namevaluec CsTztjddd|d|gdddWd Stjtfy)}z td|d|d }~ww) zuSet a GitHub Actions secret via the gh CLI. Raises: GitHubSecretError: If setting the secret fails. r#secretsetz--bodyT)r+r-zFailed to set GitHub secret ''N)r0r1r4r5r )r9r:errr_set_github_secretSs r?app_id token_namecCsht'}|jd|d|tdd}||}|d|ddWdS1s-wYdS) z`Create a new deploy token. Returns token_data dict with 'value' and 'expired_at' keys. z/apps/z/tokens)r9expires_in_days)jsonr: expired_at)r:rDN)rpostTOKEN_EXPIRES_DAYSraise_for_statusrC)r@rAclientresponsedatarrr _create_tokencs $rKc Cs>ztjgddddd}|jWStjtfyYdSw)z-Get the default branch of the Git repository.)r#r%r&r'defaultBranchRefr)z.defaultBranchRef.nameTr*mainr/r6rrr_get_default_branchrs  rNbranch workflow_pathcCs*d|d}|jjddd||dS)Nz9name: Deploy to FastAPI Cloud on: push: branches: [a4] jobs: deploy: runs-on: ubuntu-latest steps: - uses: actions/checkout@v5 - uses: astral-sh/setup-uv@v7 - run: uv run fastapi deploy env: FASTAPI_CLOUD_TOKEN: ${{ secrets.FASTAPI_CLOUD_TOKEN }} FASTAPI_CLOUD_APP_ID: ${{ secrets.FASTAPI_CLOUD_APP_ID }} T)parentsexist_ok)parentmkdir write_text)rOrPworkflow_contentrrr_write_workflow_files rWz--branchz-bzDBranch that triggers deploys (defaults to the repo's default branch))helpFz--secrets-onlyz-szBProvisions token and sets secrets, skips writing the workflow fileT)rX show_defaultz --dry-runz-dzAPrints steps that would be taken without actually performing themz--filez-fz8Custom workflow filename (written to .github/workflows/)pathzEPath to the folder containing the app (defaults to current directory) secrets_onlydry_runfilec Cst}t<}|s|jdddtd|pt}t|}|s0|jdddtdt s?|jdddtdzt } Wnt j yY|jdddtdd wd | vrl|jd ddtdt| pq| } t| } t} |s~t}|r|d ||jd dd||d| d|d|d} d}|ptj}d|d|d}d}|r|| |||s|| Wd d Sddlm}m}||jd}d| d|d}|jdd*}t|ddt|j|}|| Wd n 1swYWd n 1s"wY|| rn|jdd0}zt d |d!t d"|jWnt!yW|"d#tdd w||Wd n 1shwYn4d$| d%| d&}|jd'd(d||d)|d*||d+|d!|d,|j||s#|rtd-|}nt}d.}|s|#r|j$d/|d0d1d2d3}|s|j%d4d1d&}|rtd-|}n |d5|d2}||r#d6|d|d}|jd7d}t'||||Wd n 1swY|||||d8|d9d d:d;Wd d S1sGwYd S)[yellow]This is a dry run — no changes will be made[/yellow]zConfiguring CIFastAPIzSetting up CI for [bold]z[/bold] (branch: )zCreated deploy tokenzJSet [bold]FASTAPI_CLOUD_TOKEN[/bold] and [bold]FASTAPI_CLOUD_APP_ID[/bold]zWrote [bold].github/workflows/u,Done — commit and push to start deploying.r)datetimetimezonez%Y-%m-%d %H:%M UTCuGitHub Actions — z (zGenerating deploy token...)titlezError creating deploy token.)default_messagezSetting repo secrets...FASTAPI_CLOUD_TOKENr:FASTAPI_CLOUD_APP_IDz(Failed to set GitHub secrets via gh CLI.zhttps:///z/settings/secrets/actionsz>[yellow]gh CLI not found. Set these secrets manually:[/yellow]infoz Repository: [blue]z[/]z% [bold]FASTAPI_CLOUD_TOKEN[/bold] = z& [bold]FASTAPI_CLOUD_APP_ID[/bold] = z.github/workflows/TzWorkflow file [bold]z"[/bold] already exists. Overwrite?workflowF)r_defaultzUEnter a new filename (without path) or leave blank to skip writing the workflow file:zSkipped writing workflow file.z Wrote [bold]zWriting workflow file...z#Your deploy token expires on [bold]rD zM[/bold]. Regenerate it from the dashboard or re-run this command before then.)(rr is_logged_inprinttyperExitrcwdrr!r8r0r4lowerrrr$rN print_line print_titleDEFAULT_WORKFLOW_PATHr9rdrenowutcstrftimeprogressrrKr@logr?r set_errorexistsconfirminputr3rW)rZrOr[r\r]identitytoolkitapp_path app_configr repo_slug github_hosthas_gh msg_token msg_secrets workflow_file msg_workflowmsg_donerdre timestamprAr{ token_data secrets_urlrPwrite_workflow overwritenew_namerrrsetup_cis,              N             $r)(loggingrrr0pathlibrtypingrrqfastapi_cloud_cli.utils.apirfastapi_cloud_cli.utils.appsrfastapi_cloud_cli.utils.authrfastapi_cloud_cli.utils.clirr getLoggerr loggerrFrw Exceptionr strrrboolr!r$r8r?dictrKrNrWOptionArgumentrrrrrs        !