feat: 优化 Windows 缓存加载和图片显示

This commit is contained in:
电摇小子
2026-07-14 11:07:35 +08:00
committed by 电摇小子
parent ad50939005
commit a776830144
12 changed files with 663 additions and 93 deletions
+203 -39
View File
@@ -1,4 +1,4 @@
import React, { useState } from 'react'
import React, { useState } from 'react'
import { Sidebar } from './components/Sidebar'
import ChatWindow from './components/ChatWindow'
import { SettingsPanel } from './components/SettingsPanel'
@@ -13,6 +13,7 @@ interface SelfInfo {
const MAC_KEY_FAQ_URL = 'https://github.com/hicccc77/WeFlow/blob/main/docs/MAC-KEY-FAQ.md'
const MESSAGE_MONITOR_DEBOUNCE_MS = 8000
const VIEW_MESSAGE_LIMIT = 600
const getMessageIdentity = (message: Message): string => {
if (message.localId) return `local:${message.localId}`
@@ -36,6 +37,12 @@ type GroupSnapshot = {
}
type GroupMemberMeta = { nickname: string; avatar: string }
type StartupProgress = {
title: string
subtitle: string
detail?: string
percent?: number
}
const formatGroupMemberName = (member: GroupSnapshot['members'][number]): string =>
member.nickname || member.wxid
@@ -100,8 +107,9 @@ function App(): React.ReactElement {
const [contacts, setContacts] = useState<Contact[]>([])
const [selectedContact, setSelectedContact] = useState<Contact | null>(null)
const [messages, setMessages] = useState<Message[]>([])
const [isMessagesLoading, setIsMessagesLoading] = useState(false)
const [filteredContacts, setFilteredContacts] = useState<Contact[]>([])
const [dateRange, setDateRange] = useState('today') // 默认今天
const [dateRange, setDateRange] = useState('today') // 默认今天
const [contentFilter, setContentFilter] = useState('')
const [isFetchingDbKey, setIsFetchingDbKey] = useState(false)
const [dbKeyStatus, setDbKeyStatus] = useState('')
@@ -113,10 +121,15 @@ function App(): React.ReactElement {
const [isNativeMonitorActive, setIsNativeMonitorActive] = useState(false)
const [bootState, setBootState] = useState<'loading' | 'connecting' | 'login'>('loading')
const [autoConnectSource, setAutoConnectSource] = useState<'env' | 'saved' | null>(null)
const [startupProgress, setStartupProgress] = useState<StartupProgress | null>(null)
const currentGroupSnapshotRef = React.useRef<GroupSnapshot | null>(null)
const syntheticGroupMessagesRef = React.useRef<Record<string, Message[]>>({})
const groupMemberMetaRef = React.useRef<Record<string, Map<string, GroupMemberMeta>>>({})
const selectedContactMd5Ref = React.useRef<string>('')
const contactAvatarHydrationRunRef = React.useRef(0)
const waitForPaint = (): Promise<void> =>
new Promise((resolve) => window.setTimeout(resolve, 80))
const refreshSelfInfo = async (): Promise<void> => {
try {
@@ -132,27 +145,80 @@ function App(): React.ReactElement {
}
}
const loadContacts = async (): Promise<void> => {
const loadBootstrapCache = async (): Promise<boolean> => {
try {
const cache = await window.api.getBootstrapCache()
if (!cache) return false
if (cache.contacts.length) {
setContacts(cache.contacts)
setFilteredContacts(cache.contacts)
}
if (cache.self) setSelfInfo(cache.self)
return Boolean(cache.contacts.length || cache.self)
} catch (error) {
console.warn('[BootstrapCache] 加载失败:', error)
return false
}
}
const loadContacts = async (options?: {
waitForAvatars?: boolean
onProgress?: (message: string, percent?: number) => void
}): Promise<void> => {
options?.onProgress?.('正在加载联系人...', 35)
const list = await window.api.getContacts()
setContacts(list)
setFilteredContacts(list)
void hydrateContactAvatars(list)
const runId = ++contactAvatarHydrationRunRef.current
const hydrate = (): Promise<void> => hydrateContactAvatars(list, runId, options?.onProgress)
if (options?.waitForAvatars) {
await hydrate()
} else {
window.setTimeout(() => {
void hydrate()
}, 1500)
}
}
const hydrateContactAvatars = async (list: Contact[]): Promise<void> => {
const hydrateContactAvatars = async (
list: Contact[],
runId: number,
onProgress?: (message: string, percent?: number) => void
): Promise<void> => {
const usernames = Array.from(
new Set(
list
.map((contact) => contact.m_nsUsrName)
.filter((username) => username && !username.startsWith('Group_') && !username.startsWith('Unknown_'))
.filter((username, index) => {
const contact = list[index]
return (
username &&
!contact.avatar &&
!username.startsWith('Group_') &&
!username.startsWith('Unknown_')
)
})
)
)
const chunkSize = 60
onProgress?.(
usernames.length ? `正在加载头像 0/${usernames.length}...` : '头像缓存已就绪',
usernames.length ? 55 : 90
)
let loadedCount = 0
const chunkSize = 8
for (let index = 0; index < usernames.length; index += chunkSize) {
if (runId !== contactAvatarHydrationRunRef.current) return
const chunk = usernames.slice(index, index + chunkSize)
if (chunk.length === 0) continue
try {
const avatars = await window.api.getContactAvatars(chunk)
if (runId !== contactAvatarHydrationRunRef.current) return
loadedCount += chunk.length
onProgress?.(
`正在加载头像 ${Math.min(loadedCount, usernames.length)}/${usernames.length}...`,
55 + Math.round((Math.min(loadedCount, usernames.length) / usernames.length) * 35)
)
if (Object.keys(avatars).length === 0) continue
setContacts((current) =>
current.map((contact) => avatars[contact.m_nsUsrName] ? { ...contact, avatar: avatars[contact.m_nsUsrName] } : contact)
)
@@ -162,16 +228,17 @@ function App(): React.ReactElement {
} catch (error) {
console.warn('[Contacts] avatar hydrate failed:', error)
}
await new Promise((resolve) => window.setTimeout(resolve, 50))
await new Promise((resolve) => window.setTimeout(resolve, 150))
}
if (usernames.length) onProgress?.('头像加载完成', 90)
}
React.useEffect(() => {
let active = true
const attemptAutoConnect = async (): Promise<void> => {
// 优先级 1:构建期环境变量 VITE_DB_KEY(本地开发/打包时硬编码的密钥)
// 优先级 1: 构建期环境变量 VITE_DB_KEY本地开发用)
const envKey = String(import.meta.env.VITE_DB_KEY || '').trim()
// 优先级 2:上一次保存到 safeStorage 的密钥
// 优先级 2: 上一次保存到 safeStorage 的密钥
let savedKey = ''
if (!envKey) {
const result = await window.api.getSavedDbKey()
@@ -197,7 +264,7 @@ function App(): React.ReactElement {
setDbKey(key)
setAutoConnectSource(envKey ? 'env' : 'saved')
setDbKeyStatus(
envKey ? '检测到环境变量中的密钥,正在自动连接...' : '已加载安全保存的密钥,正在自动连接...'
envKey ? '检测到环境变量中的密钥正在自动连接...' : '已加载安全保存的密钥正在自动连接...'
)
setDbKeyStatusKind('normal')
}
@@ -215,7 +282,7 @@ function App(): React.ReactElement {
} else {
const error = typeof result === 'boolean' ? '' : result.error
setDbKeyStatus(
`自动连接失败,请重新输入${error ? `: ${error}` : ''}`
`自动连接失败请重新输入${error ? `: ${error}` : ''}`
)
setDbKeyStatusKind('error')
setBootState('login')
@@ -249,22 +316,62 @@ function App(): React.ReactElement {
const handleLogin = async (keyInput?: string): Promise<void> => {
const keyToUse = keyInput || dbKey
if (!keyToUse) return
setBootState('connecting')
setStartupProgress({
title: '正在连接数据库...',
subtitle: '正在初始化微信数据',
detail: '请稍候',
percent: 8
})
try {
setStartupProgress({
title: '正在连接数据库...',
subtitle: '正在初始化微信数据',
detail: '正在打开 WCDB 数据库',
percent: 15
})
const result = await window.api.initDb(keyToUse)
const success = typeof result === 'boolean' ? result : result.success
if (success) {
setIsNativeMonitorActive(typeof result !== 'boolean' && result.monitoring === true)
setIsAuthenticated(true)
// 手动输入也持久化,下次启动可自动连接(参考 WeFlow)
setStartupProgress({
title: '正在读取缓存...',
subtitle: '正在恢复上次联系人和头像',
detail: '正在读取本地缓存',
percent: 25
})
const hasBootstrapCache = await loadBootstrapCache()
// 持久化手动输入的密钥,供下次启动继续使用
void window.api.saveDbKey(keyToUse).catch(() => undefined)
loadContacts()
setStartupProgress({
title: '正在加载账号信息...',
subtitle: '即将进入 WechatExplorer',
detail: '正在读取当前账号信息',
percent: 95
})
void refreshSelfInfo()
setStartupProgress({
title: '加载完成',
subtitle: '正在进入主页面',
detail: '联系人和头像已准备好',
percent: 100
})
setIsAuthenticated(true)
setBootState('login')
window.setTimeout(() => {
setStartupProgress(null)
if (!hasBootstrapCache) void loadContacts({ waitForAvatars: false })
}, 500)
} else {
const error = typeof result === 'boolean' ? '' : result.error
setBootState('login')
setStartupProgress(null)
alert(`Failed to open database.${error ? `\n\n${error}` : '\nCheck your key.'}`)
}
} catch (error) {
console.error(error)
setBootState('login')
setStartupProgress(null)
alert('Error connecting to database')
}
}
@@ -307,10 +414,17 @@ function App(): React.ReactElement {
return baseMessages.map((message) => {
const senderId = String(message.senderId || message.name || '').trim()
if (!senderId || !senderId.startsWith('wxid_')) return message
if (!senderId) return message
const member = memberMap.get(senderId)
if (!member) return message
const nickname = member.nickname && !member.nickname.startsWith('wxid_') ? member.nickname : senderId
const rawNickname = String(member.nickname || '').trim()
const nickname =
rawNickname &&
rawNickname !== senderId &&
!rawNickname.startsWith('wxid_') &&
!/^[a-z]{2,}\d{4,}$/i.test(rawNickname)
? rawNickname
: senderId
return {
...message,
name: nickname,
@@ -423,19 +537,41 @@ function App(): React.ReactElement {
setSelectedContact(contact)
selectedContactMd5Ref.current = contact.md5
currentGroupSnapshotRef.current = null
setIsMessagesLoading(true)
const { startTime, endTime } = getDateRangeParams(dateRange)
const msgs = await window.api.getMessages(contact.md5, startTime, endTime)
const cachedMsgs = await window.api.getCachedMessages(contact.md5, startTime, endTime)
if (selectedContactMd5Ref.current !== contact.md5) return
const cachedMessages = applyGroupMemberMeta(contact, mergeSyntheticMessages(contact, msgs))
setMessages(cachedMessages)
if (contact.type === 'group') {
void loadGroupMemberMeta(contact).then((snapshot) => {
if (selectedContactMd5Ref.current !== contact.md5) return
if (!snapshot) return
setMessages((current) =>
applyGroupMemberMeta(contact, mergeSyntheticMessages(contact, current, snapshot.roomId))
if (cachedMsgs.length) {
setMessages(
applyGroupMemberMeta(
contact,
mergeSyntheticMessages(contact, cachedMsgs.slice(-VIEW_MESSAGE_LIMIT))
)
)
} else {
setMessages([])
}
await waitForPaint()
try {
const msgs = await window.api.getMessages(contact.md5, startTime, endTime, {
limit: VIEW_MESSAGE_LIMIT
})
if (selectedContactMd5Ref.current !== contact.md5) return
const cachedMessages = applyGroupMemberMeta(contact, mergeSyntheticMessages(contact, msgs))
setMessages(cachedMessages)
if (contact.type === 'group') {
window.setTimeout(() => {
void loadGroupMemberMeta(contact).then((snapshot) => {
if (selectedContactMd5Ref.current !== contact.md5) return
if (!snapshot) return
setMessages((current) =>
applyGroupMemberMeta(contact, mergeSyntheticMessages(contact, current, snapshot.roomId))
)
})
}, 120)
}
} finally {
if (selectedContactMd5Ref.current === contact.md5) setIsMessagesLoading(false)
}
}
@@ -443,8 +579,22 @@ function App(): React.ReactElement {
setDateRange(range)
if (selectedContact) {
const { startTime, endTime } = getDateRangeParams(range)
window.api.getMessages(selectedContact.md5, startTime, endTime).then((nextMessages) => {
window.api.getCachedMessages(selectedContact.md5, startTime, endTime).then((cachedMessages) => {
if (!cachedMessages.length) return
setMessages(
applyGroupMemberMeta(
selectedContact,
mergeSyntheticMessages(selectedContact, cachedMessages.slice(-VIEW_MESSAGE_LIMIT))
)
)
})
setIsMessagesLoading(true)
window.api.getMessages(selectedContact.md5, startTime, endTime, { limit: VIEW_MESSAGE_LIMIT }).then((nextMessages) => {
setMessages(applyGroupMemberMeta(selectedContact, mergeSyntheticMessages(selectedContact, nextMessages)))
setIsMessagesLoading(false)
}).catch((error) => {
console.warn('[Messages] date range load failed:', error)
setIsMessagesLoading(false)
})
}
}
@@ -469,7 +619,8 @@ function App(): React.ReactElement {
const latestMessages = await window.api.getMessages(
contactMd5,
range.startTime,
range.endTime
range.endTime,
{ limit: VIEW_MESSAGE_LIMIT }
)
const nextMessages = applyGroupMemberMeta(
selectedContact,
@@ -561,19 +712,31 @@ function App(): React.ReactElement {
}, [resize, stopResizing])
if (!isAuthenticated && bootState !== 'login') {
const title =
startupProgress?.title || (bootState === 'connecting' ? '正在自动连接数据库...' : '正在准备...')
const subtitle =
startupProgress?.subtitle ||
(bootState === 'connecting'
? autoConnectSource === 'env'
? '检测到环境变量中的密钥'
: '使用上次安全保存的密钥'
: 'WechatExplorer')
return (
<div className="boot-splash">
<div className="boot-splash-spinner" aria-hidden />
<div className="boot-splash-title">
{bootState === 'connecting' ? '正在自动连接数据库...' : '正在准备...'}
</div>
<div className="boot-splash-subtitle">
{bootState === 'connecting'
? autoConnectSource === 'env'
? '检测到环境变量中的密钥'
: '使用上次安全保存的密钥'
: 'WechatExplorer'}
</div>
<div className="boot-splash-title">{title}</div>
<div className="boot-splash-subtitle">{subtitle}</div>
{startupProgress?.detail && (
<div className="boot-splash-detail">{startupProgress.detail}</div>
)}
{typeof startupProgress?.percent === 'number' && (
<div className="boot-splash-progress" aria-label="加载进度">
<div
className="boot-splash-progress-bar"
style={{ width: `${Math.max(0, Math.min(100, startupProgress.percent))}%` }}
/>
</div>
)}
</div>
)
}
@@ -597,7 +760,7 @@ function App(): React.ReactElement {
onClick={() => setShowDbKey(!showDbKey)}
title={showDbKey ? '隐藏密钥' : '显示密钥'}
>
{showDbKey ? '👁️' : '👁️‍🗨️'}
{showDbKey ? '隐藏' : '显示'}
</button>
</div>
<button
@@ -654,6 +817,7 @@ function App(): React.ReactElement {
key={`${selectedContact?.md5}-${contentFilter}`}
contact={selectedContact}
messages={messages}
isLoadingMessages={isMessagesLoading}
contentFilter={contentFilter}
onRefresh={() => selectedContact && handleSelectContact(selectedContact)}
onRefreshData={loadContacts}
+43 -5
View File
@@ -244,6 +244,18 @@ body {
margin-bottom: 14px;
}
.message-loading-pill {
align-self: center;
margin: 0 0 12px;
padding: 5px 12px;
border-radius: 999px;
background: rgba(255, 255, 255, 0.78);
color: #7a858b;
font-size: 12px;
line-height: 1.4;
box-shadow: 0 1px 4px rgba(15, 23, 42, 0.08);
}
.wechat-message-row {
display: flex;
align-items: flex-start;
@@ -947,11 +959,12 @@ body {
justify-content: center;
background: rgba(16, 24, 28, 0.28);
backdrop-filter: blur(1px);
overscroll-behavior: contain;
}
.image-viewer-window {
width: min(900px, 86vw);
height: min(700px, 82vh);
width: min(1280px, 96vw);
height: min(900px, 92vh);
display: flex;
flex-direction: column;
overflow: hidden;
@@ -1018,14 +1031,17 @@ body {
align-items: center;
justify-content: center;
padding: 28px;
overflow: auto;
overflow: hidden;
overscroll-behavior: contain;
cursor: grab;
user-select: none;
}
.image-viewer-stage img {
max-width: 100%;
max-height: 100%;
max-width: none;
max-height: none;
width: auto;
height: auto;
object-fit: contain;
box-shadow: 0 3px 16px rgba(0, 0, 0, 0.12);
transform-origin: center center;
@@ -1329,6 +1345,28 @@ body {
color: #6f767c;
}
.boot-splash-detail {
min-height: 18px;
font-size: 12px;
color: #8a9298;
}
.boot-splash-progress {
width: min(320px, 72vw);
height: 6px;
margin-top: 2px;
overflow: hidden;
border-radius: 999px;
background: rgba(7, 193, 96, 0.12);
}
.boot-splash-progress-bar {
height: 100%;
border-radius: inherit;
background: #07c160;
transition: width 0.18s ease-out;
}
/* 设置面板 */
.settings-overlay {
position: fixed;
+29 -3
View File
@@ -13,6 +13,7 @@ import {
interface ChatWindowProps {
contact: Contact | null
messages: Message[]
isLoadingMessages?: boolean
contentFilter?: string
onRefresh?: () => void
onRefreshData?: () => void
@@ -76,6 +77,7 @@ const getSummaryDateRange = (range: SummaryDateRange): { startTime: number; endT
const ChatWindow: React.FC<ChatWindowProps> = ({
contact,
messages,
isLoadingMessages,
contentFilter,
onRefresh,
onRefreshData
@@ -90,6 +92,7 @@ const ChatWindow: React.FC<ChatWindowProps> = ({
const [imageScale, setImageScale] = useState(0.75)
const [imageRotation, setImageRotation] = useState(0)
const [imageOffset, setImageOffset] = useState({ x: 0, y: 0 })
const imageViewerStageRef = useRef<HTMLDivElement>(null)
const imageDragRef = useRef<{ x: number; y: number; offsetX: number; offsetY: number } | null>(
null
)
@@ -133,7 +136,7 @@ const ChatWindow: React.FC<ChatWindowProps> = ({
const openImagePreview = (imageUrl: string): void => {
setPreviewImage(imageUrl)
setImageScale(0.75)
setImageScale(1)
setImageRotation(0)
setImageOffset({ x: 0, y: 0 })
}
@@ -144,17 +147,18 @@ const ChatWindow: React.FC<ChatWindowProps> = ({
}
const zoomImage = (delta: number): void => {
setImageScale((prev) => Math.min(3, Math.max(0.25, Number((prev + delta).toFixed(2)))))
setImageScale((prev) => Math.min(8, Math.max(0.1, Number((prev + delta).toFixed(2)))))
}
const resetImageTransform = (): void => {
setImageScale(0.75)
setImageScale(1)
setImageRotation(0)
setImageOffset({ x: 0, y: 0 })
}
const handleViewerWheel = (event: React.WheelEvent): void => {
event.preventDefault()
event.stopPropagation()
zoomImage(event.deltaY > 0 ? -0.1 : 0.1)
}
@@ -181,6 +185,24 @@ const ChatWindow: React.FC<ChatWindowProps> = ({
imageDragRef.current = null
}
useEffect(() => {
if (!previewImage) return
const previousOverflow = document.body.style.overflow
document.body.style.overflow = 'hidden'
const stage = imageViewerStageRef.current
const preventBackgroundWheel = (event: WheelEvent): void => {
event.preventDefault()
}
stage?.addEventListener('wheel', preventBackgroundWheel, { passive: false })
return () => {
document.body.style.overflow = previousOverflow
stage?.removeEventListener('wheel', preventBackgroundWheel)
}
}, [previewImage])
const handleExport = (days: number | 'all'): void => {
if (!messages.length) return
@@ -354,6 +376,9 @@ const ChatWindow: React.FC<ChatWindowProps> = ({
</div>
<div className="message-list wechat-message-list">
{isLoadingMessages && (
<div className="message-loading-pill">...</div>
)}
{hiddenMessageCount > 0 && (
<div className="wechat-system-message-row">
<div className="wechat-system-message">
@@ -562,6 +587,7 @@ const ChatWindow: React.FC<ChatWindowProps> = ({
</button>
</div>
<div
ref={imageViewerStageRef}
className="image-viewer-stage"
onWheel={handleViewerWheel}
onMouseDown={handleViewerMouseDown}
+14 -2
View File
@@ -63,7 +63,13 @@ export const Sidebar: React.FC<SidebarProps> = ({
>
<div className="contact-avatar">
{contact.avatar ? (
<img src={contact.avatar} alt={contact.m_nsNickName} referrerPolicy="no-referrer" />
<img
src={contact.avatar}
alt={contact.m_nsNickName}
referrerPolicy="no-referrer"
loading="lazy"
decoding="async"
/>
) : (
(contact.m_nsNickName || contact.m_nsUsrName || '?').charAt(0)
)}
@@ -131,7 +137,13 @@ export const Sidebar: React.FC<SidebarProps> = ({
<div className="sidebar-footer" onClick={onOpenSettings} title="设置">
<div className="sidebar-self-avatar">
{selfInfo?.avatar ? (
<img src={selfInfo.avatar} alt={selfInfo.nickname} referrerPolicy="no-referrer" />
<img
src={selfInfo.avatar}
alt={selfInfo.nickname}
referrerPolicy="no-referrer"
loading="lazy"
decoding="async"
/>
) : (
((selfInfo?.nickname || selfInfo?.wxid || '我').charAt(0))
)}