}\dZddlZddlZddlZddlmZddlmZdZdZ dZ dZ d Z d Z y) zEHelper functions to interact with git and github for iac remediation.N)errors)filesc|dtjd|dtjd|dtjdy)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  d}tj|ddtjtjtjd}d|j j fS#tj$rYywxYw)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?Bctjddd||gdtjtjtjd}t |j jS)aHelper 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"9sU >> k9k;? ryy{ __Z__   & fmm!!# $$r cxd}tjgdjd}|jD]}||vs|j d}n|t j }tjddd|d|gd tjtjtj t||s;tjdd d ||gd |tjtj tjddgd |tjtj |S)aReturns 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!++-ddZZ\!_l . ##%LNN  E<{C z [ 9nn &*K E< NN z r cjt\}}~t||}|jD]\}}tjj ||} tjj || } tj| |tjdd| gd|tjtjtjddd|gd|tjtjtjddgd|tjtjy ) 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;?,(..0iwwy-8HGGLLx8M M95NN }% z 1 .. hn- | __Z__  .. fo | __Z__r c t||}d|d|d|d|d }tj|dd|tjtjtjd d d d |gd|tjtj y)aDCreates 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