cc
2e41a03c96
fix: 修复了虚拟机上无法索引到wxkey的问题；修复图片密钥扫描的问题；修复年度报告错误；修复了年度报告和数据分析中的发送者错误问题；修复了部分页面偶发的未渲染名称问题；修复了头像偶发渲染失败的问题；修复了部分图片无法解密的问题
564 lines
20 KiB
TypeScript
564 lines
20 KiB
TypeScript
import { useState, useEffect } from 'react'
|
|
import { useNavigate } from 'react-router-dom'
|
|
import { useAppStore } from '../stores/appStore'
|
|
import { dialog } from '../services/ipc'
|
|
import * as configService from '../services/config'
|
|
import {
|
|
ArrowLeft, ArrowRight, CheckCircle2, Database, Eye, EyeOff,
|
|
FolderOpen, FolderSearch, KeyRound, ShieldCheck, Sparkles,
|
|
UserRound, Wand2, Minus, X, HardDrive, RotateCcw
|
|
} from 'lucide-react'
|
|
import './WelcomePage.scss'
|
|
|
|
const steps = [
|
|
{ id: 'intro', title: '欢迎', desc: '准备开始你的本地数据探索' },
|
|
{ id: 'db', title: '数据库目录', desc: '定位 xwechat_files 目录' },
|
|
{ id: 'cache', title: '缓存目录', desc: '设置本地缓存存储位置（可选）' },
|
|
{ id: 'key', title: '解密密钥', desc: '获取密钥与自动识别账号' },
|
|
{ id: 'image', title: '图片密钥', desc: '获取 XOR 与 AES 密钥' }
|
|
]
|
|
|
|
interface WelcomePageProps {
|
|
standalone?: boolean
|
|
}
|
|
|
|
function WelcomePage({ standalone = false }: WelcomePageProps) {
|
|
const navigate = useNavigate()
|
|
const { isDbConnected, setDbConnected, setLoading } = useAppStore()
|
|
|
|
const [stepIndex, setStepIndex] = useState(0)
|
|
const [dbPath, setDbPath] = useState('')
|
|
const [decryptKey, setDecryptKey] = useState('')
|
|
const [imageXorKey, setImageXorKey] = useState('')
|
|
const [imageAesKey, setImageAesKey] = useState('')
|
|
const [cachePath, setCachePath] = useState('')
|
|
const [wxid, setWxid] = useState('')
|
|
const [wxidOptions, setWxidOptions] = useState<Array<{ wxid: string; modifiedTime: number }>>([])
|
|
const [error, setError] = useState('')
|
|
const [isConnecting, setIsConnecting] = useState(false)
|
|
const [isDetectingPath, setIsDetectingPath] = useState(false)
|
|
const [isScanningWxid, setIsScanningWxid] = useState(false)
|
|
const [isFetchingDbKey, setIsFetchingDbKey] = useState(false)
|
|
const [isFetchingImageKey, setIsFetchingImageKey] = useState(false)
|
|
const [showDecryptKey, setShowDecryptKey] = useState(false)
|
|
const [isClosing, setIsClosing] = useState(false)
|
|
const [dbKeyStatus, setDbKeyStatus] = useState('')
|
|
const [imageKeyStatus, setImageKeyStatus] = useState('')
|
|
const [isManualStartPrompt, setIsManualStartPrompt] = useState(false)
|
|
|
|
useEffect(() => {
|
|
const removeDb = window.electronAPI.key.onDbKeyStatus((payload) => {
|
|
setDbKeyStatus(payload.message)
|
|
})
|
|
const removeImage = window.electronAPI.key.onImageKeyStatus((payload) => {
|
|
setImageKeyStatus(payload.message)
|
|
})
|
|
return () => {
|
|
removeDb?.()
|
|
removeImage?.()
|
|
}
|
|
}, [])
|
|
|
|
useEffect(() => {
|
|
if (isDbConnected && !standalone) {
|
|
navigate('/home')
|
|
}
|
|
}, [isDbConnected, standalone, navigate])
|
|
|
|
useEffect(() => {
|
|
setWxidOptions([])
|
|
setWxid('')
|
|
}, [dbPath])
|
|
|
|
const currentStep = steps[stepIndex]
|
|
const rootClassName = `welcome-page${isClosing ? ' is-closing' : ''}${standalone ? ' is-standalone' : ''}`
|
|
const showWindowControls = standalone
|
|
|
|
const handleMinimize = () => {
|
|
window.electronAPI.window.minimize()
|
|
}
|
|
|
|
const handleCloseWindow = () => {
|
|
window.electronAPI.window.close()
|
|
}
|
|
|
|
const handleSelectPath = async () => {
|
|
try {
|
|
const result = await dialog.openFile({
|
|
title: '选择微信数据库目录',
|
|
properties: ['openDirectory']
|
|
})
|
|
|
|
if (!result.canceled && result.filePaths.length > 0) {
|
|
setDbPath(result.filePaths[0])
|
|
setError('')
|
|
}
|
|
} catch (e) {
|
|
setError('选择目录失败')
|
|
}
|
|
}
|
|
|
|
const handleAutoDetectPath = async () => {
|
|
if (isDetectingPath) return
|
|
setIsDetectingPath(true)
|
|
setError('')
|
|
try {
|
|
const result = await window.electronAPI.dbPath.autoDetect()
|
|
if (result.success && result.path) {
|
|
setDbPath(result.path)
|
|
setError('')
|
|
} else {
|
|
setError(result.error || '未能检测到数据库目录')
|
|
}
|
|
} catch (e) {
|
|
setError(`自动检测失败: ${e}`)
|
|
} finally {
|
|
setIsDetectingPath(false)
|
|
}
|
|
}
|
|
|
|
const handleSelectCachePath = async () => {
|
|
try {
|
|
const result = await dialog.openFile({
|
|
title: '选择缓存目录',
|
|
properties: ['openDirectory']
|
|
})
|
|
|
|
if (!result.canceled && result.filePaths.length > 0) {
|
|
setCachePath(result.filePaths[0])
|
|
setError('')
|
|
}
|
|
} catch (e) {
|
|
setError('选择缓存目录失败')
|
|
}
|
|
}
|
|
|
|
const handleScanWxid = async (silent = false) => {
|
|
if (!dbPath) {
|
|
if (!silent) setError('请先选择数据库目录')
|
|
return
|
|
}
|
|
if (isScanningWxid) return
|
|
setIsScanningWxid(true)
|
|
if (!silent) setError('')
|
|
try {
|
|
const wxids = await window.electronAPI.dbPath.scanWxids(dbPath)
|
|
setWxidOptions(wxids)
|
|
if (wxids.length > 0) {
|
|
// scanWxids 已经按时间排过序了，直接取第一个
|
|
setWxid(wxids[0].wxid)
|
|
if (!silent) setError('')
|
|
} else {
|
|
if (!silent) setError('未检测到账号目录，请检查路径')
|
|
}
|
|
} catch (e) {
|
|
if (!silent) setError(`扫描失败: ${e}`)
|
|
} finally {
|
|
setIsScanningWxid(false)
|
|
}
|
|
}
|
|
|
|
const handleAutoGetDbKey = async () => {
|
|
if (isFetchingDbKey) return
|
|
setIsFetchingDbKey(true)
|
|
setError('')
|
|
setIsManualStartPrompt(false)
|
|
setDbKeyStatus('正在连接微信进程...')
|
|
try {
|
|
const result = await window.electronAPI.key.autoGetDbKey()
|
|
if (result.success && result.key) {
|
|
setDecryptKey(result.key)
|
|
setDbKeyStatus('密钥获取成功')
|
|
setError('')
|
|
// 获取成功后自动扫描并填入 wxid
|
|
await handleScanWxid(true)
|
|
} else {
|
|
if (result.error?.includes('未找到微信安装路径') || result.error?.includes('启动微信失败')) {
|
|
setIsManualStartPrompt(true)
|
|
setDbKeyStatus('需要手动启动微信')
|
|
} else {
|
|
setError(result.error || '自动获取密钥失败')
|
|
}
|
|
}
|
|
} catch (e) {
|
|
setError(`自动获取密钥失败: ${e}`)
|
|
} finally {
|
|
setIsFetchingDbKey(false)
|
|
}
|
|
}
|
|
|
|
const handleManualConfirm = async () => {
|
|
setIsManualStartPrompt(false)
|
|
handleAutoGetDbKey()
|
|
}
|
|
|
|
const handleAutoGetImageKey = async () => {
|
|
if (isFetchingImageKey) return
|
|
if (!dbPath) {
|
|
setError('请先选择数据库目录')
|
|
return
|
|
}
|
|
setIsFetchingImageKey(true)
|
|
setError('')
|
|
setImageKeyStatus('正在准备获取图片密钥...')
|
|
try {
|
|
// 拼接完整的账号目录，确保 KeyService 能准确找到模板文件
|
|
const accountPath = wxid ? `${dbPath}/${wxid}` : dbPath
|
|
const result = await window.electronAPI.key.autoGetImageKey(accountPath)
|
|
if (result.success && result.aesKey) {
|
|
if (typeof result.xorKey === 'number') {
|
|
setImageXorKey(`0x${result.xorKey.toString(16).toUpperCase().padStart(2, '0')}`)
|
|
}
|
|
setImageAesKey(result.aesKey)
|
|
setImageKeyStatus('已获取图片密钥')
|
|
} else {
|
|
setError(result.error || '自动获取图片密钥失败')
|
|
}
|
|
} catch (e) {
|
|
setError(`自动获取图片密钥失败: ${e}`)
|
|
} finally {
|
|
setIsFetchingImageKey(false)
|
|
}
|
|
}
|
|
|
|
const canGoNext = () => {
|
|
if (currentStep.id === 'intro') return true
|
|
if (currentStep.id === 'db') return Boolean(dbPath)
|
|
if (currentStep.id === 'cache') return true
|
|
if (currentStep.id === 'key') return decryptKey.length === 64 && Boolean(wxid)
|
|
if (currentStep.id === 'image') return true
|
|
return false
|
|
}
|
|
|
|
const handleNext = () => {
|
|
if (!canGoNext()) {
|
|
if (currentStep.id === 'db' && !dbPath) setError('请先选择数据库目录')
|
|
if (currentStep.id === 'key') {
|
|
if (decryptKey.length !== 64) setError('密钥长度必须为 64 个字符')
|
|
else if (!wxid) setError('未能自动识别 wxid，请尝试重新获取或检查目录')
|
|
}
|
|
return
|
|
}
|
|
setError('')
|
|
setStepIndex((prev) => Math.min(prev + 1, steps.length - 1))
|
|
}
|
|
|
|
const handleBack = () => {
|
|
setError('')
|
|
setStepIndex((prev) => Math.max(prev - 1, 0))
|
|
}
|
|
|
|
const handleConnect = async () => {
|
|
if (!dbPath) { setError('请先选择数据库目录'); return }
|
|
if (!wxid) { setError('请填写微信ID'); return }
|
|
if (!decryptKey || decryptKey.length !== 64) { setError('请填写 64 位解密密钥'); return }
|
|
|
|
setIsConnecting(true)
|
|
setError('')
|
|
setLoading(true, '正在连接数据库...')
|
|
|
|
try {
|
|
const result = await window.electronAPI.wcdb.testConnection(dbPath, decryptKey, wxid)
|
|
if (!result.success) {
|
|
setError(result.error || 'WCDB 连接失败')
|
|
setLoading(false)
|
|
return
|
|
}
|
|
|
|
await configService.setDbPath(dbPath)
|
|
await configService.setDecryptKey(decryptKey)
|
|
await configService.setMyWxid(wxid)
|
|
await configService.setCachePath(cachePath)
|
|
if (imageXorKey) {
|
|
const parsed = parseInt(imageXorKey.replace(/^0x/i, ''), 16)
|
|
if (!Number.isNaN(parsed)) {
|
|
await configService.setImageXorKey(parsed)
|
|
}
|
|
}
|
|
if (imageAesKey) {
|
|
await configService.setImageAesKey(imageAesKey)
|
|
}
|
|
await configService.setOnboardingDone(true)
|
|
|
|
setDbConnected(true, dbPath)
|
|
setLoading(false)
|
|
|
|
if (standalone) {
|
|
setIsClosing(true)
|
|
setTimeout(() => {
|
|
window.electronAPI.window.completeOnboarding()
|
|
}, 450)
|
|
} else {
|
|
navigate('/home')
|
|
}
|
|
} catch (e) {
|
|
setError(`连接失败: ${e}`)
|
|
setLoading(false)
|
|
} finally {
|
|
setIsConnecting(false)
|
|
}
|
|
}
|
|
|
|
const formatModifiedTime = (time: number) => {
|
|
if (!time) return '未知时间'
|
|
const date = new Date(time)
|
|
const year = date.getFullYear()
|
|
const month = String(date.getMonth() + 1).padStart(2, '0')
|
|
const day = String(date.getDate()).padStart(2, '0')
|
|
const hours = String(date.getHours()).padStart(2, '0')
|
|
const minutes = String(date.getMinutes()).padStart(2, '0')
|
|
return `${year}-${month}-${day} ${hours}:${minutes}`
|
|
}
|
|
|
|
if (isDbConnected) {
|
|
return (
|
|
<div className={rootClassName}>
|
|
{showWindowControls && (
|
|
<div className="window-controls">
|
|
<button type="button" className="window-btn" onClick={handleMinimize} aria-label="最小化">
|
|
<Minus size={14} />
|
|
</button>
|
|
<button type="button" className="window-btn is-close" onClick={handleCloseWindow} aria-label="关闭">
|
|
<X size={14} />
|
|
</button>
|
|
</div>
|
|
)}
|
|
<div className="welcome-shell">
|
|
<div className="welcome-panel">
|
|
<div className="panel-header">
|
|
<img src="./logo.png" alt="WeFlow" className="panel-logo" />
|
|
<div>
|
|
<p className="panel-kicker">WeFlow</p>
|
|
<h1>已连接数据库</h1>
|
|
</div>
|
|
</div>
|
|
<div className="panel-note">
|
|
<CheckCircle2 size={16} />
|
|
<span>配置已完成，可直接进入首页</span>
|
|
</div>
|
|
<button
|
|
className="btn btn-primary btn-full"
|
|
onClick={() => {
|
|
if (standalone) {
|
|
setIsClosing(true)
|
|
setTimeout(() => {
|
|
window.electronAPI.window.completeOnboarding()
|
|
}, 450)
|
|
} else {
|
|
navigate('/home')
|
|
}
|
|
}}
|
|
>
|
|
进入首页
|
|
</button>
|
|
</div>
|
|
</div>
|
|
</div>
|
|
)
|
|
}
|
|
|
|
return (
|
|
<div className={rootClassName}>
|
|
{showWindowControls && (
|
|
<div className="window-controls">
|
|
<button type="button" className="window-btn" onClick={handleMinimize} aria-label="最小化">
|
|
<Minus size={14} />
|
|
</button>
|
|
<button type="button" className="window-btn is-close" onClick={handleCloseWindow} aria-label="关闭">
|
|
<X size={14} />
|
|
</button>
|
|
</div>
|
|
)}
|
|
<div className="welcome-shell">
|
|
<div className="welcome-panel">
|
|
<div className="panel-header">
|
|
<img src="./logo.png" alt="WeFlow" className="panel-logo" />
|
|
<div>
|
|
<p className="panel-kicker">首次配置</p>
|
|
<h1>WeFlow 初始引导</h1>
|
|
<p className="panel-subtitle">一步一步完成数据库与密钥设置</p>
|
|
</div>
|
|
</div>
|
|
<div className="step-list">
|
|
{steps.map((step, index) => (
|
|
<div key={step.id} className={`step-item ${index === stepIndex ? 'active' : ''} ${index < stepIndex ? 'done' : ''}`}>
|
|
<div className="step-index">{index < stepIndex ? <CheckCircle2 size={14} /> : index + 1}</div>
|
|
<div>
|
|
<div className="step-title">{step.title}</div>
|
|
<div className="step-desc">{step.desc}</div>
|
|
</div>
|
|
</div>
|
|
))}
|
|
</div>
|
|
<div className="panel-foot">
|
|
<ShieldCheck size={16} />
|
|
<span>数据仅在本地处理，不上传服务器</span>
|
|
</div>
|
|
</div>
|
|
|
|
<div className="setup-card">
|
|
<div className="setup-header">
|
|
<div className="setup-icon">
|
|
{currentStep.id === 'intro' && <Sparkles size={18} />}
|
|
{currentStep.id === 'db' && <Database size={18} />}
|
|
{currentStep.id === 'cache' && <HardDrive size={18} />}
|
|
{currentStep.id === 'key' && <KeyRound size={18} />}
|
|
{currentStep.id === 'image' && <ShieldCheck size={18} />}
|
|
</div>
|
|
<div>
|
|
<h2>{currentStep.title}</h2>
|
|
<p>{currentStep.desc}</p>
|
|
</div>
|
|
</div>
|
|
|
|
{currentStep.id === 'intro' && (
|
|
<div className="setup-body">
|
|
<div className="intro-card">
|
|
<Wand2 size={18} />
|
|
<div>
|
|
<h3>准备好了吗？</h3>
|
|
<p>接下来只需配置数据库目录和获取解密密钥。</p>
|
|
</div>
|
|
</div>
|
|
</div>
|
|
)}
|
|
|
|
{currentStep.id === 'db' && (
|
|
<div className="setup-body">
|
|
<label className="field-label">数据库根目录</label>
|
|
<input
|
|
type="text"
|
|
className="field-input"
|
|
placeholder="例如：C:\\Users\\xxx\\Documents\\xwechat_files"
|
|
value={dbPath}
|
|
onChange={(e) => setDbPath(e.target.value)}
|
|
/>
|
|
<div className="button-row">
|
|
<button className="btn btn-secondary" onClick={handleAutoDetectPath} disabled={isDetectingPath}>
|
|
<FolderSearch size={16} /> {isDetectingPath ? '检测中...' : '自动检测'}
|
|
</button>
|
|
<button className="btn btn-primary" onClick={handleSelectPath}>
|
|
<FolderOpen size={16} /> 浏览选择
|
|
</button>
|
|
</div>
|
|
<div className="field-hint">建议选择包含 xwechat_files 的目录</div>
|
|
</div>
|
|
)}
|
|
|
|
{currentStep.id === 'cache' && (
|
|
<div className="setup-body">
|
|
<label className="field-label">缓存目录</label>
|
|
<input
|
|
type="text"
|
|
className="field-input"
|
|
placeholder="留空使用默认目录"
|
|
value={cachePath}
|
|
onChange={(e) => setCachePath(e.target.value)}
|
|
/>
|
|
<div className="button-row">
|
|
<button className="btn btn-primary" onClick={handleSelectCachePath}>
|
|
<FolderOpen size={16} /> 浏览选择
|
|
</button>
|
|
<button className="btn btn-secondary" onClick={() => setCachePath('')}>
|
|
<RotateCcw size={16} /> 使用默认
|
|
</button>
|
|
</div>
|
|
<div className="field-hint">用于头像、表情与图片缓存，留空使用默认目录</div>
|
|
</div>
|
|
)}
|
|
|
|
{currentStep.id === 'key' && (
|
|
<div className="setup-body">
|
|
<label className="field-label">微信账号 wxid</label>
|
|
<input
|
|
type="text"
|
|
className="field-input"
|
|
placeholder="获取密钥后将自动填充"
|
|
value={wxid}
|
|
onChange={(e) => setWxid(e.target.value)}
|
|
/>
|
|
<label className="field-label">解密密钥</label>
|
|
<div className="field-with-toggle">
|
|
<input
|
|
type={showDecryptKey ? 'text' : 'password'}
|
|
className="field-input"
|
|
placeholder="64 位十六进制密钥"
|
|
value={decryptKey}
|
|
onChange={(e) => setDecryptKey(e.target.value.trim())}
|
|
/>
|
|
<button type="button" className="toggle-btn" onClick={() => setShowDecryptKey(!showDecryptKey)}>
|
|
{showDecryptKey ? <EyeOff size={14} /> : <Eye size={14} />}
|
|
</button>
|
|
</div>
|
|
|
|
{isManualStartPrompt ? (
|
|
<div className="manual-prompt">
|
|
<p className="prompt-text">未能自动启动微信，请手动启动并登录后点击下方确认</p>
|
|
<button className="btn btn-primary" onClick={handleManualConfirm}>
|
|
我已启动微信，继续检测
|
|
</button>
|
|
</div>
|
|
) : (
|
|
<button className="btn btn-secondary btn-inline" onClick={handleAutoGetDbKey} disabled={isFetchingDbKey}>
|
|
{isFetchingDbKey ? '获取中...' : '自动获取密钥'}
|
|
</button>
|
|
)}
|
|
|
|
{dbKeyStatus && <div className="field-hint status-text">{dbKeyStatus}</div>}
|
|
<div className="field-hint">获取密钥会自动识别最近登录的账号</div>
|
|
<div className="field-hint">点击自动获取后微信将重新启动，当页面提示可以登录微信了再点击登录</div>
|
|
</div>
|
|
)}
|
|
|
|
{currentStep.id === 'image' && (
|
|
<div className="setup-body">
|
|
<label className="field-label">图片 XOR 密钥</label>
|
|
<input
|
|
type="text"
|
|
className="field-input"
|
|
placeholder="例如：0xA4"
|
|
value={imageXorKey}
|
|
onChange={(e) => setImageXorKey(e.target.value)}
|
|
/>
|
|
<label className="field-label">图片 AES 密钥</label>
|
|
<input
|
|
type="text"
|
|
className="field-input"
|
|
placeholder="16 位密钥"
|
|
value={imageAesKey}
|
|
onChange={(e) => setImageAesKey(e.target.value)}
|
|
/>
|
|
<button className="btn btn-secondary btn-inline" onClick={handleAutoGetImageKey} disabled={isFetchingImageKey}>
|
|
{isFetchingImageKey ? '获取中...' : '自动获取图片密钥'}
|
|
</button>
|
|
{imageKeyStatus && <div className="field-hint status-text">{imageKeyStatus}</div>}
|
|
<div className="field-hint">如获取失败，请先打开朋友圈图片再重试</div>
|
|
{isFetchingImageKey && <div className="field-hint status-text">正在扫描内存，请稍候...</div>}
|
|
</div>
|
|
)}
|
|
|
|
{error && <div className="error-message">{error}</div>}
|
|
|
|
<div className="setup-actions">
|
|
<button className="btn btn-tertiary" onClick={handleBack} disabled={stepIndex === 0}>
|
|
<ArrowLeft size={16} /> 上一步
|
|
</button>
|
|
{stepIndex < steps.length - 1 ? (
|
|
<button className="btn btn-primary" onClick={handleNext} disabled={!canGoNext()}>
|
|
下一步 <ArrowRight size={16} />
|
|
</button>
|
|
) : (
|
|
<button className="btn btn-primary" onClick={handleConnect} disabled={isConnecting || !canGoNext()}>
|
|
{isConnecting ? '连接中...' : '测试并完成'}
|
|
</button>
|
|
)}
|
|
</div>
|
|
</div>
|
|
</div>
|
|
</div>
|
|
)
|
|
}
|
|
|
|
export default WelcomePage
|
|
|