o
    ¾ÔiÈJ  ã                   @  s  d dl mZ d dlZd dlZd dlZd dlZd dlZd dl	Zd dl
mZ d dlmZ d dlmZmZ d dlmZmZmZ d dlmZ d dlmZmZmZmZmZmZmZ ejj Z ejj!Z!e "e#¡Z$e %d	¡Z&e %d
¡Z'e %dej(¡Z)ddd„Z*G dd„ deƒZ+e ,e+¡ dS )é    )ÚannotationsN)Úreplace)ÚAny)Ú
BadCommandÚInstallationError)Ú
HiddenTextÚdisplay_pathÚhide_url)Úmake_command)ÚAuthInfoÚRemoteNotFoundErrorÚRemoteNotValidErrorÚ
RevOptionsÚVersionControlÚ(find_path_to_project_root_from_repo_rootÚvcsz(^git version (\d+)\.(\d+)(?:\.(\d+))?.*$z^[a-fA-F0-9]{40}$a/  ^
    # Optional user, e.g. 'git@'
    (\w+@)?
    # Server, e.g. 'github.com'.
    ([^/:]+):
    # The server-side path. e.g. 'user/project.git'. Must start with an
    # alphanumeric character so as not to be confusable with a Windows paths
    # like 'C:/foo/bar' or 'C:\foo\bar'.
    (\w[^:]*)
    $ÚshaÚstrÚreturnÚboolc                 C  s   t t | ¡ƒS ©N)r   Ú
HASH_REGEXÚmatch)r   © r   ú5lib/python3.10/site-packages/pip/_internal/vcs/git.pyÚlooks_like_hash:   s   r   c                      sR  e Zd ZdZdZdZdZdZdZe	dJdd„ƒZ
edK‡ fdd„ƒZdLdd„ZdMdd„ZedNdd„ƒZedOdd „ƒZedPd!d"„ƒZedQd&d'„ƒZedRd)d*„ƒZdSd.d/„Z	0dTdSd1d2„Z	0dTdSd3d4„ZedUd5d6„ƒZe	dVd7d8„ƒZedWd9d:„ƒZedXdYd<d=„ƒZedNd>d?„ƒZedZ‡ fdAdB„ƒZedTd[dCdD„ƒZedN‡ fdEdF„ƒZe	d\dHdI„ƒZ‡  Z S )]ÚGitÚgitz.gitÚclone)zgit+httpz	git+httpszgit+sshzgit+gitzgit+file)ZGIT_DIRZGIT_WORK_TREEÚHEADÚrevr   r   ú	list[str]c                 C  s   | gS r   r   ©r    r   r   r   Úget_base_rev_argsN   s   zGit.get_base_rev_argsÚargsr   Úkwargsc                   sB   t j d¡r| di ¡}d|d< d|d< ||d< tƒ j|i |¤ŽS )NZPIP_NO_INPUTÚextra_environÚ0ZGIT_TERMINAL_PROMPTzssh -oBatchMode=yesZGIT_SSH_COMMAND)ÚosÚenvironÚgetÚsuperÚrun_command)Úclsr$   r%   r&   ©Ú	__class__r   r   r,   R   s   zGit.run_commandÚurlÚdestr   c                 C  sJ   |   t|ƒ¡\}}|jsdS |  ||j¡sdS t|  ||j¡d ƒ}| S )NFr   )Zget_url_rev_optionsr	   r    Úis_commit_id_equalr   Úget_revision_sha)Úselfr0   r1   Ú_Úrev_optionsZis_tag_or_branchr   r   r   Úis_immutable_rev_checkout[   s   zGit.is_immutable_rev_checkoutútuple[int, ...]c                 C  sN   | j dgdddd}t |¡}|st d|¡ dS t| d¡ƒt| d	¡ƒfS )
NÚversionzgit versionFT)Zcommand_descÚshow_stdoutÚstdout_onlyzCan't parse git version: %sr   é   é   )r,   ÚGIT_VERSION_REGEXr   ÚloggerZwarningÚintÚgroup)r4   r9   r   r   r   r   Úget_git_versioni   s   ü
zGit.get_git_versionÚlocationú
str | Nonec                 C  sB   g d¢}| j |ddd|d}| ¡ }| d¡r|tdƒd… S dS )zl
        Return the current branch, or None if HEAD isn't at a branch
        (e.g. detached HEAD).
        )zsymbolic-refú-qr   ©r<   FT©Zextra_ok_returncodesr:   r;   Úcwdzrefs/heads/N)r,   ÚstripÚ
startswithÚlen)r-   rC   r$   ÚoutputÚrefr   r   r   Úget_current_branchv   s   
û
zGit.get_current_branchútuple[str | None, bool]c              	   C  s¼   | j d|g|dddd}i }| ¡  d¡D ](}| d¡}|sqz|jdd	d
\}}W n ty8   td|›ƒ‚w |||< qd|› }d|› }	| |¡}
|
durU|
dfS | |	¡}
|
dfS )zö
        Return (sha_or_none, is_branch), where sha_or_none is a commit hash
        if the revision names a remote branch or tag, otherwise None.

        Args:
          dest: the repository directory.
          rev: the revision name.
        zshow-refFTÚignore)rH   r:   r;   Úon_returncodeÚ
úú r=   )Úmaxsplitzunexpected show-ref line: zrefs/remotes/origin/z
refs/tags/N)r,   rI   ÚsplitÚrstripÚ
ValueErrorr*   )r-   r1   r    rL   ZrefsÚlineZref_shaZref_nameZ
branch_refZtag_refr   r   r   r   r3      s2   û
ý




zGit.get_revision_shac                 C  s.   |  d¡rdS t|ƒsdS |  ||¡rdS dS )a$  
        Return true if rev is a ref or is a commit that we don't have locally.

        Branches and tags are not considered in this method because they are
        assumed to be always available locally (which is a normal outcome of
        ``git clone`` and ``git fetch --tags``).
        zrefs/TF)rJ   r   Ú
has_commit)r-   r1   r    r   r   r   Ú_should_fetch½   s   
	zGit._should_fetchr   r6   r   c                 C  s¨   |j }|dus	J ‚|  ||¡\}}|dur&| |¡}t||r |ndd}|S t|ƒs0t d|¡ |  ||¡s8|S | jt	dd|| 
¡ ƒ|d | j|dd}| |¡}|S )	zµ
        Resolve a revision to a new RevOptions object with the SHA1 of the
        branch, tag, or ref if found.

        Args:
          rev_options: a RevOptions object.
        N)Úbranch_namez:Did not find branch or tag '%s', assuming revision or ref.ÚfetchrE   ©rH   Z
FETCH_HEADr"   )Zarg_revr3   Úmake_newr   r   r?   Úinfor[   r,   r
   Úto_argsÚget_revision)r-   r1   r0   r6   r    r   Z	is_branchr   r   r   Úresolve_revisionÔ   s*   
þþ
zGit.resolve_revisionÚnamec                 C  s   |sdS |   |¡|kS )z§
        Return whether the current commit hash equals the given name.

        Args:
          dest: the repository directory.
          name: a string name.
        F)rb   )r-   r1   rd   r   r   r   r2     s   	zGit.is_commit_id_equalÚ	verbosityr@   ÚNonec                 C  sX  |  ¡ }t d||t|ƒ¡ |dkrd}n	|dkrd}nd}|  ¡ dkr5|  tdd	g|¢|‘|‘R Ž ¡ n|  tdg|¢|‘|‘R Ž ¡ |jr‘|  |||¡}t	|d
d ƒ}t 
d||¡ |d u rv|  ||j¡sutdd| ¡ ƒ}| j||d n%|  |¡|krd|› }	dd|d|	g}| j||d n
|  |¡}
| |
¡}t d||j¡ | j||d d S )NzCloning %s%s to %sr   )z--quietr<   r   )z	--verbosez
--progress)r=   é   r   z--filter=blob:noner\   zRev options %s, branch_name %sÚcheckoutrE   r^   zorigin/z-bz--trackzResolved %s to commit %s©re   )Z
to_displayr?   r`   r   rB   r,   r
   r    rc   ÚgetattrÚdebugr2   ra   rN   rb   r_   Úupdate_submodules)r4   r1   r0   r6   re   Zrev_displayÚflagsr\   Úcmd_argsZtrack_branchr   r   r   r   Ú	fetch_new  s`   þýüûÿ
ý€
û€

zGit.fetch_newr   c                 C  sd   | j tdd|ƒ|d g }|dkr| d¡ tdg|¢| ¡ ‘R Ž }| j ||d | j||d d S )NÚconfigzremote.origin.urlr^   r   rE   rh   ri   )r,   r
   Úappendra   rl   ©r4   r1   r0   r6   re   Úextra_flagsrn   r   r   r   ÚswitchP  s   
þ
z
Git.switchc                 C  s–   g }|dkr|  d¡ |  ¡ dkr| jddg|¢|d n
| jdg|¢|d |  |||¡}tddg|¢| ¡ ‘R Ž }| j||d | j||d	 d S )
Nr   rE   )r<   é	   r]   z--tagsr^   Úresetz--hardri   )rq   rB   r,   rc   r
   ra   rl   rr   r   r   r   Úupdatef  s"   
þýüz
Git.updatec                 C  sx   | j g d¢ddd|d}| ¡ }z|d }W n	 ty   t‚w |D ]}| d¡r-|} nq"| d¡d	 }|  | ¡ ¡S )
z¢
        Return URL of the first remote encountered.

        Raises RemoteNotFoundError if the repository does not have a remote
        url configured.
        )rp   z--get-regexpzremote\..*\.urlrF   FTrG   r   zremote.origin.url rT   r<   )r,   Ú
splitlinesÚ
IndexErrorr   rJ   rV   Ú_git_remote_to_pip_urlrI   )r-   rC   ÚstdoutZremotesZfound_remoteZremoter0   r   r   r   Úget_remote_url„  s(   
ûÿ
þzGit.get_remote_urlc                 C  sJ   t  d| ¡r| S tj | ¡rt | ¡ ¡ S t | ¡}|r!| 	d¡S t
| ƒ‚)a8  
        Convert a remote url from what git uses to what pip accepts.

        There are 3 legal forms **url** may take:

            1. A fully qualified url: ssh://git@example.com/foo/bar.git
            2. A local project.git folder: /path/to/bare/repository.git
            3. SCP shorthand for form 1: git@example.com:foo/bar.git

        Form 1 is output as-is. Form 2 must be converted to URI and form 3 must
        be converted to form 1.

        See the corresponding test test_git_remote_url_to_pip() for examples of
        sample inputs/outputs.
        z\w+://zssh://\1\2/\3)Úrer   r(   ÚpathÚexistsÚpathlibÚPurePathÚas_uriÚ	SCP_REGEXÚexpandr   )r0   Z	scp_matchr   r   r   rz   ¢  s   

zGit._git_remote_to_pip_urlc                 C  s8   z| j dddd| g|dd W dS  ty   Y dS w )zU
        Check if rev is a commit that is available in the local repository.
        ú	rev-parserE   z--verifyzsha^F)rH   Úlog_failed_cmdT)r,   r   )r-   rC   r    r   r   r   rZ   Á  s   ýýÿzGit.has_commitNc                 C  s*   |d u rd}| j d|gdd|d}| ¡ S )Nr   r…   FT©r:   r;   rH   )r,   rI   )r-   rC   r    Zcurrent_revr   r   r   rb   Ñ  s   üzGit.get_revisionc                 C  sT   | j ddgdd|d ¡ }tj |¡stj ||¡}tj tj |d¡¡}t||ƒS )z‘
        Return the path to Python project root, relative to the repo root.
        Return None if the project root is in the repo root.
        r…   z	--git-dirFTr‡   z..)r,   rI   r(   r~   ÚisabsÚjoinÚabspathr   )r-   rC   Zgit_dirZ	repo_rootr   r   r   Úget_subdirectoryÝ  s   üû
zGit.get_subdirectoryú tuple[str, str | None, AuthInfo]c                   sâ   t |ƒ\}}}}}| d¡rC|dt| d¡ƒ … }|tj |¡ dd¡ d¡ }| d¡d }	|d|	… t	||	d… ||||fƒ }d|vrcd|vsMJ ‚| d	d
¡}t
ƒ  |¡\}}
}| dd¡}n	t
ƒ  |¡\}}
}||
|fS )a9  
        Prefixes stub URLs like 'user@hostname:user/repo.git' with 'ssh://'.
        That's required because although they use SSH they sometimes don't
        work with a ssh:// scheme (e.g. GitHub). But we need a scheme for
        parsing. Hence we remove it again afterwards and return it as a stub.
        ÚfileNú/ú\ú+r<   z://zfile:zgit+z
git+ssh://zssh://Ú )ÚurlsplitÚendswithrK   ÚlstripÚurllibZrequestZurl2pathnamer   ÚfindÚ
urlunsplitr+   Úget_url_rev_and_auth)r-   r0   ÚschemeÚnetlocr~   ÚqueryÚfragmentÚinitial_slashesÚnewpathZ
after_plusr    Z	user_passr.   r   r   r˜   ï  s&   

ÿþÿ
zGit.get_url_rev_and_authc                 C  sF   g d¢}|dkr|  d¡ tj tj |d¡¡sd S | j||d d S )N)Z	submodulerw   z--initz--recursiver   rE   z.gitmodulesr^   )rq   r(   r~   r   r‰   r,   )r-   rC   re   Úargvr   r   r   rl     s   

þzGit.update_submodulesc                   sx   t ƒ  |¡}|r
|S z| jddg|ddddd}W n ty)   t d|¡ Y d S  ty2   Y d S w tj 	| 
d¡¡S )	Nr…   z--show-toplevelFTÚraise)rH   r:   r;   rQ   r†   zKcould not determine if %s is under git control because git is not availablez
)r+   Úget_repository_rootr,   r   r?   rk   r   r(   r~   ÚnormpathrW   )r-   rC   ÚlocÚrr.   r   r   r¡     s,   
úýÿzGit.get_repository_rootÚrepo_urlc                 C  s   dS )zEIn either https or ssh form, requirements must be prefixed with git+.Tr   )r¥   r   r   r   Úshould_add_vcs_url_prefix5  s   zGit.should_add_vcs_url_prefix)r    r   r   r!   )r$   r   r%   r   r   r   )r0   r   r1   r   r   r   )r   r8   )rC   r   r   rD   )r1   r   r    r   r   rO   )r1   r   r    r   r   r   )r1   r   r0   r   r6   r   r   r   )r1   r   rd   rD   r   r   )
r1   r   r0   r   r6   r   re   r@   r   rf   )r   )rC   r   r   r   )r0   r   r   r   )rC   r   r    r   r   r   r   )rC   r   r    rD   r   r   )r0   r   r   rŒ   )rC   r   re   r@   r   rf   )r¥   r   r   r   )!Ú__name__Ú
__module__Ú__qualname__rd   ÚdirnameZ	repo_nameZschemesZunset_environZdefault_arg_revÚstaticmethodr#   Úclassmethodr,   r7   rB   rN   r3   r[   rc   r2   ro   rt   rw   r|   rz   rZ   rb   r‹   r˜   rl   r¡   r¦   Ú__classcell__r   r   r.   r   r   >   s\    	

--
Dûûr   )r   r   r   r   )-Z
__future__r   ZloggingZos.pathr(   r€   r}   Úurllib.parser•   Zurllib.requestZdataclassesr   Útypingr   Zpip._internal.exceptionsr   r   Zpip._internal.utils.miscr   r   r	   Zpip._internal.utils.subprocessr
   Z pip._internal.vcs.versioncontrolr   r   r   r   r   r   r   Úparser’   r—   Z	getLoggerr§   r?   Úcompiler>   r   ÚVERBOSErƒ   r   r   Úregisterr   r   r   r   Ú<module>   s>    $

ÿ

õ
    