o
    i2                     @  s  d Z ddlmZ 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	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 ddlmZ eeZee Zz
ddlZee7 ZW n eyi   ed	 Y nw z
ddlZee7 ZW n ey   ed
 Y nw d;ddZd<ddZd=ddZ d>ddZ!d;ddZ"d?d d!Z#d@d$d%Z$dAdBd*d+Z%dCd,d-Z&dDd2d3Z'dEd5d6Z(	dFdGd9d:Z)dS )HzUtilities related archives.    )annotationsN)Iterable)ZipInfo)InstallationError)BZ2_EXTENSIONSTAR_EXTENSIONSXZ_EXTENSIONSZIP_EXTENSIONS)
ensure_dirzbz2 module is not availablezlzma module is not availablereturnintc                  C  s   t d} t |  | S )zBGet the current umask which involves having to set it temporarily.r   )osumask)mask r   =lib/python3.10/site-packages/pip/_internal/utils/unpacking.pycurrent_umask-   s   

r   pathstr	list[str]c                 C  sd   |  d d} d| v r$d| v r| d| dk sd| vr$| ddS d| v r.| ddS | dgS )N/\    )lstripfindsplitr   r   r   r   split_leading_dir4   s   $r   pathsIterable[str]boolc                 C  sD   d}| D ]}t |\}}|s dS |du r|}q||kr dS qdS )zyReturns true if all the paths have the same leading path name
    (i.e., everything is in one subdirectory in an archive)NFT)r   )r   Zcommon_prefixr   prefixrestr   r   r   has_leading_dir@   s   r$   	directorytargetc                 C  s0   t j| }t j|}t j||g}||kS )zL
    Return true if the absolute path of target is within the directory
    )r   r   abspathcommonprefix)r%   r&   Zabs_directoryZ
abs_targetr"   r   r   r   is_within_directoryO   s   r)   c                   C  s   dt   @ dB S )Ni  I   )r   r   r   r   r   !_get_default_mode_plus_executableZ   s   r+   Nonec                 C  s   t | t  dS )zx
    Make file present at path have execute for user/group/world
    (chmod +x) is no-op on windows per python docs
    N)r   chmodr+   r   r   r   r   2set_extracted_file_to_default_mode_plus_executable^   s   r.   infor   c                 C  s$   | j d? }t|ot|o|d@ S )N   r*   )Zexternal_attrr!   statS_ISREG)r/   moder   r   r   zip_item_is_executablef   s   
r4   Tfilenamelocationflattenc              	   C  s@  t | t| d}ztj|dd}t| o|}| D ]v}|j}|}|r,t|d }t	j
||}t	j
|}	t||sId}
t|
| |||drSt | qt |	 ||}z)t|d}t|| W d   n1 ssw   Y  W |  t|rt| q|  t|rt| w w W |  dS |  w )	a  
    Unzip the file (with path `filename`) to the destination `location`.  All
    files are written based on system defaults and umask (i.e. permissions are
    not preserved), except that regular file members with any execute
    permissions (user, group, or world) have "chmod +x" applied after being
    written. Note that for windows, any execute changes using os.chmod are
    no-ops per the python docs.
    rbT)Z
allowZip64r   zQThe zip file ({}) has a file ({}) trying to install outside target directory ({}))r   r   wbN)r
   openzipfileZipFiler$   Znamelistinfolistr5   r   r   r   joindirnamer)   r   formatendswithshutilcopyfileobjcloser4   r.   )r5   r6   r7   Zzipfpzipleadingr/   namefndirmessagefpdestfpr   r   r   
unzip_filem   sF   	





rM   c                   sb  t   ds drd}n& trd}n tr)d}n dr3d}ntd d	}tj|d
d}zht	dd |
 D }ztj W n tyd   t|| Y nAw t |r|
 D ]}t|j\}}||_| rt|j\}}	||kr|	|_qnd fdd}
|j|
d W |  dS W |  dS |  w )a  
    Untar the file (with path `filename`) to the destination `location`.
    All files are written based on system defaults and umask (i.e. permissions
    are not preserved), except that regular file members with any execute
    permissions (user, group, or world) have "chmod +x" applied on top of the
    default.  Note that for windows, any execute changes using os.chmod are
    no-ops per the python docs.
    z.gzz.tgzzr:gzzr:bz2zr:xzz.tarrz-Cannot determine compression type for file %szr:*zutf-8)encodingc                 S  s   g | ]}|j qS r   )rG   ).0memberr   r   r   
<listcomp>   s    zuntar_file.<locals>.<listcomp>rQ   tarfile.TarInfor   r   r   c              
     s   | j }z%z | } W n tjy&   tjd d dv r#t| } n Y nw W n tjy? } z
d}t||d }~ww | 	 rM|d@ rM| _ | S d | _ | S )N   >   )rT   	      )rT         )rT   
      z%Invalid member in the tar file {}: {}r*   )
r3   tarfileZLinkOutsideDestinationErrorsysversion_infoZ
tar_filterTarErrorr   r@   isfile)rQ   r   Z	orig_modeexcrJ   data_filterZdefault_mode_plus_executabler5   r6   r   r   
pip_filter   s4   

zuntar_file.<locals>.pip_filter)filterN)rQ   rS   r   r   r   rS   )r
   lowerrA   r   r   loggerwarningr[   r:   r$   
getmembersrb   AttributeError_untar_without_filterr+   r   rG   Zislnklinkname
extractallrD   )r5   r6   r3   tarrF   rQ   Z	name_leadZ	name_restZlnk_leadZlnk_restrc   r   ra   r   
untar_file   sJ   	
':rn   rm   tarfile.TarFiletarinforS   c                 C  sX   t jt j|j|j}t j|}|dd}z| | W dS  t	y+   Y dS w )zGCheck if the file pointed to by the symbolic link is in the tar archiver   r   TF)
r   r   r>   r?   rG   rk   normpathreplaceZ	getmemberKeyError)rm   rp   rk   r   r   r   is_symlink_target_in_tar   s   
rt   rF   c                 C  s  |  D ]}|j}|rt|d }tj||}t||s(d}t|| |||	 r1t
| q| rkt||sGd}t|| |j|jz||| W q tyj } ztd| |j| W Y d}~qd}~ww z||}	W n ttfy } ztd| |j| W Y d}~qd}~ww t
tj| |	dusJ t|d}
t|	|
 W d   n1 sw   Y  |	  ||| |jd@ rt| qdS )z/Fallback for Python without tarfile.data_filterr   zQThe tar file ({}) has a file ({}) trying to install outside target directory ({})z/In the tar file %s the member %s is invalid: %sNr9   r*   )rh   rG   r   r   r   r>   r)   r   r@   isdirr
   Zissymrt   rk   Z_extract_member	Exceptionrf   rg   Zextractfilers   ri   r?   r:   rB   rC   rD   utimer3   r.   )r5   r6   rm   rF   rQ   rH   r   rJ   r`   rK   rL   r   r   r   rj   	  sj   





rj   content_type
str | Nonec                 C  s   t j| } |dks|  tst| r#t| || d d d S |dks7t	
| s7|  tt t r>t| | d S td| || td| )Nzapplication/zipz.whl)r7   zapplication/x-gzipzZCannot unpack file %s (downloaded from %s, content-type: %s); cannot detect archive formatz#Cannot determine archive format of )r   r   realpathre   rA   r	   r;   
is_zipfilerM   r[   Z
is_tarfiler   r   r   rn   rf   Zcriticalr   )r5   r6   rx   r   r   r   unpack_fileN  s(   r|   )r   r   )r   r   r   r   )r   r    r   r!   )r%   r   r&   r   r   r!   )r   r   r   r,   )r/   r   r   r!   )T)r5   r   r6   r   r7   r!   r   r,   )r5   r   r6   r   r   r,   )rm   ro   rp   rS   r   r!   )
r5   r   r6   r   rm   ro   rF   r!   r   r,   )N)r5   r   r6   r   rx   ry   r   r,   )*__doc__Z
__future__r   Zloggingr   rB   r1   r\   r[   r;   Zcollections.abcr   r   Zpip._internal.exceptionsr   Zpip._internal.utils.filetypesr   r   r   r	   Zpip._internal.utils.miscr
   Z	getLogger__name__rf   ZSUPPORTED_EXTENSIONSbz2ImportErrordebuglzmar   r   r$   r)   r+   r.   r4   rM   rn   rt   rj   r|   r   r   r   r   <module>   sR    








.
`
H