diff --git a/src/equicordplugins/keywordNotify/index.tsx b/src/equicordplugins/keywordNotify/index.tsx index d0efcc2a..3c906a51 100644 --- a/src/equicordplugins/keywordNotify/index.tsx +++ b/src/equicordplugins/keywordNotify/index.tsx @@ -11,6 +11,7 @@ import { definePluginSettings } from "@api/Settings"; import { classNameFactory } from "@api/Styles"; import { Flex } from "@components/Flex"; import { DeleteIcon } from "@components/Icons"; +import { onceDefined } from "@shared/onceDefined"; import { EquicordDevs } from "@utils/constants"; import { Margins } from "@utils/margins"; import { classes } from "@utils/misc"; @@ -32,8 +33,6 @@ let keywordLog: Array = []; const recentMentionsPopoutClass = findByPropsLazy("recentMentionsPopout"); const tabClass = findByPropsLazy("tab"); const buttonClass = findByPropsLazy("size36"); - -const MenuHeader = findByCodeLazy(".getOverdueMessageReminderCount())"); const Popout = findByCodeLazy(".Messages.UNBLOCK_TO_JUMP_TITLE", "canCloseAllMessages:"); const createMessageRecord = findByCodeLazy(".createFromServer(", ".isBlockedForMessage", "messageReference:"); const KEYWORD_ENTRIES_KEY = "KeywordNotify_keywordEntries"; @@ -41,6 +40,15 @@ const KEYWORD_LOG_KEY = "KeywordNotify_log"; const cl = classNameFactory("vc-keywordnotify-"); +let MenuHeader; +onceDefined(window, "GLOBAL_ENV", v => { + if (v.RELEASE_CHANNEL === "canary") { + MenuHeader = findByCodeLazy(".getOverdueMessageReminderCount())"); + } else { + MenuHeader = findByCodeLazy(".getMessageReminders()).length"); + } +}); + async function addKeywordEntry(forceUpdate: () => void) { keywordEntries.push({ regex: "", listIds: [], listType: ListType.BlackList, ignoreCase: false }); await DataStore.set(KEYWORD_ENTRIES_KEY, keywordEntries);