mirror of
https://github.com/jackwener/wx-cli.git
synced 2026-10-08 11:45:46 +00:00
feat: daemon + CLI 架构 (wx_daemon.py + wx.py)
- wx_daemon.py: Unix socket server，mtime 感知 DB 缓存，WAL 监听，实时推送 - wx.py: Click CLI，自动拉起 daemon，sessions/history/search/contacts/watch - pyproject.toml + uv.lock: uv 依赖管理
This commit is contained in:
jackwener
@@ -0,0 +1,355 @@
|
||||
"""
|
||||
wx - 微信本地数据 CLI
|
||||
|
||||
自动管理 daemon 生命周期，无需用户手动启动。
|
||||
|
||||
用法:
|
||||
wx sessions 最近会话
|
||||
wx history "张三" 聊天记录
|
||||
wx search "关键词" 搜索消息
|
||||
wx contacts 联系人列表
|
||||
wx watch 实时监听新消息
|
||||
wx daemon status/stop/logs daemon 管理
|
||||
"""
|
||||
|
||||
import json
|
||||
import os
|
||||
import socket
|
||||
import subprocess
|
||||
import sys
|
||||
import time
|
||||
|
||||
import click
|
||||
|
||||
CLI_DIR = os.path.join(os.path.expanduser("~"), ".wechat-cli")
|
||||
SOCK_PATH = os.path.join(CLI_DIR, "daemon.sock")
|
||||
PID_PATH = os.path.join(CLI_DIR, "daemon.pid")
|
||||
LOG_PATH = os.path.join(CLI_DIR, "daemon.log")
|
||||
DAEMON_SCRIPT = os.path.join(os.path.dirname(os.path.abspath(__file__)), "wx_daemon.py")
|
||||
STARTUP_TIMEOUT = 15 # 等待 daemon 启动的最长秒数
|
||||
|
||||
# ─── daemon 管理 ─────────────────────────────────────────────────────────────
|
||||
|
||||
def _is_alive() -> bool:
|
||||
if not os.path.exists(SOCK_PATH):
|
||||
return False
|
||||
try:
|
||||
s = socket.socket(socket.AF_UNIX, socket.SOCK_STREAM)
|
||||
s.settimeout(2)
|
||||
s.connect(SOCK_PATH)
|
||||
s.sendall(b'{"cmd":"ping"}\n')
|
||||
resp = json.loads(s.makefile().readline())
|
||||
s.close()
|
||||
return resp.get("pong") is True
|
||||
except Exception:
|
||||
return False
|
||||
|
||||
|
||||
def _start_daemon() -> None:
|
||||
subprocess.Popen(
|
||||
[sys.executable, DAEMON_SCRIPT],
|
||||
stdout=subprocess.DEVNULL,
|
||||
stderr=subprocess.DEVNULL,
|
||||
start_new_session=True,
|
||||
)
|
||||
deadline = time.time() + STARTUP_TIMEOUT
|
||||
while time.time() < deadline:
|
||||
time.sleep(0.3)
|
||||
if _is_alive():
|
||||
return
|
||||
raise click.ClickException(
|
||||
f"wx-daemon 启动超时（>{STARTUP_TIMEOUT}s）\n"
|
||||
f"请查看日志: {LOG_PATH}"
|
||||
)
|
||||
|
||||
|
||||
def _ensure_daemon() -> None:
|
||||
if not _is_alive():
|
||||
click.echo("⏳ 启动 wx-daemon...", err=True)
|
||||
_start_daemon()
|
||||
|
||||
# ─── 通信 ────────────────────────────────────────────────────────────────────
|
||||
|
||||
def _send(req: dict, timeout: int = 30) -> dict:
|
||||
_ensure_daemon()
|
||||
s = socket.socket(socket.AF_UNIX, socket.SOCK_STREAM)
|
||||
s.settimeout(timeout)
|
||||
s.connect(SOCK_PATH)
|
||||
s.sendall((json.dumps(req, ensure_ascii=False) + '\n').encode())
|
||||
resp = json.loads(s.makefile().readline())
|
||||
s.close()
|
||||
if not resp.get("ok"):
|
||||
raise click.ClickException(resp.get("error", "未知错误"))
|
||||
return resp
|
||||
|
||||
# ─── 时间解析 ────────────────────────────────────────────────────────────────
|
||||
|
||||
def _parse_time(value: str, is_end: bool = False) -> int:
|
||||
from datetime import datetime
|
||||
for fmt in ('%Y-%m-%d %H:%M:%S', '%Y-%m-%d %H:%M', '%Y-%m-%d'):
|
||||
try:
|
||||
dt = datetime.strptime(value, fmt)
|
||||
if fmt == '%Y-%m-%d' and is_end:
|
||||
dt = dt.replace(hour=23, minute=59, second=59)
|
||||
return int(dt.timestamp())
|
||||
except ValueError:
|
||||
continue
|
||||
raise click.BadParameter(
|
||||
f"无法解析时间 '{value}'，支持 YYYY-MM-DD / YYYY-MM-DD HH:MM / YYYY-MM-DD HH:MM:SS"
|
||||
)
|
||||
|
||||
# ─── CLI ─────────────────────────────────────────────────────────────────────
|
||||
|
||||
@click.group(context_settings={"help_option_names": ["-h", "--help"]})
|
||||
@click.version_option("0.1.0", prog_name="wx")
|
||||
def cli():
|
||||
"""wx — 微信本地数据 CLI"""
|
||||
|
||||
|
||||
# ─── sessions ────────────────────────────────────────────────────────────────
|
||||
|
||||
@cli.command()
|
||||
@click.option('-n', '--limit', default=20, show_default=True, help='会话数量')
|
||||
@click.option('--json', 'as_json', is_flag=True, help='输出原始 JSON')
|
||||
def sessions(limit, as_json):
|
||||
"""列出最近会话"""
|
||||
resp = _send({"cmd": "sessions", "limit": limit})
|
||||
data = resp.get("sessions", [])
|
||||
|
||||
if as_json:
|
||||
click.echo(json.dumps(data, ensure_ascii=False, indent=2))
|
||||
return
|
||||
|
||||
for s in data:
|
||||
unread = f" \033[31m({s['unread']}未读)\033[0m" if s.get('unread', 0) > 0 else ''
|
||||
group = ' [群]' if s['is_group'] else ''
|
||||
sender = f"{s['last_sender']}: " if s.get('last_sender') else ''
|
||||
click.echo(f"\033[90m[{s['time']}]\033[0m \033[1m{s['chat']}\033[0m{group}{unread}")
|
||||
click.echo(f" {s['last_msg_type']}: {sender}{s['summary']}")
|
||||
click.echo()
|
||||
|
||||
|
||||
# ─── history ─────────────────────────────────────────────────────────────────
|
||||
|
||||
@cli.command()
|
||||
@click.argument('chat')
|
||||
@click.option('-n', '--limit', default=50, show_default=True, help='消息数量')
|
||||
@click.option('--offset', default=0, help='分页偏移')
|
||||
@click.option('--since', default=None, metavar='DATE', help='起始时间 YYYY-MM-DD')
|
||||
@click.option('--until', default=None, metavar='DATE', help='结束时间 YYYY-MM-DD')
|
||||
@click.option('--json', 'as_json', is_flag=True, help='输出原始 JSON')
|
||||
def history(chat, limit, offset, since, until, as_json):
|
||||
"""查看聊天记录
|
||||
|
||||
\b
|
||||
示例:
|
||||
wx history "张三"
|
||||
wx history "AI群" --since 2026-04-01 --until 2026-04-15
|
||||
wx history "张三" -n 100 --offset 50
|
||||
"""
|
||||
req = {"cmd": "history", "chat": chat, "limit": limit, "offset": offset}
|
||||
if since:
|
||||
req["since"] = _parse_time(since)
|
||||
if until:
|
||||
req["until"] = _parse_time(until, is_end=True)
|
||||
|
||||
resp = _send(req)
|
||||
|
||||
if as_json:
|
||||
click.echo(json.dumps(resp.get("messages", []), ensure_ascii=False, indent=2))
|
||||
return
|
||||
|
||||
group = ' [群]' if resp.get('is_group') else ''
|
||||
click.echo(f"=== {resp['chat']}{group} ({resp['count']} 条) ===\n")
|
||||
for m in resp.get("messages", []):
|
||||
sender = f"\033[33m{m['sender']}\033[0m: " if m.get('sender') else ''
|
||||
click.echo(f"\033[90m[{m['time']}]\033[0m {sender}{m['content']}")
|
||||
|
||||
|
||||
# ─── search ──────────────────────────────────────────────────────────────────
|
||||
|
||||
@cli.command()
|
||||
@click.argument('keyword')
|
||||
@click.option('--in', 'chats', multiple=True, metavar='CHAT', help='限定聊天（可多次指定）')
|
||||
@click.option('-n', '--limit', default=20, show_default=True)
|
||||
@click.option('--since', default=None, metavar='DATE')
|
||||
@click.option('--until', default=None, metavar='DATE')
|
||||
@click.option('--json', 'as_json', is_flag=True)
|
||||
def search(keyword, chats, limit, since, until, as_json):
|
||||
"""搜索消息
|
||||
|
||||
\b
|
||||
示例:
|
||||
wx search "Claude"
|
||||
wx search "deadline" --in "TeamA" --in "TeamB"
|
||||
wx search "会议" --since 2026-04-01
|
||||
"""
|
||||
req = {"cmd": "search", "keyword": keyword, "limit": limit}
|
||||
if chats:
|
||||
req["chats"] = list(chats)
|
||||
if since:
|
||||
req["since"] = _parse_time(since)
|
||||
if until:
|
||||
req["until"] = _parse_time(until, is_end=True)
|
||||
|
||||
resp = _send(req)
|
||||
results = resp.get("results", [])
|
||||
|
||||
if as_json:
|
||||
click.echo(json.dumps(results, ensure_ascii=False, indent=2))
|
||||
return
|
||||
|
||||
click.echo(f'搜索 "{keyword}"，找到 {resp["count"]} 条:\n')
|
||||
for r in results:
|
||||
sender = f"\033[33m{r['sender']}\033[0m: " if r.get('sender') else ''
|
||||
chat = f"\033[36m[{r['chat']}]\033[0m " if r.get('chat') else ''
|
||||
click.echo(f"\033[90m[{r['time']}]\033[0m {chat}{sender}{r['content']}")
|
||||
|
||||
|
||||
# ─── contacts ────────────────────────────────────────────────────────────────
|
||||
|
||||
@cli.command()
|
||||
@click.option('-q', '--query', default=None, help='按名字过滤')
|
||||
@click.option('-n', '--limit', default=50, show_default=True)
|
||||
@click.option('--json', 'as_json', is_flag=True)
|
||||
def contacts(query, limit, as_json):
|
||||
"""查看联系人
|
||||
|
||||
\b
|
||||
示例:
|
||||
wx contacts
|
||||
wx contacts -q "李"
|
||||
"""
|
||||
resp = _send({"cmd": "contacts", "query": query, "limit": limit})
|
||||
data = resp.get("contacts", [])
|
||||
|
||||
if as_json:
|
||||
click.echo(json.dumps(data, ensure_ascii=False, indent=2))
|
||||
return
|
||||
|
||||
click.echo(f"共 {resp.get('total', len(data))} 个联系人（显示 {len(data)} 个）:\n")
|
||||
for c in data:
|
||||
click.echo(f" {c['display']:<20} {c['username']}")
|
||||
|
||||
|
||||
# ─── watch ───────────────────────────────────────────────────────────────────
|
||||
|
||||
@cli.command()
|
||||
@click.option('--chat', default=None, help='只显示指定聊天的消息')
|
||||
@click.option('--json', 'as_json', is_flag=True, help='输出 JSON lines（方便 jq 处理）')
|
||||
def watch(chat, as_json):
|
||||
"""实时监听新消息（Ctrl+C 退出）
|
||||
|
||||
\b
|
||||
示例:
|
||||
wx watch
|
||||
wx watch --chat "AI交流群"
|
||||
wx watch --json | jq .content
|
||||
"""
|
||||
_ensure_daemon()
|
||||
s = socket.socket(socket.AF_UNIX, socket.SOCK_STREAM)
|
||||
s.connect(SOCK_PATH)
|
||||
s.sendall((json.dumps({"cmd": "watch"}) + '\n').encode())
|
||||
|
||||
if not as_json:
|
||||
click.echo("监听中（Ctrl+C 退出）...\n", err=True)
|
||||
|
||||
try:
|
||||
for line in s.makefile():
|
||||
line = line.strip()
|
||||
if not line:
|
||||
continue
|
||||
try:
|
||||
event = json.loads(line)
|
||||
except Exception:
|
||||
continue
|
||||
|
||||
evt = event.get("event", "")
|
||||
if evt in ("connected", "heartbeat"):
|
||||
continue
|
||||
|
||||
# 过滤指定聊天
|
||||
if chat and event.get("chat") != chat and event.get("username") != chat:
|
||||
continue
|
||||
|
||||
if as_json:
|
||||
click.echo(line)
|
||||
continue
|
||||
|
||||
time_s = event.get('time', '')
|
||||
chat_s = event.get('chat', '')
|
||||
is_group = event.get('is_group', False)
|
||||
sender = event.get('sender', '')
|
||||
content = event.get('content', '')
|
||||
|
||||
chat_part = f"\033[36m[{chat_s}]\033[0m " if is_group else f"\033[1m{chat_s}\033[0m "
|
||||
sender_part = f"\033[33m{sender}\033[0m: " if sender else ''
|
||||
click.echo(f"\033[90m[{time_s}]\033[0m {chat_part}{sender_part}{content}")
|
||||
|
||||
except KeyboardInterrupt:
|
||||
pass
|
||||
finally:
|
||||
try:
|
||||
s.close()
|
||||
except Exception:
|
||||
pass
|
||||
|
||||
|
||||
# ─── daemon 子命令组 ──────────────────────────────────────────────────────────
|
||||
|
||||
@cli.group()
|
||||
def daemon():
|
||||
"""管理 wx-daemon"""
|
||||
|
||||
|
||||
@daemon.command()
|
||||
def status():
|
||||
"""查看 daemon 运行状态"""
|
||||
if _is_alive():
|
||||
pid = open(PID_PATH).read().strip() if os.path.exists(PID_PATH) else '?'
|
||||
click.echo(f"✓ wx-daemon 运行中 (PID {pid})")
|
||||
else:
|
||||
click.echo("✗ wx-daemon 未运行")
|
||||
|
||||
|
||||
@daemon.command()
|
||||
def stop():
|
||||
"""停止 daemon"""
|
||||
if not os.path.exists(PID_PATH):
|
||||
click.echo("daemon 未运行")
|
||||
return
|
||||
try:
|
||||
pid = int(open(PID_PATH).read().strip())
|
||||
import signal
|
||||
os.kill(pid, signal.SIGTERM)
|
||||
click.echo(f"✓ 已停止 wx-daemon (PID {pid})")
|
||||
except (ValueError, ProcessLookupError):
|
||||
click.echo("daemon 进程不存在，清理残留文件")
|
||||
for p in (SOCK_PATH, PID_PATH):
|
||||
try:
|
||||
os.unlink(p)
|
||||
except OSError:
|
||||
pass
|
||||
|
||||
|
||||
@daemon.command()
|
||||
@click.option('-f', '--follow', is_flag=True, help='持续输出（tail -f）')
|
||||
@click.option('-n', '--lines', default=50, show_default=True, help='显示最近 N 行')
|
||||
def logs(follow, lines):
|
||||
"""查看 daemon 日志"""
|
||||
if not os.path.exists(LOG_PATH):
|
||||
click.echo("暂无日志")
|
||||
return
|
||||
if follow:
|
||||
import subprocess as sp
|
||||
sp.run(['tail', f'-{lines}', '-f', LOG_PATH])
|
||||
else:
|
||||
with open(LOG_PATH) as f:
|
||||
all_lines = f.readlines()
|
||||
click.echo(''.join(all_lines[-lines:]), nl=False)
|
||||
|
||||
|
||||
# ─── 入口 ────────────────────────────────────────────────────────────────────
|
||||
|
||||
if __name__ == "__main__":
|
||||
cli()
|
||||
Reference in New Issue
Block a user