B
    HeF                 @   s  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	m
Z
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'e(e)d
ddZ*G dd deZ+e,e+ dS )    N)ListOptionalTuple)
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returnc             C   s   t t| S )N)bool
HASH_REGEXmatch)r    r   i/var/www/downstreamdata.science/rtclock/rtclock-venv/lib/python3.7/site-packages/pip/_internal/vcs/git.pylooks_like_hash7   s    r   c                   s  e Zd ZdZdZdZdZdZdZe	e
ee
 ddd	Ze
e
ed
ddZeedf dddZee
ee
 dddZee
e
eee
 ef dddZee
e
edddZee
eeedddZee
ee
 edddZe
eeedd d!d"Ze
eeddd#d$Ze
eeddd%d&Zee
e
dd'd(Ze	e
e
d)d*d+Zee
e
ed,d-d.Z ed<e
ee
 e
d,d/d0Z!ee
ee
 dd1d2Z"ee
ee
ee
 e#f d) fd3d4Z$ee
ddd5d6Z%ee
ee
 d fd7d8Z&e	e
ed9d:d;Z'  Z(S )=Gitgitz.gitclone)zgit+httpz	git+httpszgit+sshzgit+gitzgit+file)GIT_DIRGIT_WORK_TREEHEAD)revr   c             C   s   | gS )Nr   )r    r   r   r   get_base_rev_argsK   s    zGit.get_base_rev_args)urldestr   c             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)selfr"   r#   _rev_optionsis_tag_or_branchr   r   r   is_immutable_rev_checkoutO   s    zGit.is_immutable_rev_checkout.)r   c             C   sH   | j dgdddd}t|}|s2td| dS tdd	 | D S )
Nversionzgit versionFT)command_descshow_stdoutstdout_onlyzCan't parse git version: %sr   c             s   s   | ]}t |V  qd S )N)int).0cr   r   r   	<genexpr>h   s    z&Git.get_git_version.<locals>.<genexpr>)run_commandGIT_VERSION_REGEXr   loggerwarningtuplegroups)r'   r,   r   r   r   r   get_git_version]   s    
zGit.get_git_version)locationr   c             C   sD   dddg}| 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-refz-qr   )   FT)extra_ok_returncodesr.   r/   cwdzrefs/heads/N)r4   strip
startswithlen)clsr;   argsoutputrefr   r   r   get_current_branchj   s    


zGit.get_current_branch)r#   r    r   c          	   C   s   | j d|g|dddd}i }xh| dD ]V}|d}|s@q,y|jdd	d
\}}W n" tk
rx   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-refFTignore)r>   r.   r/   on_returncode
    )maxsplitzunexpected show-ref line: zrefs/remotes/origin/z
refs/tags/N)r4   r?   splitrstrip
ValueErrorget)rB   r#   r    rD   refslineref_sharef_name
branch_reftag_refr   r   r   r   r&      s.    




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)r@   r   
has_commit)rB   r#   r    r   r   r   _should_fetch   s    	
zGit._should_fetch)r#   r"   r)   r   c             C   s   |j }|dk	st| ||\}}|dk	rF||}|r<|nd|_|S t|sZtd| | ||sj|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.fetchz-q)r>   
FETCH_HEAD)r    )arg_revAssertionErrorr&   make_newbranch_namer   r6   r7   rY   r4   r
   to_argsget_revision)rB   r#   r"   r)   r    r   	is_branchr   r   r   resolve_revision   s&    

zGit.resolve_revision)r#   namer   c             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)ra   )rB   r#   rd   r   r   r   r%      s    	zGit.is_commit_id_equalN)r#   r"   r)   	verbosityr   c             C   sJ  |  }td||t| |dkr*d}n|dkr8d}nd}|  dkrb| td|||f  n| td|||f  |jr| |||}t	|d
d }t
d|| |d kr| ||jstdd| }| j||d n6| ||kr,d| }	dd|d|	g}| j||d n| |}
||
}td||j | | d S )NzCloning %s%s to %sr   )z--quietr<   r   )z	--verbosez
--progress)rL      r   --filter=blob:noner_   zRev options %s, branch_name %scheckoutz-q)r>   zorigin/z-bz--trackzResolved %s to commit %s)r   rg   )r   )
to_displayr6   infor   r:   r4   r
   r    rc   getattrdebugr%   r`   rF   ra   r^   update_submodules)r'   r#   r"   r)   re   rev_displayflagsr_   cmd_argstrack_branchr   r   r   r   	fetch_new  sN     



zGit.fetch_newc             C   sB   | j tdd||d tdd| }| j ||d | | d S )Nconfigzremote.origin.url)r>   rh   z-q)r4   r
   r`   rm   )r'   r#   r"   r)   rp   r   r   r   switchD  s    
z
Git.switchc             C   sp   |   dkr"| jdddg|d n| jddg|d | |||}tddd| }| j||d | | d S )N)r<   	   rZ   z-qz--tags)r>   resetz--hard)r:   r4   rc   r
   r`   rm   )r'   r#   r"   r)   rp   r   r   r   updateN  s    z
Git.updatec             C   s   | j dddgddd|d}| }y|d }W n tk
rF   tY nX x|D ]}|d	rN|}P qNW |d
d }| | S )z
        Return URL of the first remote encountered.

        Raises RemoteNotFoundError if the repository does not have a remote
        url configured.
        rs   z--get-regexpzremote\..*\.url)r<   FT)r=   r.   r/   r>   r   zremote.origin.url rK   r<   )r4   
splitlines
IndexErrorr   r@   rN   _git_remote_to_pip_urlr?   )rB   r;   stdoutremotesfound_remoteremoter"   r   r   r   get_remote_url\  s"    



zGit.get_remote_url)r"   r   c             C   sN   t d| r| S tj| r*t|  S t| }|rB|	dS t
| dS )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/\3N)rer   ospathexistspathlibPurePathas_uri	SCP_REGEXexpandr   )r"   	scp_matchr   r   r   rz   z  s    

zGit._git_remote_to_pip_url)r;   r    r   c             C   s>   y | j dddd| g|dd W n tk
r4   dS X dS dS )	zU
        Check if rev is a commit that is available in the local repository.
        z	rev-parsez-qz--verifyzsha^F)r>   log_failed_cmdTN)r4   r   )rB   r;   r    r   r   r   rX     s    zGit.has_commitc             C   s*   |d krd}| j d|gdd|d}| S )Nr   z	rev-parseFT)r.   r/   r>   )r4   r?   )rB   r;   r    current_revr   r   r   ra     s    zGit.get_revisionc             C   sT   | j ddgdd|d }tj|s4t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.
        z	rev-parsez	--git-dirFT)r.   r/   r>   z..)r4   r?   r   r   isabsjoinabspathr   )rB   r;   git_dir	repo_rootr   r   r   get_subdirectory  s    zGit.get_subdirectoryc                s   t |\}}}}}|dr|dt|d  }|tj|ddd }|dd }	|d|	 t	||	d ||||f }d|krd|kst
|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endswithrA   lstripurllibrequesturl2pathnamereplacefind
urlunsplitr]   superget_url_rev_and_auth)rB   r"   schemenetlocr   queryfragmentinitial_slashesnewpath
after_plusr    	user_pass)	__class__r   r   r     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	submodulerw   z--initz--recursivez-q)r>   )r   r   r   r   r4   )rB   r;   r   r   r   rm     s
    zGit.update_submodulesc                sx   t  |}|r|S y| jddg|ddddd}W n2 tk
rR   td| d S  tk
rd   d S X tj	|
dS )	Nz	rev-parsez--show-toplevelFTraise)r>   r.   r/   rH   r   zKcould not determine if %s is under git control because git is not availablez
)r   get_repository_rootr4   r   r6   rl   r   r   r   normpathrO   )rB   r;   locr)r   r   r   r     s&    zGit.get_repository_root)repo_urlr   c             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_prefix  s    zGit.should_add_vcs_url_prefix)N))__name__
__module____qualname__rd   dirname	repo_nameschemesunset_environdefault_arg_revstaticmethodstrr   r!   r   r+   r   r0   r:   classmethodr   rF   r&   rY   r   r   rc   r%   rr   rt   rw   r   rz   rX   ra   r   r   r   rm   r   r   __classcell__r   r   )r   r   r   ;   sV    -+=
$r   )-loggingos.pathr   r   r   urllib.parser   urllib.requesttypingr   r   r   pip._internal.exceptionsr   r   pip._internal.utils.miscr   r   r	   pip._internal.utils.subprocessr
    pip._internal.vcs.versioncontrolr   r   r   r   r   r   r   parser   r   	getLoggerr   r6   compiler5   r   VERBOSEr   r   r   r   r   registerr   r   r   r   <module>   s2   $



   V