
0c@_6                 @   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 r$d  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 j& e'  Z( e j) d  Z* d d   Z+ Gd d   d e  Z, e j- e,  d S)    )absolute_importN)parse)request)
BadCommandSubProcessError)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 j |    S)N)bool
HASH_REGEXmatch)sha r   4/tmp/pip-build-jynh7p1z/pip/pip/_internal/vcs/git.pylooks_like_hash*   s    r   c                   sQ  e  Z d  Z d Z d Z d Z d/ Z d0 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   S)1Gitgitz.gitclonegit+http	git+httpsgit+sshgit+git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   se   |  j  t |   \ } } | j s( d S|  j | | j  sA d St |  j | | 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  } | j |  rJ | t |  d   j   d } n d } d j | j d  d  d   } t |  S)Nzgit version versionr    .   )run_command
startswithlensplitjoinparse_version)r+   ZVERSION_PFXr1   r   r   r   get_git_versionO   s    #"zGit.get_git_versionc             C   s]   d d d g } |  j  | d d	 d | } | j   } | j d  rY | t d  d  Sd S)
zl
        Return the current branch, or None if HEAD isn't at a branch
        (e.g. detached HEAD).
        zsymbolic-refz-qr&   extra_ok_returncodes   cwdzrefs/heads/N)r=   )r5   stripr6   r7   )clslocationargsoutputrefr   r   r   get_current_branch\   s    
zGit.get_current_branchc             C   sq   | j  d  s | d } t d d  B } |  j | j d | |  j d d d d | g d	 | j Wd
 QRXd
 S)z@Export the Git repository at the url to the destination location/kindexportr,   zcheckout-indexz-az-fz--prefixr>   N)endswithr
   unpackpathr5   )r+   rA   r,   temp_dirr   r   r   rH   q   s    
z
Git.exportc       
      C   s  d } y |  j  d | g d | } Wn t k
 r9 Yn Xi  } xd | j   j   D]P } y | j   \ } } Wn' t k
 r t d j |    Yn X| | | <qS Wd j |  } d j |  }	 | j |  } | d k	 r | d f S| j |	  } | 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.
        r2   zshow-refr>   zunexpected show-ref line: {!r}zrefs/remotes/origin/{}zrefs/tags/{}NTF)r5   r   r?   
splitlinesr8   
ValueErrorformatget)
r@   r-   r'   rC   refsliner   rD   Z
branch_refZtag_refr   r   r   r*   ~   s&    
zGit.get_revision_shac             C   s   | j  } | d k	 s t  |  j | |  \ } } | d k	 rg | j |  } | rZ | n d | _ | St |  s t j d |  | j d  s | S|  j	 t
 d d | | j    d | |  j | d d } | j |  } | 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-qr>   r'   Z
FETCH_HEAD)Zarg_revAssertionErrorr*   Zmake_newbranch_namer   loggerwarningr6   r5   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|  j  |  | k S)z
        Return whether the current commit hash equals the given name.

        Args:
          dest: the repository directory.
          name: a string name.
        F)rY   )r@   r-   namer   r   r   r)      s    	zGit.is_commit_id_equalc             C   s   | j    } t j d | | t |   |  j t d d | |   | j r|  j | | |  } t | d d   } | d  k r |  j	 | | j  st d d | j
    } |  j | d | nL |  j |  | k rd j |  } d d | d	 | g } |  j | d | |  j |  d  S)
NzCloning %s%s to %sr   z-qrU   checkoutr>   z	origin/{}z-bz--track)Z
to_displayrV   infor   r5   r	   r'   rZ   getattrr)   rX   rE   rO   update_submodules)r+   r-   r,   r/   rev_displayrU   cmd_argsZtrack_branchr   r   r   	fetch_new   s     	zGit.fetch_newc             C   s[   |  j  t d d |  d | t d d | j    } |  j  | d | |  j |  d  S)Nconfigzremote.origin.urlr>   r\   z-q)r5   r	   rX   r_   )r+   r-   r,   r/   ra   r   r   r   switch   s    z
Git.switchc             C   s   |  j    t d  k r7 |  j d d d g d | n |  j d d g d | |  j | | |  } t d d d | j    } |  j | d | |  j |  d  S)Nz1.9.0rS   z-qz--tagsr>   resetz--hard)r;   r:   r5   rZ   r	   rX   r_   )r+   r-   r,   r/   ra   r   r   r   update  s    z
Git.updatec             C   s   |  j  d d d g d d
 d | } | j   } y | d } Wn t k
 rX t  Yn Xx$ | D] } | j d  r` | } Pq` W| j d	  d } | j   S)z
        Return URL of the first remote encountered.

        Raises RemoteNotFoundError if the repository does not have a remote
        url configured.
        rc   z--get-regexpzremote\..*\.urlr<   r=   r>   r   zremote.origin.url  )r=   )r5   rM   
IndexErrorr   r6   r8   r?   )r@   rA   stdoutremotesZfound_remoteZremoter,   r   r   r   get_remote_url  s    
zGit.get_remote_urlNc             C   s7   | d  k r d } |  j  d | g d | } | j   S)Nr&   z	rev-parser>   )r5   r?   )r@   rA   r'   current_revr   r   r   rY   5  s
    zGit.get_revisionc             C   sv   |  j  d d g d | j   } t j j |  sH t j j | |  } t j j t j 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-dirr>   z..)r5   r?   osrK   isabsr9   abspathr   )r@   rA   git_dirZ	repo_rootr   r   r   get_subdirectory>  s    !zGit.get_subdirectoryc                sg  t  |  \ } } } } } | j d  r | d t | j d    } | t j |  j d d  j d  } t | | | | | f  } | j d  d }	 | d |	  t | |	 d  | | | | f  } d | k r9d | k s t	  | j d	 d
  } t
 t |   j |  \ } }
 } | j d d  } n! t
 t |   j |  \ } }
 } | |
 | 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.
        fileNrF   \+r=   z://zfile:zgit+z
git+ssh://zssh://r2   )urlsplitrI   r7   lstripurllib_requesturl2pathnamereplace
urlunsplitfindrT   superr   get_url_rev_and_auth)r@   r,   schemenetlocrK   queryfragmentinitial_slashesnewpath
after_plusr'   	user_pass)	__class__r   r   r}   M  s"     #!!zGit.get_url_rev_and_authc             C   sK   t  j j t  j j | d   s% d  S|  j d d d d d g d | d  S)Nz.gitmodules	submodulerf   z--initz--recursivez-qr>   )rm   rK   existsr9   r5   )r@   rA   r   r   r   r_   p  s
    !zGit.update_submodulesc                s   t  t |   j |  } | r" | Sy% |  j d d g d | d d } Wn; t k
 ro t j d |  d  SYn t k
 r d  SYn Xt j	 j
 | j d   S)Nz	rev-parsez--show-toplevelr>   Zlog_failed_cmdFzKcould not determine if %s is under git control because git is not availablez
)r|   r   get_repository_rootr5   r   rV   debugr   rm   rK   normpathrstrip)r@   rA   locr)r   r   r   r   y  s    		zGit.get_repository_root)zgitr   r    r!   r"   r#   )r$   r%   )__name__
__module____qualname__r[   dirname	repo_nameschemesZunset_environZdefault_arg_revstaticmethodr(   r0   r;   classmethodrE   rH   r*   rZ   r)   rb   rd   rf   rk   rY   rq   r}   r_   r   r   r   )r   r   r   .   s0   (-#	r   ).
__future__r   loggingos.pathrm   reZpip._vendor.packaging.versionr   r:   Zpip._vendor.six.moves.urlliburllib_parser   rw   Zpip._internal.exceptionsr   r   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   Ztypingr   r   r   r   r   ru   rz   	getLoggerr   rV   compiler   r   r   registerr   r   r   r   <module>   s0   "		 `