}\SrSSKrSSKrSSKrSSKJr SSKJr SrSr Sr Sr S r S r g) zEHelper functions to interact with git and github for iac remediation.N)errors)filescUSc[R"S5eUSc[R"S5eUSc[R"S5eg)zValidates the git config file.remoteNzmain-branch-namez branch-prefix)rInvalidGitConfigError)git_config_files 5lib/googlecloudsdk/api_lib/scc/iac_remediation/git.pyvalidate_git_configr s]X&  & &x 00'(0  & &'9 ::_%-  & & 77.c Sn[R"USS[R"5[R[RSS9nSUR R 54$![Ra gf=f)zCheck whether the current directory is a git repo or not. Returns: True, repo_root_path if the current directory is a git repo False, None otherwise. zgit rev-parse --show-toplevelT)shellcheckcwdstdoutstderrtext)FN) subprocessrunosgetcwdPIPErstripCalledProcessError) git_check_cmdresults r is_git_repor%sl 4M ^^$BIIKz  F $$& &&  & & sA&A))B?Bc [R"SSSX/S[R"5[R[RSS9n[ UR R55$)zHelper function to check if a branch exists in the remote. Args: remote_name: Name of the remote of the repo at which to check. branch_name: Name of the branch to check. Returns: Boolean indicating whether the branch exists in the remote. gitz ls-remotez--headsFT)rrrrr)rrrrrboolrr) remote_name branch_namers r branch_remote_existsr"9sS >> k9k? ryy{ __Z__   & fmm!!# $$r cSn[R"/SQ5RS5nUR5HnX;dM UR 5Sn O Uc[ R "5n[R"SSSUSU/S [R"5[R[RS 9 [X5(d8[R"SS S X/S U[R[RS 9 [R"SS/S U[R[RS 9 U$)a Returns the working tree directory for the branch. Args: remote_name: Name of the remote of the repo at which to check. branch_name: Name of the branch for which the working tree directory is required. Returns: Working tree directory path for the branch. N)rworktreelistzutf-8rrr$addz-BTrrrrpushz--set-upstreamFpull) r check_outputdecode splitlinessplittempfilemkdtemprrrrr")r r! worktree_direxisting_worktreeslines r get_working_tree_dirr3Ls,!..! F7O!++-dZZ\!_l . ##%LNN  E<{C z 9 9nn &*K E< NN z r c X[5upEA[X#5nUR5Hupx[RR Xu5n [RR Xi5n [R"X5 [R"SSU /SU[R[RS9 M [R"SSSU/SU[R[RS9 [R"SS/SU[R[RS9 g ) aPushes the commit to the given branch with the given files data and commit message. Args: files_data: Dictionary of file path (absolute path of the files in original repo) and file data to be written commit_message: Message to be added to the commit remote_name: Name of the remote of the repo at which to check. branch_name: Name of the branch where commit needs to be pushed rr&Tr'commitz-mFr(N) rr3itemsrpathrelpathjoinrWriteFileContentsrrr) files_datacommit_messager r! is_repo_flag repo_root_dirr0 file_path file_datarel_path abs_file_paths r push_commitrCxs!, ,%k?,(..0iwwy8HGGLL8M M5NN }% z 1 .. hn- | __Z__  .. fo | __Z__r c [X#5nSUSUSUSUS3 n[R"USSU[R[RS9 [R"S S S S U/SU[R[RS 9 g)a6Creates a PR for the given branch to the main base branch. Args: pr_title: PR title pr_desc: PR description remote_name: Name of the remote of the repo at which to check. branch_name: The branch from which PR needs to be created. base_branch: The main branch name to be which PR needs to be merged. zgh pr create --base z --head z --title "z " --body ""TF)r rrrrrr$removez--forcer'N)r3rrr)pr_titlepr_descr r! base_branchr0 pr_commands r raise_prrKs&k?,[M+? *Jwiq*  ..E| __Z__ .. j(I|< | __Z__r )__doc__rrr..googlecloudsdk.command_lib.scc.iac_remediationrgooglecloudsdk.core.utilrr rr"r3rCrKr r rPs:L A* 8(%&)X"Jr