JjPdZddlmZddlZddlZddlZddlZddlmZm Z m Z ej e Z GddZd d ZdS) u Transport-agnostic WhatsApp behavior shared by the Baileys bridge adapter and the official WhatsApp Cloud API adapter. The mixin provides: - Allow-list / DM / group gating - Mention detection (explicit @-mentions + configurable regex patterns) - Quoted-reply-to-bot detection - Broadcast / Channel / Newsletter filtering - WhatsApp-flavored markdown conversion - Outgoing chunk length budgeting It is the *behavior layer*. Transport-specific concerns (subprocess management, HTTP webhooks, Graph API calls, media upload protocols) live in each adapter. Mixin contract — the adapter must set these on ``self`` before any of the mixin's methods are called (typically in ``__init__``): self.config # gateway.config.PlatformConfig self.name # str — adapter name (used in log lines) self._dm_policy # str: "open" | "allowlist" | "disabled" self._allow_from # set[str] self._group_policy # str: "open" | "allowlist" | "disabled" self._group_allow_from # set[str] self._mention_patterns # list[re.Pattern] self._reply_prefix # Optional[str] Class attributes ``MAX_MESSAGE_LENGTH`` and ``DEFAULT_REPLY_PREFIX`` are defined on the mixin and may be overridden per-adapter if needed. ) annotationsN)AnyDictOptionalceZdZUdZdZded<dZdZded<ej d Z ej d Z e d.d Z ed/dZd0dZd1dZd/dZd2dZed2dZed3dZed4dZd/dZed5dZd6dZd6d Zd4d!Zd"Zd7d%Zd8d&Zd8d'Z d8d(Z!d9d*Z"d8d+Z#d.d,Z$d-S):WhatsAppBehaviorMixinu"Shared behavior for all WhatsApp adapters (Baileys + Cloud API). See module docstring for the attribute contract the host adapter must satisfy. This mixin owns no state of its own — every value it touches is either a class attribute or set by the adapter's ``__init__``. iintMAX_MESSAGE_LENGTHTu8⚕ *Hermes Agent* ──────────── strDEFAULT_REPLY_PREFIXz[\u200b\u2060\u2063\ufeff]z3[\u00a0\u1680\u180e\u2000-\u200a\u202f\u205f\u3000]contentreturncv|s|S|jd|}|jd|S)uRemove invisible formatting chars that leak badly in WhatsApp. Some provider/gateway formatting paths can emit unicode like WORD JOINER (U+2060) plus NARROW NO-BREAK SPACE (U+202F). WhatsApp may render those as mojibake-looking prefixes (``⁠ text``) instead of invisible spacing. Keep normal text and emoji joiners intact, but strip known zero-width format chars and normalize odd unicode spaces.  )_OUTBOUND_INVISIBLE_CHARS_REsub_OUTBOUND_ODD_SPACE_RE)clsr s G/home/rurouni/.hermes/hermes-agent/gateway/platforms/whatsapp_common.py_sanitize_outbound_textz-WhatsAppBehaviorMixin._sanitize_outbound_text>sB N266r7CC)--c7;;;boolcdS)zDWhatsApp gates DM/group access at intake via dm_policy/group_policy.Tselfs renforces_own_access_policyz0WhatsAppBehaviorMixin.enforces_own_access_policyMs trctjdd}|dkrdS|j|jddStjd}||ddS|jS)zReturn the prefix to add to outgoing replies in self-chat mode. Subclasses that don't have a self-chat concept (the Cloud API adapter) can override this to always return ``""`` or apply a different policy. WHATSAPP_MODEz self-chatrNz\n WHATSAPP_REPLY_PREFIX)osgetenv _reply_prefixreplacer )r whatsapp_mode env_prefixs r_effective_reply_prefixz-WhatsAppBehaviorMixin._effective_reply_prefixSs| /;?? K ' '2   )%--eT:: :Y677  !%%eT22 2((rctt|}td|j|z S)z>>rc |jjd}|:t|tr|dvSt |StjdddvS)Nrequire_mention>1onyestrueWHATSAPP_REQUIRE_MENTIONfalse) configextraget isinstancer lowerrr#r$)r configureds r_whatsapp_require_mentionz/WhatsAppBehaviorMixin._whatsapp_require_mentionks[&**+<==  !*c** H!''))-GGG ## #y3W==CCEEJ   rset[str]c|jjd}|tjdd}t |t r d|DSdt|dDS)Nfree_response_chatsWHATSAPP_FREE_RESPONSE_CHATSrch|]D}t|#t|ESrr strip.0parts r zFWhatsAppBehaviorMixin._whatsapp_free_response_chats..}=KKK$T9J9JKCIIOO%%KKKrc^h|]*}||+SrrDrEs rrHzFWhatsAppBehaviorMixin._whatsapp_free_response_chats..~-MMM M MMMr,) r7r8r9r#r$r:listr split)rraws r_whatsapp_free_response_chatsz3WhatsAppBehaviorMixin._whatsapp_free_response_chatsxsk##$9:: ;):B??C c4  LKK#KKK KMMS)<)<MMMMrc|tSt|tr d|DSdt|dDS)z;Parse allow_from / group_allow_from from config or env var.Nch|]D}t|#t|ESrrCrEs rrHz;WhatsAppBehaviorMixin._coerce_allow_list..rIrc^h|]*}||+SrrKrEs rrHz;WhatsAppBehaviorMixin._coerce_allow_list..rLrrM)setr:rNr rO)rPs r_coerce_allow_listz(WhatsAppBehaviorMixin._coerce_allow_lists_ ;55L c4  LKK#KKK KMMS)<)<MMMMrvalue Optional[str]c|sdSt|}d|vrd|vr|ddd}|S)Nr:@)r rDr&)rW normalizeds r_normalize_whatsapp_idz,WhatsAppBehaviorMixin._normalize_whatsapp_idsX 2ZZ%%'' *   !2!2#++Ca88Jrchat_idc|sdS|}|dkrdS|ds|drdSdS)upTrue for WhatsApp pseudo-chats that aren't real conversations. Covers Status updates (Stories) and Channel/Newsletter broadcasts. These show up as inbound messages on Baileys but the agent should never reply — answering a Story update spams the contact's status feed, and Channel posts aren't addressable in the first place. Fzstatus@broadcastTz @broadcastz @newsletter)rDr;endswith)r_cids r_is_broadcast_chatz(WhatsAppBehaviorMixin._is_broadcast_chatsn 5mmoo##%% $ $ $4 << % % m)D)D 4urctjdddvrdStjdddvS)NGATEWAY_ALLOW_ALL_USERSr>r1r3r4TWHATSAPP_ALLOW_ALL_USERS)r#r$r;rs r_open_dm_opted_inz'WhatsAppBehaviorMixin._open_dm_opted_insQ 9. 3 3 9 9 ; ;?S S S4y3R88>>@@DXXXr candidatec|sdS||vrdSddlm}m}||}|sdS|D],}|dkrdS|||vrdS|||zrdS-dS)amMatch a WhatsApp identifier against an allowlist across phone/LID forms. WhatsApp delivers inbound senders in LID form (``@lid``) while operators usually configure allowlists with phone numbers, and vice versa. A raw set-membership check therefore never matches a known contact. Resolve both the candidate and each allowlist entry through the bridge's ``lid-mapping-*.json`` files (the shared ``gateway.whatsapp_identity`` helper that the gateway authz and session-key paths already use) so either configured form resolves to the inbound form. FTr)expand_whatsapp_aliasesnormalize_whatsapp_identifier*)gateway.whatsapp_identityrjrk)rh allow_fromrjrkcandidate_aliasesentrys r_matches_whatsapp_allowlistz1WhatsAppBehaviorMixin._matches_whatsapp_allowlists 5  " "4        43I>>  5  E||tt,,U337HHHtt'&u--0AA tt ur sender_idc|jdkrdS|jdkr|||jS|jdkr|SdS)u:Strict DM authorization — pairing does not imply access.disabledF allowlistopen) _dm_policyrq _allow_fromrg)rrrs r_is_dm_allowedz$WhatsAppBehaviorMixin._is_dm_alloweds^ ?j ( (5 ?k ) )33It?OPP P ?f $ $))++ +urct|pd}|sdS|jdkrdS|jdkr|||jS|jdkrdS|jdkr|SdS)zCWhether a DM may reach the gateway intake (pairing handshake path).rFrtrupairingTrv)r rDrwrqrxrg)rrr principals r_is_dm_intake_allowedz+WhatsAppBehaviorMixin._is_dm_intake_alloweds R((..00  5 ?j ( (5 ?k ) )33It?OPP P ?i ' '4 ?f $ $))++ +urc|jdkrdS|jdkr|||jS|jdkrdS|jdkrdSdS)z/Check whether a group chat should be processed.rtFrur{rvT) _group_policyrq_group_allow_from)rr_s r_is_group_allowedz'WhatsAppBehaviorMixin._is_group_allowedsg   + +5   , ,33GT=STT T   * *5   ' '4urc|jjd}|tjdd}|re t j|}nO#t$rBd| D}|sd| dD}YnwxYw|gSt|tr|g}t|ts5td|jt#|jgSg}|D]}t|tr|s, |t)j|t(j`#t(j$r,}td|j||Yd}~d}~wwxYw|r.td |jt3||S) Nmention_patternsWHATSAPP_MENTION_PATTERNSrc^g|]*}||+SrrKrEs r zCWhatsAppBehaviorMixin._compile_mention_patterns..s=   )-TZZ\\     rc^g|]*}||+SrrKrEs rrzCWhatsAppBehaviorMixin._compile_mention_patterns..s=$$$-1tzz||$ JJLL$$$rrMz?[%s] whatsapp mention_patterns must be a list or string; got %sz,[%s] Invalid WhatsApp mention pattern %r: %sz*[%s] Loaded %d WhatsApp mention pattern(s))r7r8r9r#r$rDjsonloads Exception splitlinesrOr:r rNloggerwarningnametype__name__appendrecompile IGNORECASEerrorinfor+)rpatternsrPcompiledpatternexcs r_compile_mention_patternsz/WhatsAppBehaviorMixin._compile_mention_patternss/;$(();<<  )7<<BBDDC  #z#HH   141A1A   H$$$58YYs^^$$$   I h $ $ " zH(D))  NNQ X'    I  Ggs++ 7==??   7BM B BCCCC8   BI     KKWhatsAppBehaviorMixin._message_mentions_bot..9sB   229===    r mentionedIdsTbodyrr[r\r)rr9r r;rO) rrr mentioned_idsr lower_bodybot_idbare_idrs ` @r_message_mentions_botz+WhatsAppBehaviorMixin._message_mentions_bot5s,,T22 5     "hh~66<"   7 " 4488F##)r**ZZ\\   Fll3**1-3355G MMMZ777j;P;Ptturc|jsdSt|dpdtfd|jDS)NFrrc3BK|]}|VdS)N)search)rFrrs r zJWhatsAppBehaviorMixin._message_matches_mention_patterns..Ms/NNG7>>$''NNNNNNr)_mention_patternsr r9any)rrrs @r!_message_matches_mention_patternsz7WhatsAppBehaviorMixin._message_matches_mention_patternsIsW% 5488F##)r**NNNNt7MNNNNNNrtextc|s|S||}|}|D]L}|ddd}|r,tjdtj|dd|}M|p|S)Nr[r\rz \b[,:\-]*\s*r)rrOrrescaperD)rrrrcleanedrrs r_clean_bot_mention_textz-WhatsAppBehaviorMixin._clean_bot_mention_textOs K,,T22  Fll3**1-G &97++9992w}}&$&rcDt|dpd}||rdS|dd}|r|}||sdSnRt|dp|dpd}||sdSdSt|dpd}||vrdS|sdSt|dpd}|d rdS| |rdS| |rdS| |S) NchatIdrFisGroupsenderIdfromTr/) r r9rcrr}rQr=rD startswithrrr)rr chat_id_rawis_groupr_rrrs r_should_process_messagez-WhatsAppBehaviorMixin._should_process_message\s$((8,,233  " "; / / 588Iu--  !G))'22 u DHHZ00JDHHV4D4DJKKI--i88 u4dhhx((.B// d88:: : :4--// 4488F##)r**0022 ??3   4  ( ( . . 4  % %d + + 455d;;;rc |s|S||}d g d fd }tjd||}d g d fd }tjd ||}tjd d |}tjd d|}tjdd|}tjdd|}dd}tjd||tj}tjdd|}t D] \}}| |d|}!t D] \}}| |d|}!|S)aConvert standard markdown to WhatsApp-compatible formatting. WhatsApp supports: *bold*, _italic_, ~strikethrough~, ```code```, and monospaced `inline`. Standard markdown uses different syntax for bold/italic/strikethrough, so we convert here. Code blocks (``` fenced) and inline code (`) are protected from conversion via placeholder substitution. zFENCEmre.Matchrr c|dtdz dSNrr\rgroupr+)r _FENCE_PHfencess r _save_fencez9WhatsAppBehaviorMixin.format_message.._save_fences= MM!''!** % % %6Vq666 6rz```[\s\S]*?```zCODEc|dtdz dSrr)r_CODE_PHcodess r _save_codez8WhatsAppBehaviorMixin.format_message.._save_codes= LL $ $ $4E Q444 4rz `[^`\n]+`z.(?._header_to_boldsGGAJJ$$&&Ee**q..U%5%5c%:%:.u~~c?R?R.ad ))++e**q..U%5%5c%:%:.u~~c?R?R.u<<< rz^#{1,6}\s+(.+)$)flagsz\[([^\]]+)\]\(([^)]+)\)z\1 (\2)r)rrrr )rrr MULTILINE enumerater&) rr rresultrrifencecoderrrrs @@@@rformat_messagez$WhatsAppBehaviorMixin.format_messages N..w77  7 7 7 7 7 7 7);@@ 5 5 5 5 5 5 5 j&99  =     ('6:: gv66 gv66     r|    2JGG"&)) B BHAu^^y$9!$9$9$95AAFF '' @ @GAt^^x$8$8$8$8$??FF rN)r r rr )rr)rr )rr )rr>)rWrXrr )r_r rr)rhr rr)rrr rr)rrrr>)rrrr)rr rrrr )%r __module__ __qualname____doc__r __annotations__supports_code_blocksr rrrr classmethodrpropertyrr)r.r=rQ staticmethodrVr^rcrgrqryr}rrrrrrrrrrrrrr,s#"""" \\\\\#-2:.K#L#L 'RZ(^__ < < <[ <X ))))"????     NNNNNNN\N\\(YYYY $$$\$L        ,,,\FFFF (OOOO ' ' ' ' < < < r s>#"""""  &&&&&&&&&&  8 $ $]]]]]]]]H ......r