߲jdZddlmZddlZddlZddlZddlmZddlm Z eje Z ejdZejdZd dZd d Zd d Zdd Zdd Zy)aShared helpers for canonicalising WhatsApp sender identity. The bridge can surface one human as a LID (``999...@lid``) or a phone JID (``1555...@s.whatsapp.net``) within one conversation. Authorisation (:mod:`gateway.run`) and session keys (:mod:`gateway.session`) both resolve aliases here so they never drift apart; plugins should use :func:`canonical_whatsapp_identifier` to line up with Hermes' session keys. ) annotationsN)Set)get_hermes_dirz^[A-Za-z0-9@.+\-]+$z^\+?[\d\s().\-]+$ct|xsdjjdddjdddjdddS)zStrip JID/LID/device/plus syntax down to the bare numeric identifier: ``"6012:47@s.whatsapp.net"``, ``"6012@lid"`` and ``"+6012"`` all become ``"6012"``.+:r@)strstripreplacesplit)values >/home/ubuntu/.hermes/hermes-agent/gateway/whatsapp_identity.pynormalize_whatsapp_identifierrsQ u{  ! ! # + +CQ 7 = =c1 Ea H N NsTU VWX YYc|syt|j}d|vr3d|vr/|jd\}}}|jdddd|}d|vr|Stj |rt jdd|}|r|dS|S)aUNormalize an *outbound* target to a bridge-safe JID (inverse of normalize). Baileys' ``jidDecode`` crashes on a bare phone, so bare phones become ``@s.whatsapp.net``; ``user:device@domain`` collapses to ``user@domain``; anything else is returned unchanged so the bridge can surface a real error. ``""`` for empty input.rr r r rz\D+z@s.whatsapp.net)r r partitionr_BARE_PHONE_RE fullmatchresub)r normalizedprefix_domaindigitss rto_whatsapp_jidr#s U!!#J jSJ.&0056 S!,Q/0&:  j +J/ X_- - rct|}|jd\}}}|r&|jr|jr|dvr|SdS)zBReturn a valid participant JID for an outbound mention, or ``""``.r >s.whatsapp.netlidr)rrisasciiisdigit)rjiduser separatorrs rnormalize_whatsapp_mention_jidr(7sU % C!mmC0D)V  LLN LLN / /   rc@t|}|s tStdd}t}|g}|r|jd}|r||vstj |s/|j |dD]h}|d||dz }|js tj|jd}t|} | sS| |vsX|j| j|r|S#ttjf$r!} tjd || Yd } ~ d } ~ wwxYw) zAll identifiers transitively reachable via the bridge's ``lid-mapping-*.json`` files; always includes the normalized input itself (empty set if it normalizes to empty).zplatforms/whatsapp/sessionzwhatsapp/sessionr)r_reversez lid-mapping-z.jsonzutf-8)encodingz(whatsapp_identity: failed to read %s: %sN)rsetrpop_SAFE_IDENTIFIER_REmatchaddexistsjsonloads read_textOSErrorJSONDecodeErrorloggerdebugappend) identifierr session_dirresolvedqueuecurrentsuffix mapping_pathrawmappedexcs rexpand_whatsapp_aliasesrDEs#/z:J u  !=?QRKH LE ))A,'X-5H5N5Nw5W  W& %F&<y)NNL&&( jj!7!7!7!IJ6s;&0 V$ % & O T112  GWZ[ s0C##D<DDcPt|}|syt|}t|dS)aStable sender identity across phone-JID/LID variants (DM ``chat_id`` and group ``participant_id`` alike): the shortest alias from :func:`expand_whatsapp_aliases`, which degrades to the normalized input when no mapping files exist. ``""`` for empty input.rct||fS)N)len)cs rz/canonical_whatsapp_identifier..nss1vqkr)key)rrDmin)r:raliasess rcanonical_whatsapp_identifierrMds./z:J &j1G w1 22r)rr returnr )r:r rNzSet[str])r:r rNr )__doc__ __future__rr2loggingrtypingrhermes_constantsr getLogger__name__r7compiler.rrrr(rDrMrrrXsq#  +   8 $!bjj!7801Z ( > 3r