43 lines
1.3 KiB
Python
43 lines
1.3 KiB
Python
|
from __future__ import annotations
|
||
|
|
|
||
|
|
import logging
|
||
|
|
|
||
|
|
logger = logging.getLogger(__name__)
|
||
|
|
|
||
|
|
|
||
|
|
class KakaoTalkSecurityAdapter:
|
||
|
|
DM_POLICY_OPTIONS = ["pairing", "allowlist", "open", "disabled"]
|
||
|
|
|
||
|
|
def resolve_dm_policy(self) -> dict:
|
||
|
|
return {"mode": "pairing", "allow_from": []}
|
||
|
|
|
||
|
|
def resolve_dm_policy_for_account(self, account: dict) -> dict:
|
||
|
|
mode = account.get("dm_policy", "pairing")
|
||
|
|
return {"mode": mode, "allow_from": account.get("allow_from", [])}
|
||
|
|
|
||
|
|
def is_allowed_dm(self, peer_id: str, allow_from: list[str], policy: str) -> bool:
|
||
|
|
if "*" in allow_from:
|
||
|
|
return True
|
||
|
|
if policy == "open":
|
||
|
|
return True
|
||
|
|
if policy == "disabled":
|
||
|
|
return False
|
||
|
|
if policy == "allowlist":
|
||
|
|
return self._match_peer(peer_id, allow_from)
|
||
|
|
if policy == "pairing":
|
||
|
|
return self._match_peer(peer_id, allow_from)
|
||
|
|
return False
|
||
|
|
|
||
|
|
@staticmethod
|
||
|
|
def _match_peer(peer_id: str, allow_from: list[str]) -> bool:
|
||
|
|
for entry in allow_from:
|
||
|
|
entry = entry.strip()
|
||
|
|
if entry == "*":
|
||
|
|
return True
|
||
|
|
if entry.startswith("kakaotalk:"):
|
||
|
|
target = entry[len("kakaotalk:"):]
|
||
|
|
else:
|
||
|
|
target = entry
|
||
|
|
if target == peer_id:
|
||
|
|
return True
|
||
|
|
return False
|