mirror of
https://github.com/pandorafuture/wx-cli.git
synced 2026-08-29 04:00:55 +00:00
pandorafuture
96ccbbdc3d
WeChat macOS database decryption and query tool, supporting key extraction, message querying, full-text search, conversation export, real-time monitoring, media handling, and HTTP server mode with REST API and SSE.
66 lines
2.2 KiB
Rust
66 lines
2.2 KiB
Rust
//! WeChat database context layer — account resolution, decrypt orchestration,
|
|
//! and encrypted direct-open support.
|
|
//!
|
|
//! When `raw_key` is available, use [`open_encrypted_db`] or
|
|
//! [`open_encrypted_db_with_pool`] to open encrypted databases directly
|
|
//! via SQLCipher's `sqlite3_key()` C API, bypassing the decrypt-cache pipeline.
|
|
|
|
mod account;
|
|
mod cache;
|
|
mod contact;
|
|
mod db_category;
|
|
mod decrypt_request;
|
|
mod decrypt_scope;
|
|
mod error;
|
|
mod file_lock;
|
|
mod fts_tokenizer;
|
|
mod kdf_cache;
|
|
mod patch_state;
|
|
mod progress;
|
|
pub mod shard_routing;
|
|
pub mod tokenizer;
|
|
mod visibility;
|
|
mod wal_patch;
|
|
|
|
pub use account::{AccountContext, ResolveParams};
|
|
pub use cache::PersistentCache;
|
|
pub use contact::{ContactResolver, Direction};
|
|
pub use db_category::{discover_db_files, DbCategory, DbFile};
|
|
pub use decrypt_request::DecryptRequest;
|
|
pub use decrypt_scope::DecryptScope;
|
|
pub use error::ContextError;
|
|
pub use fts_tokenizer::{
|
|
open_fts_connection, open_fts_connection_with_key, register_mm_fts_tokenizer,
|
|
};
|
|
pub use progress::{DecryptProgress, DecryptStats};
|
|
pub use shard_routing::{route_shards_for_query, write_shard_metadata_sidecar};
|
|
pub use visibility::VisibilityIndex;
|
|
|
|
/// Open encrypted WeChat DB directory directly (no pool, no FTS).
|
|
/// For one-shot commands: contacts, sessions, query, search, export.
|
|
pub fn open_encrypted_db(account: &AccountContext) -> Result<wx_db::WechatDb, ContextError> {
|
|
let raw_key = account
|
|
.raw_key
|
|
.ok_or_else(|| ContextError::Cache("raw_key required for encrypted direct open".into()))?;
|
|
let encrypted_root = account.data_dir.join("db_storage");
|
|
let db = wx_db::WechatDb::open_encrypted(&encrypted_root, raw_key)?;
|
|
Ok(db)
|
|
}
|
|
|
|
/// Open encrypted WeChat DB directory with connection pool and FTS tokenizer.
|
|
/// For long-running serve mode only.
|
|
pub fn open_encrypted_db_with_pool(
|
|
account: &AccountContext,
|
|
) -> Result<wx_db::WechatDb, ContextError> {
|
|
let raw_key = account
|
|
.raw_key
|
|
.ok_or_else(|| ContextError::Cache("raw_key required for encrypted direct open".into()))?;
|
|
let encrypted_root = account.data_dir.join("db_storage");
|
|
let db = wx_db::WechatDb::open_encrypted_with_pool(
|
|
&encrypted_root,
|
|
raw_key,
|
|
register_mm_fts_tokenizer,
|
|
)?;
|
|
Ok(db)
|
|
}
|