HEX

File: //lib/python3/dist-packages/mercurial/__pycache__/encoding.cpython-310.pyc
o

�]Lb�[�@stddlmZmZddlZddlZddlZddlZddlmZddl	m
Z
mZmZddlm
ZejrXddlmZmZmZmZmZmZmZeeeeeefD]ZesQJ�qKedd	d
�Ze�d�Z
e
jZe
jZe
jZe
jZej Z!ej"rqe#Z$dd
�d�%�D�Z&e'dd�e&D��s�J�dd�Z(ej"p�ej)Z*ej"s�ej+Z+ne*r�ej,Z+n
dd�ej+�-�D�Z+ddd�Z.ej/r�ej"s�de.d<ze+�0d�Z1e1s�e�2��3d�p�dZ1e.�0e1e1�Z1Wnej4y�dZ1Ynwe+�0dd�Z5dZ6Gdd	�d	e7�Z8Gdd �d e7�Z9d!d"�Z:d#d$�Z;d%d&�Z<d'd(�Z=d)d*�Z>ej"�re<Z?e=Z@e>ZAnd+d,�Z?d-d.�Z@ejBZAd/d0�ZCd1d2�ZDd3d4�ZEe*�sXej/�r?ej"�r?Gd5d6�d6eF�ZGeG�Z+ej+�-�D]\ZHZIe:eI�3d7��e+e:eH�3d7��<�qDe�Jd8�ZKej"�rnej/�rjd9d:�ZLnejMZLnejLZLe!e+�0d;d<�d=k�r~d>�pd?�ZNd@dA�ZOdBdC�ZPdDdE�ZQdWdHdI�ZRGdJdK�dKeS�ZTdXdLdM�ZUej"�r�dNZVndOZVgdP�ZWdQdR�ZXdSdT�ZYdUdV�ZZdS)Y�)�absolute_import�print_functionN�)�getattr)�error�policy�pycompat)�
charencode)�Any�Callable�List�Text�Type�TypeVar�Union�
_Tlocalstr�localstr)�boundr	cCs g|]}tt|d���d��qS)��utf-8)�unichr�int�encode)�.0�x�r�4/usr/lib/python3/dist-packages/mercurial/encoding.py�
<listcomp>8s��rsO200c 200d 200e 200f 202a 202b 202c 202d 202e 206a 206b 206c 206d 206e 206f feffccs�|]}|�d�VqdS))����N)�
startswith)r�irrr�	<genexpr>>s�r"cCs*d|vsd|vrtD]}|�|d�}q
|S)u�Remove codepoints ignored by HFS+ from s.

    >>> hfsignoreclean(u'.h‌g'.encode('utf-8'))
    '.hg'
    >>> hfsignoreclean(u'.hg'.encode('utf-8'))
    '.hg'
    rr�)�_ignore�replace)�s�crrr�hfsignorecleanAs	r(cCs"i|]
\}}|�d�|�d��qS)r)r)r�k�vrrr�
<dictcomp>Zs��r+sascii)s646sANSI_X3.4-1968sutf-8scp65001s
HGENCODING�asciisHGENCODINGMODEsstricts
ISO-8859-1cs6eZdZdZdd�Zejr�fdd�Zdd�Z�Z	S)rzdThis class allows strings that are unmodified to be
    round-tripped to the local encoding and backcCst�||�}||_|S�N)�bytes�__new__�_utf8)�cls�u�lr&rrrr/xszlocalstr.__new__cstt|��|�||_dSr-)�superr�__init__r0)�selfr2r3��	__class__rrr5s
zlocalstr.__init__cCs
t|j�Sr-)�hashr0)r6rrr�__hash__�s
zlocalstr.__hash__)
�__name__�
__module__�__qualname__�__doc__r/r�
TYPE_CHECKINGr5r:�
__classcell__rrr7rrtsc@seZdZdZdS)�safelocalstraDTagged string denoting it was previously an internal UTF-8 string,
    and can be converted back to UTF-8 losslessly

    >>> assert safelocalstr(b'\xc3') == b'\xc3'
    >>> assert b'\xc3' == safelocalstr(b'\xc3')
    >>> assert b'\xc3' in {safelocalstr(b'\xc3'): 0}
    >>> assert safelocalstr(b'\xc3') in {b'\xc3': 0}
    N)r;r<r=r>rrrrrA�srAcCs.t|�r|Szzz+|�d�}tdkr|WWS|�tt�d�}||�tt��kr,t|�WWSt||�WWSty�z,|�tt��}|�tt�d�}||�tt��krZt|�WYWSt|�d�|�WYWSty|�dd�}|�tt�d�YYWSwwt	y�}z
t
jt�
|�dd��d}~ww)a�
    Convert a string from internal UTF-8 to local encoding

    All internal strings should be UTF-8 but some repos before the
    implementation of locale support may contain latin1 or possibly
    other character sets. We attempt to decode everything strictly
    using UTF-8, then Latin-1, and failing that, we use UTF-8 and
    replace unknown characters.

    The localstr class is used to cache the known UTF-8 encoding of
    strings next to their local representation to allow lossless
    round-trip conversion back to UTF-8.

    >>> u = b'foo: \xc3\xa4' # utf-8
    >>> l = tolocal(u)
    >>> l
    'foo: ?'
    >>> fromlocal(l)
    'foo: \xc3\xa4'
    >>> u2 = b'foo: \xc3\xa1'
    >>> d = { l: 1, tolocal(u2): 2 }
    >>> len(d) # no collision
    2
    >>> b'foo: ?' in d
    False
    >>> l1 = b'foo: \xe4' # historical latin1 fallback
    >>> l = tolocal(l1)
    >>> l
    'foo: ?'
    >>> fromlocal(l) # magically in utf-8
    'foo: \xc3\xa4'
    zUTF-8sUTF-8r%r�!please check your locale settings��hintN)�
isasciistr�decode�encodingr�_sysstrrAr�UnicodeDecodeError�fallbackencoding�LookupErrorr�Abortr�bytestr)r&r2�rr)rrr�tolocal�s:#
��
���rOc
Cs�t|t�r|jSt|�r|Sz|�tt�tt��}|�d�WSt	yE}z|t
d|jd�|jd�}t�
d|t�|�f��d}~wtyZ}z
tj
t�|�dd��d}~ww)a�
    Convert a string from the local character encoding to UTF-8

    We attempt to decode strings using the encoding mode set by
    HGENCODINGMODE, which defaults to 'strict'. In this mode, unknown
    characters will cause an error message. Other modes include
    'replace', which replaces unknown characters with a special
    Unicode character, and 'ignore', which drops the character.
    rr�
sdecoding near '%s': %s!NrBrC)�
isinstancerr0rErFrHrG�encodingmoderrI�max�startrrLrrMrK)r&r2�inst�subr)rrr�	fromlocal�s&

��
���rWcCst|�d��S)z;Convert a unicode string to a byte string of local encodingr)rOr)r2rrr�
unitolocal��rXcCst|��d�S)z;Convert a byte string of local encoding to a unicode stringr)rWrF�r&rrr�unifromlocal�rYr[cs�fdd�}|S)z^Create a proxy method that forwards __unicode__() and __str__() of
    Python 3 to __bytes__()cst�|��Sr-)r[)�obj��	bytesfuncrr�unifunc	szunimethod.<locals>.unifuncr)r^r_rr]r�	unimethodsr`cC�|Sr-rrZrrr�
strtolocal�rbcCrar-rrZrrr�strfromlocalrcrdc
Cs�zt|�WStyYnwz)t|t�r|j�d�}n
|�tt�tt��}|�	�}||kr1|WS|�
tt��WStyE|�	�YStyZ}z
t
jt�|�dd��d}~ww)�9best-effort encoding-aware case-folding of local string srrBrCN)�
asciilowerrIrQrr0rFrHrGrR�lowerr�UnicodeErrorrKrrLrrM)r&r2�lur)rrrrg#s*
�

���rgcCs&zt|�WStyt|�YSw)re)�
asciiupperrI�
upperfallbackrZrrr�upper<s

�rlc
Cs�z)t|t�r
|j�d�}n
|�tt�tt��}|��}||kr"|WS|�tt��WSt	y6|��YSt
yK}z
tjt
�|�dd��d}~ww)NrrBrC)rQrr0rFrHrGrRrlrrhrKrrLrrM)r&r2�uur)rrrrkEs 

���rkcs"eZdZdZd�fdd�	Z�ZS)�WindowsEnvironzE`os.environ` normalizes environment variables to uppercase on windowsNcst��t|�|�Sr-)r4�getrl)r6�key�defaultr7rrroaszWindowsEnviron.getr-)r;r<r=r>ror@rrr7rrn^srnrs^[a-z]:cCsFt��}tj�|�}t|�}t�|�r!|dd���|dd�}|S)Nrr)�os�getcwd�path�realpathrb�DRIVE_RE�matchrl)�cwdrrrrs~s
rssHGENCODINGAMBIGUOUSsnarrowswidesWFAsWFcCst|�tt�d��S)zCFind the column width of a string for display in the local encodingr%)�	ucolwidthrFrHrGrZrrr�colwidth�srzcs2ttdd���durt�fdd�|D��St|�S)z5Find the column width of a Unicode string for display�east_asian_widthNcs g|]}�|�tvrdp
d�qS)�r)�_wide)rr'��eawrrr�s zucolwidth.<locals>.<listcomp>)r�unicodedata�sum�len)�drr~rry�srycCsBt�||t|��D]}|||�}t|�|kr|Sq
td��)zOUse colwidth to find a c-column substring of s starting at byte
    index startzsubstring not found)r�xranger�rz�
ValueError)r&rTr'r�trrr�getcols�s�r�r#Fc	CsJz	|�tt��}Wn>tyGt|�|kr|YS|t|�8}|dkr0|d|t|��YS|r=|||d�YS|d|�|YSwt|�|krP|S|t|�8}|dkrd|d|t|��St|�}|rn|��d}t|�D]\}}|t|�7}||kr�nqt|d|�}|r�|��d�	|��
tt��}|r�||S||S)u�Trim string 's' to at most 'width' columns (including 'ellipsis').

    If 'leftside' is True, left side of string 's' is trimmed.
    'ellipsis' is always placed at trimmed side.

    >>> from .node import bin
    >>> def bprint(s):
    ...     print(pycompat.sysstr(s))
    >>> ellipsis = b'+++'
    >>> from . import encoding
    >>> encoding.encoding = b'utf-8'
    >>> t = b'1234567890'
    >>> bprint(trim(t, 12, ellipsis=ellipsis))
    1234567890
    >>> bprint(trim(t, 10, ellipsis=ellipsis))
    1234567890
    >>> bprint(trim(t, 8, ellipsis=ellipsis))
    12345+++
    >>> bprint(trim(t, 8, ellipsis=ellipsis, leftside=True))
    +++67890
    >>> bprint(trim(t, 8))
    12345678
    >>> bprint(trim(t, 8, leftside=True))
    34567890
    >>> bprint(trim(t, 3, ellipsis=ellipsis))
    +++
    >>> bprint(trim(t, 1, ellipsis=ellipsis))
    +
    >>> u = u'あいうえお' # 2 x 5 = 10 columns
    >>> t = u.encode(pycompat.sysstr(encoding.encoding))
    >>> bprint(trim(t, 12, ellipsis=ellipsis))
    ããããã
    >>> bprint(trim(t, 10, ellipsis=ellipsis))
    ããããã
    >>> bprint(trim(t, 8, ellipsis=ellipsis))
    ãã+++
    >>> bprint(trim(t, 8, ellipsis=ellipsis, leftside=True))
    +++ãã
    >>> bprint(trim(t, 5))
    ãã
    >>> bprint(trim(t, 5, leftside=True))
    ãã
    >>> bprint(trim(t, 4, ellipsis=ellipsis))
    +++
    >>> bprint(trim(t, 4, ellipsis=ellipsis, leftside=True))
    +++
    >>> t = bin(b'112233445566778899aa') # invalid byte sequence
    >>> bprint(trim(t, 12, ellipsis=ellipsis))
    "3DUfwª
    >>> bprint(trim(t, 10, ellipsis=ellipsis))
    "3DUfwª
    >>> bprint(trim(t, 8, ellipsis=ellipsis))
    "3DU+++
    >>> bprint(trim(t, 8, ellipsis=ellipsis, leftside=True))
    +++fwª
    >>> bprint(trim(t, 8))
    "3DUfw
    >>> bprint(trim(t, 8, leftside=True))
    3DUfwª
    >>> bprint(trim(t, 3, ellipsis=ellipsis))
    +++
    >>> bprint(trim(t, 1, ellipsis=ellipsis))
    +
    rN�)rFrHrGrIr�ry�list�reverse�	enumerate�joinr)	r&�width�ellipsis�leftsider2�chars�width_so_farr!r'rrr�trim�sBB�
�r�c@seZdZdZdZdZdZdS)�
normcasespecsaxwhat a platform's normcase does to ASCII strings

    This is specified per platform, and should be consistent with what normcase
    on that platform actually does.

    lower: normcase lowercases ASCII strings
    upper: normcase uppercases ASCII strings
    other: the fallback function should always be called

    This should be kept in sync with normcase_spec in util.h.���rrN)r;r<r=r>rgrl�otherrrrrr�s
r�cCs4t|�}zt||�WStyYnwt�||�S)a�returns a string suitable for JSON

    JSON is problematic for us because it doesn't support non-Unicode
    bytes. To deal with this, we take the following approach:

    - localstr/safelocalstr objects are converted back to UTF-8
    - valid UTF-8/ASCII strings are passed as-is
    - other strings are converted to UTF-8b surrogate encoding
    - apply JSON-specified string escaping

    (escapes are doubled in these tests)

    >>> jsonescape(b'this is a test')
    'this is a test'
    >>> jsonescape(b'escape characters: \0 \x0b \x7f')
    'escape characters: \\u0000 \\u000b \\u007f'
    >>> jsonescape(b'escape characters: \b \t \n \f \r \" \\')
    'escape characters: \\b \\t \\n \\f \\r \\" \\\\'
    >>> jsonescape(b'a weird byte: \xdd')
    'a weird byte: \xed\xb3\x9d'
    >>> jsonescape(b'utf-8: caf\xc3\xa9')
    'utf-8: caf\xc3\xa9'
    >>> jsonescape(b'')
    ''

    If paranoid, non-ascii and common troublesome characters are also escaped.
    This is suitable for web output.

    >>> s = b'escape characters: \0 \x0b \x7f'
    >>> assert jsonescape(s) == jsonescape(s, paranoid=True)
    >>> s = b'escape characters: \b \t \n \f \r \" \\'
    >>> assert jsonescape(s) == jsonescape(s, paranoid=True)
    >>> jsonescape(b'escape boundary: \x7e \x7f \xc2\x80', paranoid=True)
    'escape boundary: ~ \\u007f \\u0080'
    >>> jsonescape(b'a weird byte: \xdd', paranoid=True)
    'a weird byte: \\udcdd'
    >>> jsonescape(b'utf-8: caf\xc3\xa9', paranoid=True)
    'utf-8: caf\\u00e9'
    >>> jsonescape(b'non-BMP: \xf0\x9d\x84\x9e', paranoid=True)
    'non-BMP: \\ud834\\udd1e'
    >>> jsonescape(b'<foo@example.org>', paranoid=True)
    '\\u003cfoo@example.org\\u003e'
    )�toutf8b�_jsonescapeu8fastr��charencodepure�jsonescapeu8fallback)r&�paranoid�u8charsrrr�
jsonescape$s.�r��
surrogatepass�strict)rrrrrrrrrrrrr|r|��cCsPtt|||d��d?}|s|||d�S||||�}|�dt�|S)z�get the next full utf-8 character in the given string, starting at pos

    Raises a UnicodeError if the given location does not start a valid
    utf-8 character.
    rr�r)�_utf8len�ordrF�_utf8strict)r&�posr3r'rrr�getutf8chards	r�cCs"t|t�r|jSt|t�rt|�St|�r|Sd|vr.z	|�dt�|WSty-Ynwt	�
|�}d}d}t|�}||kr�z,t||�}d|krOdkrdnnt
dt||���dt�}|d7}n|t|�7}Wnty�t
dt||���dt�}|d7}Ynw||7}||ks?|S)	aBconvert a local, possibly-binary string into UTF-8b

    This is intended as a generic method to preserve data when working
    with schemes like JSON and XML that have no provision for
    arbitrary byte strings. As Mercurial often doesn't know
    what encoding data is in, we use so-called UTF-8b.

    If a string is already valid UTF-8 (or ASCII), it passes unmodified.
    Otherwise, unsupported bytes are mapped to UTF-16 surrogate range,
    uDC00-uDCFF.

    Principles of operation:

    - ASCII and UTF-8 data successfully round-trips and is understood
      by Unicode-oriented clients
    - filenames and file contents in arbitrary other encodings can have
      be round-tripped or recovered by clueful clients
    - local strings that have a cached known UTF-8 encoding (aka
      localstr) get sent as UTF-8 so Unicode-oriented clients get the
      Unicode data they want
    - non-lossy local strings (aka safelocalstr) get sent as UTF-8 as well
    - because we must preserve UTF-8 bytestring in places such as
      filenames, metadata can't be roundtripped without help

    (Note: "UTF-8b" often refers to decoding a mix of valid UTF-8 and
    arbitrary bytes into an internal Unicode format that can be
    re-encoded back into the original. Here we are exposing the
    internal surrogate encoding as a UTF-8 string.)
    ��rr#r����i�r)rQrr0rArWrErFr�rIrrMr�r�rr�r�r&rNr�r3r'rrrr�ws@
 
�


���
r�cCs�t|�r|Sd|vr|St�|�}d}d}t|�}||krIt||�}|t|�7}d|kr2dkrAnn
t�t|�dt��d@�}||7}||ks|S)aWGiven a UTF-8b string, return a local, possibly-binary string.

    return the original binary string. This
    is a round-trip process for strings like filenames, but metadata
    that's was passed through tolocal will remain in UTF-8.

    >>> roundtrip = lambda x: fromutf8b(toutf8b(x)) == x
    >>> m = b"\xc3\xa9\x99abcd"
    >>> toutf8b(m)
    '\xc3\xa9\xed\xb2\x99abcd'
    >>> roundtrip(m)
    True
    >>> roundtrip(b"\xc2\xc2\x80")
    True
    >>> roundtrip(b"\xef\xbf\xbd")
    True
    >>> roundtrip(b"\xef\xef\xbf\xbd")
    True
    >>> roundtrip(b"\xf1\x80\x80\x80\x80")
    True
    r�r#rr�r�r�)	rErrMr�r��bytechrr�rFr�r�rrr�	fromutf8b�s 

�r�)r#F)F)[�
__future__rr�localerr�rer�rrr�rr�purer	r�r?�typingr
rrr
rrrr�r�	importmodrErfrj�jsonescapeu8fastr��sysstrrH�ispy3�chrr�splitr$�allr(�supports_bytes_environ�_nativeenviron�environ�environb�items�_encodingrewrites�	iswindowsrorG�getpreferredencodingr�ErrorrRrJr.rrArOrWrXr[r`rbrd�	strmethod�identityrgrlrk�dictrnr)r*�compilervrs�getcwdbr}rzryr�r��objectr�r�r�r�r�r�r�rrrr�<module>s�$

���
��E 	 

	��	
f
8E