a
    њhJ                     @  s   d dl mZ d dlZd dlZd dlZddlmZmZ ddgZ	dZ
ejG dd dZd	dd
ddZejG dd dZd	ddddZddd
ddZdS )    )annotationsN   )InvalidProxy
InvalidURI	parse_uriWebSocketURIz:/?#[]@!$&'()*+,;=c                   @  sv   e Zd ZU dZded< ded< ded< ded< ded	< d
Zded< d
Zded< eddddZeddddZ	d
S )r   a  
    WebSocket URI.

    Attributes:
        secure: :obj:`True` for a ``wss`` URI, :obj:`False` for a ``ws`` URI.
        host: Normalized to lower case.
        port: Always set even if it's the default.
        path: May be empty.
        query: May be empty if the URI doesn't include a query component.
        username: Available when the URI contains `User Information`_.
        password: Available when the URI contains `User Information`_.

    .. _User Information: https://datatracker.ietf.org/doc/html/rfc3986#section-3.2.1

    boolsecurestrhostintportpathqueryN
str | Noneusernamepasswordreturnc                 C  s*   | j r| j }nd}| jr&|d| j 7 }|S )N/?)r   r   )selfresource_name r   I/wd/webapps/venvs/v2025_4um/lib/python3.9/site-packages/websockets/uri.pyr   +   s    zWebSocketURI.resource_nametuple[str, str] | Nonec                 C  s(   | j d u rd S | jd usJ | j | jfS Nr   r   r   r   r   r   	user_info5   s    
zWebSocketURI.user_info)
__name__
__module____qualname____doc____annotations__r   r   propertyr   r   r   r   r   r   r      s   
	r
   )urir   c           	      C  s8  t j| }|jdvr t| d|jdu r4t| d|jdkrHt| d|jdk}|j}|jph|rfdnd	}|j}|j	}|j
}|j}|dur|du rt| d
z| d W nx ty"   |d }t jj|td}t jj|td}|dur|dusJ t jj|td}t jj|td}Y n0 t|||||||S )z
    Parse and validate a WebSocket URI.

    Args:
        uri: WebSocket URI.

    Returns:
        Parsed WebSocket URI.

    Raises:
        InvalidURI: If ``uri`` isn't a valid WebSocket URI.

    )wswsszscheme isn't ws or wssNhostname isn't provided z"fragment identifier is meaninglessr(     P   "username provided without passwordasciiidnasafe)urllibparseurlparseschemer   hostnamefragmentr   r   r   r   r   encodeUnicodeEncodeErrordecodequoteDELIMSr   )	r&   parsedr	   r   r   r   r   r   r   r   r   r   r   =   s6    








c                   @  sT   e Zd ZU dZded< ded< ded< dZded	< dZded
< eddddZdS )Proxya  
    Proxy.

    Attributes:
        scheme: ``"socks5h"``, ``"socks5"``, ``"socks4a"``, ``"socks4"``,
            ``"https"``, or ``"http"``.
        host: Normalized to lower case.
        port: Always set even if it's the default.
        username: Available when the proxy address contains `User Information`_.
        password: Available when the proxy address contains `User Information`_.

    .. _User Information: https://datatracker.ietf.org/doc/html/rfc3986#section-3.2.1

    r
   r5   r   r   r   Nr   r   r   r   r   c                 C  s(   | j d u rd S | jd usJ | j | jfS r   r   r   r   r   r   r      s    
zProxy.user_info)	r    r!   r"   r#   r$   r   r   r%   r   r   r   r   r   r>   o   s   
r>   )proxyr   c                 C  s>  t j| }|jdvr*t| d|j d|jdu r>t| d|jdvrRt| d|jdkrft| d	|jdkrzt| d
|j}|j}|j	p|jdkrdnd}|j
}|j}|dur|du rt| dz| d W nZ ty,   |d }|dur(|dusJ t jj|td}t jj|td}Y n0 t|||||S )z
    Parse and validate a proxy.

    Args:
        proxy: proxy.

    Returns:
        Parsed proxy.

    Raises:
        InvalidProxy: If ``proxy`` isn't a valid proxy.

    )Zsocks5hZsocks5Zsocks4aZsocks4httpshttpzscheme z isn't supportedNr)   )r*   r   zpath is meaninglessr*   zquery is meaninglesszfragment is meaninglessr@   r+   r,   r-   r.   r/   r0   )r2   r3   r4   r5   r   r6   r   r   r7   r   r   r   r8   r9   r:   r;   r<   r>   )r?   r=   r5   r   r   r   r   r   r   r   parse_proxy   s6    










rB   r   c                 C  s   t j| j d| j rdS t j }| jr8g d}ng d}|D ]@}||}|durD|dkr||dr|d|dd  }|  S qDdS )	zU
    Return the proxy to use for connecting to the given WebSocket URI, if any.

    :N)r(   socksr@   )r'   rD   r@   rA   rD   zhttp://z
socks5h://   )	r2   requestproxy_bypassr   r   
getproxiesr	   get
startswith)r&   proxiesschemesr5   r?   r   r   r   	get_proxy   s    



rM   )
__future__r   dataclassesurllib.parser2   urllib.request
exceptionsr   r   __all__r<   	dataclassr   r   r>   rB   rM   r   r   r   r   <module>   s   +22