U
    Eh8                     @   s  d dl m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 d dlmZ d dlmZmZ d dlmZ d dlmZ d d	lmZ d d
lmZmZmZmZ erd dlmZmZ d dlm Z  d dlm!Z!m"Z" e
j#Z#e
j$Z$e%e&Z'e(dZ)dd Z*G dd deZ+e,e+ dS )    )absolute_importN)parse)request)
BadCommand)display_pathhide_url)make_command)TempDirectory)MYPY_CHECK_RUNNING)RemoteNotFoundErrorVersionControl!find_path_to_setup_from_repo_rootvcs)OptionalTuple)
HiddenText)AuthInfo
RevOptionsz^[a-fA-F0-9]{40}$c                 C   s   t t| S N)bool
HASH_REGEXmatch)sha r   ;./venv/lib/python3.8/site-packages/pip/_internal/vcs/git.pylooks_like_hash*   s    r   c                       s   e Zd ZdZdZdZdZdZdZe	dd Z
d	d
 Zdd Zedd Zdd Zedd Zedd Zedd Zdd Zdd Zdd Zedd Zed*d d!Zed"d# Ze fd$d%Zed&d' Ze fd(d)Z  ZS )+Gitgitz.gitclone)r   zgit+httpz	git+httpszgit+sshzgit+gitzgit+file)ZGIT_DIRZGIT_WORK_TREEHEADc                 C   s   | gS r   r   revr   r   r   get_base_rev_args:   s    zGit.get_base_rev_argsc                 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)selfurldest_rev_optionsZis_tag_or_branchr   r   r   is_immutable_rev_checkout>   s    zGit.is_immutable_rev_checkoutc                 C   s\   d}| j dgdd}||r8|t|d   d }nd}d|dd d }t|S )	Nzgit version versionF)show_stdoutr    .   )run_command
startswithlensplitjoinparse_version)r%   ZVERSION_PFXr+   r   r   r   get_git_versionO   s    
zGit.get_git_versionc                 C   sB   dddg}| j |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      FZextra_ok_returncodesr,   cwdzrefs/heads/N)r0   stripr1   r2   )clslocationargsoutputrefr   r   r   get_current_branch\   s    

   
zGit.get_current_branchc              	   C   sX   | ds|d }tdd2}| j|j|d | jdddd|gd	|jd
 W 5 Q R X dS )z@Export the Git repository at the url to the destination location/export)kind)r&   zcheckout-indexz-az-fz--prefixFr,   r;   N)endswithr	   unpackpathr0   )r%   r>   r&   Ztemp_dirr   r   r   rD   q   s    
 z
Git.exportc           
   	   C   s   | j d|g|ddd}i }| dD ]V}|d}|s<q(z|jddd	\}}W n" tk
rt   td
|Y nX |||< q(d|}d|}	||}|dk	r|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-refFignore)r;   r,   on_returncode
    )maxsplitzunexpected show-ref line: {!r}zrefs/remotes/origin/{}zrefs/tags/{}NT)r0   r<   r3   rstrip
ValueErrorformatget)
r=   r'   r!   r@   Zrefsliner   rA   Z
branch_refZtag_refr   r   r   r$   ~   s*     





zGit.get_revision_shac                 C   s   |j }|dk	st| ||\}}|dk	rF||}|r<|nd|_|S t|sZtd| |dsh|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.
        Nz:Did not find branch or tag '%s', assuming revision or ref.zrefs/fetchr7   r;   Z
FETCH_HEADr    )Zarg_revAssertionErrorr$   Zmake_newbranch_namer   loggerZwarningr1   r0   r   to_argsget_revision)r=   r'   r&   r)   r!   r   Z	is_branchr   r   r   resolve_revision   s*    



zGit.resolve_revisionc                 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)r\   )r=   r'   namer   r   r   r#      s    	zGit.is_commit_id_equalc                 C   s   |  }td||t| | tdd|| |jr| |||}t|dd }|d kr| 	||jstdd|
 }| j||d n4| ||krd|}dd|d	|g}| j||d | | d S )
NzCloning %s%s to %sr   r7   rY   checkoutrW   z	origin/{}z-bz--track)Z
to_displayrZ   infor   r0   r   r!   r]   getattrr#   r[   rB   rS   update_submodules)r%   r'   r&   r)   Zrev_displayrY   cmd_argsZtrack_branchr   r   r   	fetch_new   s0      
    zGit.fetch_newc                 C   sB   | j tdd||d tdd| }| j ||d | | d S )Nconfigzremote.origin.urlrW   r_   r7   )r0   r   r[   rb   r%   r'   r&   r)   rc   r   r   r   switch  s    
z
Git.switchc                 C   st   |   tdkr&| jdddg|d n| jddg|d | |||}tddd| }| j||d | | d S )Nz1.9.0rV   r7   z--tagsrW   resetz--hard)r6   r5   r0   r]   r   r[   rb   rf   r   r   r   update  s    z
Git.updatec                 C   sx   | j dddgdd|d}| }z|d }W n tk
rD   tY nX |D ]}|drJ|} qbqJ|d	d
 }| S )z
        Return URL of the first remote encountered.

        Raises RemoteNotFoundError if the repository does not have a remote
        url configured.
        re   z--get-regexpzremote\..*\.urlr8   Fr:   r   zremote.origin.url rN   r9   )r0   
splitlines
IndexErrorr   r1   r3   r<   )r=   r>   stdoutZremotesZfound_remoteZremoter&   r   r   r   get_remote_url  s"    
  

zGit.get_remote_urlNc                 C   s(   |d krd}| j d|gd|d}| S )Nr   	rev-parseFrF   )r0   r<   )r=   r>   r!   Zcurrent_revr   r   r   r\   6  s      zGit.get_revisionc                 C   sR   | j ddgd|d }tj|s2tj||}tjtj|d}t||S )z~
        Return the path to setup.py, relative to the repo root.
        Return None if setup.py is in the repo root.
        rn   z	--git-dirFrF   z..)r0   r<   osrI   isabsr4   abspathr   )r=   r>   Zgit_dirZ	repo_rootr   r   r   get_subdirectory?  s     
zGit.get_subdirectoryc                    s   t |\}}}}}|dr|dt|d  }|t|ddd }t|||||f}|dd }	|d|	 t||	d ||||f }d|krd|kst	|d	d
}t
t| |\}}
}|dd}nt
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.
        fileNrC   \+r9   z://zfile:zgit+z
git+ssh://zssh://r-   )urlsplitrG   r2   lstripurllib_requestZurl2pathnamereplace
urlunsplitfindrX   superr   get_url_rev_and_auth)r=   r&   ZschemeZnetlocrI   ZqueryZfragmentinitial_slashesnewpathZ
after_plusr!   Z	user_pass	__class__r   r   r}   N  s0    

 zGit.get_url_rev_and_authc                 C   s6   t jt j|dsd S | jdddddg|d d S )Nz.gitmodulesZ	submoduleri   z--initz--recursiver7   rW   )ro   rI   existsr4   r0   )r=   r>   r   r   r   rb   q  s    zGit.update_submodulesc                    sZ   t t| |rdS z| jdg|dddd}| W S  tk
rT   td| Y dS X d S )NTrn   FrJ   )r;   r,   rK   Zlog_failed_cmdzKcould not determine if %s is under git control because git is not available)r|   r   controls_locationr0   r   rZ   debug)r=   r>   rr   r   r   r   z  s    zGit.controls_location)N)__name__
__module____qualname__r^   dirnameZ	repo_nameZschemesZunset_environZdefault_arg_revstaticmethodr"   r*   r6   classmethodrB   rD   r$   r]   r#   rd   rg   ri   rm   r\   rr   r}   rb   r   __classcell__r   r   r   r   r   .   sD   


(
,


"
r   )-Z
__future__r   ZloggingZos.pathro   reZpip._vendor.packaging.versionr   r5   Zpip._vendor.six.moves.urllibZurllib_parser   rx   Zpip._internal.exceptionsr   Zpip._internal.utils.miscr   r   Zpip._internal.utils.subprocessr   Zpip._internal.utils.temp_dirr	   Zpip._internal.utils.typingr
   Z pip._internal.vcs.versioncontrolr   r   r   r   typingr   r   r   r   r   rv   rz   Z	getLoggerr   rZ   compiler   r   r   registerr   r   r   r   <module>   s2   

  _