SheWhoReignsSupreme/revenge-plugins
1
0
Fork 0
Code Issues 1 Pull requests 1 Projects Releases Packages Wiki Activity Actions

MessageLogger: cap edit history and tracked-message maps to bound memory under heavy channel activity

Browse source
This commit is contained in:
Ally Founder 2026-08-12 12:09:18 -04:00
parent 17ffd75a34
commit e6609fbc11
1 changed files with 17 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

17
plugins/message-logger/src/patches/fluxIntercept.ts
View file

@ -11,6 +11,12 @@ const TAG = "[MessageLogger]";
const FAKE_DELETE_FLAG = "__msgLoggerDeleted"; const FAKE_DELETE_FLAG = "__msgLoggerDeleted";
const CLEANUP_FLAG = "__msgLoggerCleanup"; const CLEANUP_FLAG = "__msgLoggerCleanup";
// A busy, high-traffic channel (flooding, spam, price-ticker bots editing the same message
// repeatedly) can grow these maps without bound over a long session - capped so memory and the
// per-row content-rebuild in rowStyling.ts stay bounded regardless of channel activity.
const MAX_EDIT_HISTORY_PER_MESSAGE = 10;
const MAX_TRACKED_MESSAGES = 500;
const ChannelMessages = findByProps("_channelMessages"); const ChannelMessages = findByProps("_channelMessages");
const UserStore = findByStoreName("UserStore"); const UserStore = findByStoreName("UserStore");
@ -21,6 +27,14 @@ export const fakedMessages = new Map<string, string>();
// Past content versions per message, oldest first, current content not included. // Past content versions per message, oldest first, current content not included.
export const editHistory = new Map<string, string[]>(); export const editHistory = new Map<string, string[]>();
function evictOldest<K, V>(map: Map<K, V>, max: number): void {
while (map.size > max) {
const oldestKey = map.keys().next().value;
if (oldestKey === undefined) break;
map.delete(oldestKey);
}
}
function currentUserId(): string | undefined { function currentUserId(): string | undefined {
return UserStore?.getCurrentUser?.()?.id; return UserStore?.getCurrentUser?.()?.id;
} }
@ -97,6 +111,7 @@ function captureAndMaybeKeep(original: any, channelId: string, id: string): any
if (!keepInline) return undefined; if (!keepInline) return undefined;
fakedMessages.set(id, channelId); fakedMessages.set(id, channelId);
evictOldest(fakedMessages, MAX_TRACKED_MESSAGES);
return fakeUpdateFor(original); return fakeUpdateFor(original);
} }
@ -164,7 +179,9 @@ function handleUpdate(event: any): void {
const history = editHistory.get(updated.id) ?? []; const history = editHistory.get(updated.id) ?? [];
history.push(original.content ?? ""); history.push(original.content ?? "");
if (history.length > MAX_EDIT_HISTORY_PER_MESSAGE) history.shift();
editHistory.set(updated.id, history); editHistory.set(updated.id, history);
evictOldest(editHistory, MAX_TRACKED_MESSAGES);
addLogEntry(snapshotMessage(original, "edited", newContent, false), limitsFromStorage()); addLogEntry(snapshotMessage(original, "edited", newContent, false), limitsFromStorage());
} }