o
    [hH                     @   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	 d dl
mZ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*de+de,fddZ-G dd deZ.e/e. dS )    N)replace)Any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   |/home/ubuntu/experiments/live_experiments/Pythonexperiments/Otree/venv/lib/python3.10/site-packages/pip/_internal/vcs/git.pylooks_like_hash8   s   r   c                
       sf  e Zd ZdZdZdZdZdZdZe	de
dee
 fd	d
Zededede
f fddZde
de
defddZdeedf fddZede
dee
 fddZede
de
deee
 ef fddZede
de
defddZede
dededefddZede
d ee
 defd!d"Zde
deded#edd$f
d%d&Zde
dededd$fd'd(Zde
dededd$fd)d*Zede
de
fd+d,Z e	de
de
fd-d.Z!ede
de
defd/d0Z"ed>de
dee
 de
fd1d2Z#ede
dee
 fd3d4Z$ede
dee
ee
 e%f f fd5d6Z&ede
dd$fd7d8Z'ede
dee
 f fd9d:Z(e	d;e
def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 r   r   r#   r   r   r   get_base_rev_argsL   s   zGit.get_base_rev_argsargskwargsc                    sB   t jdr|di }d|d< d|d< ||d< t j|i |S )N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/   P   s   zGit.run_commandurldestc                 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)selfr3   r4   _rev_optionsis_tag_or_branchr   r   r   is_immutable_rev_checkoutY   s   zGit.is_immutable_rev_checkout.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)command_descshow_stdoutstdout_onlyzCan't parse git version: %sr         )r/   GIT_VERSION_REGEXr   loggerwarningintgroup)r8   r=   r   r   r   r   get_git_versiong   s   
zGit.get_git_versionlocationc                 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"   rA   FTextra_ok_returncodesr?   r@   cwdzrefs/heads/N)r/   strip
startswithlen)r0   rI   r&   outputrefr   r   r   get_current_brancht   s   

zGit.get_current_branchc              	   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)rN   r?   r@   on_returncode
 rB   )maxsplitzunexpected show-ref line: zrefs/remotes/origin/z
refs/tags/N)r/   rO   splitrstrip
ValueErrorr-   )r0   r4   r#   rR   refslineref_sharef_name
branch_reftag_refr   r   r   r   r7      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)rP   r   
has_commit)r0   r4   r#   r   r   r   _should_fetch   s   
	zGit._should_fetchr:   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.fetchrJ   rN   
FETCH_HEADr$   )arg_revr7   make_newr   r   rD   rE   re   r/   r   to_argsget_revision)r0   r4   r3   r:   r#   r   	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)rm   )r0   r4   rp   r   r   r   r6      s   	zGit.is_commit_id_equal	verbosityNc                 C   sT  |  }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 | | d S )NzCloning %s%s to %sr   )z--quietrA   r   )z	--verbosez
--progress)rB      r   z--filter=blob:nonerf   zRev options %s, branch_name %scheckoutrJ   rh   zorigin/z-bz--trackzResolved %s to commit %s)
to_displayrD   infor
   rH   r/   r   r#   ro   getattrdebugr6   rl   rT   rm   rk   update_submodules)r8   r4   r3   r:   rq   rev_displayflagsrf   cmd_argstrack_branchr   r   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.urlrh   rs   rJ   )r/   r   rl   rx   r8   r4   r3   r:   r{   r   r   r   switchN  s   
z
Git.switchc                 C   sn   |   dkr| jg d|d n	| jddg|d | |||}tddd| }| j||d | | d S )N)rA   	   )rg   rJ   z--tagsrh   rg   rJ   resetz--hard)rH   r/   ro   r   rl   rx   r   r   r   r   updateX  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.
        )r~   z--get-regexpzremote\..*\.urlrK   FTrL   r   zremote.origin.url rY   rA   )r/   
splitlines
IndexErrorr   rP   r[   _git_remote_to_pip_urlrO   )r0   rI   stdoutremotesfound_remoteremoter3   r   r   r   get_remote_urlf  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   )r3   	scp_matchr   r   r   r     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-parserJ   z--verifyzsha^F)rN   log_failed_cmdT)r/   r   )r0   rI   r#   r   r   r   rd     s   zGit.has_commitc                 C   s*   |d u rd}| j d|gdd|d}| S )Nr"   r   FTr?   r@   rN   )r/   rO   )r0   rI   r#   current_revr   r   r   rm     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/   rO   r+   r   isabsjoinabspathr   )r0   rI   git_dir	repo_rootr   r   r   get_subdirectory  s   
zGit.get_subdirectoryc                    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/\+rA   z://zfile:zgit+z
git+ssh://zssh:// )urlsplitendswithrQ   lstripurllibrequesturl2pathnamer   find
urlunsplitr.   get_url_rev_and_auth)r0   r3   schemenetlocr   queryfragmentinitial_slashesnewpath
after_plusr#   	user_passr1   r   r   r     s&   


zGit.get_url_rev_and_authc                 C   s0   t jt j|dsd S | jg d|d d S )Nz.gitmodules)	submoduler   z--initz--recursiverJ   rh   )r+   r   r   r   r/   )r0   rI   r   r   r   rx     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)rN   r?   r@   rV   r   zKcould not determine if %s is under git control because git is not availablez
)r.   get_repository_rootr/   r   rD   rw   r   r+   r   normpathr\   )r0   rI   locrr1   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_prefix  s   zGit.should_add_vcs_url_prefixr   )+__name__
__module____qualname__rp   dirname	repo_nameschemesunset_environdefault_arg_revstaticmethodstrr   r%   classmethodr   r/   r   r<   r   rF   rH   r   rT   r7   re   r	   r   ro   r6   r}   r   r   r   r   rd   rm   r   r   r   rx   r   r   __classcell__r   r   r1   r   r   <   s|    	$--
?
&r   )0loggingZos.pathr+   r   r   urllib.parser   urllib.requestdataclassesr   typingr   r   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   rD   compilerC   r   VERBOSEr   r   r   r   r   registerr   r   r   r   <module>   s:    $



   _