diff --git a/src/debug/Tracer.ts b/src/debug/Tracer.ts
new file mode 100644
index 000000000..86b5786d5
--- /dev/null
+++ b/src/debug/Tracer.ts
@@ -0,0 +1,63 @@
+/*
+ * Vencord, a modification for Discord's desktop app
+ * Copyright (c) 2022 Vendicated and contributors
+ *
+ * This program is free software: you can redistribute it and/or modify
+ * it under the terms of the GNU General Public License as published by
+ * the Free Software Foundation, either version 3 of the License, or
+ * (at your option) any later version.
+ *
+ * This program is distributed in the hope that it will be useful,
+ * but WITHOUT ANY WARRANTY; without even the implied warranty of
+ * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
+ * GNU General Public License for more details.
+ *
+ * You should have received a copy of the GNU General Public License
+ * along with this program. If not, see .
+*/
+
+import Logger from "../utils/logger";
+
+if (IS_DEV) {
+ var traces = {} as Record;
+ var logger = new Logger("Tracer", "#FFD166");
+}
+
+const noop = function () { };
+
+export const beginTrace = !IS_DEV ? noop :
+ function beginTrace(name: string, ...args: any[]) {
+ if (name in traces)
+ throw new Error(`Trace ${name} already exists!`);
+
+ traces[name] = [performance.now(), args];
+ };
+
+export const finishTrace = !IS_DEV ? noop : function finishTrace(name: string) {
+ const end = performance.now();
+
+ const [start, args] = traces[name];
+ delete traces[name];
+
+ logger.debug(`${name} took ${end - start}ms`, args);
+};
+
+type Func = (...args: any[]) => any;
+type TraceNameMapper = (...args: Parameters) => string;
+
+const noopTracer =
+ (name: string, f: F, mapper?: TraceNameMapper) => f;
+
+export const traceFunction = !IS_DEV
+ ? noopTracer
+ : function traceFunction(name: string, f: F, mapper?: TraceNameMapper): F {
+ return function (this: any, ...args: Parameters) {
+ const traceName = mapper?.(...args) ?? name;
+
+ beginTrace(traceName, ...arguments);
+ const result = f.apply(this, args);
+ finishTrace(traceName);
+
+ return result;
+ } as F;
+ };
diff --git a/src/plugins/index.ts b/src/plugins/index.ts
index 86d44b4f6..d3842aa53 100644
--- a/src/plugins/index.ts
+++ b/src/plugins/index.ts
@@ -20,6 +20,7 @@ import Plugins from "~plugins";
import { registerCommand, unregisterCommand } from "../api/Commands";
import { Settings } from "../api/settings";
+import { traceFunction } from "../debug/Tracer";
import Logger from "../utils/logger";
import { Patch, Plugin } from "../utils/types";
@@ -43,12 +44,12 @@ for (const p of Object.values(Plugins))
}
}
-export function startAllPlugins() {
+export const startAllPlugins = traceFunction("startAllPlugins", function startAllPlugins() {
for (const name in Plugins)
if (isPluginEnabled(name)) {
startPlugin(Plugins[name]);
}
-}
+});
export function startDependenciesRecursive(p: Plugin) {
let restartNeeded = false;
@@ -70,7 +71,7 @@ export function startDependenciesRecursive(p: Plugin) {
return { restartNeeded, failures };
}
-export function startPlugin(p: Plugin) {
+export const startPlugin = traceFunction("startPlugin", function startPlugin(p: Plugin) {
if (p.start) {
logger.info("Starting plugin", p.name);
if (p.started) {
@@ -100,9 +101,9 @@ export function startPlugin(p: Plugin) {
}
return true;
-}
+}, p => `startPlugin ${p.name}`);
-export function stopPlugin(p: Plugin) {
+export const stopPlugin = traceFunction("stopPlugin", function stopPlugin(p: Plugin) {
if (p.stop) {
logger.info("Stopping plugin", p.name);
if (!p.started) {
@@ -131,4 +132,4 @@ export function stopPlugin(p: Plugin) {
}
return true;
-}
+}, p => `stopPlugin ${p.name}`);
diff --git a/src/plugins/messageActions.ts b/src/plugins/messageActions.ts
index 5fe06c030..98c920e06 100644
--- a/src/plugins/messageActions.ts
+++ b/src/plugins/messageActions.ts
@@ -17,9 +17,10 @@
*/
import { addClickListener, removeClickListener } from "../api/MessageEvents";
+import { lazyWebpack } from "../utils";
import { Devs } from "../utils/constants";
import definePlugin from "../utils/types";
-import { find, findByProps } from "../webpack";
+import { filters } from "../webpack";
import { UserStore } from "../webpack/common";
let isDeletePressed = false;
@@ -33,10 +34,10 @@ export default definePlugin({
dependencies: ["MessageEventsAPI"],
start() {
- const { deleteMessage, startEditMessage } = findByProps("deleteMessage", "startEditMessage");
- const { can } = findByProps("can", "initialize");
- const { MANAGE_MESSAGES } = find(m => typeof m.MANAGE_MESSAGES === "bigint");
- const { isEditing } = findByProps("isEditing", "isEditingAny");
+ const MessageActions = lazyWebpack(filters.byProps("deleteMessage", "startEditMessage"));
+ const PermissionStore = lazyWebpack(filters.byProps("can", "initialize"));
+ const Permissions = lazyWebpack(m => typeof m.MANAGE_MESSAGES === "bigint");
+ const EditStore = lazyWebpack(filters.byProps("isEditing", "isEditingAny"));
document.addEventListener("keydown", keydown);
document.addEventListener("keyup", keyup);
@@ -44,12 +45,12 @@ export default definePlugin({
this.onClick = addClickListener((msg, chan, event) => {
const isMe = msg.author.id === UserStore.getCurrentUser().id;
if (!isDeletePressed) {
- if (isMe && event.detail >= 2 && !isEditing(chan.id, msg.id)) {
- startEditMessage(chan.id, msg.id, msg.content);
+ if (isMe && event.detail >= 2 && !EditStore.isEditing(chan.id, msg.id)) {
+ MessageActions.startEditMessage(chan.id, msg.id, msg.content);
event.preventDefault();
}
- } else if (isMe || can(MANAGE_MESSAGES, chan)) {
- deleteMessage(chan.id, msg.id);
+ } else if (isMe || PermissionStore.can(Permissions.MANAGE_MESSAGES, chan)) {
+ MessageActions.deleteMessage(chan.id, msg.id);
event.preventDefault();
}
});
diff --git a/src/webpack/webpack.ts b/src/webpack/webpack.ts
index 1a3e15d5c..b78afd084 100644
--- a/src/webpack/webpack.ts
+++ b/src/webpack/webpack.ts
@@ -18,6 +18,7 @@
import type { WebpackInstance } from "discord-types/other";
+import { traceFunction } from "../debug/Tracer";
import Logger from "../utils/logger";
import { proxyLazy } from "../utils/proxyLazy";
@@ -74,7 +75,7 @@ if (IS_DEV && !IS_WEB) {
}, 0);
}
-export function find(filter: FilterFn, getDefault = true, isWaitFor = false) {
+export const find = traceFunction("find", function find(filter: FilterFn, getDefault = true, isWaitFor = false) {
if (typeof filter !== "function")
throw new Error("Invalid filter. Expected a function got " + typeof filter);
@@ -109,7 +110,7 @@ export function find(filter: FilterFn, getDefault = true, isWaitFor = false) {
}
return null;
-}
+});
export function findAll(filter: FilterFn, getDefault = true) {
if (typeof filter !== "function")
@@ -222,7 +223,7 @@ export function bulk(...filterFns: FilterFn[]) {
* @param code Code
* @returns number or null
*/
-export function findModuleId(code: string) {
+export const findModuleId = traceFunction("findModuleId", function findModuleId(code: string) {
for (const id in wreq.m) {
if (wreq.m[id].toString().includes(code)) {
return Number(id);
@@ -239,7 +240,7 @@ export function findModuleId(code: string) {
}
return null;
-}
+});
/**
* Finds a mangled module by the provided code "code" (must be unique and can be anywhere in the module)
@@ -253,7 +254,7 @@ export function findModuleId(code: string) {
* closeModal: filters.byCode("key==")
* })
*/
-export function mapMangledModule(code: string, mappers: Record): Record {
+export const mapMangledModule = traceFunction("mapMangledModule", function mapMangledModule(code: string, mappers: Record): Record {
const exports = {} as Record;
const id = findModuleId(code);
@@ -273,7 +274,7 @@ export function mapMangledModule(code: string, mappers: Record
}
}
return exports;
-}
+});
/**
* Same as {@link mapMangledModule} but lazy