vj;>ddlmZddlZddlZddlZddlZddlmZmZm Z m Z ddl m Z m Z ddlmZmZmZmZmZmZgdZe dZd*d Zd+d Zej0d Zd,d Zej0d Zd-dZej0dZej0dZd-dZej0dZ ej0dZ!d.dZ" d/dZ# d0dZ$d1dZ%ej0dZ& d2dZ'd3dZ( d4dZ) d5dZ*d6dZ+e+Z, d7dZ-d8dZ.e.Z/ d9d Z0d:d!Z1e1Z2d;d"Z3e3Z4dd'Z9d?d(Z:d@d)Z;y)A) annotationsN)CallableSequenceTypeVarcast)InvalidHeaderFormatInvalidHeaderValue)ConnectionOptionExtensionHeader ExtensionNameExtensionParameter SubprotocolUpgradeProtocol) build_hostparse_connection parse_upgradeparse_extensionbuild_extensionparse_subprotocolbuild_subprotocolvalidate_subprotocolsbuild_www_authenticate_basicparse_authorization_basicbuild_authorization_basicTc tj|}|jdk(rd|d} ||rdndk7r|d|}|S#t$rYwxYw)z# Build a ``Host`` header. []iP:) ipaddress ip_addressversion ValueError)hostportsecureaddresss c/workspace/moltex-experiments/.pixi/envs/default/lib/python3.12/site-packages/websockets/headers.pyrr&sg&&t, ??a tfA;D v2&q K    s? A  A c,|t|k(rdS||S)z Return the next character from ``header`` at the given position. Return :obj:`None` at the end of ``header``. We never need to peek more than one character ahead. N)len)headerposs r+ peek_aheadr0Bs#f+%466#;6z[\t ]*cVtj||}|J|jS)z Parse optional whitespace from ``header`` at the given position. Return the new position. The whitespace itself isn't returned because it isn't significant. )_OWS_rematchend)r.r/r4s r+ parse_OWSr6Qs, MM&# &E    99;r1z[-!#$%&\'*+.^_`|~0-9a-zA-Z]+ctj||}|t|d|||j|j fS)z Parse a token from ``header`` at the given position. Return the token value and the new position. Raises: InvalidHeaderFormat: On invalid inputs. zexpected token) _token_rer4r groupr5r.r/ header_namer4s r+ parse_tokenr<csB OOFC (E }!+/?MM ;;=%))+ %%r1zC"(?:[\x09\x20-\x21\x23-\x5b\x5d-\x7e]|\\[\x09\x20-\x7e\x80-\xff])*"z\\([\x09\x20-\x7e\x80-\xff])ctj||}|t|d||tj d|j dd|j fS)z Parse a quoted string from ``header`` at the given position. Return the unquoted value and the new position. Raises: InvalidHeaderFormat: On invalid inputs. zexpected quoted stringz\1r)_quoted_string_rer4r _unquote_resubr9r5r:s r+parse_quoted_stringrB{sW  # #FC 0E }!+/GQTUU ??5%++-""5 6 CCr1z[\x09\x20-\x7e\x80-\xff]*z ([\x22\x5c])c~tj|}| tddtj d|zdzS)zh Format ``value`` as a quoted string. This is the reverse of :func:`parse_quoted_string`. z-invalid characters for quoted-string encoding"z\\\1) _quotable_re fullmatchr& _quote_rerA)valuer4s r+build_quoted_stringrIs@  " "5 )E }HII w. . 44r1ct||dk(rt||dz}t||dk(rg} ||||\}}|j|t||}|t|k(rnkt||dk(rt||dz}nt |d||t||dk(rt||dz}t||dk(r|t|k(rn|t|k(sJ|S)a Parse a comma-separated list from ``header`` at the given position. This is appropriate for parsing values with the following grammar: 1#item ``parse_item`` parses one item. ``header`` is assumed not to start or end with whitespace. (This function is designed for parsing an entire header value and :func:`~websockets.http.read_headers` strips whitespace from values.) Return a list of items. Raises: InvalidHeaderFormat: On invalid inputs. ,rzexpected comma)r0r6appendr-r ) parse_itemr.r/r;itemsitems r+ parse_listrPs > VS !S (a( VS !S ( E vsK8 c T$ #f+   fc "c )FC!G,C%k3CVSQ Q%,FC!G,C%, #f+  - 4 #f+   Lr1cFt|||\}}tt||fS)z Parse a Connection option from ``header`` at the given position. Return the protocol value and the new position. Raises: InvalidHeaderFormat: On invalid inputs. )r<rr r.r/r;rOs r+parse_connection_optionrSs)FC5ID#  $ ' ,,r1c&tt|ddS)z Parse a ``Connection`` header. Return a list of HTTP connection options. Args header: value of the ``Connection`` header. Raises: InvalidHeaderFormat: On invalid inputs. r Connection)rPrSr.s r+rrs -vq, GGr1z>[-!#$%&\'*+.^_`|~0-9a-zA-Z]+(?:/[-!#$%&\'*+.^_`|~0-9a-zA-Z]+)?ctj||}|t|d||tt|j |j fS)z Parse an Upgrade protocol from ``header`` at the given position. Return the protocol value and the new position. Raises: InvalidHeaderFormat: On invalid inputs. zexpected protocol) _protocol_rer4r rrr9r5r:s r+parse_upgrade_protocolrYsK   vs +E }!+/BFCPP  / <JE3$ %=# r1ct|||\}}t||}g}t||dk(r@t||dz}t|||\}}|j |t||dk(r@t t ||f|fS)a Parse an extension definition from ``header`` at the given position. Return an ``(extension name, parameters)`` pair, where ``parameters`` is a list of ``(name, value)`` pairs, and the new position. Raises: InvalidHeaderFormat: On invalid inputs. ;r)r<r6r0r`rLrr )r.r/r;r^ parameters parameters r+parse_extension_itemreKsFC5ID# FC CJ VS !S (a(3FCM 3)$ VS !S (  %z 2C 77r1c&tt|ddS)a Parse a ``Sec-WebSocket-Extensions`` header. Return a list of WebSocket extensions and their parameters in this format:: [ ( 'extension name', [ ('parameter name', 'parameter value'), .... ] ), ... ] Parameter values are :obj:`None` when no value is provided. Raises: InvalidHeaderFormat: On invalid inputs. rzSec-WebSocket-Extensions)rPrerVs r+rrds. *FA7Q RRr1c djtt|g|Dcgc]\}}||n|d|c}}zScc}}w)zc Build an extension definition. This is the reverse of :func:`parse_extension_item`. z; r])joinrstr)r^rcrHs r+build_extension_itemrjsb 99 c4 *  * eMD$q'8 8)    sA c2djd|DS)zl Build a ``Sec-WebSocket-Extensions`` header. This is the reverse of :func:`parse_extension`. , c3:K|]\}}t||yw)N)rj).0r^rcs r+ z"build_extension..s!GQ3C4T:.zsrh) extensionss r+rrs" 99GQ r1cFt|||\}}tt||fS)z Parse a subprotocol from ``header`` at the given position. Return the subprotocol value and the new position. Raises: InvalidHeaderFormat: On invalid inputs. )r<rrrRs r+parse_subprotocol_itemrss(FC5ID#  T "C ''r1c&tt|ddS)z Parse a ``Sec-WebSocket-Protocol`` header. Return a list of WebSocket subprotocols. Raises: InvalidHeaderFormat: On invalid inputs. rzSec-WebSocket-Protocol)rPrsrVs r+rrs ,fa9Q RRr1c$dj|S)zl Build a ``Sec-WebSocket-Protocol`` header. This is the reverse of :func:`parse_subprotocol`. rlrp) subprotocolss r+rrs 99\ ""r1ct|ts tdt|tr td|D]%}tj |rt d|y)zT Validate that ``subprotocols`` is suitable for :func:`build_subprotocol`. zsubprotocols must be a listz&subprotocols must be a list, not a strzinvalid subprotocol: N) isinstancer TypeErrorrir8rFr&)rv subprotocols r+rrs[ lH -566,$@AA# "";/4[MBC C$r1c>t|}td}d|d|S)z Build a ``WWW-Authenticate`` header for HTTP Basic Auth. Args: realm: Identifier of the protection space. zUTF-8z Basic realm=z , charset=)rI)realmcharsets r+rrs+  &E!'*G % 7) 44r1z[A-Za-z0-9-._~+/]+=*ctj||}|t|d|||j|j fS)z Parse a token68 from ``header`` at the given position. Return the token value and the new position. Raises: InvalidHeaderFormat: On invalid inputs. zexpected token68) _token68_rer4r r9r5r:s r+ parse_token68rsD   fc *E }!+/A63OO ;;=%))+ %%r1c<|t|krt|d||y)z8 Check that parsing reached the end of header. z trailing dataN)r-r )r.r/r;s r+ parse_endrs&  S[!+LLr1ct|dd\}}|jdk7rtdd|t||dk7rt dd|||dz }t ||d\}}t ||d tj|jj} |jd d\}}||fS#tj$rtddd wxYw#t$rtdd d wxYw) a! Parse an ``Authorization`` header for HTTP Basic Auth. Return a ``(username, password)`` tuple. Args: header: Value of the ``Authorization`` header. Raises: InvalidHeaderFormat: On invalid inputs. InvalidHeaderValue: On unsupported inputs. r Authorizationbasiczunsupported scheme:  zexpected space after schemerz#expected base64-encoded credentialsNr"z&expected username:password credentials)r<lowerr r0r rrbase64 b64decodeencodedecodebinasciiErrorsplitr&)r.schemer/basic_credentials user_passusernamepasswords r+rrs5 fa9KFC ||~  "6( +  &##%! :FC  1HC*63Hs fc?+$$%6%=%=%?@GGI &__S!4( X  >>  1     4  s31B>%C">!C"C9cd|vsJ|d|}tj|jj}d|zS)z Build an ``Authorization`` header for HTTP Basic Auth. This is the reverse of :func:`parse_authorization_basic`. r"zBasic )r b64encoderr)rrrrs r+rr8sQ h  *AhZ(I(()9)9);<CCE ' ''r1)r'rir(intr)boolreturnri)r.rir/rrz str | None)r.rir/rrr)r.rir/rr;rirztuple[str, int])rHrirri) rMz(Callable[[str, int, str], tuple[T, int]]r.rir/rr;rirzlist[T])r.rir/rr;rirztuple[ConnectionOption, int])r.rirzlist[ConnectionOption])r.rir/rr;rirztuple[UpgradeProtocol, int])r.rirzlist[UpgradeProtocol])r.rir/rr;rirztuple[ExtensionParameter, int])r.rir/rr;rirztuple[ExtensionHeader, int])r.rirzlist[ExtensionHeader])r^r rczlist[ExtensionParameter]rri)rqzSequence[ExtensionHeader]rri)r.rir/rr;rirztuple[Subprotocol, int])r.rirzlist[Subprotocol])rvSequence[Subprotocol]rri)rvrrNone)r|rirri)r.rir/rr;rirr)r.rirztuple[str, str])rrirrirri)< __future__rrrr#retypingrrrr exceptionsr r r r r rrr__all__rrr0compiler3r6r8r<r?r@rBrErGrIrPrSrrXrYrr`rerparse_extension_listrjrbuild_extension_listrsrparse_subprotocol_listrbuild_subprotocol_listrrrrrrrr1r+rsa"  44?  CL8 7 "**Y   BJJ6 7  & BJJJ bjj89  D rzz67  BJJ '  5?8? ? ? ?  ?D -  - -(+ -! - H rzzE = ==(+= =$ D ! !!(+!#!H8 88(+8 82S4' %=& ' (  ( ((+ ( ( S+#+ D 5bjj01  & M-` (r1