digraph wechat_exporter {
    "开始" [shape=doublecircle];
    "工具目录存在?" [shape=diamond];
    "首次安装" [shape=box];
    "检测微信版本变化?" [shape=diamond];
    "重签名 + 重提取 key" [shape=box];
    "直接解密" [shape=box];
    "更新版本记录" [shape=box];
    "运行 decrypt_db.py" [shape=box];
    "输出摘要" [shape=box];
    "完成" [shape=doublecircle];

    "开始" -> "工具目录存在?";
    "工具目录存在?" -> "首次安装" [label="否"];
    "工具目录存在?" -> "检测微信版本变化?" [label="是"];
    "首次安装" -> "检测微信版本变化?";
    "检测微信版本变化?" -> "重签名 + 重提取 key" [label="版本变更或无 key"];
    "检测微信版本变化?" -> "直接解密" [label="版本未变"];
    "重签名 + 重提取 key" -> "更新版本记录";
    "直接解密" -> "运行 decrypt_db.py";
    "更新版本记录" -> "运行 decrypt_db.py";
    "运行 decrypt_db.py" -> "输出摘要";
    "输出摘要" -> "完成";
}

ls ~/wechat-decrypt-macos/

# 2.1 安装系统依赖
brew install sqlcipher

# 2.2 克隆主仓库
git clone https://github.com/cocohahaha/wechat-decrypt-macos.git ~/wechat-decrypt-macos
cd ~/wechat-decrypt-macos

# 2.3 下载解密脚本（ylytdeng/wechat-decrypt）
curl -sL https://raw.githubusercontent.com/ylytdeng/wechat-decrypt/main/find_all_keys_macos.c -o find_all_keys_macos.c
curl -sL https://raw.githubusercontent.com/ylytdeng/wechat-decrypt/main/decrypt_db.py -o decrypt_db.py
curl -sL https://raw.githubusercontent.com/ylytdeng/wechat-decrypt/main/config.py -o config.py
curl -sL https://raw.githubusercontent.com/ylytdeng/wechat-decrypt/main/key_utils.py -o key_utils.py

# 2.4 创建 Python 虚拟环境并安装依赖
python3 -m venv .venv
.venv/bin/pip install "mcp[cli]>=1.0.0" pycryptodome

# 2.5 编译内存扫描器
cc -O2 -o find_all_keys_macos find_all_keys_macos.c -framework Foundation

# 2.6 修改 server.py 读取明文数据库（见"server.py 必要修改"一节）

# 2.7 注册 MCP Server
claude mcp add -s user wechat \
  ~/wechat-decrypt-macos/.venv/bin/python \
  ~/wechat-decrypt-macos/server.py

CURRENT=$(defaults read /Applications/WeChat.app/Contents/Info.plist CFBundleShortVersionString)
STORED=$(cat ~/wechat-decrypt-macos/.wechat_version 2>/dev/null || echo "")
echo "当前版本: $CURRENT  存档版本: $STORED"

# 4.1 确认微信已登录
pgrep -l WeChat  # 必须有进程，否则提示用户先登录

# 4.2 重签名（如果是微信更新后第一次运行）
# ⚠️ 重签名前需退出微信，然后重新启动登录
sudo codesign --force --deep --sign - /Applications/WeChat.app
# 重新启动微信并等待用户登录

# 4.3 提取密钥（需要 sudo，必须让用户手动在终端执行）
cd ~/wechat-decrypt-macos
sudo ./find_all_keys_macos
# 输出 all_keys.json，包含各数据库的加密密钥

# 4.4 更新版本记录
echo "$CURRENT" > ~/wechat-decrypt-macos/.wechat_version

# 找到最近活跃的账号目录（当前登录账号）
ls -lt ~/Library/Containers/com.tencent.xinWeChat/Data/Documents/xwechat_files/ | \
  grep -v "^total\|Backup\|all_users\|old_backup" | head -5

ACCOUNT_DIR="<确认的账号目录名>"  # 例如 robbinfankai_f79d
cat > ~/wechat-decrypt-macos/config.json << EOF
{
    "db_dir": "$HOME/Library/Containers/com.tencent.xinWeChat/Data/Documents/xwechat_files/${ACCOUNT_DIR}/db_storage",
    "keys_file": "$HOME/wechat-decrypt-macos/all_keys.json",
    "decrypted_dir": "$HOME/wechat-decrypt-macos/decrypted",
    "wechat_process": "WeChat"
}
EOF

cd ~/wechat-decrypt-macos
.venv/bin/python decrypt_db.py

# 个人好友数（local_type=1 = 真实好友，排除公众号和群）
/usr/bin/sqlite3 ~/wechat-decrypt-macos/decrypted/contact/contact.db \
"SELECT count(*) FROM contact
 WHERE delete_flag=0 AND local_type=1
 AND username NOT LIKE 'gh_%'
 AND username NOT LIKE '%@chatroom';"

# 消息数据库文件数及总大小
ls ~/wechat-decrypt-macos/decrypted/message/message_[0-9]*.db | wc -l
du -sh ~/wechat-decrypt-macos/decrypted/message/

# 最新消息时间（从 session.db 查，比从 Msg_ 表查更可靠）
/usr/bin/sqlite3 ~/wechat-decrypt-macos/decrypted/session/session.db \
"SELECT datetime(max(last_timestamp), 'unixepoch', '+8 hours') FROM SessionTable;"

✅ 微信聊天记录导出完成
━━━━━━━━━━━━━━━━━━━━━━━━
个人联系人：2,540 人
消息数据库：12 个分片
数据库总大小：约 380MB
最新消息时间：2026-04-17 11:28
━━━━━━━━━━━━━━━━━━━━━━━━
MCP 工具已就绪，可直接查询。

import hashlib, sqlite3

conn = sqlite3.connect("message_0.db")
cur = conn.cursor()
cur.execute("SELECT user_name FROM Name2Id")
md5_to_user = {
    hashlib.md5(row[0].encode()).hexdigest(): row[0]
    for row in cur.fetchall()
}
# md5_to_user["6925d39..."] → "wxid_bc95l9h68t6z22"

# 改为：
SQLCIPHER_PATH = "/usr/bin/sqlite3"
DECRYPTED_DIR = os.path.join(os.path.dirname(__file__), "decrypted")

def _load_key() -> str:
    return ""

def _find_data_dir() -> str:
    if not os.path.isdir(DECRYPTED_DIR):
        raise FileNotFoundError(f"解密目录不存在: {DECRYPTED_DIR}")
    return DECRYPTED_DIR

def _preamble(key: str) -> str:
    return ""

def _test_key(key: str, db_path: str) -> bool:
    if not os.path.exists(db_path):
        return False
    try:
        result = subprocess.run(
            [SQLCIPHER_PATH, db_path, "SELECT count(*) FROM sqlite_master;"],
            capture_output=True, timeout=5,
        )
        return result.returncode == 0
    except Exception:
        return False

def _get_message_dbs() -> list[str]:
    data_dir = _find_data_dir()
    pattern = os.path.join(data_dir, "message", "message_*.db")
    dbs = [
        db for db in glob.glob(pattern)
        if os.path.basename(db).replace("message_", "").replace(".db", "").isdigit()
    ]
    return sorted(dbs, key=lambda p: int(os.path.basename(p).replace("message_", "").replace(".db", "")))