o l>jxV@sNdZddlmZddlZddlZddlZddlmZmZm Z m Z m Z ddl m Z ddlmZddlmZdd lmZd d lmZd d lmZd d lmZmZmZe ZeeZ dBddZ!dCddZ" dDdEddZ#dddFd(d)Z$dddFd*d+Z%dddFd,d-Z&dddFd.d/Z' dDdGd1d2Z(dHd5d6Z)d7dd8dId>d?Z*Gd@dAdAZ+dS)Ju+ Exporter helpers for Apple Notes → HTML. These functions are thin, testable wrappers around the existing decoding, datasource hydration, rendering, and file I/O utilities. They are intentionally pure (no global state) and perform only the minimal work needed by CLI tools and higher-level APIs. ) annotationsN)DictIterableListOptionalTuple)Console)CKRecord) BodyDecoder) notes_pb2)CloudKitNoteDataSource) ExportConfig) NoteRendererrender_note_fragmentrender_note_pagerecordr returnOptional[pb.Note]cCst|tsdS|jd}|sdSz&t|}|r t|dds#WdSt}| |j tt|ddddWSt yAYdSw)zDecode a Note CKRecord's TextDataEncrypted and return a parsed pb.Note. Returns None if body is missing or cannot be parsed. NTextDataEncryptedbytesdocumentnote) isinstancer fields get_valuer decodegetattrpbNoteStoreProtoParseFromStringr Exception)rrawnbmsgr&_/home/thesage/.local/lib/python3.10/site-packages/pyicloud/services/notes/rendering/exporter.pydecode_and_parse_note s     r(rpb.Note List[str]c Csg}|jd}|r't|dr't|dgpgD]}t|dd}|r&||qg}t|dgp0gD]}|drI|jdrI|jj}|rI||q1t} g} ||D]} | | vrc| | | | qS| S)N Attachmentsvalue recordName attribute_runattachment_infoattachment_identifier) r get_fieldhasattrrappendHasFieldr/r0setadd) rridsfldrefrn ids_from_runsrattraidseenmergedar&r&r'$_attachment_ids_from_record_and_runs5s2       rA note_recordconfigOptional[ExportConfig](Tuple[CloudKitNoteDataSource, List[str]]c Cst|d}t||}|r||}i}tt|dd}t|jD]F\} } |r4td| t | t | t rg| | z| j d} | rMt| ddnd} t| dd} | r\| j|| <Wq!tyfYq!wq!|rz|t|}tt|ddrztd t |td |Wn tyYnw|jD]}t |t sqd}z,tt|d d D]}|j |} t| dd}t|d d}t |tr|r|}nqWn tyd}Ynw|rl||j}|rlz ||pd}Wn tyd}Ynw|pt}||}d|vp d|vp d|vp d|v}t|ddo,|p,|}d}z||}Wn ty@Ynw|rG|rlz||}Wn tyZd}Ynw|rf|rl||krl|||qW||fSty~Y||fSw||fS)zBuild a CloudKit-backed Note datasource for a single note. Returns (datasource, attachment_ids) where attachment_ids is the merged list of attachment record names to which the datasource has been hydrated. )_configdebugFzrec_idx Mediar,Nr-zmedia lookup responsezattachment media resp: %srr& downloadURLaudiovideomoviempegprefer_media_for_imagesT)rrAlookupboolr enumeraterecordsconsoleruleprintrr add_attachment_recordrr1r-r"listkeysLOGGERinfostrgetget_attachment_utilowerr is_image_utiget_primary_asset_urlget_thumbnail_urlset_primary_asset_url) ck_clientrBrrCdsatt_idsresp media_maprGrec_idxrecr8r9r:mrespmrecurlkvaluparent parent_uticonfis_imageis_avis_media_upgrade cur_primary cur_thumbr&r&r'build_datasourcePs                            CryrCrerrf Iterable[str] assets_dirr\out_dirDict[str, str]c Csdtj|ddi}ddd}tj|}|D]} || pd } || s&q|| } | r7| d s8| d s8qzn|| |} | } zBt | d  }| d }Wdn1sWwY|dr|  ds| d}z t | ||} Wn t y| } YnwWn t y| } Ynwtjj| tj|d}|| |||| <Wqt yYqw|S)uDownload PDFs for attachments and rewrite datasource URLs to local paths. Returns a mapping of attachment id → relative path used in HTML. Only applies to PDF UTIs. Files are renamed with `.pdf` extension when the magic header is present. Texist_oks Optional[str]rrQcSst|o|dvS)N)z com.adobe.pdfz public.pdfzcom.apple.paper.doc.pdf)rQr_rr&r&r' _is_pdf_utis z(download_pdf_assets.._is_pdf_utirJhttp://https://rbNs%PDF-z.pdfstart)rrrrQosmakedirspathabspathr^r_ra startswithdownload_asset_toopenreadendswithreplacer"relpathrc)rdrerfr|r}rCupdatedr note_subdirr=utirm saved_pathresolvedfhheadnew_pathrelr&r&r'download_pdf_assetssN            rc Cstj|ddi}|p t}ddd}tj|} |D]} || p#d } || s,q|| } | sFz| | } Wn t yEd } Ynw| rR| d sS| d sSqzo| | | } | }zCt | d  }|d}Wd n1srwY||}|r| |s| |}z t| ||}Wn t y| }YnwWn t y| }Ynwtjj|tj|d}|| |||| <Wqt yYqw|S)uDownload image attachments and rewrite datasource URLs to local paths. Returns a mapping of attachment id → relative path used in HTML. Applies to common image UTIs (jpeg/png/heic/webp/gif/bmp/tiff...). TrrrrrcSsz`|dr WdS|drWdS|ds|drWdS|dr.|d d d kr.Wd St|d krI|d d dkrI|d d }|dvrIWdS|drQWdS|ds[|dr^WdSWdStyjYdSw)Nsz.jpgsPNG  z.pngsGIF87asGIF89az.gifRIFF sWEBPz.webpftyp)sheicsheifsmif1smsf1shevcz.heicsBMz.bmpsII*sMM*z.tiff)rlenr"rbrandr&r&r'_infer_image_exts.     z/download_image_assets.._infer_image_extrJNrrrrrrrr)rrrrrr^r_r`rarbr"rrrrrrrrc)rdrerfr|r}rCrrsrrr=rrmrrrrextrrr&r&r'download_image_assetssZ               rc Cstj|ddi}ddd}tj|}|D]} || pd } d | vp4d | vp4d | vp4d | vp4d| v} | s8q|| } | rI| dsJ| dsJqz~|| |} | }zRt | d }| d}Wdn1siwY||}|sd| vr{d}nd| vrd}|r|  |s| |}z t | ||}Wn t y| }YnwWn t y| }Ynwtjj|tj|d}|| |||| <Wqt yYqw|S)uDownload audio/video attachments and rewrite datasource URLs to local paths. Returns a mapping of attachment id → relative path used in HTML. TrrrrrcSszvt|dkr&|dddkr&|dd}|dvrWdS|dvr#WdSWd St|dkr7|ddd kr7WdS|d r?Wd S|d sN|dsN|drQWd S|dra|dddkraWdS|drq|dddkrtWdSWdSWdStyYdSw)Nrrrr)sM4A smp42sisom.m4a)sqt moov.movz.mp4rsID3z.mp3sssrsWAVEz.wavsAVI z.avi)rrr"rr&r&r' _infer_av_ext`s>   z)download_av_assets.._infer_av_extrJrKrLrNrM quicktimerrrrNzcom.apple.m4a-audiorrrrr)rdrerfr|r}rCrrrr=rrurmrrrrrrrr&r&r'download_av_assetsPsh   $          rc Cstj|ddi}tj|}|D]l}||pd} d| vr!q||} | r2| ds3| ds3qzA|| |} | } |  ds]| d} z t | | | } Wn t y\| } Ynwtjj | tj|d}| |||||<Wqt y}Yqw|S) uDownload VCard (contact) attachments and rewrite datasource URLs to local paths. Returns a mapping of attachment id → relative path used in HTML. TrrJz public.vcardrrz.vcfr)rrrrr^r_rarrrrr"rrc)rdrerfr|r}rCrrr=rrmrrrrr&r&r'download_vcard_assetss8         r Optional[CloudKitNoteDataSource]cCst|||dS)Nrz)r)rrerCr&r&r'render_fragmentsrrrcCs8|sdStdd|}tdd|}|ddpdS)Nuntitledz\s+ z [^\w\- ]+-<)resubstriprr&r&r' _safe_names rF full_pagefilenametitle html_fragmentrrQrc Cstj|dd|rt||n|}|pt|d}tj|}tjtj||}tj||g|kr8tdt |ddd} | |Wd|S1sPwY|S)NTrz.htmlz!filename must stay within out_dirwutf-8)encoding) rrrrrrjoin commonpath ValueErrorrwrite) rrr}rrpagefnamerootrfr&r&r' write_htmls   rc@s*eZdZdZddddZ ddddZdS) NoteExporterz5Orchestrator for exporting notes to HTML with assets.NrCrDcCs"||_|pt|_t|j|_dSN)clientrrCrrenderer)selfrdrCr&r&r'__init__ s zNoteExporter.__init__rBr output_dirr\rrrcCsjt|}|sdSt|j|||j\}}tt|jddpd}|}|dkrnt|jddp5tj |d}tj ||j } t |j||| ||jdt |j||| ||jdt|j||| ||jdt|j||| ||jd|j||} d} |jd} | rzt| tr| d } nt| tr| } Wn tyYnwt|jd d} | durd } t| | |t| |d S) z Export a single note record to HTML in the output directory. Returns the path to the written HTML file, or None if export failed (e.g. no body). N export_modearchivalr|assets)r|r}rCUntitledTitleEncryptedrrTr)r(ryrrCr\rrr_rrrr-rrrrrrenderrrrrrr"rrQ)rrBrrrrerfr assets_rootr|rr title_encrr&r&r'exports        zNoteExporter.exportr)rCrD)rBr rr\rrrr)__name__ __module__ __qualname____doc__rrr&r&r&r'r s   r)rr rr)rr rr)rr*r)rBr rr)rCrDrrE) rerrfr{r|r\r}r\rCrDrr~)rr)rerrCrDrr\)rrrr\) rr\rr\r}r\rrQrrrr\),r __future__rloggingrrtypingrrrrr rich.consolerpyicloud.common.cloudkitr decodingr protobufr r ck_datasourceroptionsrrrrrrT getLoggerrrZr(rAryrrrrrrrrr&r&r&r'sD           xB[m3