o l>j@sUdZddlmZddlZddlZddlZddlZddlmZm Z m Z m Z e e ZehdZded<dd d Zddd ddZd!ddZd"ddZdS)#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]c Cs|durdSt|tr-z tj|dd}Wt |Stjtfy,tdt |YdSwt|t t frft |}t |dd}|durD|Sz tj|dd}Wt |Stjtfyetdt |YdSwdS)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)r rawparseddecodedr$U/home/thesage/.local/lib/python3.10/site-packages/pyicloud/services/invites/codecs.pydecode_json_bytess.   r&Trr!rrboolcCspz|d}Wnty|rtdt|YdSwzt|WStjy7|r4tdt|YdSw)Nutf-8z&invites.codecs.utf8_decode_fail len=%dz*invites.codecs.json_parse_fail text_len=%d)decodeUnicodeDecodeErrorrrrjsonloadsJSONDecodeError)r!rtextr$r$r%rCs  rrrcCs&tj|ddd}t|ddS)zCEncode a Python value as a base64-encoded JSON string for the wire.),:F) separators ensure_asciir(ascii)r+dumpsr b64encodeencoder))r r.r$r$r%encode_json_bytesRsr7blobOptional[Mapping[str, Any]]tuple[str, ...]cCsdt|tsdS|d}t|tsdSg}|D]}t|tr-|d}t|tr-||qt|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. r$datatype)rrgetlistrappendtuple)r8r;outentrytr$r$r%decode_integrationsXs       rD)r rrr)r!rrr'rr)r rrr)r8r9rr:)__doc__ __future__rrrr+loggingtypingrrrr getLogger__name__r frozensetr __annotations__r&rr7rDr$r$r$r%s   &