BjdZddlmZddlZddlZddlZddlmZeje Z ejdZ ddl mZd dZd dZd d Zy) uShared helpers for canonicalising WhatsApp sender identity. WhatsApp's bridge can surface the same human under two different JID shapes within a single conversation: - LID form: ``999999999999999@lid`` - Phone form: ``15551234567@s.whatsapp.net`` Both the authorisation path (:mod:`gateway.run`) and the session-key path (:mod:`gateway.session`) need to collapse these aliases to a single stable identity. This module is the single source of truth for that resolution so the two paths can never drift apart. Public helpers: - :func:`normalize_whatsapp_identifier` — strip JID/LID/device/plus syntax down to the bare numeric identifier. - :func:`canonical_whatsapp_identifier` — walk the bridge's ``lid-mapping-*.json`` files and return a stable canonical identity across phone/LID variants. - :func:`expand_whatsapp_aliases` — return the full alias set for an identifier. Used by authorisation code that needs to match any known form of a sender against an allow-list. Plugins that need per-sender behaviour on WhatsApp (role-based routing, per-contact authorisation, policy gating in a gateway hook) should use ``canonical_whatsapp_identifier`` so their bookkeeping lines up with Hermes' own session keys. ) annotationsN)Setz^[A-Za-z0-9@.+\-]+$)get_hermes_homect|xsdjjdddjdddjdddS)aNStrip WhatsApp JID/LID syntax down to its stable numeric identifier. Accepts any of the identifier shapes the WhatsApp bridge may emit: ``"60123456789@s.whatsapp.net"``, ``"60123456789:47@s.whatsapp.net"``, ``"60123456789@lid"``, or a bare ``"+601****6789"`` / ``"60123456789"``. Returns just the numeric identifier (``"60123456789"``) suitable for equality comparisons. Useful for plugins that want to match sender IDs against user-supplied config (phone numbers in ``config.yaml``) without worrying about which variant the bridge happens to deliver. +:r@)strstripreplacesplit)values >/home/ubuntu/.hermes/hermes-agent/gateway/whatsapp_identity.pynormalize_whatsapp_identifierr0sP EKR  b!  sAq  sAq  cFt|}|s tStdz dz }t}|g}|r|jd}|r||vrtj |s0|j |dD]f}|d||dz }|js ttj|jd}|sQ||vsV|j|h|r|S#ttjf$r!} tjd || Yd } ~ d } ~ wwxYw) aResolve WhatsApp phone/LID aliases via bridge session mapping files. Returns the set of all identifiers transitively reachable through the bridge's ``$HERMES_HOME/whatsapp/session/lid-mapping-*.json`` files, starting from ``identifier``. The result always includes the normalized input itself, so callers can safely ``in`` check against the return value without a separate fallback branch. Returns an empty set if ``identifier`` normalizes to empty. whatsappsessionr)r_reversez lid-mapping-z.jsonzutf-8)encodingz(whatsapp_identity: failed to read %s: %sN)rsetrpop_SAFE_IDENTIFIER_REmatchaddexistsjsonloads read_textOSErrorJSONDecodeErrorloggerdebugappend) identifier normalized session_dirresolvedqueuecurrentsuffix mapping_pathmappedexcs rexpand_whatsapp_aliasesr1Fs(/z:J u !#j09z/canonical_whatsapp_identifier..ss9~y.Ir)key)rr1min)r'r(aliasess rcanonical_whatsapp_identifierr:zs.2/z:J  &j1G wI JJr)rr returnr )r'r r;zSet[str])r'r r;r )__doc__ __future__rrloggingretypingr getLogger__name__r$compilerhermes_constantsrrr1r:rrrFsU<#     8 $ !bjj!78,,1h!Kr