B
    `dU7                 @   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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InstallationError)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   9/tmp/pip-unpacked-wheel-hu9qqap1/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)GIT_DIRGIT_WORK_TREEHEADc             C   s   | gS )Nr   )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   )get_url_rev_optionsr   r"   is_commit_id_equalr   get_revision_sha)selfurldest_rev_options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'   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-refz-qr!   )   F)extra_ok_returncodesr/   cwdzrefs/heads/N)r3   stripr4   r5   )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 d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pathr3   )r'   r@   r(   temp_dirr   r   r   rF   q   s    
z
Git.exportc       
   	   C   s   | j d|g|ddd}i }xP|  D ]@}y| \}}W n" tk
r^   td|Y nX |||< q(W 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_returncodezunexpected show-ref line: {!r}zrefs/remotes/origin/{}zrefs/tags/{}NT)r3   r>   
splitlinesr6   
ValueErrorformatget)
r?   r)   r"   rB   refsliner   rC   
branch_ref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/fetchz-q)r=   
FETCH_HEAD)r"   )arg_revAssertionErrorr&   make_newbranch_namer   loggerwarningr4   r3   r	   to_argsget_revision)r?   r)   r(   r+   r"   r   	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   z-qr[   checkout)r=   z	origin/{}z-bz--track)
to_displayr\   infor   r3   r	   r"   ra   getattrr%   r^   rD   rP   update_submodules)r'   r)   r(   r+   rev_displayr[   cmd_argstrack_branchr   r   r   	fetch_new   s     
zGit.fetch_newc             C   sB   | j tdd||d tdd| }| j ||d | | d S )Nconfigzremote.origin.url)r=   rc   z-q)r3   r	   r^   rg   )r'   r)   r(   r+   ri   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   z-qz--tags)r=   resetz--hard)r:   r8   r3   ra   r	   r^   rg   )r'   r)   r(   r+   ri   r   r   r   update  s    z
Git.updatec             C   sz   | j dddgdd|d}| }y|d }W n tk
rD   tY nX x|D ]}|drL|}P qLW |d	d
 }| S )z
        Return URL of the first remote encountered.

        Raises RemoteNotFoundError if the repository does not have a remote
        url configured.
        rl   z--get-regexpzremote\..*\.url)r;   F)r<   r/   r=   r   zremote.origin.url  r;   )r3   rN   
IndexErrorr   r4   r6   r>   )r?   r@   stdoutremotesfound_remote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!   z	rev-parseF)r/   r=   )r3   r>   )r?   r@   r"   current_revr   r   r   r_   0  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.
        z	rev-parsez	--git-dirF)r/   r=   z..)r3   r>   osrJ   isabsr7   abspathr   )r?   r@   git_dir	repo_rootr   r   r   get_subdirectory9  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.
        fileNrE   \+r;   z://zfile:zgit+z
git+ssh://zssh://r0   )urlsplitrH   r5   lstripurllib_requesturl2pathnamereplace
urlunsplitfindrY   superr   get_url_rev_and_auth)r?   r(   schemenetlocrJ   queryfragmentinitial_slashesnewpath
after_plusr"   	user_pass)	__class__r   r   r   H  s"    

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.gitmodules	submodulero   z--initz--recursivez-q)r=   )rx   rJ   existsr7   r3   )r?   r@   r   r   r   rg   k  s
    zGit.update_submodulesc                sz   t t| |}|r|S y| jddg|dddd}W n2 tk
rT   td| d S  tk
rf   d S X tj	
|dS )Nz	rev-parsez--show-toplevelFraise)r=   r/   rM   log_failed_cmdzKcould not determine if %s is under git control because git is not availablez
)r   r   get_repository_rootr3   r   r\   debugr   rx   rJ   normpathrstrip)r?   r@   locr)r   r   r   r   t  s"    zGit.get_repository_root)N)__name__
__module____qualname__rb   dirname	repo_nameschemesunset_environdefault_arg_revstaticmethodr#   r-   r:   classmethodrD   rF   r&   ra   r%   rk   rm   ro   rv   r_   r}   r   rg   r   __classcell__r   r   )r   r   r   .   s0   #-#	r   ).
__future__r   loggingos.pathrx   repip._vendor.packaging.versionr   r8   Zpip._vendor.six.moves.urlliburllib_parser   r   pip._internal.exceptionsr   r   pip._internal.utils.miscr   r   pip._internal.utils.subprocessr	   pip._internal.utils.temp_dirr
   pip._internal.utils.typingr    pip._internal.vcs.versioncontrolr   r   r   r   typingr   r   r   r   r   r   r   	getLoggerr   r\   compiler   r   r   registerr   r   r   r   <module>   s2   

  ^