�
i 0i� � � � d dl mZ d dlZd dlZd dlZd dlZd dlZd dlZd dl Z d dl
m Z d dl
m Z d dl
m
Z
d dl
mZ d dl
mZ d dlmZ d d lmZ d d
lmZ d d lmZ d d lmZ ed
� Z ed� Zdd�Zd d d�Zdd�Z d d�Z G d� de � Z d d d�Z ejB d d�� Z"dd e � d� d!d�Z#y)"� )� annotationsN)�Callable)� Generator)�Iterable)�MutableMapping)�Sequence)�Any)�TypeVar)�
parse_shebang)� cmd_output_b)� cmd_output_p�TArg�TRetc � � t t j d� � S # t $ r Y nw xY w t j
� S # t $ r Y yw xY w)Nr � )�len�os�sched_getaffinity�AttributeError�multiprocessing� cpu_count�NotImplementedError� � �?/opt/hc_python/lib/python3.12/site-packages/pre_commit/xargs.pyr r sW � �
� �2�'�'��*�+�+�� �
� �
��� �(�(�*�*�� � ���s � � ,�,�A � A�Ac �� � | �| n t t dt j � }dt |� z }|j � D ]" \ }}|t |� t |� z dz z
}�$ |S )N�environb� � )�getattrr �environr �items)�_envr! �size�k�vs r �
_environ_sizer' ) s] � ��&�d�G�B�
�B�J�J�,O�G� �s�7�|� �D��
�
����1� ��A���Q���!�#�#�� � �Kr c �� � t j dk( r;t j d� dz
t � z
} t t | d� d� } | S t j dk( ryy)N�posix�
SC_ARG_MAXi i i �nti x )r �name�sysconfr' �max�min)�maximums r �_get_platform_max_lengthr1 1 sS � � �w�w�'���*�*�\�*�T�1�M�O�C���c�'�7�+�W�5���� ���D� � � r c �� � dj | � }t j dk( r t |j d� � dz S t |j t j
� � � S )N� �win32zutf-16ler )�join�sys�platformr �encode�getfilesystemencoding)�cmd�full_cmds r �_command_lengthr< = sS � ��x�x��}�H� �|�|�w� ��8�?�?�:�.�/�1�4�4��8�?�?�3�#<�#<�#>�?�@�@r c � � e Zd Zy)�ArgumentTooLongErrorN)�__name__�
__module__� __qualname__r r r r> r> H s � �r r> c �P � |xs
t � }t dt j t |� |z � � }t | � } g }g }t
t |� � }t | � dz }|r�|j � }t |� dz } || z |k r%t |� |k r|j |� || z
}nH|s t |� �|j | t |� z � g }t | � dz }|j |� |r��|j | t |� z � t |� S )N� r ) r1 r. �math�ceilr �tuple�list�reversedr< �pop�appendr> )
r: �varargs�target_concurrency� _max_length�max_args�ret�ret_cmd� total_length�arg�
arg_lengths
r � partitionrT L s � � �;�!9�!;�K� �1�d�i�i��G� �/A� A�B�C�H�
��*�C�
�C��G��8�G�$�%�G�"�C�(�1�,�L�
��k�k�m��$�S�)�A�-�
� �z�)�[�8��G� �x�'� �N�N�3� � �J� &�L��&�s�+� +�
�J�J�s�U�7�^�+� ,��G�*�C�0�1�4�L� �N�N�3� �# �& �J�J�s�U�7�^�#�$� ��:�r c # � K � | dk( r t �� y t j j | � 5 }|j �� d d d � y # 1 sw Y y xY w�w)Nr )�map�
concurrent�futures�ThreadPoolExecutor)�maxsize�exs r �_thread_mapperr\ x s> � �� � �!�|�� �
�
�
�
2�
2�7�
;�r��&�&�L� <�
;�
;�s �.A�A� A�A�
AFr )�colorrL rM c � ��� |rt nt �d}d} t j | � } t j dk( rT| d j � j d� r2t j d� } dt | � z
t d � z
d
z
}t | |||� }
d ��fd � } t t |
� |� } t | � 5 }
|
| |
� }|D ]$ \ }}}t |� t |� kD r|}||z
}�& ddd� ||fS # t j $ r }|j � dd cY d}~S d}~ww xY w# 1 sw Y ||fS xY w)
z�A simplified implementation of xargs.
color: Make a pty if on a platform that supports it
target_concurrency: Target number of partitions to run concurrently
r r Nr r4 )z.batz.cmdzcmd.exei z /c i c �6 �� �| dt j d����S )NF)�check�stderr)�
subprocess�STDOUT)�run_cmd�cmd_fn�kwargss ��r �run_cmd_partitionz xargs.<locals>.run_cmd_partition� s' �� � �
�E�*�*;�*;�
�?E�
�
r )rd �tuple[str, ...]�returnz tuple[int, bytes, bytes | None])r
r r �
normalize_cmd�ExecutableNotFoundError� to_outputr6 r7 �lower�endswith�find_executabler rT r/ r\ �abs)r: rK r] rL rM rf �retcode�stdout�e�cmd_exe�
partitionsrg �threads�
thread_map�results� proc_retcode�proc_out�_re s ` @r �xargsr| � sT �� � #�\� �F��G�
�F�!��)�)�#�.��
�L�L�G� #� ��F�L�L�N� #� #�$4� 5� �/�/� �:�� �S��\�)�C��K�7�$�>� ��3��);�[�I�J�
�$�
� (�
� �#�j�/�#5�6�G� �� �J� � .�
�;��)0� %�L�(�A��<� �3�w�<�/�&�� �h� �F� *1�
!� �F�?���C � 0� 0� !��{�{�}�R�a� � ��!��2
!� �F�?��s) �D � 3D>� D;� D6�0D;�6D;�>E
)ri �int)N)r# z MutableMapping[str, str] | Noneri r} )r: �strri r} )
r: �
Sequence[str]rK r rL r} rM z
int | Noneri ztuple[tuple[str, ...], ...])rZ r} ri zNGenerator[Callable[[Callable[[TArg], TRet], Iterable[TArg]], Iterable[TRet]],])r: rh rK r r] �boolrL r} rM r} rf r ri ztuple[int, bytes])$�
__future__r �concurrent.futuresrW �
contextlibrD r r rb r6 �collections.abcr r r r r �typingr r
�
pre_commitr �pre_commit.utilr r
r r r r' r1 r< � RuntimeErrorr> rT �contextmanagerr\ r| r r r �<module>r� s �� "� � � � � � �
� $� %� $� *� $� � � $� (� (��v����v��� � � �A� �<� � #'� )�
�)� �)� �)� � )�
!� )�X ���� �� �"#�3�5�
5�
�5� �5� � 5�
� 5� �
5� �5� �5r