SHT
9f40d3bdd0
Route FEISHU_ALLOW_BOTS, FEISHU_GROUP_POLICY, FEISHU_ALLOWED_USERS,
FEISHU_BOT_*, FEISHU_APP_ID and FEISHU_REQUIRE_MENTION through
_get_scoped_secret, and snapshot FEISHU_ALLOW_ALL_USERS /
GATEWAY_ALLOW_ALL_USERS into settings.allow_all_dm at construct time so
_admit (running on the lark_oapi WS thread, no secret scope) no longer
reads the default profile's os.environ. Feishu open_ids are app-scoped,
so a secondary bot's DMs were always dm_policy_rejected against the
default profile's allow-list.
Salvaged from PR #86908 trimmed to the adapter call-site migration; the
fail-closed _get_scoped_secret / _auth_env rewrites are already on main
(2912c36aa4, agent/secret_scope.py get_secret policy).
68 lines
2.1 KiB
Python
68 lines
2.1 KiB
Python
"""Shared fixtures for Feishu adapter tests (admission, group policy, dispatch)."""
|
|
|
|
from __future__ import annotations
|
|
|
|
import threading
|
|
from types import SimpleNamespace
|
|
from typing import Any, Optional
|
|
|
|
|
|
def make_sender(sender_type: str = "user", open_id: str = "ou_human",
|
|
user_id: Optional[str] = None, union_id: Optional[str] = None) -> Any:
|
|
return SimpleNamespace(
|
|
sender_type=sender_type,
|
|
sender_id=SimpleNamespace(open_id=open_id, user_id=user_id, union_id=union_id),
|
|
)
|
|
|
|
|
|
def make_message(message_id: str = "om_xxx", chat_type: str = "p2p",
|
|
chat_id: str = "oc_1", mentions: Optional[list] = None) -> Any:
|
|
return SimpleNamespace(
|
|
message_id=message_id,
|
|
chat_type=chat_type,
|
|
chat_id=chat_id,
|
|
mentions=mentions,
|
|
content="",
|
|
message_type="text",
|
|
)
|
|
|
|
|
|
def make_adapter_skeleton(
|
|
*,
|
|
bot_open_id: str = "ou_me",
|
|
bot_user_id: str = "",
|
|
allow_bots: str = "none",
|
|
require_mention: bool = True,
|
|
group_policy: str = "allowlist",
|
|
allow_all_dm: bool = False,
|
|
) -> Any:
|
|
from plugins.platforms.feishu.adapter import FeishuAdapter
|
|
|
|
adapter = object.__new__(FeishuAdapter)
|
|
adapter._bot_open_id = bot_open_id
|
|
adapter._bot_user_id = bot_user_id
|
|
adapter._bot_name = ""
|
|
adapter._app_id = ""
|
|
adapter._admins = set()
|
|
adapter._group_rules = {}
|
|
adapter._group_policy = group_policy
|
|
adapter._default_group_policy = group_policy
|
|
adapter._allowed_group_users = frozenset()
|
|
adapter._allow_bots = allow_bots
|
|
adapter._allow_all_dm = allow_all_dm
|
|
adapter._require_mention = require_mention
|
|
return adapter
|
|
|
|
|
|
def install_dedup_state(adapter: Any, seen: Optional[dict] = None) -> None:
|
|
adapter._seen_message_ids = dict(seen) if seen else {}
|
|
adapter._seen_message_order = list((seen or {}).keys())
|
|
adapter._dedup_cache_size = 100
|
|
adapter._dedup_lock = threading.Lock()
|
|
adapter._dedup_state_path = None
|
|
adapter._persist_seen_message_ids = lambda: None
|
|
|
|
|
|
def stub_mention(adapter: Any, mentions_self: bool) -> None:
|
|
adapter._mentions_self = lambda _message: mentions_self
|