From ceaaf9ab8a3bb87494573a8442f31cc4a2396bbf Mon Sep 17 00:00:00 2001
From: Nuckyz <61953774+Nuckyz@users.noreply.github.com>
Date: Thu, 20 Jun 2024 01:00:07 -0300
Subject: [PATCH] Reporter: Test mapMangledModule

---
 src/debug/runReporter.ts | 15 ++++++++++++---
 src/webpack/webpack.ts   |  4 +++-
 2 files changed, 15 insertions(+), 4 deletions(-)

diff --git a/src/debug/runReporter.ts b/src/debug/runReporter.ts
index 6c7a2a03f..ddd5e5f18 100644
--- a/src/debug/runReporter.ts
+++ b/src/debug/runReporter.ts
@@ -39,9 +39,8 @@ async function runReporter() {
             }
             if (searchType === "waitForStore") method = "findStore";
 
+            let result: any;
             try {
-                let result: any;
-
                 if (method === "proxyLazyWebpack" || method === "LazyComponentWebpack") {
                     const [factory] = args;
                     result = factory();
@@ -50,16 +49,26 @@ async function runReporter() {
 
                     result = await Webpack.extractAndLoadChunks(code, matcher);
                     if (result === false) result = null;
+                } else if (method === "mapMangledModule") {
+                    const [code, mapper] = args;
+
+                    result = Webpack.mapMangledModule(code, mapper);
+                    if (Object.keys(result).length !== Object.keys(mapper).length) throw new Error("Webpack Find Fail");
                 } else {
                     // @ts-ignore
                     result = Webpack[method](...args);
                 }
 
-                if (result == null || (result.$$vencordInternal != null && result.$$vencordInternal() == null)) throw "a rock at ben shapiro";
+                if (result == null || (result.$$vencordInternal != null && result.$$vencordInternal() == null)) throw new Error("Webpack Find Fail");
             } catch (e) {
                 let logMessage = searchType;
                 if (method === "find" || method === "proxyLazyWebpack" || method === "LazyComponentWebpack") logMessage += `(${args[0].toString().slice(0, 147)}...)`;
                 else if (method === "extractAndLoadChunks") logMessage += `([${args[0].map(arg => `"${arg}"`).join(", ")}], ${args[1].toString()})`;
+                else if (method === "mapMangledModule") {
+                    const failedMappings = Object.keys(args[1]).filter(key => result?.[key] == null);
+
+                    logMessage += `("${args[0]}", {\n${failedMappings.map(mapping => `\t${mapping}: ${args[1][mapping].toString().slice(0, 147)}...`).join(",\n")}\n})`;
+                }
                 else logMessage += `(${args.map(arg => `"${arg}"`).join(", ")})`;
 
                 ReporterLogger.log("Webpack Find Fail:", logMessage);
diff --git a/src/webpack/webpack.ts b/src/webpack/webpack.ts
index b536063e8..f776ab1c3 100644
--- a/src/webpack/webpack.ts
+++ b/src/webpack/webpack.ts
@@ -279,7 +279,7 @@ export function findModuleFactory(...code: string[]) {
     return wreq.m[id];
 }
 
-export const lazyWebpackSearchHistory = [] as Array<["find" | "findByProps" | "findByCode" | "findStore" | "findComponent" | "findComponentByCode" | "findExportedComponent" | "waitFor" | "waitForComponent" | "waitForStore" | "proxyLazyWebpack" | "LazyComponentWebpack" | "extractAndLoadChunks", any[]]>;
+export const lazyWebpackSearchHistory = [] as Array<["find" | "findByProps" | "findByCode" | "findStore" | "findComponent" | "findComponentByCode" | "findExportedComponent" | "waitFor" | "waitForComponent" | "waitForStore" | "proxyLazyWebpack" | "LazyComponentWebpack" | "extractAndLoadChunks" | "mapMangledModule", any[]]>;
 
 /**
  * This is just a wrapper around {@link proxyLazy} to make our reporter test for your webpack finds.
@@ -483,6 +483,8 @@ export const mapMangledModule = traceFunction("mapMangledModule", function mapMa
  *          })
  */
 export function mapMangledModuleLazy<S extends string>(code: string, mappers: Record<S, FilterFn>): Record<S, any> {
+    if (IS_REPORTER) lazyWebpackSearchHistory.push(["mapMangledModule", [code, mappers]]);
+
     return proxyLazy(() => mapMangledModule(code, mappers));
 }