=[jUdZddlmZddlZddlZddlZddlZddlmZm Z m Z m Z ej e ZehdZded<dd Zd dddZddZddZdS)aCodecs for base64-JSON Invites fields. Several `EventDetails` and `cloudkit.share` fields are typed as ``BYTES`` (or ``ENCRYPTED_BYTES``) on the wire but carry base64-encoded JSON. CloudKit Web Services hands the inner JSON back decrypted to authenticated sessions, so decoding is just ``base64 -> utf-8 -> json``. The decoder is permissive (returns ``None`` for malformed input rather than raising) so a single malformed record can't poison a page of results. ) annotationsN)AnyMappingOptionalUnion>timeplacestyle background integrationszfrozenset[str]JSON_BYTES_FIELDSvalue&Optional[Union[str, bytes, bytearray]]return Optional[Any]ch|dSt|trl tj|d}nE#tjt f$r,tdt|YdSwxYwt|St|ttfrt|}t|d}||S tj|d}nE#tjt f$r,tdt|YdSwxYwt|SdS)aDecode a base64-encoded JSON blob to its Python value. Accepts either a base64 ``str`` (the wire form) or ``bytes``/``bytearray``. For ``bytes`` we try parsing as raw UTF-8 JSON first (Pydantic's ``Base64Bytes`` decodes the wire base64 for us, so the value reaches this codec as already-decoded JSON bytes) and fall back to a second base64-decode attempt if that fails (handles callers passing the wire base64 form as ``bytes``). Returns ``None`` when ``value`` is missing or cannot be decoded. NT)validatez%invites.codecs.b64_decode_fail len=%dF log_failuresz+invites.codecs.json_parse_fail bytes_len=%d) isinstancestrbase64 b64decodebinasciiError ValueErrorLOGGERdebuglen_parse_json_bytesbytes bytearray)rrawparseddecodeds h/home/thesage/.hermes/hermes-agent/venv/lib/python3.11/site-packages/pyicloud/services/invites/codecs.pydecode_json_bytesr'sD }t%& "54888CC +    LL@#e** M M M44 !%%%%%+,, *Ell"3U;;;  M &sT:::GG +    LLFC Q Q Q44 !))) 4s!2>A43A4C>D D Trr#r!rboolcP |d}n;#t$r.|r(tdt |YdSwxYw t j|S#t j$r.|r(tdt |YdSwxYw)Nutf-8z&invites.codecs.utf8_decode_fail len=%dz*invites.codecs.json_parse_fail text_len=%d)decodeUnicodeDecodeErrorrrrjsonloadsJSONDecodeError)r#rtexts r&r r Cszz'""   M LLA3s88 L L Lttz$    R LLEs4yy Q Q Qtts!4AAA((9B%$B%rrctj|dd}tj|ddS)zCEncode a Python value as a base64-encoded JSON string for the wire.),:F) separators ensure_asciir*ascii)r-dumpsr b64encodeencoder+)rr0s r&encode_json_bytesr:RsC :e  G G GD  DKK00 1 1 8 8 A AAblobOptional[Mapping[str, Any]]tuple[str, ...]c\t|tsdS|d}t|tsdSg}|D]V}t|tr?|d}t|tr||Wt |S)zExtract the list of widget type strings from a decoded ``integrations`` blob. Observed shape: ``{"version": "1", "data": [{"type": "com.apple.widget.weather"}, ...]}``. Returns an empty tuple when the blob is missing or malformed. datatype)rrgetlistrappendtuple)r<rAoutentryts r&decode_integrationsrJXs dG $ $r 88F  D dD ! !rC eW % %  &!!A!S!!  1 ::r;)rrrr)r#r!rr(rr)rrrr)r<r=rr>)__doc__ __future__rrrr-loggingtypingrrrr getLogger__name__r frozensetr __annotations__r'r r:rJr@r;r&rSs!   #"""""  000000000000  8 $ $%.I<<<%% ####L;?      BBBB r;