wehub-resource-sync
454 lines
23 KiB
Python
454 lines
23 KiB
Python
import os, sys, re, threading, queue, time, socket, json, struct, base64, uuid, hashlib, math
|
||
from pathlib import Path
|
||
from urllib.parse import quote
|
||
import requests, qrcode
|
||
from Crypto.Cipher import AES
|
||
sys.path.insert(0, os.path.dirname(os.path.dirname(os.path.abspath(__file__))))
|
||
_TEMP_DIR = os.path.join(os.path.dirname(os.path.dirname(os.path.abspath(__file__))), 'temp')
|
||
from agentmain import GeneraticAgent
|
||
|
||
# ── AuthExpired (errcode -14 from getUpdates) ──
|
||
class AuthExpired(Exception):
|
||
"""Bot token expired or invalid (errcode=-14)."""
|
||
pass
|
||
|
||
# ── Per-user abort flags (shared between on_message invocations) ──
|
||
_task_aborted: dict = {} # uid -> True (set by /stop, read by _handle)
|
||
|
||
# ── WxBotClient (inline from wx_bot_client.py) ──
|
||
for _k in ('HTTPS_PROXY', 'https_proxy'):
|
||
os.environ.pop(_k, None) # avoid inherited proxy breaking WeChat long-poll SSL
|
||
API = 'https://ilinkai.weixin.qq.com'
|
||
TOKEN_FILE = Path.home() / '.wxbot' / 'token.json'
|
||
TOKEN_FILE.parent.mkdir(exist_ok=True)
|
||
VER, MSG_USER, MSG_BOT, ITEM_TEXT, STATE_FINISH = '2.1.10', 1, 2, 1, 2
|
||
ILINK_APP_ID = 'bot'
|
||
ILINK_APP_CLIENT_VERSION = (2 << 16) | (1 << 8) | 10
|
||
UA = f'openclaw-weixin/{VER}'
|
||
ITEM_IMAGE, ITEM_FILE, ITEM_VIDEO = 2, 4, 5
|
||
CDN_BASE = 'https://novac2c.cdn.weixin.qq.com/c2c'
|
||
|
||
def _uin():
|
||
return base64.b64encode(str(struct.unpack('>I', os.urandom(4))[0]).encode()).decode()
|
||
|
||
class WxBotClient:
|
||
def __init__(self, token=None, token_file=None):
|
||
self._tf = Path(token_file) if token_file else TOKEN_FILE
|
||
self.token = token
|
||
self.bot_id = None
|
||
self._buf = ''
|
||
if not self.token: self._load()
|
||
|
||
def _load(self):
|
||
if self._tf.exists():
|
||
d = json.loads(self._tf.read_text('utf-8'))
|
||
self.token, self.bot_id, self._buf = d.get('bot_token',''), d.get('ilink_bot_id',''), d.get('updates_buf','')
|
||
|
||
def _save(self, **kw):
|
||
d = {'bot_token': self.token or '', 'ilink_bot_id': self.bot_id or '',
|
||
'updates_buf': self._buf or '', **kw}
|
||
self._tf.write_text(json.dumps(d, ensure_ascii=False, indent=2), 'utf-8')
|
||
|
||
def _post(self, ep, body, timeout=15):
|
||
data = json.dumps(body, ensure_ascii=False, separators=(',', ':')).encode('utf-8')
|
||
h = {'Content-Type': 'application/json', 'AuthorizationType': 'ilink_bot_token',
|
||
'Content-Length': str(len(data)), 'X-WECHAT-UIN': _uin(),
|
||
'iLink-App-Id': ILINK_APP_ID,
|
||
'iLink-App-ClientVersion': str(ILINK_APP_CLIENT_VERSION),
|
||
'User-Agent': UA}
|
||
tok = (self.token or '').strip()
|
||
if tok: h['Authorization'] = f'Bearer {tok}'
|
||
r = requests.post(f'{API}/{ep}', data=data, headers=h, timeout=timeout)
|
||
r.raise_for_status()
|
||
return r.json()
|
||
|
||
def login_qr(self, poll_interval=2):
|
||
# 获取二维码：对限流/缺字段（无 'qrcode'）退避重试，避免崩溃→重启→更狠地打接口的死亡螺旋
|
||
d = {}
|
||
for attempt in range(6):
|
||
try:
|
||
r = requests.get(f'{API}/ilink/bot/get_bot_qrcode', params={'bot_type': 3}, headers={'User-Agent': UA}, timeout=10)
|
||
r.raise_for_status()
|
||
d = r.json()
|
||
except requests.exceptions.RequestException as e:
|
||
print(f'[QR登录] 获取二维码失败（{e}），{2 ** attempt}s 后重试...'); time.sleep(2 ** attempt); continue
|
||
if d.get('qrcode') and d.get('qrcode_img_content'): # 二维码 ID + 可扫图都就绪才算成功
|
||
break
|
||
print(f'[QR登录] 二维码未就绪（可能被限流，ret={d.get("ret")}），{2 ** attempt}s 后重试...')
|
||
time.sleep(2 ** attempt)
|
||
if not (d.get('qrcode') and d.get('qrcode_img_content')):
|
||
raise RuntimeError('多次重试仍未获取到可扫二维码（疑似限流），请稍后重试')
|
||
qr_id, url = d['qrcode'], d.get('qrcode_img_content', '')
|
||
print(f'[QR登录] ID: {qr_id}')
|
||
if url:
|
||
# 先打 ASCII 二维码（纯文本，无需 PIL；容器/无头环境靠它扫码）
|
||
qr = qrcode.QRCode(border=1); qr.add_data(url); qr.make(fit=True); qr.print_ascii(invert=True)
|
||
# 再尝试存 PNG 兜底——依赖 PIL，缺失/失败不应让登录崩溃
|
||
try:
|
||
qrcode.make(url).save(str(self._tf.parent / 'wx_qr.png'))
|
||
except Exception as e:
|
||
print(f'[QR登录] PNG 兜底保存失败（{e}），用上方 ASCII 二维码扫码即可')
|
||
last = ''
|
||
while True:
|
||
time.sleep(poll_interval)
|
||
# 轮询状态：对所有网络异常 / 非 JSON（被限流时常返回 HTML）容错重试，
|
||
# 不让单次抖动把进程打崩——配合 restart:unless-stopped 否则会死亡螺旋
|
||
try:
|
||
s = requests.get(f'{API}/ilink/bot/get_qrcode_status', params={'qrcode': qr_id}, headers={'User-Agent': UA}, timeout=60).json()
|
||
except requests.exceptions.RequestException:
|
||
continue
|
||
except ValueError: # 响应非 JSON（限流/网关页）
|
||
time.sleep(poll_interval); continue
|
||
st = s.get('status', '')
|
||
if st != last: print(f' 状态: {st}'); last = st
|
||
if st == 'confirmed':
|
||
self.token, self.bot_id = s.get('bot_token', ''), s.get('ilink_bot_id', '')
|
||
self._save(login_time=time.strftime('%Y-%m-%d %H:%M:%S'))
|
||
print(f'[QR登录] 成功! bot_id={self.bot_id}')
|
||
return s
|
||
if st == 'expired': raise RuntimeError('二维码过期')
|
||
|
||
def get_updates(self, timeout=30):
|
||
try:
|
||
resp = self._post('ilink/bot/getupdates',
|
||
{'get_updates_buf': self._buf or '',
|
||
'base_info': {'channel_version': VER}},
|
||
timeout=timeout + 5)
|
||
except requests.exceptions.ReadTimeout:
|
||
return []
|
||
if resp.get('errcode'):
|
||
print(f'[getUpdates] err: {resp.get("errcode")} {resp.get("errmsg","")}')
|
||
if resp['errcode'] == -14:
|
||
self._buf = ''; self.token = ''; self.bot_id = ''
|
||
self._save(bot_token='', ilink_bot_id='')
|
||
raise AuthExpired(resp.get('errmsg',''))
|
||
return []
|
||
nb = resp.get('get_updates_buf', '')
|
||
if nb: self._buf = nb; self._save()
|
||
return resp.get('msgs') or []
|
||
|
||
def send_text(self, to_user_id, text, context_token=''):
|
||
msg = {'from_user_id': '', 'to_user_id': to_user_id,
|
||
'client_id': f'pyclient-{uuid.uuid4().hex[:16]}',
|
||
'message_type': MSG_BOT, 'message_state': STATE_FINISH,
|
||
'item_list': [{'type': ITEM_TEXT, 'text_item': {'text': text}}]}
|
||
if context_token: msg['context_token'] = context_token
|
||
return self._post('ilink/bot/sendmessage', {'msg': msg, 'base_info': {'channel_version': VER}})
|
||
|
||
def send_typing(self, to_user_id, typing_ticket='', cancel=False):
|
||
return self._post('ilink/bot/sendtyping', {
|
||
'ilink_user_id': to_user_id, 'typing_ticket': typing_ticket,
|
||
'status': 2 if cancel else 1,
|
||
'base_info': {'channel_version': VER}})
|
||
|
||
def get_typing_ticket(self, to_user_id, context_token=''):
|
||
payload = {'ilink_user_id': to_user_id}
|
||
if context_token: payload['context_token'] = context_token
|
||
return self._post('ilink/bot/getconfig', payload).get('typing_ticket', '')
|
||
|
||
def _enc(self, raw, aes_key):
|
||
pad = 16 - (len(raw) % 16)
|
||
return AES.new(aes_key, AES.MODE_ECB).encrypt(raw + bytes([pad] * pad))
|
||
|
||
def _upload(self, filekey, upload_param, raw, aes_key, timeout=120, upload_url=''):
|
||
url = upload_url.strip() if upload_url else f'{CDN_BASE}/upload?encrypted_query_param={quote(upload_param)}&filekey={filekey}'
|
||
data = self._enc(raw, aes_key)
|
||
last_err = None
|
||
for attempt in range(1, 4):
|
||
try:
|
||
r = requests.post(url, data=data, headers={'Content-Type': 'application/octet-stream', 'User-Agent': UA}, timeout=timeout)
|
||
if 400 <= r.status_code < 500:
|
||
msg = r.headers.get('x-error-message') or r.text[:300]
|
||
raise RuntimeError(f'CDN upload client error {r.status_code}: {msg}')
|
||
if r.status_code != 200:
|
||
msg = r.headers.get('x-error-message') or f'status {r.status_code}'
|
||
raise RuntimeError(f'CDN upload server error: {msg}')
|
||
eq = r.headers.get('x-encrypted-param', '')
|
||
if not eq: raise RuntimeError('CDN upload response missing x-encrypted-param header')
|
||
return {'encrypt_query_param': eq,
|
||
'aes_key': base64.b64encode(aes_key.hex().encode()).decode(), 'encrypt_type': 1}
|
||
except Exception as e:
|
||
last_err = e
|
||
if 'client error' in str(e) or attempt >= 3: break
|
||
print(f'[WX] CDN upload retry {attempt}: {e}', file=sys.__stdout__)
|
||
raise last_err
|
||
|
||
def _send_media(self, to_user_id, file_path, media_type, item_type, item_key, context_token=''):
|
||
fp = Path(file_path)
|
||
raw = fp.read_bytes()
|
||
filekey = uuid.uuid4().hex
|
||
aes_key = os.urandom(16)
|
||
ciphertext_size = ((len(raw) // 16) + 1) * 16
|
||
thumb_raw = b''; thumb_w = thumb_h = 0; thumb_ciphertext_size = 0
|
||
if item_key == 'image_item':
|
||
from io import BytesIO
|
||
from PIL import Image
|
||
im = Image.open(fp); im.thumbnail((240, 240))
|
||
thumb_w, thumb_h = im.size
|
||
if im.mode not in ('RGB', 'L'):
|
||
im = im.convert('RGB')
|
||
bio = BytesIO(); im.save(bio, format='JPEG', quality=85)
|
||
thumb_raw = bio.getvalue()
|
||
thumb_ciphertext_size = ((len(thumb_raw) // 16) + 1) * 16
|
||
body = {
|
||
'filekey': filekey, 'media_type': media_type, 'to_user_id': to_user_id,
|
||
'rawsize': len(raw), 'rawfilemd5': hashlib.md5(raw).hexdigest(),
|
||
'filesize': ciphertext_size,
|
||
'no_need_thumb': item_key not in ('image_item', 'video_item'),
|
||
'aeskey': aes_key.hex(), 'base_info': {'channel_version': VER}}
|
||
if thumb_raw:
|
||
body.update({'thumb_rawsize': len(thumb_raw),
|
||
'thumb_rawfilemd5': hashlib.md5(thumb_raw).hexdigest(),
|
||
'thumb_filesize': thumb_ciphertext_size})
|
||
resp = self._post('ilink/bot/getuploadurl', body)
|
||
upload_param = resp.get('upload_param', '')
|
||
upload_url = resp.get('upload_full_url', '')
|
||
if not (upload_param or upload_url): raise RuntimeError(f'getuploadurl failed: {resp}')
|
||
media = self._upload(filekey, upload_param, raw, aes_key=aes_key, upload_url=upload_url)
|
||
item = {'media': media}
|
||
if item_key == 'file_item':
|
||
item.update({'file_name': fp.name, 'len': str(len(raw))})
|
||
elif item_key == 'image_item':
|
||
thumb_param = resp.get('thumb_upload_param', '')
|
||
thumb_url = resp.get('thumb_upload_full_url', '')
|
||
if thumb_param or thumb_url:
|
||
thumb_media = self._upload(filekey, thumb_param, thumb_raw, aes_key=aes_key, upload_url=thumb_url)
|
||
thumb_size = thumb_ciphertext_size
|
||
else:
|
||
# Some getuploadurl responses only return a single upload_full_url for IMAGE.
|
||
# Keep ImageItem structurally complete by reusing the original CDN media as thumb_media.
|
||
thumb_media = media
|
||
thumb_size = ciphertext_size
|
||
item.update({'mid_size': ciphertext_size, 'thumb_media': thumb_media,
|
||
'thumb_size': thumb_size,
|
||
'thumb_width': thumb_w, 'thumb_height': thumb_h})
|
||
elif item_key == 'video_item':
|
||
item.update({'video_size': ciphertext_size})
|
||
msg = {'from_user_id': '', 'to_user_id': to_user_id,
|
||
'client_id': f'pyclient-{uuid.uuid4().hex[:16]}',
|
||
'message_type': MSG_BOT, 'message_state': STATE_FINISH,
|
||
'item_list': [{'type': item_type, item_key: item}]}
|
||
if context_token: msg['context_token'] = context_token
|
||
return self._post('ilink/bot/sendmessage', {'msg': msg, 'base_info': {'channel_version': VER}})
|
||
|
||
def send_file(self, to_user_id, file_path, context_token=''):
|
||
return self._send_media(to_user_id, file_path, 3, ITEM_FILE, 'file_item', context_token)
|
||
|
||
def send_image(self, to_user_id, file_path, context_token=''):
|
||
return self._send_media(to_user_id, file_path, 1, ITEM_IMAGE, 'image_item', context_token)
|
||
|
||
def send_video(self, to_user_id, file_path, context_token=''):
|
||
return self._send_media(to_user_id, file_path, 2, ITEM_VIDEO, 'video_item', context_token)
|
||
|
||
@staticmethod
|
||
def extract_text(msg):
|
||
return '\n'.join(it['text_item'].get('text', '')
|
||
for it in msg.get('item_list', [])
|
||
if it.get('type') == ITEM_TEXT and it.get('text_item'))
|
||
|
||
@staticmethod
|
||
def is_user_msg(msg): return msg.get('message_type') == MSG_USER
|
||
|
||
def run_loop(self, on_message, poll_timeout=30):
|
||
print(f'[Bot] 监听中... (bot_id={self.bot_id})')
|
||
seen = set()
|
||
while True:
|
||
try:
|
||
for msg in self.get_updates(poll_timeout):
|
||
mid = msg.get('message_id', 0)
|
||
if not self.is_user_msg(msg) or mid in seen: continue
|
||
seen.add(mid)
|
||
if len(seen) > 5000: seen = set(list(seen)[-2000:])
|
||
try: on_message(self, msg)
|
||
except Exception as e: print(f'[Bot] 回调异常: {e}')
|
||
except KeyboardInterrupt: print('[Bot] 退出'); break
|
||
except AuthExpired: raise
|
||
except Exception as e: print(f'[Bot] 异常: {e}，5s重试'); time.sleep(5)
|
||
|
||
# ── Unified media download (IMAGE/VIDEO/FILE/VOICE) ──
|
||
_MEDIA_KEYS = {'image_item': '.jpg', 'video_item': '.mp4', 'file_item': '', 'voice_item': '.silk'}
|
||
|
||
def _dl_media(items):
|
||
"""Download & decrypt all media items → list of local file paths."""
|
||
paths = []
|
||
for item in items:
|
||
for key, ext in _MEDIA_KEYS.items():
|
||
sub = item.get(key)
|
||
if not sub: continue
|
||
eq = (sub.get('media') or {}).get('encrypt_query_param')
|
||
if not eq: continue
|
||
ak = (sub.get('media') or {}).get('aes_key', '') or sub.get('aeskey', '')
|
||
if not ak: continue
|
||
try:
|
||
aes_key = (bytes.fromhex(base64.b64decode(ak).decode())
|
||
if sub.get('media', {}).get('aes_key') else bytes.fromhex(ak))
|
||
ct = requests.get(f'{CDN_BASE}/download?encrypted_query_param={quote(eq)}', headers={'User-Agent': UA}, timeout=60).content
|
||
pt = AES.new(aes_key, AES.MODE_ECB).decrypt(ct); pt = pt[:-pt[-1]]
|
||
fname = sub.get('file_name') or f'{uuid.uuid4().hex[:8]}{ext or ".bin"}'
|
||
p = os.path.join(_TEMP_DIR, fname); open(p, 'wb').write(pt)
|
||
paths.append(p); print(f'[WX] media saved: {fname}', file=sys.__stdout__)
|
||
except Exception as e:
|
||
print(f'[WX] media dl err ({key}): {e}', file=sys.__stdout__)
|
||
break # one media per item
|
||
return paths
|
||
|
||
agent = GeneraticAgent()
|
||
agent.verbose = False
|
||
|
||
_TAG_PATS = [r'<' + t + r'>.*?</' + t + r'>' for t in ('thinking', 'tool_use')]
|
||
_TAG_PATS.append(r'<file_content>.*?</file_content>')
|
||
|
||
def _strip_md(t):
|
||
"""Filter markdown for WeChat rich-text rendering.
|
||
WeChat natively renders: code fences, inline code, bold, italic,
|
||
H1-H4 headings, horizontal rules, tables. We only strip unsupported syntax."""
|
||
def _trunc_code(m):
|
||
full = m.group()
|
||
fence = re.match(r'`{3,}', full).group()
|
||
rest = full[len(fence):-len(fence)]
|
||
if '\n' not in rest: return full # single-line, keep as-is
|
||
lang_line, _, body = rest.partition('\n')
|
||
lines = body.split('\n')
|
||
if len(lines) > 10:
|
||
return f'{fence}{lang_line}\n' + '\n'.join(lines[:10]) + '\n...\n' + fence
|
||
return full # keep intact
|
||
t = re.sub(r'(`{3,})[\s\S]*?\1', _trunc_code, t)
|
||
# inline code: keep (WeChat renders it)
|
||
# bold/italic (*/**/***): keep (WeChat renders it)
|
||
t = re.sub(r'!\[.*?\]\(.*?\)', '', t) # images: remove
|
||
t = re.sub(r'\[([^\]]+)\]\([^\)]+\)', r'\1', t) # links: text only
|
||
t = re.sub(r'^#{5,6}\s+', '', t, flags=re.M) # H5-H6: strip (H1-H4 kept)
|
||
t = re.sub(r'^\s*[-*+]\s+', '• ', t, flags=re.M) # unordered list: bullet
|
||
t = re.sub(r'^\s*\d+\.\s+', '', t, flags=re.M) # ordered list: strip num
|
||
t = re.sub(r'^\s*>\s?', '', t, flags=re.M) # blockquote: strip
|
||
# horizontal rules (---): keep (WeChat renders it)
|
||
return re.sub(r'\n{3,}', '\n\n', t).strip()
|
||
|
||
def _clean(t):
|
||
t = re.sub(r'^\s*LLM Running \(Turn \d+\) \.{3}\s*$', '', t, flags=re.M)
|
||
t = re.sub(r'^\s*🛠️\s*[A-Za-z_][A-Za-z0-9_]*\(.*$', '', t, flags=re.M)
|
||
for p in _TAG_PATS:
|
||
t = re.sub(p, '', t, flags=re.DOTALL)
|
||
t = re.sub(r'</?summary>', '', t)
|
||
return re.sub(r'\n{3,}', '\n\n', _strip_md(t)).strip()
|
||
|
||
def on_message(bot, msg):
|
||
text = bot.extract_text(msg).strip()
|
||
uid = msg.get('from_user_id', '')
|
||
ctx = msg.get('context_token', '')
|
||
media_paths = _dl_media(msg.get('item_list', []))
|
||
if not text and not media_paths: return
|
||
if media_paths:
|
||
text = (text + '\n' if text else '') + '\n'.join(f'[用户发送文件: {p}]' for p in media_paths)
|
||
print(f'[WX] 收到: {text[:80]}', file=sys.__stdout__)
|
||
|
||
# Commands
|
||
if text in ('/stop', '/abort'):
|
||
agent.abort()
|
||
_task_aborted[uid] = True
|
||
print(f'[WX] /stop set _task_aborted[{uid}]', file=sys.__stdout__)
|
||
return
|
||
if text.startswith('/llm'):
|
||
args = text.split()
|
||
if len(args) > 1:
|
||
try:
|
||
n = int(args[1]); agent.next_llm(n)
|
||
bot.send_text(uid, f'切换到 [{agent.llm_no}] {agent.get_llm_name()}', context_token=ctx)
|
||
except (ValueError, IndexError):
|
||
bot.send_text(uid, f'用法: /llm <0-{len(agent.list_llms())-1}>', context_token=ctx)
|
||
else:
|
||
lines = [f"{'→' if cur else ' '} [{i}] {name}" for i, name, cur in agent.list_llms()]
|
||
bot.send_text(uid, 'LLMs:\n' + '\n'.join(lines), context_token=ctx)
|
||
return
|
||
|
||
def _handle():
|
||
prompt = text if text.startswith('/') else f"If you need to show files to user, use [FILE:filepath] in your response.\n\n{text}"
|
||
dq = agent.put_task(prompt, source="wechat")
|
||
_typing_stop = threading.Event()
|
||
def _keep_typing():
|
||
ticket = bot.get_typing_ticket(uid, ctx)
|
||
if not ticket: return
|
||
while not _typing_stop.is_set():
|
||
try: bot.send_typing(uid, ticket)
|
||
except: pass
|
||
_typing_stop.wait(2.0)
|
||
threading.Thread(target=_keep_typing, daemon=True).start()
|
||
result = ''; sent = 0; mi = 0; last_send = 0; item = {}
|
||
def _wx_send(text):
|
||
s = text.strip(); t0 = time.time()
|
||
try:
|
||
bot.send_text(uid, s, context_token=ctx)
|
||
print(f'[WX] send ok len={len(s)} dt={time.time()-t0:.1f}s', file=sys.__stdout__)
|
||
return True
|
||
except Exception as e:
|
||
print(f'[WX] send err len={len(s)} dt={time.time()-t0:.1f}s {type(e).__name__}: {e}', file=sys.__stdout__)
|
||
return False
|
||
def _send(show):
|
||
nonlocal mi, last_send
|
||
now = time.time()
|
||
if mi >= 9 or not show.strip(): return False
|
||
if mi and now - last_send < 6 * mi: return None
|
||
if _wx_send(show[:3000]): mi += 1; last_send = time.time(); return True
|
||
return False
|
||
try:
|
||
done = []; turn = 1
|
||
while True:
|
||
item = dq.get(timeout=300)
|
||
if 'done' in item: break
|
||
if item.get('turn', turn) > turn:
|
||
outputs = item.get('outputs', [])
|
||
lastdone = outputs[-2] if len(outputs) >= 2 else ''
|
||
turn = item['turn']; done.append(lastdone)
|
||
if len(done) > sent:
|
||
merged = _clean('\n\n'.join(done[sent:]))
|
||
print(f'[WX] turns={len(done)}/{len(done)+1} sent={sent} sending={len(done)-sent}', file=sys.__stdout__)
|
||
if _send(merged): sent = len(done)
|
||
except queue.Empty: result = '[超时]'
|
||
_typing_stop.set()
|
||
|
||
if 'done' in item: result, done = item['done'], item.get('outputs', [])
|
||
aborted = _task_aborted.pop(uid, False)
|
||
tag = '[已停止]' if aborted else '[任务已完成]'
|
||
rest = _clean('\n\n'.join(done[sent:] + ['\n\n' + tag]).strip())
|
||
if rest: _wx_send(rest[-3000:])
|
||
|
||
files = re.findall(r'\[FILE:([^\]]+)\]', result)
|
||
bad = {'filepath', '<filepath>', 'path', '<path>', 'file_path', '<file_path>', '...'}
|
||
files = [f for f in files if f.strip().lower() not in bad and (f if os.path.isabs(f) else os.path.join(_TEMP_DIR, f)) not in media_paths]
|
||
for fpath in set(files):
|
||
if not os.path.isabs(fpath): fpath = os.path.join(_TEMP_DIR, fpath)
|
||
try:
|
||
if not os.path.exists(fpath): raise FileNotFoundError(f"文件不存在: {fpath}")
|
||
ext = os.path.splitext(fpath)[1].lower()
|
||
sender = bot.send_video if ext in {'.mp4', '.mov', '.m4v', '.webm'} else \
|
||
bot.send_image if ext in {'.jpg', '.jpeg', '.png', '.gif', '.webp', '.bmp'} else bot.send_file
|
||
sender(uid, fpath, context_token=ctx)
|
||
print(f'[WX] sent media: {fpath}', file=sys.__stdout__)
|
||
except Exception as e: print(f'[WX] send media err: {e}', file=sys.__stdout__)
|
||
|
||
threading.Thread(target=_handle, daemon=True).start()
|
||
|
||
if __name__ == '__main__':
|
||
_do_relogin = '--relogin' in sys.argv
|
||
try: _lock = socket.socket(socket.AF_INET, socket.SOCK_STREAM); _lock.bind(('127.0.0.1', 19531))
|
||
except OSError: print('[WeChat] Another instance running, exiting.'); sys.exit(1)
|
||
_logf = open(os.path.join(os.path.dirname(os.path.dirname(__file__)), 'temp', 'wechatapp.log'), 'a', encoding='utf-8', buffering=1)
|
||
sys.stdout = sys.stderr = _logf
|
||
print(f'[NEW] Process starting {time.strftime("%m-%d %H:%M")}')
|
||
bot = WxBotClient()
|
||
if _do_relogin or not bot.token:
|
||
# QR 登录在无 TTY 的容器里也可用：把二维码打到真实 stdout（docker logs
|
||
# 可见），而不是日志文件——之前在重定向后才判 isatty()，文件句柄恒 false
|
||
# 导致容器内必然退出，无法首次登录。PNG 仍存 ~/.wxbot/wx_qr.png 作兜底。
|
||
sys.stdout = sys.stderr = sys.__stdout__ # restore for QR display (real stdout / container log)
|
||
try:
|
||
bot.login_qr()
|
||
finally:
|
||
sys.stdout = sys.stderr = _logf
|
||
threading.Thread(target=agent.run, daemon=True).start()
|
||
print(f'WeChat Bot 已启动 (bot_id={bot.bot_id})', file=sys.__stdout__)
|
||
try:
|
||
bot.run_loop(on_message)
|
||
except AuthExpired:
|
||
print('[Bot] token expired, exit.', file=sys.__stdout__)
|
||
sys.exit(2) |