HEX

File: //usr/lib/python3/dist-packages/mercurial/__pycache__/pathutil.cpython-310.pyc
o

�]Lb�0�@sddlmZddlZddlZddlZddlZddlZddlmZddl	m
Z
mZmZm
Z
mZe
jr=ddlmZmZmZmZe�dd�Ze�d	�Zd
d�ZGdd
�d
e�Zddd�Zdd�Zdd�ZGdd�de�Ze� ed�rrejZedurxeZej!Z!ej"Z"dS)�)�absolute_importN�)�_)�encoding�error�policy�pycompat�util)�Any�Callable�Iterator�Optional�dirstate�Dirs�parserscCst�|���S�N)r�hfsignoreclean�lower)�s�r�4/usr/lib/python3/dist-packages/mercurial/pathutil.py�_lowercleansrc@sBeZdZdZddd�Zddd�Zd	d
�Zdd�Zej	d
d��Z
dS)�pathauditora^ensure that a filesystem path contains no banned components.
    the following properties of a path are checked:

    - ends with a directory separator
    - under top-level .hg
    - starts at the root of a windows drive
    - contains ".."

    More check are also done about the file system states:
    - traverses a symlink (e.g. a/symlink_here/b)
    - inside a nested repository (a callback can be used to approve
      some nested repositories, e.g., subrepositories)

    The file system checks are only done when 'realfs' is set to True (the
    default). They should be disable then we are auditing path for operation on
    stored history.

    If 'cached' is set to True, audited paths and sub-directories are cached.
    Be careful to not keep the cache of unmanaged directories for long because
    audited paths may be replaced with symlinks.
    NTFcCsXt�|_t�|_||_||_||_||_tj�	|�r%t
�|�s%t
j|_dSdd�|_dS)NcSs|Srr)�xrrr�<lambda>Esz&pathauditor.__init__.<locals>.<lambda>)
�set�audited�
auditeddir�root�_realfs�_cached�callback�os�path�lexistsr	�fscasesensitive�normcase)�selfrr!�realfs�cachedrrr�__init__;szpathauditor.__init__cCs�t�|�}|�|�}||jvrdSt�|�rt�td�|��t�|�}t	j
�|�ds9t|d�dvs9t
j|vrBt�td�|��|D]!}d|vre|�dd�\}}|��re|��dvret�td�|��qDd	t|�vr�d
d�|D�}dD]'}||dd�vr�|�|�}	t	j
j|d|	��}
t�td
�|t
�|
�f��qut�|�}t|�t|�ks�J�|��|��tt|��D]2}t
j�|d|d��}
t
j�|d|d��}||jvr�q�|jr�|�|
|�|jr�|j�|�q�|jr�|j�|�dSdS)zPCheck the relative path.
        path may contain a pattern (e.g. foodir/**.txt)Ns$path ends in directory separator: %sr)�.hg�.hg.�s#path contains illegal component: %s�~r)sHGsHG8B6Cr+cSsg|]}t|��qSr)r)�.0�prrr�
<listcomp>gsz(pathauditor.__call__.<locals>.<listcomp>)r+r,�"path '%s' is inside nested repo %r)r	�	localpathr&r�endswithsepr�
InputErrorr�	splitpathr"r#�
splitdriverr�ospardir�split�isdigit�upper�index�join�bytestr�len�pop�range�osseprr�_checkfsr �add)r'r#�mode�normpath�partsr0�first�last�lparts�pos�base�	normparts�i�prefix�
normprefixrrr�__call__Gsj




�
�

�
��
���

��zpathauditor.__call__c
Cs�tj�|j|�}zt�|�}Wnty.}z|jtjtjtj	fvr#�WYd}~dSd}~wwt
�|j�rHt
d�t�|�t�|�f}t�|��t
�|j�rqtj�tj�|d��rs|jra|�|�sut
d�}t�||t�|�f��dSdSdS)z3raise exception if a file system backed check failsNs"path %r traverses symbolic link %rr+r2)r"r#r=r�lstat�OSError�errno�ENOENT�ENOTDIR�EINVAL�stat�S_ISLNK�st_moderrr>r�Abort�S_ISDIR�isdirr!)r'rOr#�curpath�st�err�msgrrrrC�s.����
��zpathauditor._checkfsc	Cs*z||�WdSttjfyYdSw�NTF)rSrr[)r'r#rrr�check�s�zpathauditor.checkccs\�|jr	dVdSzd|_dVW|j��|j��d|_dS|j��|j��d|_wrb)r r�clearr�r'rrrr)�s�




�
zpathauditor.cachedrbr)�__name__�
__module__�__qualname__�__doc__r*rQrCrc�
contextlib�contextmanagerr)rrrrr$s

ArcCs�t�|�r|}n|tj}|}tj�|�stj�|||�}tj�|�}|dur+t	|�}||krE|�
|�rE|t|�d�}||�t�|�S||krKdSg}	zt�
||�}Wntyad}Ynw|r{|shdS|��tjj|�}||�t�|�St�|�\}}	|�|	�||kr�n|}qNd}
z&||kr�t||||�t�||d�}|�tj�r�|dd�}td�|}
Wn
tjy�Ynwtjtd�||f|
d��)	a�return the canonical path of myname, given cwd and root

    >>> def check(root, cwd, myname):
    ...     a = pathauditor(root, realfs=False)
    ...     try:
    ...         return canonpath(root, cwd, myname, a)
    ...     except error.Abort:
    ...         return 'aborted'
    >>> def unixonly(root, cwd, myname, expected='aborted'):
    ...     if pycompat.iswindows:
    ...         return expected
    ...     return check(root, cwd, myname)
    >>> def winonly(root, cwd, myname, expected='aborted'):
    ...     if not pycompat.iswindows:
    ...         return expected
    ...     return check(root, cwd, myname)
    >>> winonly(b'd:\\repo', b'c:\\dir', b'filename')
    'aborted'
    >>> winonly(b'c:\\repo', b'c:\\dir', b'filename')
    'aborted'
    >>> winonly(b'c:\\repo', b'c:\\', b'filename')
    'aborted'
    >>> winonly(b'c:\\repo', b'c:\\', b'repo\\filename',
    ...         b'filename')
    'filename'
    >>> winonly(b'c:\\repo', b'c:\\repo', b'filename', b'filename')
    'filename'
    >>> winonly(b'c:\\repo', b'c:\\repo\\subdir', b'filename',
    ...         b'subdir/filename')
    'subdir/filename'
    >>> unixonly(b'/repo', b'/dir', b'filename')
    'aborted'
    >>> unixonly(b'/repo', b'/', b'filename')
    'aborted'
    >>> unixonly(b'/repo', b'/', b'repo/filename', b'filename')
    'filename'
    >>> unixonly(b'/repo', b'/repo', b'filename', b'filename')
    'filename'
    >>> unixonly(b'/repo', b'/repo/subdir', b'filename', b'subdir/filename')
    'subdir/filename'
    Nr-TF���sconsider using '--cwd %s's%s not under root '%s')�hint)r	r4rrBr"r#�isabsr=rFr�
startswithr?�pconvert�samefilerS�reverser9�append�	canonpath�pathto�endswithrrr[)r�cwd�myname�auditor�rootsep�name�relr�dirname�basenamerm�relpathrrrrt�sd
+

�

����rtcCs0tj�|�\}}t|�ttj�kr|tjS|S)a�normalize the specified path as path prefix

    Returned value can be used safely for "p.startswith(prefix)",
    "p[len(prefix):]", and so on.

    For efficiency, this expects "path" argument to be already
    normalized by "os.path.normpath", "os.path.realpath", and so on.

    See also issue3033 for detail about need of this function.

    >>> normasprefix(b'/foo/bar').replace(pycompat.ossep, b'/')
    '/foo/bar/'
    >>> normasprefix(b'/').replace(pycompat.ossep, b'/')
    '/'
    )r"r#r7r?rrB)r#�dr0rrr�normasprefixs
r�ccsB�|�d�}|dkr|d|�V|�dd|�}|dks
dVdS)N�/rlrr-)�rfind)r#rKrrr�finddirs6s�
�
r�c@s:eZdZdZddd�Zdd�Zdd�Zd	d
�Zdd�Zd
S)�dirsz6a multiset of directory names from a set of file pathsFcCsli|_|j}t|t�r"|r"t�|�D]
\}}|jdkr||�qdS|r+d}t�|��|D]}||�q-dS)zS
        a dict map indicates a dirstate while a list indicates a manifest
        �rs3`only_tracked` is only supported with a dict sourceN)	�_dirs�addpath�
isinstance�dictr�	iteritems�stater�ProgrammingError)r'�map�only_trackedr��frrarrrr*Bs
��

�z
dirs.__init__cCsT|j}t|�D] }|�d�rtd|��||vr#||d7<dSd||<qdS)Nr�z-found invalid consecutive slashes in path: %rr)r�r�rv�
ValueError�r'r#r�rLrrrr�Ss
�
�zdirs.addpathcCs@|j}t|�D]}||dkr||d8<dS||=qdS)Nr)r�r�r�rrr�delpath`s�zdirs.delpathcCs
t|j�Sr)�iterr�rerrr�__iter__is
z
dirs.__iter__cCs
||jvSr)r�)r'r�rrr�__contains__ls
zdirs.__contains__N)F)	rfrgrhrir*r�r�r�r�rrrrr�?s

	r�r)#�
__future__rrjrTr"�	posixpathrX�i18nr�rrrrr	�
TYPE_CHECKING�typingr
rrr
�
importrust�rustdirs�	importmodrr�objectrrtr�r�r��safehasattrr=r}rrrr�<module>s2

g	2