ForcePilot/backend/package/yuxi/channels/adapters/wechat/mp/crypto.py
Kris 05abecc02b feat(wechat): 新增完整微信渠道适配器实现
该提交实现了支持企业微信、微信公众号、个人微信桥接三种模式的完整微信渠道适配器,包含以下核心模块:
1. 基础认证与配置相关:auth_adapter、config_reload、setup_contract等
2. 消息处理与格式转换:format、attachment_adapter、outbound_adapter等
3. 多模式客户端支持:wecom/mp子模块,包含加解密、消息收发能力
4. 辅助能力:限速器、防抖、会话绑定、事件映射、模板渲染等
5. 扩展能力:二维码登录、消息读取、特权用户、心跳监控等

实现了完整的微信生态对接能力,支持消息收发、事件处理、API调用限流、配置热重载等功能。
2026-05-12 00:51:04 +08:00

43 lines
1.2 KiB
Python

from __future__ import annotations
import base64
import hashlib
import struct
def verify_signature(token: str, timestamp: str, nonce: str, signature: str) -> bool:
params = sorted([token, timestamp, nonce])
sha1 = hashlib.sha1("".join(params).encode()).hexdigest()
return sha1 == signature
def decrypt_message(
encrypted_msg: str,
encoding_aes_key: str,
app_id: str,
) -> str:
import socket
from cryptography.hazmat.primitives.ciphers import Cipher, algorithms, modes
key = base64.b64decode(encoding_aes_key + "=")
ciphertext = base64.b64decode(encrypted_msg)
cipher = Cipher(algorithms.AES(key), modes.CBC(key[:16]))
decryptor = cipher.decryptor()
plaintext = decryptor.update(ciphertext) + decryptor.finalize()
pad = plaintext[-1]
plaintext = plaintext[:-pad]
content_len = socket.ntohl(struct.unpack("I", plaintext[16:20])[0])
content = plaintext[20 : 20 + content_len].decode("utf-8")
return content
def verify_url_echostr(token: str, timestamp: str, nonce: str, echostr: str, signature: str) -> tuple[bool, str]:
if not verify_signature(token, timestamp, nonce, signature):
return False, ""
return True, echostr