feat: 实现退群监控

This commit is contained in:
电摇小子
2026-06-30 16:04:01 +08:00
parent a50e903e55
commit f96ada906c
9 changed files with 319 additions and 12 deletions
+37 -4
View File
@@ -265,19 +265,22 @@ app.whenReady().then(() => {
/<appmsg\b|<refermsg\b|&lt;appmsg\b|&lt;refermsg\b/i.test(content) /<appmsg\b|<refermsg\b|&lt;appmsg\b|&lt;refermsg\b/i.test(content)
? 49 ? 49
: msgType : msgType
if ([3, 42, 47, 48, 49, 50].includes(inferredMsgType)) { if ([3, 42, 47, 48, 49, 50, 10000, 10002].includes(inferredMsgType)) {
try { try {
const parsed = const parsed =
inferredMsgType === 47 inferredMsgType === 47
? parseStickerMessageFromRow(msg, content) ? parseStickerMessageFromRow(msg, content)
: parseMessageContent(content, inferredMsgType) : parseMessageContent(content, inferredMsgType)
if (parsed.type !== 'unknown') { if (parsed.type === 'system') {
content = parsed.content
contentData = parsed
} else if (parsed.type !== 'unknown') {
content = '' content = ''
} }
if (parsed.type === 'image') { if (parsed.type === 'image') {
const imageDatName = parseImageDatNameFromRow(msg) const imageDatName = parseImageDatNameFromRow(msg)
contentData = { ...parsed, datName: parsed.datName || imageDatName } contentData = { ...parsed, datName: parsed.datName || imageDatName }
} else { } else if (parsed.type !== 'system') {
if (parsed.type === 'sticker' && !parsed.url && parsed.md5 && wcdb4Client) { if (parsed.type === 'sticker' && !parsed.url && parsed.md5 && wcdb4Client) {
parsed.url = wcdb4Client.resolveEmoticonCdnUrl(parsed.md5) parsed.url = wcdb4Client.resolveEmoticonCdnUrl(parsed.md5)
} }
@@ -313,7 +316,7 @@ app.whenReady().then(() => {
return { return {
id: msg.mesLocalID || Math.random().toString(), id: msg.mesLocalID || Math.random().toString(),
from: isMine ? 'assistant' : 'user', from: contentData?.type === 'system' ? 'system' : isMine ? 'assistant' : 'user',
type: displayType, type: displayType,
datetime: date.toLocaleString('zh-CN', { hour12: false }), datetime: date.toLocaleString('zh-CN', { hour12: false }),
content: content, content: content,
@@ -327,6 +330,36 @@ app.whenReady().then(() => {
}) })
}) })
ipcMain.handle('db:getGroupSnapshot', (_, userMd5: string) => {
if (!wechatDb) return null
const wcdb4Client = wechatDb.getWcdb4Client()
if (!wcdb4Client) return null
const roomId = wcdb4Client.getUsernameByMd5(userMd5)
if (!roomId || !roomId.endsWith('@chatroom')) return null
const members = wcdb4Client
.getGroupMembers(roomId)
.filter((member) => member?.m_nsUsrName)
.map((member) => ({
wxid: member.m_nsUsrName,
nickname: member.nickname || '',
avatar: member.m_nsHeadImgUrl || ''
}))
console.log(
`[GroupSnapshot] roomId=${roomId} memberCount=${members.length} members=${members
.map((member) => `${member.nickname || member.wxid}(${member.wxid})`)
.join(', ')}`
)
return {
roomId,
memberCount: members.length,
members
}
})
ipcMain.handle('db:search', (_, keyword: string) => { ipcMain.handle('db:search', (_, keyword: string) => {
if (!wechatDb) return null if (!wechatDb) return null
return wechatDb.searchAllMessages(keyword) return wechatDb.searchAllMessages(keyword)
+82 -2
View File
@@ -41,7 +41,7 @@ type QuoteContent = {
quotedSender?: string quotedSender?: string
quotedType?: string quotedType?: string
} }
type SystemContent = { type: 'system'; content: string } type SystemContent = { type: 'system'; content: string; raw?: string }
type UnknownContent = { type: 'unknown'; raw: string } type UnknownContent = { type: 'unknown'; raw: string }
export type ParsedContent = export type ParsedContent =
@@ -79,12 +79,40 @@ export function parseMessageContent(content: string, messageType: number): Parse
return parseVoipMessage(normalized) return parseVoipMessage(normalized)
case 10000: case 10000:
case 10002: case 10002:
return { type: 'system', content: normalized } return parseSystemMessage(normalized)
default: default:
return { type: 'text', content: normalized } return { type: 'text', content: normalized }
} }
} }
function parseSystemMessage(content: string): ParsedContent {
const stripped = stripChatroomPrefix(content)
const decoded = decodeXmlEntities(stripped)
const delChatroomMemberText = extractDelChatroomMemberText(decoded)
if (delChatroomMemberText) {
return {
type: 'system',
content: normalizeSystemText(delChatroomMemberText),
raw: content
}
}
const plainText =
extractXmlNodeText(decoded, 'plain') ||
extractXmlNodeText(decoded, 'text') ||
extractXmlNodeText(decoded, 'title') ||
extractXmlValue(decoded, 'plain') ||
extractXmlValue(decoded, 'text') ||
extractXmlValue(decoded, 'title') ||
''
const normalized = normalizeSystemText(plainText || fallbackSystemText(decoded))
return {
type: 'system',
content: normalized || '[系统消息]',
raw: content
}
}
function parseImageMessage(content: string): ParsedContent { function parseImageMessage(content: string): ParsedContent {
// 尝试 XML 格式: <img md5="..." aeskey="..."/> // 尝试 XML 格式: <img md5="..." aeskey="..."/>
let md5 = extractXmlAttribute(content, 'img', 'md5') || extractXmlValue(content, 'md5') || '' let md5 = extractXmlAttribute(content, 'img', 'md5') || extractXmlValue(content, 'md5') || ''
@@ -356,6 +384,20 @@ function sanitizeQuotedContent(content: string): string {
return decoded return decoded
} }
function stripChatroomPrefix(content: string): string {
return String(content || '')
.replace(/^[0-9a-z_-]+@chatroom:\s*/i, '')
.replace(/^wxid_[^:\n]+:\s*/i, '')
.trim()
}
function normalizeSystemText(content: string): string {
return String(content || '')
.replace(/\s+/g, ' ')
.replace(/\s+([,.;!?])/g, '$1')
.trim()
}
function parseVoipMessage(content: string): ParsedContent { function parseVoipMessage(content: string): ParsedContent {
const roomTypeStr = extractXmlValue(content, 'room_type') const roomTypeStr = extractXmlValue(content, 'room_type')
const msg = extractXmlValue(content, 'msg') || '' const msg = extractXmlValue(content, 'msg') || ''
@@ -416,6 +458,44 @@ function decodeXmlUrl(value: string): string {
} }
} }
function decodeXmlEntities(value: string): string {
return String(value || '')
.replace(/&lt;/g, '<')
.replace(/&gt;/g, '>')
.replace(/&quot;/g, '"')
.replace(/&#39;/g, "'")
.replace(/&amp;/g, '&')
}
function extractXmlNodeText(xml: string, tagName: string): string {
const match = new RegExp(`<${tagName}[^>]*>([\\s\\S]*?)<\\/${tagName}>`, 'i').exec(xml)
if (!match?.[1]) return ''
return normalizeSystemText(
decodeXmlEntities(
match[1].replace(/<!\[CDATA\[([\s\S]*?)\]\]>/g, '$1').replace(/<[^>]+>/g, ' ')
)
)
}
function fallbackSystemText(xml: string): string {
return normalizeSystemText(
decodeXmlEntities(
String(xml || '')
.replace(/<!\[CDATA\[([\s\S]*?)\]\]>/g, '$1')
.replace(/<[^>]+>/g, ' ')
)
)
}
function extractDelChatroomMemberText(xml: string): string {
if (!/<sysmsg[^>]+delchatroommember/i.test(xml)) return ''
const plainMatch = /<plain[^>]*><!\[CDATA\[([\s\S]*?)\]\]><\/plain>/i.exec(xml)
if (plainMatch?.[1]) return plainMatch[1].trim()
const textMatch = /<text[^>]*><!\[CDATA\[([\s\S]*?)\]\]><\/text>/i.exec(xml)
if (textMatch?.[1]) return textMatch[1].trim()
return ''
}
function normalizeMd5(value: unknown): string | undefined { function normalizeMd5(value: unknown): string | undefined {
const md5 = String(value || '') const md5 = String(value || '')
.trim() .trim()
+5
View File
@@ -39,6 +39,11 @@ declare global {
) => Promise<boolean | { success: boolean; error?: string; monitoring?: boolean }> ) => Promise<boolean | { success: boolean; error?: string; monitoring?: boolean }>
getContacts: (filter?: string) => Promise<Contact[]> getContacts: (filter?: string) => Promise<Contact[]>
getMessages: (userMd5: string, startTime?: number, endTime?: number) => Promise<Message[]> getMessages: (userMd5: string, startTime?: number, endTime?: number) => Promise<Message[]>
getGroupSnapshot: (userMd5: string) => Promise<{
roomId: string
memberCount: number
members: { wxid: string; nickname: string; avatar: string }[]
} | null>
search: (keyword: string) => Promise<string | null> search: (keyword: string) => Promise<string | null>
aiChat: ( aiChat: (
messages: { role: string; content: string }[], messages: { role: string; content: string }[],
+1
View File
@@ -8,6 +8,7 @@ const api = {
getContacts: (filter?: string) => ipcRenderer.invoke('db:getContacts', filter), getContacts: (filter?: string) => ipcRenderer.invoke('db:getContacts', filter),
getMessages: (userMd5: string, startTime?: number, endTime?: number) => getMessages: (userMd5: string, startTime?: number, endTime?: number) =>
ipcRenderer.invoke('db:getMessages', userMd5, startTime, endTime), ipcRenderer.invoke('db:getMessages', userMd5, startTime, endTime),
getGroupSnapshot: (userMd5: string) => ipcRenderer.invoke('db:getGroupSnapshot', userMd5),
search: (keyword: string) => ipcRenderer.invoke('db:search', keyword), search: (keyword: string) => ipcRenderer.invoke('db:search', keyword),
aiChat: ( aiChat: (
messages: { role: string; content: string }[], messages: { role: string; content: string }[],
+149 -4
View File
@@ -21,6 +21,67 @@ const areMessagesEquivalent = (left: Message[], right: Message[]): boolean => {
return true return true
} }
type GroupSnapshot = {
roomId: string
memberCount: number
members: { wxid: string; nickname: string; avatar: string }[]
}
const formatGroupMemberName = (member: GroupSnapshot['members'][number]): string =>
member.nickname || member.wxid
const buildSyntheticGroupMessages = (
previous: GroupSnapshot | null,
next: GroupSnapshot | null,
referenceMessages: Message[]
): Message[] => {
if (!previous || !next || previous.roomId !== next.roomId) return []
const previousMap = new Map(previous.members.map((member) => [member.wxid, member]))
const nextMap = new Map(next.members.map((member) => [member.wxid, member]))
const latestMessageTime = referenceMessages.reduce(
(max, message) => Math.max(max, message.createTime || 0),
0
)
const fallbackNow = Math.floor(Date.now() / 1000)
const events: Message[] = []
let offset = 1
for (const [wxid, member] of previousMap.entries()) {
if (nextMap.has(wxid)) continue
const name = formatGroupMemberName(member)
const eventTime = Math.max(latestMessageTime + offset, fallbackNow)
const eventDate = new Date(eventTime * 1000)
offset += 1
events.push({
id: `synthetic-leave:${next.roomId}:${wxid}:${eventTime}`,
from: 'system',
type: '系统消息',
datetime: eventDate.toLocaleString('zh-CN', { hour12: false }),
content: `${name} 退出了群聊`,
isSender: false,
createTime: eventTime
})
}
if (events.length) {
console.log(
`[GroupMonitor] synthetic leave detected roomId=${next.roomId} events=${events
.map((event) => event.content)
.join(' | ')}`
)
}
return events
}
const sortMessagesChronologically = (items: Message[]): Message[] =>
[...items].sort((left, right) => {
const timeDelta = (left.createTime || 0) - (right.createTime || 0)
if (timeDelta !== 0) return timeDelta
return getMessageIdentity(left).localeCompare(getMessageIdentity(right))
})
function App(): React.ReactElement { function App(): React.ReactElement {
const [isAuthenticated, setIsAuthenticated] = useState(false) const [isAuthenticated, setIsAuthenticated] = useState(false)
const [dbKey, setDbKey] = useState(import.meta.env.VITE_DB_KEY || '') const [dbKey, setDbKey] = useState(import.meta.env.VITE_DB_KEY || '')
@@ -36,6 +97,8 @@ function App(): React.ReactElement {
const [showDbKey, setShowDbKey] = useState(false) const [showDbKey, setShowDbKey] = useState(false)
const [showMacKeyFaq, setShowMacKeyFaq] = useState(false) const [showMacKeyFaq, setShowMacKeyFaq] = useState(false)
const [isNativeMonitorActive, setIsNativeMonitorActive] = useState(false) const [isNativeMonitorActive, setIsNativeMonitorActive] = useState(false)
const currentGroupSnapshotRef = React.useRef<GroupSnapshot | null>(null)
const syntheticGroupMessagesRef = React.useRef<Record<string, Message[]>>({})
React.useEffect(() => { React.useEffect(() => {
let active = true let active = true
@@ -84,6 +147,41 @@ function App(): React.ReactElement {
} }
} }
const logGroupSnapshot = React.useCallback(
async (contact: Contact | null, reason: string): Promise<GroupSnapshot | null> => {
if (!contact || contact.type !== 'group') return null
try {
const snapshot = (await window.api.getGroupSnapshot(contact.md5)) as GroupSnapshot | null
if (!snapshot) {
console.log(`[GroupSnapshot] reason=${reason} name=${contact.m_nsNickName} snapshot=null`)
return null
}
console.log(
`[GroupSnapshot] reason=${reason} name=${contact.m_nsNickName} roomId=${snapshot.roomId} memberCount=${snapshot.memberCount}`,
snapshot.members
)
return snapshot
} catch (error) {
console.warn(`[GroupSnapshot] reason=${reason} name=${contact.m_nsNickName} failed:`, error)
return null
}
},
[]
)
const mergeSyntheticMessages = React.useCallback(
(contact: Contact | null, baseMessages: Message[], roomId?: string): Message[] => {
if (!contact || contact.type !== 'group') return baseMessages
const resolvedRoomId = roomId || currentGroupSnapshotRef.current?.roomId
if (!resolvedRoomId) return baseMessages
const synthetic = syntheticGroupMessagesRef.current[resolvedRoomId] || []
return synthetic.length
? sortMessagesChronologically([...baseMessages, ...synthetic])
: baseMessages
},
[]
)
const handleAutoGetDbKey = async (): Promise<void> => { const handleAutoGetDbKey = async (): Promise<void> => {
if (isFetchingDbKey) return if (isFetchingDbKey) return
setIsFetchingDbKey(true) setIsFetchingDbKey(true)
@@ -175,14 +273,18 @@ function App(): React.ReactElement {
setSelectedContact(contact) setSelectedContact(contact)
const { startTime, endTime } = getDateRangeParams(dateRange) const { startTime, endTime } = getDateRangeParams(dateRange)
const msgs = await window.api.getMessages(contact.md5, startTime, endTime) const msgs = await window.api.getMessages(contact.md5, startTime, endTime)
setMessages(msgs) const snapshot = await logGroupSnapshot(contact, 'select-contact')
currentGroupSnapshotRef.current = snapshot
setMessages(mergeSyntheticMessages(contact, msgs, snapshot?.roomId))
} }
const handleDateRangeChange = (range: string): void => { const handleDateRangeChange = (range: string): void => {
setDateRange(range) setDateRange(range)
if (selectedContact) { if (selectedContact) {
const { startTime, endTime } = getDateRangeParams(range) const { startTime, endTime } = getDateRangeParams(range)
window.api.getMessages(selectedContact.md5, startTime, endTime).then(setMessages) window.api.getMessages(selectedContact.md5, startTime, endTime).then((nextMessages) => {
setMessages(mergeSyntheticMessages(selectedContact, nextMessages))
})
} }
} }
@@ -201,9 +303,45 @@ function App(): React.ReactElement {
range.startTime, range.startTime,
range.endTime range.endTime
) )
const latestSnapshot = await logGroupSnapshot(selectedContact, 'wcdb-change')
const syntheticEvents = buildSyntheticGroupMessages(
currentGroupSnapshotRef.current,
latestSnapshot,
latestMessages
)
if (latestSnapshot) {
currentGroupSnapshotRef.current = latestSnapshot
if (syntheticEvents.length) {
const existing = syntheticGroupMessagesRef.current[latestSnapshot.roomId] || []
const existingIds = new Set(existing.map((message) => message.id))
const appended = syntheticEvents.filter((message) => !existingIds.has(message.id))
if (appended.length) {
syntheticGroupMessagesRef.current[latestSnapshot.roomId] = [...existing, ...appended]
console.log(
`[GroupMonitor] merged synthetic messages roomId=${latestSnapshot.roomId} total=${syntheticGroupMessagesRef.current[latestSnapshot.roomId].length}`
)
if (!disposed) { if (!disposed) {
setMessages((current) => setMessages((current) =>
areMessagesEquivalent(current, latestMessages) ? current : latestMessages sortMessagesChronologically([
...current,
...appended.filter(
(message) =>
!current.some((existingMessage) => existingMessage.id === message.id)
)
])
)
}
}
}
}
const nextMessages = mergeSyntheticMessages(
selectedContact,
latestMessages,
latestSnapshot?.roomId
)
if (!disposed) {
setMessages((current) =>
areMessagesEquivalent(current, nextMessages) ? current : nextMessages
) )
} }
} catch (error) { } catch (error) {
@@ -224,7 +362,14 @@ function App(): React.ReactElement {
if (refreshTimer) window.clearTimeout(refreshTimer) if (refreshTimer) window.clearTimeout(refreshTimer)
unsubscribe() unsubscribe()
} }
}, [dateRange, isAuthenticated, isNativeMonitorActive, selectedContact]) }, [
dateRange,
isAuthenticated,
isNativeMonitorActive,
selectedContact,
logGroupSnapshot,
mergeSyntheticMessages
])
const handleSearchContacts = (keyword: string): void => { const handleSearchContacts = (keyword: string): void => {
if (!keyword) { if (!keyword) {
+25
View File
@@ -252,6 +252,31 @@ body {
justify-content: flex-end; justify-content: flex-end;
} }
.wechat-system-message-row {
display: flex;
flex-direction: column;
align-items: center;
gap: 4px;
margin: 2px 0 6px;
}
.wechat-system-message {
max-width: min(78%, 760px);
padding: 6px 12px;
border-radius: 999px;
background: rgba(227, 86, 86, 0.1);
color: #c84a4a;
font-size: 12px;
line-height: 1.5;
text-align: center;
word-break: break-word;
}
.wechat-system-message-meta {
color: #a0a7ab;
font-size: 11px;
}
.message-avatar { .message-avatar {
width: 38px; width: 38px;
height: 38px; height: 38px;
+13 -1
View File
@@ -304,6 +304,7 @@ const ChatWindow: React.FC<ChatWindowProps> = ({
<div className="message-list wechat-message-list"> <div className="message-list wechat-message-list">
{filteredMessages.map((msg) => { {filteredMessages.map((msg) => {
const isMine = msg.from === 'assistant' const isMine = msg.from === 'assistant'
const isSystem = msg.from === 'system' || msg.type === '系统消息'
const displayName = isMine const displayName = isMine
? '我' ? '我'
: isGroupChat : isGroupChat
@@ -312,7 +313,18 @@ const ChatWindow: React.FC<ChatWindowProps> = ({
const avatarSrc = isMine ? msg.img : msg.img || contact.avatar const avatarSrc = isMine ? msg.img : msg.img || contact.avatar
const isVoice = msg.type === '语音' const isVoice = msg.type === '语音'
const isImage = msg.type === '图片' const isImage = msg.type === '图片'
const isRichMedia = ['名片', '位置', '分享消息', '通话', '表情包'].includes(msg.type) const isRichMedia = ['名片', '位置', '分享消息', '通话', '表情包', '系统消息'].includes(
msg.type
)
if (isSystem) {
return (
<div key={msg.id} className="wechat-system-message-row">
<div className="wechat-system-message">{msg.content}</div>
<div className="wechat-system-message-meta">{msg.datetime}</div>
</div>
)
}
return ( return (
<div key={msg.id} className={`wechat-message-row ${isMine ? 'mine' : 'other'}`}> <div key={msg.id} className={`wechat-message-row ${isMine ? 'mine' : 'other'}`}>
@@ -22,6 +22,8 @@ export function RichMessageBubble({ contentData }: RichMessageBubbleProps): JSX.
return <StickerBubble data={contentData} /> return <StickerBubble data={contentData} />
case 'quote': case 'quote':
return <QuoteBubble data={contentData} /> return <QuoteBubble data={contentData} />
case 'system':
return <SystemBubble data={contentData} />
case 'unknown': case 'unknown':
return ( return (
<div className="message-text">{(contentData as { raw?: string }).raw || '[未知消息]'}</div> <div className="message-text">{(contentData as { raw?: string }).raw || '[未知消息]'}</div>
@@ -212,6 +214,10 @@ function QuoteBubble({ data }: { data: Extract<ParsedContent, { type: 'quote' }>
) )
} }
function SystemBubble({ data }: { data: Extract<ParsedContent, { type: 'system' }> }): JSX.Element {
return <div className="message-text">{data.content || '[系统消息]'}</div>
}
function getUrlHost(url?: string): string { function getUrlHost(url?: string): string {
if (!url) return '' if (!url) return ''
try { try {
+1 -1
View File
@@ -66,7 +66,7 @@ type QuoteContent = {
quotedSender?: string quotedSender?: string
quotedType?: string quotedType?: string
} }
type SystemContent = { type: 'system'; content: string } type SystemContent = { type: 'system'; content: string; raw?: string }
type UnknownContent = { type: 'unknown'; raw: string } type UnknownContent = { type: 'unknown'; raw: string }
export type ParsedContent = export type ParsedContent =