feat(account-data): 新增账号信息查询与项目数据删除能力

新增 /api/chat/account_info 与 DELETE /api/chat/account

删除账号时清理编辑记录、账号密钥缓存与导出/数据库目录

桌面端新增 app:getAccountInfo、app:deleteAccountData IPC 能力

前端 API 层新增 getChatAccountInfo、deleteChatAccount

Author: 2977094657

desktop/src/main.cjs

@@ -215,6 +215,161 @@ function getUserDataDir() {
   return resolveDataDir();
 }
 
+function sanitizeAccountName(account) {
+  const name = String(account || "").trim();
+  if (!name) throw new Error("缺少账号参数");
+  if (name === "." || name === "..") throw new Error("账号参数非法");
+  if (name.includes("/") || name.includes("\\")) throw new Error("账号参数非法");
+  return name;
+}
+
+function listDecryptedAccountsOnDisk(databasesDir) {
+  try {
+    if (!fs.existsSync(databasesDir)) return [];
+  } catch {
+    return [];
+  }
+
+  let entries = [];
+  try {
+    entries = fs.readdirSync(databasesDir, { withFileTypes: true });
+  } catch {
+    return [];
+  }
+
+  const accounts = [];
+  for (const entry of entries) {
+    try {
+      if (!entry || !entry.isDirectory()) continue;
+      const accountDir = path.join(databasesDir, entry.name);
+      const hasSession = fs.existsSync(path.join(accountDir, "session.db"));
+      const hasContact = fs.existsSync(path.join(accountDir, "contact.db"));
+      if (hasSession && hasContact) accounts.push(String(entry.name || ""));
+    } catch {}
+  }
+  accounts.sort((a, b) => a.localeCompare(b));
+  return accounts;
+}
+
+function resolveAccountDirInOutput(account) {
+  const dataDir = resolveDataDir();
+  if (!dataDir) throw new Error("无法定位数据目录");
+
+  const outputDir = path.join(dataDir, "output");
+  const databasesDir = path.join(outputDir, "databases");
+  const accountName = sanitizeAccountName(account);
+
+  const base = path.resolve(databasesDir);
+  const accountDir = path.resolve(path.join(databasesDir, accountName));
+  if (accountDir !== base && !accountDir.startsWith(base + path.sep)) {
+    throw new Error("账号路径非法");
+  }
+
+  return {
+    dataDir,
+    outputDir,
+    databasesDir,
+    accountName,
+    accountDir,
+  };
+}
+
+function getAccountInfoFromDisk(account) {
+  const { accountName, accountDir } = resolveAccountDirInOutput(account);
+  if (!fs.existsSync(accountDir) || !fs.statSync(accountDir).isDirectory()) {
+    throw new Error("账号数据不存在");
+  }
+
+  let entries = [];
+  try {
+    entries = fs.readdirSync(accountDir, { withFileTypes: true });
+  } catch {}
+  const dbFiles = entries
+    .filter((e) => !!e && e.isFile() && String(e.name || "").toLowerCase().endsWith(".db"))
+    .map((e) => String(e.name || ""))
+    .sort((a, b) => a.localeCompare(b));
+
+  let sessionUpdatedAt = 0;
+  try {
+    const st = fs.statSync(path.join(accountDir, "session.db"));
+    sessionUpdatedAt = Math.floor(Number(st?.mtimeMs || 0) / 1000);
+  } catch {}
+
+  return {
+    status: "success",
+    account: accountName,
+    path: accountDir,
+    database_count: dbFiles.length,
+    databases: dbFiles,
+    session_updated_at: sessionUpdatedAt,
+  };
+}
+
+function removeAccountFromKeyStore(dataDir, accountName) {
+  const keyStorePath = path.join(dataDir, "output", "account_keys.json");
+  try {
+    if (!fs.existsSync(keyStorePath)) return false;
+    const raw = fs.readFileSync(keyStorePath, { encoding: "utf8" });
+    const parsed = JSON.parse(raw || "{}");
+    if (!parsed || typeof parsed !== "object" || Array.isArray(parsed)) return false;
+    if (!Object.prototype.hasOwnProperty.call(parsed, accountName)) return false;
+    delete parsed[accountName];
+    fs.writeFileSync(keyStorePath, JSON.stringify(parsed, null, 2), { encoding: "utf8" });
+    return true;
+  } catch {
+    return false;
+  }
+}
+
+async function deleteAccountDataFromDisk(account) {
+  const { dataDir, outputDir, databasesDir, accountName, accountDir } = resolveAccountDirInOutput(account);
+  if (!fs.existsSync(accountDir) || !fs.statSync(accountDir).isDirectory()) {
+    throw new Error("账号数据不存在");
+  }
+
+  const wasBackendRunning = !!backendProc;
+  let restartError = null;
+  let result = null;
+
+  if (wasBackendRunning) {
+    await stopBackendAndWait({ timeoutMs: 10_000 });
+  }
+
+  try {
+    const exportsDir = path.join(outputDir, "exports", accountName);
+    try {
+      fs.rmSync(exportsDir, { recursive: true, force: true });
+    } catch {}
+
+    fs.rmSync(accountDir, { recursive: true, force: true });
+    const removedKeyCache = removeAccountFromKeyStore(dataDir, accountName);
+    const accounts = listDecryptedAccountsOnDisk(databasesDir);
+    result = {
+      status: "success",
+      deleted_account: accountName,
+      accounts,
+      default_account: accounts.length ? accounts[0] : null,
+      removed_key_cache: removedKeyCache,
+    };
+  } finally {
+    if (wasBackendRunning) {
+      try {
+        startBackend();
+        await waitForBackend({ timeoutMs: 30_000 });
+      } catch (err) {
+        restartError = err;
+        logMain(`[main] failed to restart backend after deleteAccountData: ${err?.message || err}`);
+      }
+    }
+  }
+
+  if (restartError) {
+    throw new Error(`删除完成，但后端重启失败：${restartError?.message || restartError}`);
+  }
+  if (!result) throw new Error("删除账号数据失败");
+  return result;
+}
+
 function getExeDir() {
   try {
     return path.dirname(process.execPath);
@@ -1384,6 +1539,22 @@ function registerWindowIpc() {
     }
   });
 
+  ipcMain.handle("app:getAccountInfo", async (_event, account) => {
+    try {
+      return getAccountInfoFromDisk(account);
+    } catch (e) {
+      throw new Error(e?.message || String(e));
+    }
+  });
+
+  ipcMain.handle("app:deleteAccountData", async (_event, account) => {
+    try {
+      return await deleteAccountDataFromDisk(account);
+    } catch (e) {
+      throw new Error(e?.message || String(e));
+    }
+  });
+
   ipcMain.handle("app:checkForUpdates", async () => {
     return await checkForUpdatesInternal();
   });

desktop/src/preload.cjs

@@ -22,6 +22,8 @@ contextBridge.exposeInMainWorld("wechatDesktop", {
   // Data/output folder helpers
   getOutputDir: () => ipcRenderer.invoke("app:getOutputDir"),
   openOutputDir: () => ipcRenderer.invoke("app:openOutputDir"),
+  getAccountInfo: (account) => ipcRenderer.invoke("app:getAccountInfo", String(account || "")),
+  deleteAccountData: (account) => ipcRenderer.invoke("app:deleteAccountData", String(account || "")),
 
   // Auto update
   getVersion: () => ipcRenderer.invoke("app:getVersion"),

frontend/composables/useApi.js

@@ -60,6 +60,22 @@ export const useApi = () => {
     return await request('/chat/accounts')
   }
 
+  const getChatAccountInfo = async (params = {}) => {
+    const query = new URLSearchParams()
+    if (params && params.account) query.set('account', params.account)
+    const url = '/chat/account_info' + (query.toString() ? `?${query.toString()}` : '')
+    return await request(url)
+  }
+
+  const deleteChatAccount = async (params = {}) => {
+    const account = String(params?.account || '').trim()
+    if (!account) throw new Error('Missing account')
+    const query = new URLSearchParams()
+    query.set('account', account)
+    const url = '/chat/account' + (query.toString() ? `?${query.toString()}` : '')
+    return await request(url, { method: 'DELETE' })
+  }
+
   const listChatSessions = async (params = {}) => {
     const query = new URLSearchParams()
     if (params && params.account) query.set('account', params.account)
@@ -540,6 +556,8 @@ export const useApi = () => {
     decryptDatabase,
     healthCheck,
     listChatAccounts,
+    getChatAccountInfo,
+    deleteChatAccount,
     listChatSessions,
     listChatMessages,
     getChatMessageRaw,

src/wechat_decrypt_tool/chat_edit_store.py

@@ -485,3 +485,30 @@ def update_message_edit_local_id(
                 conn.close()
         except Exception:
             pass
+
+
+def delete_account_edits(account: str) -> int:
+    a = str(account or "").strip()
+    if not a:
+        return 0
+
+    conn: Optional[sqlite3.Connection] = None
+    try:
+        conn = _connect()
+        cur = conn.execute(
+            """
+            DELETE FROM message_edits
+            WHERE account = ?
+            """,
+            (a,),
+        )
+        conn.commit()
+        return int(getattr(cur, "rowcount", 0) or 0)
+    except Exception:
+        return 0
+    finally:
+        try:
+            if conn is not None:
+                conn.close()
+        except Exception:
+            pass

src/wechat_decrypt_tool/key_store.py

@@ -67,3 +67,20 @@ def upsert_account_keys_in_store(
         pass
 
     return item
+
+
+def remove_account_keys_from_store(account: str) -> bool:
+    account = str(account or "").strip()
+    if not account:
+        return False
+
+    store = load_account_keys_store()
+    if account not in store:
+        return False
+
+    try:
+        store.pop(account, None)
+        _atomic_write_json(_KEY_STORE_PATH, store)
+        return True
+    except Exception:
+        return False

src/wechat_decrypt_tool/routers/chat.py

@@ -3,6 +3,7 @@
 import sqlite3
 import asyncio
 import json
+import shutil
 import time
 import threading
 from datetime import datetime, timedelta
@@ -67,6 +68,8 @@
 )
 from ..media_helpers import _resolve_account_db_storage_dir, _try_find_decrypted_resource
 from .. import chat_edit_store
+from ..app_paths import get_output_dir
+from ..key_store import remove_account_keys_from_store
 from ..path_fix import PathFixRoute
 from ..session_last_message import (
     build_session_last_message_table,
@@ -3496,6 +3499,85 @@ async def list_chat_accounts():
     }
 
 
+@router.get("/api/chat/account_info", summary="获取当前账号信息")
+def get_chat_account_info(account: Optional[str] = None):
+    account_dir = _resolve_account_dir(account)
+    db_files = sorted([p.name for p in account_dir.glob("*.db") if p.is_file()])
+
+    session_db = account_dir / "session.db"
+    session_updated_at = 0
+    try:
+        session_updated_at = int(session_db.stat().st_mtime)
+    except Exception:
+        session_updated_at = 0
+
+    return {
+        "status": "success",
+        "account": account_dir.name,
+        "path": str(account_dir),
+        "database_count": len(db_files),
+        "databases": db_files,
+        "session_updated_at": session_updated_at,
+    }
+
+
+@router.delete("/api/chat/account", summary="删除当前账号在本项目中的数据")
+def delete_chat_account(account: str):
+    account_name = str(account or "").strip()
+    if not account_name:
+        raise HTTPException(status_code=400, detail="Missing account.")
+
+    account_dir = _resolve_account_dir(account_name)
+
+    # Best-effort: close realtime connections first, otherwise Windows may keep db files locked.
+    try:
+        WCDB_REALTIME.disconnect(account_name)
+    except Exception:
+        pass
+
+    with _REALTIME_SYNC_MU:
+        _REALTIME_SYNC_ALL_LOCKS.pop(account_name, None)
+        stale_lock_keys = [k for k in _REALTIME_SYNC_LOCKS.keys() if k and k[0] == account_name]
+        for k in stale_lock_keys:
+            _REALTIME_SYNC_LOCKS.pop(k, None)
+
+    removed_edit_count = 0
+    try:
+        removed_edit_count = int(chat_edit_store.delete_account_edits(account_name) or 0)
+    except Exception:
+        removed_edit_count = 0
+
+    removed_key_cache = False
+    try:
+        removed_key_cache = bool(remove_account_keys_from_store(account_name))
+    except Exception:
+        removed_key_cache = False
+
+    output_dir = get_output_dir()
+    exports_dir = output_dir / "exports" / account_name
+    if exports_dir.exists():
+        try:
+            shutil.rmtree(exports_dir)
+        except Exception:
+            # Ignore export cleanup failure; account dir removal is the core operation.
+            pass
+
+    try:
+        shutil.rmtree(account_dir)
+    except Exception as e:
+        raise HTTPException(status_code=500, detail=f"删除账号数据失败：{e}")
+
+    accounts = _list_decrypted_accounts()
+    return {
+        "status": "success",
+        "deleted_account": account_name,
+        "accounts": accounts,
+        "default_account": accounts[0] if accounts else None,
+        "removed_edit_count": removed_edit_count,
+        "removed_key_cache": removed_key_cache,
+    }
+
+
 @router.get("/api/chat/sessions", summary="获取会话列表（聊天左侧列表）")
 def list_chat_sessions(
     request: Request,