ulysia/Vencord
1
0
Fork 0
Code Pull requests Activity

Support regex in webpack finds

Browse source
This commit is contained in:
Nuckyz 2024-07-02 20:00:05 -03:00
parent cc536bdbf4
commit bc34d72b57
No known key found for this signature in database
GPG key ID: 440BF8296E1C4AD9
2 changed files with 47 additions and 48 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

4
src/debug/runReporter.ts
View file

@ -140,7 +140,7 @@ async function runReporter() {
if (filter.$$vencordProps != null) { if (filter.$$vencordProps != null) {
const filterName = filter.$$vencordProps[0]; const filterName = filter.$$vencordProps[0];
parsedFilter = `${filterName}(${filter.$$vencordProps.slice(1).map((arg: any) => JSON.stringify(arg)).join(", ")})`; parsedFilter = `${filterName}(${filter.$$vencordProps.slice(1).map((arg: any) => arg instanceof RegExp ? String(arg) : JSON.stringify(arg)).join(", ")})`;
} else { } else {
parsedFilter = String(filter).slice(0, 147) + "..."; parsedFilter = String(filter).slice(0, 147) + "...";
} }
@ -150,7 +150,7 @@ async function runReporter() {
logMessage += `(${JSON.stringify(code)}, {\n${parsedFailedMappers.map(([key, parsedFilter]) => `\t${key}: ${parsedFilter}`).join(",\n")}\n})`; logMessage += `(${JSON.stringify(code)}, {\n${parsedFailedMappers.map(([key, parsedFilter]) => `\t${key}: ${parsedFilter}`).join(",\n")}\n})`;
} else { } else {
logMessage += `(${filterName.length ? `${filterName}(` : ""}${parsedArgs.map(arg => JSON.stringify(arg)).join(", ")})${filterName.length ? ")" : ""}`; logMessage += `(${filterName.length ? `${filterName}(` : ""}${parsedArgs.map(arg => arg instanceof RegExp ? String(arg) : JSON.stringify(arg)).join(", ")})${filterName.length ? ")" : ""}`;
} }
ReporterLogger.log("Webpack Find Fail:", logMessage); ReporterLogger.log("Webpack Find Fail:", logMessage);

91
src/webpack/api.tsx
View file

@ -62,13 +62,24 @@ if (IS_DEV && IS_DISCORD_DESKTOP) {
}, 0); }, 0);
} }
export type PropsFilter = Array<string>;
export type CodeFilter = Array<string | RegExp>;
export type StoreNameFilter = string;
export type FilterFn = ((module: any) => boolean) & { export type FilterFn = ((module: any) => boolean) & {
$$vencordProps?: string[]; $$vencordProps?: Array<string | RegExp>;
$$vencordIsFactoryFilter?: boolean; $$vencordIsFactoryFilter?: boolean;
}; };
const stringMatches = (s: string, filter: CodeFilter) =>
filter.every(f =>
typeof f === "string"
? s.includes(f)
: (f.global && (f.lastIndex = 0), f.test(s))
);
export const filters = { export const filters = {
byProps: (...props: string[]): FilterFn => { byProps: (...props: PropsFilter): FilterFn => {
const filter: FilterFn = props.length === 1 const filter: FilterFn = props.length === 1
? m => m?.[props[0]] !== void 0 ? m => m?.[props[0]] !== void 0
: m => props.every(p => m?.[p] !== void 0); : m => props.every(p => m?.[p] !== void 0);
@ -77,28 +88,25 @@ export const filters = {
return filter; return filter;
}, },
byCode: (...code: string[]): FilterFn => { byCode: (...code: CodeFilter): FilterFn => {
code = code.map(canonicalizeMatch);
const filter: FilterFn = m => { const filter: FilterFn = m => {
if (typeof m !== "function") return false; if (typeof m !== "function") return false;
const s = String(m); return stringMatches(String(m), code);
for (const c of code) {
if (!s.includes(c)) return false;
}
return true;
}; };
filter.$$vencordProps = ["byCode", ...code]; filter.$$vencordProps = ["byCode", ...code];
return filter; return filter;
}, },
byStoreName: (name: string): FilterFn => { byStoreName: (name: StoreNameFilter): FilterFn => {
const filter: FilterFn = m => m?.constructor?.displayName === name; const filter: FilterFn = m => m?.constructor?.displayName === name;
filter.$$vencordProps = ["byStoreName", name]; filter.$$vencordProps = ["byStoreName", name];
return filter; return filter;
}, },
byComponentCode: (...code: string[]): FilterFn => { byComponentCode: (...code: CodeFilter): FilterFn => {
const byCodeFilter = filters.byCode(...code); const byCodeFilter = filters.byCode(...code);
const filter: FilterFn = m => { const filter: FilterFn = m => {
let inner = m; let inner = m;
@ -118,7 +126,7 @@ export const filters = {
return filter; return filter;
}, },
byFactoryCode: (...code: string[]): FilterFn => { byFactoryCode: (...code: CodeFilter): FilterFn => {
const byCodeFilter = filters.byCode(...code); const byCodeFilter = filters.byCode(...code);
byCodeFilter.$$vencordProps = ["byFactoryCode", ...code]; byCodeFilter.$$vencordProps = ["byFactoryCode", ...code];
@ -133,7 +141,7 @@ export const webpackSearchHistory = [] as Array<["waitFor" | "find" | "findCompo
function printFilter(filter: FilterFn) { function printFilter(filter: FilterFn) {
if (filter.$$vencordProps != null) { if (filter.$$vencordProps != null) {
const props = filter.$$vencordProps; const props = filter.$$vencordProps;
return `${props[0]}(${props.slice(1).map(arg => JSON.stringify(arg)).join(", ")})`; return `${props[0]}(${props.slice(1).map(arg => arg instanceof RegExp ? String(arg) : JSON.stringify(arg)).join(", ")})`;
} }
return String(filter); return String(filter);
@ -263,9 +271,9 @@ export function findComponent<T extends object = any>(filter: FilterFn, parse: (
* @param parse A function that takes the found component as its first argument and returns a component. Useful if you want to wrap the found component in something. Defaults to the original component * @param parse A function that takes the found component as its first argument and returns a component. Useful if you want to wrap the found component in something. Defaults to the original component
* @returns The component if found, or a noop component * @returns The component if found, or a noop component
*/ */
export function findExportedComponent<T extends object = any>(...props: string[] | [...string[], (component: any) => LazyComponentType<T>]) { export function findExportedComponent<T extends object = any>(...props: PropsFilter | [...PropsFilter, (component: any) => LazyComponentType<T>]) {
const parse = (typeof props.at(-1) === "function" ? props.pop() : m => m) as (component: any) => LazyComponentType<T>; const parse = (typeof props.at(-1) === "function" ? props.pop() : m => m) as (component: any) => LazyComponentType<T>;
const newProps = props as string[]; const newProps = props as PropsFilter;
const filter = filters.byProps(...newProps); const filter = filters.byProps(...newProps);
@ -308,9 +316,9 @@ export function findExportedComponent<T extends object = any>(...props: string[]
* @param parse A function that takes the found component as its first argument and returns a component. Useful if you want to wrap the found component in something. Defaults to the original component * @param parse A function that takes the found component as its first argument and returns a component. Useful if you want to wrap the found component in something. Defaults to the original component
* @returns The component if found, or a noop component * @returns The component if found, or a noop component
*/ */
export function findComponentByCode<T extends object = any>(...code: string[] | [...string[], (component: any) => LazyComponentType<T>]) { export function findComponentByCode<T extends object = any>(...code: CodeFilter | [...CodeFilter, (component: any) => LazyComponentType<T>]) {
const parse = (typeof code.at(-1) === "function" ? code.pop() : m => m) as (component: any) => LazyComponentType<T>; const parse = (typeof code.at(-1) === "function" ? code.pop() : m => m) as (component: any) => LazyComponentType<T>;
const newCode = code as string[]; const newCode = code as CodeFilter;
const ComponentResult = findComponent<T>(filters.byComponentCode(...newCode), parse, { isIndirect: true }); const ComponentResult = findComponent<T>(filters.byComponentCode(...newCode), parse, { isIndirect: true });
@ -327,9 +335,9 @@ export function findComponentByCode<T extends object = any>(...code: string[] |
* @param props A list of props to search the module or exports for * @param props A list of props to search the module or exports for
* @param parse A function that takes the find result as its first argument and returns something. Useful if you want to use a value from the find result, instead of all of it. Defaults to the find result itself * @param parse A function that takes the find result as its first argument and returns something. Useful if you want to use a value from the find result, instead of all of it. Defaults to the find result itself
*/ */
export function findByProps<T = any>(...props: string[] | [...string[], (mod: any) => T]) { export function findByProps<T = any>(...props: PropsFilter | [...PropsFilter, (mod: any) => T]) {
const parse = (typeof props.at(-1) === "function" ? props.pop() : m => m) as (mod: any) => T; const parse = (typeof props.at(-1) === "function" ? props.pop() : m => m) as (mod: any) => T;
const newProps = props as string[]; const newProps = props as PropsFilter;
const result = find<T>(filters.byProps(...newProps), parse, { isIndirect: true }); const result = find<T>(filters.byProps(...newProps), parse, { isIndirect: true });
@ -348,9 +356,9 @@ export function findByProps<T = any>(...props: string[] | [...string[], (mod: an
* @param props A list of props to search the module or exports for * @param props A list of props to search the module or exports for
* @param parse A function that takes the find result as its first argument and returns something. Useful if you want to use a value from the find result, instead of all of it. Defaults to the find result itself * @param parse A function that takes the find result as its first argument and returns something. Useful if you want to use a value from the find result, instead of all of it. Defaults to the find result itself
*/ */
export function findByPropsAndExtract<T = any>(...props: string[] | [...string[], (mod: any) => T]) { export function findByPropsAndExtract<T = any>(...props: PropsFilter | [...PropsFilter, (mod: any) => T]) {
const parse = (typeof props.at(-1) === "function" ? props.pop() : m => m) as (mod: any) => T; const parse = (typeof props.at(-1) === "function" ? props.pop() : m => m) as (mod: any) => T;
const newProps = props as string[]; const newProps = props as PropsFilter;
const result = find<T>(filters.byProps(...newProps), m => parse(m[newProps[0]]), { isIndirect: true }); const result = find<T>(filters.byProps(...newProps), m => parse(m[newProps[0]]), { isIndirect: true });
@ -367,9 +375,9 @@ export function findByPropsAndExtract<T = any>(...props: string[] | [...string[]
* @param code A list of code to search each export for * @param code A list of code to search each export for
* @param parse A function that takes the find result as its first argument and returns something. Useful if you want to use a value from the find result, instead of all of it. Defaults to the find result itself * @param parse A function that takes the find result as its first argument and returns something. Useful if you want to use a value from the find result, instead of all of it. Defaults to the find result itself
*/ */
export function findByCode<T = any>(...code: string[] | [...string[], (mod: any) => T]) { export function findByCode<T = any>(...code: CodeFilter | [...CodeFilter, (mod: any) => T]) {
const parse = (typeof code.at(-1) === "function" ? code.pop() : m => m) as (mod: any) => T; const parse = (typeof code.at(-1) === "function" ? code.pop() : m => m) as (mod: any) => T;
const newCode = code as string[]; const newCode = code as CodeFilter;
const result = find<T>(filters.byCode(...newCode), parse, { isIndirect: true }); const result = find<T>(filters.byCode(...newCode), parse, { isIndirect: true });
@ -385,7 +393,7 @@ export function findByCode<T = any>(...code: string[] | [...string[], (mod: any)
* *
* @param name The store name * @param name The store name
*/ */
export function findStore<T = GenericStore>(name: string) { export function findStore<T = GenericStore>(name: StoreNameFilter) {
const result = find<T>(filters.byStoreName(name), m => m, { isIndirect: true }); const result = find<T>(filters.byStoreName(name), m => m, { isIndirect: true });
if (IS_REPORTER) { if (IS_REPORTER) {
@ -401,9 +409,9 @@ export function findStore<T = GenericStore>(name: string) {
* @param code A list of code to search each factory for * @param code A list of code to search each factory for
* @param parse A function that takes the find result as its first argument and returns something. Useful if you want to use a value from the find result, instead of all of it. Defaults to the find result itself * @param parse A function that takes the find result as its first argument and returns something. Useful if you want to use a value from the find result, instead of all of it. Defaults to the find result itself
*/ */
export function findByFactoryCode<T = any>(...code: string[] | [...string[], (mod: any) => T]) { export function findByFactoryCode<T = any>(...code: CodeFilter | [...CodeFilter, (mod: any) => T]) {
const parse = (typeof code.at(-1) === "function" ? code.pop() : m => m) as (mod: any) => T; const parse = (typeof code.at(-1) === "function" ? code.pop() : m => m) as (mod: any) => T;
const newCode = code as string[]; const newCode = code as CodeFilter;
const result = find<T>(filters.byFactoryCode(...newCode), parse, { isIndirect: true }); const result = find<T>(filters.byFactoryCode(...newCode), parse, { isIndirect: true });
@ -428,7 +436,7 @@ export function findByFactoryCode<T = any>(...code: string[] | [...string[], (mo
* @param mappers Mappers to create the non mangled exports object * @param mappers Mappers to create the non mangled exports object
* @returns Unmangled exports as specified in mappers * @returns Unmangled exports as specified in mappers
*/ */
export function mapMangledModule<S extends PropertyKey>(code: string | string[], mappers: Record<S, FilterFn>) { export function mapMangledModule<S extends PropertyKey>(code: string | RegExp | CodeFilter, mappers: Record<S, FilterFn>) {
const mapping = {} as Record<S, any>; const mapping = {} as Record<S, any>;
const setters = {} as Record<S, (innerValue: any) => void>; const setters = {} as Record<S, (innerValue: any) => void>;
@ -481,7 +489,7 @@ export function mapMangledModule<S extends PropertyKey>(code: string | string[],
/** /**
* Find the first module factory that includes all the given code. * Find the first module factory that includes all the given code.
*/ */
export function findModuleFactory(...code: string[]) { export function findModuleFactory(...code: CodeFilter) {
const filter = filters.byFactoryCode(...code); const filter = filters.byFactoryCode(...code);
const [proxy, setInnerValue] = proxyInner<ModuleFactory>(`Webpack module factory find matched no module. Filter: ${printFilter(filter)}`, "Webpack find with proxy called on a primitive value."); const [proxy, setInnerValue] = proxyInner<ModuleFactory>(`Webpack module factory find matched no module. Filter: ${printFilter(filter)}`, "Webpack find with proxy called on a primitive value.");
@ -533,7 +541,7 @@ export const ChunkIdsRegex = /\("([^"]+?)"\)/g;
* @param matcher A RegExp that returns the chunk ids array as the first capture group and the entry point id as the second. Defaults to a matcher that captures the first lazy chunk loading found in the module factory * @param matcher A RegExp that returns the chunk ids array as the first capture group and the entry point id as the second. Defaults to a matcher that captures the first lazy chunk loading found in the module factory
* @returns A function that returns a promise that resolves with a boolean whether the chunks were loaded, on first call * @returns A function that returns a promise that resolves with a boolean whether the chunks were loaded, on first call
*/ */
export function extractAndLoadChunksLazy(code: string | string[], matcher: RegExp = DefaultExtractAndLoadChunksRegex) { export function extractAndLoadChunksLazy(code: string | RegExp | CodeFilter, matcher: RegExp = DefaultExtractAndLoadChunksRegex) {
const module = findModuleFactory(...Array.isArray(code) ? code : [code]); const module = findModuleFactory(...Array.isArray(code) ? code : [code]);
const extractAndLoadChunks = makeLazy(async () => { const extractAndLoadChunks = makeLazy(async () => {
@ -805,16 +813,9 @@ export const cacheFindBulk = traceFunction("cacheFindBulk", function cacheFindBu
/** /**
* Find the id of the first already loaded module factory that includes all the given code. * Find the id of the first already loaded module factory that includes all the given code.
*/ */
export const cacheFindModuleId = traceFunction("cacheFindModuleId", function cacheFindModuleId(...code: string[]) { export const cacheFindModuleId = traceFunction("cacheFindModuleId", function cacheFindModuleId(...code: CodeFilter) {
outer:
for (const id in wreq.m) { for (const id in wreq.m) {
const str = String(wreq.m[id]); if (stringMatches(String(wreq.m[id]), code)) return id;
for (const c of code) {
if (!str.includes(c)) continue outer;
}
return id;
} }
const err = new Error("Didn't find module with code(s):\n" + code.join("\n")); const err = new Error("Didn't find module with code(s):\n" + code.join("\n"));
@ -829,12 +830,12 @@ export const cacheFindModuleId = traceFunction("cacheFindModuleId", function cac
/** /**
* Find the first already loaded module factory that includes all the given code. * Find the first already loaded module factory that includes all the given code.
*/ */
export function cacheFindModuleFactory(...code: string[]) { export const cacheFindModuleFactory = traceFunction("cacheFindModuleFactory", function cacheFindModuleFactory(...code: CodeFilter) {
const id = cacheFindModuleId(...code); const id = cacheFindModuleId(...code);
if (id == null) return; if (id == null) return;
return wreq.m[id]; return wreq.m[id];
} });
/** /**
* Search modules by keyword. This searches the factory methods, * Search modules by keyword. This searches the factory methods,
@ -843,18 +844,16 @@ export function cacheFindModuleFactory(...code: string[]) {
* @param filters One or more strings or regexes * @param filters One or more strings or regexes
* @returns Mapping of found modules * @returns Mapping of found modules
*/ */
export function search(...filters: Array<string | RegExp>) { export function search(...filters: CodeFilter) {
const results = {} as Record<number, Function>; const results = {} as Record<number, Function>;
const factories = wreq.m; const factories = wreq.m;
outer:
for (const id in factories) { for (const id in factories) {
const factory = factories[id]; const factory = factories[id];
const factoryStr = String(factory);
for (const filter of filters) { if (stringMatches(String(factory), filters)) {
if (typeof filter === "string" && !factoryStr.includes(filter)) continue outer; results[id] = factory;
if (filter instanceof RegExp && !filter.test(factoryStr)) continue outer;
} }
results[id] = factory;
} }
return results; return results;