o
    +ca                  	   @   s  U 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	 d dlm
Z
 d dlmZ d dlmZ d d	lmZ d d
lmZ d dlmZ d dlmZ d dlmZ d dlmZ d dlmZ ddlmZ ejdksrJ d dlZd dlmZ d dlmZ ee
ZejZejZejZejZej Z ej!Z!eedejfZ"eeeeee	dej#fZ$eeedejfZ%edZ&edZ'edZ(d Z)dZ*d Z+dZ,dZ-d Z.dZ/dZ0dZ1dZ2zd d lm3Z3 W n e4y   dZ5Y nw G d!d" d"eZ6e3j7Z7e3j8Z8d?d$d%Z5G d&d' d'e j9Z:G d(d) d)e:Z;G d*d+ d+e:Z<G d,d- d-Z=d.ej>d/ej?fd0d1Z@d.ej>d/ej?fd2d3ZAd.ej>d/ej?fd4d5ZBe@eAeBd6ZCejDeEejFej>gej?f f eGd7< d8ej?d/eHfd9d:ZId8ej?d;ejJeK d<ejJeK d/ejJej? fd=d>ZLdS )@    N)byref)c_char)c_char_p)c_int)	c_ssize_t)c_ulong)c_void_p)POINTER)	py_object)	Structure)DWORD)HANDLE)LPCWSTR)LPWSTR   )_NonClosingTextIOWrapperwin32)windll)WINFUNCTYPEGetCommandLineWCommandLineToArgvW	LocalFreeiii   i        i  )	pythonapic                   @   sR   e Zd Zdefdefdefdefdefdefdefdefd	efd
efdefgZ	dS )	Py_bufferbufobjlenitemsizereadonlyndimformatshapestrides
suboffsetsZinternalN)
__name__
__module____qualname__r   r
   r   r   r   	c_ssize_pZ_fields_ r+   r+   3/usr/lib/python3/dist-packages/click/_winconsole.pyr   K   s    r   Fc              	   C   sZ   t  }|rtnt}tt| t|| zt|j }||j	W t
t| S t
t| w N)r   PyBUF_WRITABLEPyBUF_SIMPLEPyObject_GetBufferr
   r   r   r   Zfrom_addressr   PyBuffer_Release)r   writabler   flagsZbuffer_typer+   r+   r,   
get_buffer]   s   
r4   c                       s$   e Zd Zdd Z fddZ  ZS )_WindowsConsoleRawIOBasec                 C   s
   || _ d S r-   )handle)selfr6   r+   r+   r,   __init__j      
z!_WindowsConsoleRawIOBase.__init__c                    s   t    dS NT)superisattyr7   	__class__r+   r,   r<   m   s   
z_WindowsConsoleRawIOBase.isatty)r'   r(   r)   r8   r<   __classcell__r+   r+   r>   r,   r5   i   s    r5   c                   @   s   e Zd Zdd Zdd ZdS )_WindowsConsoleReaderc                 C      dS r:   r+   r=   r+   r+   r,   readables      z_WindowsConsoleReader.readablec                 C   s   t |}|sdS |d rtdt|dd}|d }t }tt| j||t|d }t t	kr4t
d |s>tdt  |d tkrFdS d|j S )Nr   r   z>cannot read odd number of bytes from UTF-16-LE encoded consoleT)r2   g?zWindows error: )r   
ValueErrorr4   r   ReadConsoleWr   r6   r   GetLastErrorERROR_OPERATION_ABORTEDtimesleepOSErrorEOFvalue)r7   bZbytes_to_be_readbufferZcode_units_to_be_readZcode_units_readZrvr+   r+   r,   readintov   s0   


z_WindowsConsoleReader.readintoN)r'   r(   r)   rC   rP   r+   r+   r+   r,   rA   r   s    rA   c                   @   s(   e Zd Zdd Zedd Zdd ZdS )_WindowsConsoleWriterc                 C   rB   r:   r+   r=   r+   r+   r,   r2      rD   z_WindowsConsoleWriter.writablec                 C   s"   | t krdS | tkrdS d|  S )NERROR_SUCCESSERROR_NOT_ENOUGH_MEMORYzWindows error )rR   rS   )errnor+   r+   r,   _get_error_message   s
   
z(_WindowsConsoleWriter._get_error_messagec                 C   sl   t |}t|}t|td }t }tt| j||t|d  d|j	 }|dkr4|dkr4t
| t |S )Nr   r   )r   r4   minMAX_BYTES_WRITTENr   WriteConsoleWr   r6   r   rM   rK   rU   rG   )r7   rN   Zbytes_to_be_writtenr   Zcode_units_to_be_writtenZcode_units_writtenZbytes_writtenr+   r+   r,   write   s   
z_WindowsConsoleWriter.writeN)r'   r(   r)   r2   staticmethodrU   rY   r+   r+   r+   r,   rQ      s
    
rQ   c                   @   s   e Zd ZdejdejddfddZedefddZ	d	ej
defd
dZdejej
 ddfddZdedejfddZdefddZdd ZdS )ConsoleStreamtext_streambyte_streamreturnNc                 C   s   || _ || _d S r-   )_text_streamrO   )r7   r\   r]   r+   r+   r,   r8      s   
zConsoleStream.__init__c                 C   s   | j jS r-   )rO   namer=   r+   r+   r,   r`      s   zConsoleStream.namexc                 C   sB   t |tr| j|S z|   W n	 ty   Y nw | j|S r-   )
isinstancestrr_   rY   flush	ExceptionrO   )r7   ra   r+   r+   r,   rY      s   
zConsoleStream.writelinesc                 C   s   |D ]}|  | qd S r-   )rY   )r7   rf   liner+   r+   r,   
writelines   s   zConsoleStream.writelinesr`   c                 C   s   t | j|S r-   )getattrr_   )r7   r`   r+   r+   r,   __getattr__   s   zConsoleStream.__getattr__c                 C   s
   | j  S r-   )rO   r<   r=   r+   r+   r,   r<      r9   zConsoleStream.isattyc                 C   s   d| j d| jdS )Nz<ConsoleStream name=z
 encoding=>)r`   encodingr=   r+   r+   r,   __repr__   s   zConsoleStream.__repr__)r'   r(   r)   tTextIOBinaryIOr8   propertyrc   r`   ZAnyStrintrY   Iterablerh   ZAnyrj   boolr<   rm   r+   r+   r+   r,   r[      s    	r[   buffer_streamr^   c                 C   .   t tttdddd}ttjt|| S N	utf-16-lestrictT)line_buffering)	r   ioBufferedReaderrA   STDIN_HANDLErn   castro   r[   ru   r\   r+   r+   r,   _get_text_stdin      r   c                 C   rv   rw   )	r   r{   BufferedWriterrQ   STDOUT_HANDLErn   r~   ro   r[   r   r+   r+   r,   _get_text_stdout   r   r   c                 C   rv   rw   )	r   r{   r   rQ   STDERR_HANDLErn   r~   ro   r[   r   r+   r+   r,   _get_text_stderr   r   r   )r   r   r   _stream_factoriesfc              	   C   sT   t | dsdS z|  }W n ttjfy   Y dS w t|}tt|t	t
 S )NfilenoF)hasattrr   rK   r{   UnsupportedOperationmsvcrtZget_osfhandlert   GetConsoleModer   r   )r   r   r6   r+   r+   r,   _is_console   s   

r   rl   errorsc                 C   sj   t d ur+|dv r-|dv r/t| r1t|  }|d ur3t| dd }|d u r'd S ||S d S d S d S d S d S )N>   Nrx   >   Nry   rO   )r4   r   r   getr   ri   )r   rl   r   funcrN   r+   r+   r,   _get_windows_console_stream  s    r   )F)Mr{   sysrI   typingrn   Zctypesr   r   r   r   r   r   r   r	   r
   r   Zctypes.wintypesr   r   r   r   Z_compatr   platformr   r   r   r*   Zkernel32ZGetStdHandlerF   rX   r   rG   r   Zshell32r   r   r}   r   r   r/   r.   rR   rS   rH   ZSTDIN_FILENOZSTDOUT_FILENOZSTDERR_FILENOrL   rW   r   ImportErrorr4   r   r0   r1   	RawIOBaser5   rA   rQ   r[   rp   ro   r   r   r   r   Mappingrr   Callable__annotations__rt   r   ZOptionalrc   r   r+   r+   r+   r,   <module>   s   

	#  

(
