ulysia/Vencord
1
0
Fork 0
Code Pull requests Activity

Add findComponentByFields and filters.componentByFilter

Browse source
This commit is contained in:
Nuckyz 2024-08-03 14:01:52 -03:00
parent ed32b7964d
commit 4677ac779a
No known key found for this signature in database
GPG key ID: 440BF8296E1C4AD9
1 changed files with 39 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

40
src/webpack/api.tsx
View file

@ -113,6 +113,12 @@ export const filters = {
return filter;
},
// For use inside mapMangledModule
componentByFilter: (filter: FilterFn): FilterFn => {
filter.$$vencordIsComponentFilter = true;
return filter;
},
componentByCode: (...code: CodeFilter): FilterFn => {
const byCodeFilter = filters.byCode(...code);
const filter: FilterFn = m => {
@ -134,6 +140,15 @@ export const filters = {
return filter;
},
componentByFields: (...fields: PropsFilter): FilterFn => {
const byPropsFilter = filters.byProps(...fields);
const filter: FilterFn = m => m?.prototype?.render && byPropsFilter(m.prototype);
filter.$$vencordProps = ["componentByFields", ...fields];
filter.$$vencordIsComponentFilter = true;
return filter;
},
byFactoryCode: (...code: CodeFilter): FilterFn => {
const byCodeFilter = filters.byCode(...code);
@ -143,7 +158,7 @@ export const filters = {
}
};
export const webpackSearchHistory = [] as Array<["waitFor" | "find" | "findComponent" | "findExportedComponent" | "findComponentByCode" | "findByProps" | "findProp" | "findByCode" | "findStore" | "findByFactoryCode" | "mapMangledModule" | "extractAndLoadChunks" | "webpackDependantLazy" | "webpackDependantLazyComponent", any[]]>;
export const webpackSearchHistory = [] as Array<["waitFor" | "find" | "findComponent" | "findExportedComponent" | "findComponentByCode" | "findComponentByFields" | "findByProps" | "findProp" | "findByCode" | "findStore" | "findByFactoryCode" | "mapMangledModule" | "extractAndLoadChunks" | "webpackDependantLazy" | "webpackDependantLazyComponent", any[]]>;
function printFilter(filter: FilterFn) {
if (filter.$$vencordProps != null) {
@ -327,6 +342,29 @@ export function findComponentByCode<T extends object = any>(...code: CodeFilter
return ComponentResult;
}
/**
* Find the first exported class component which includes all the given fields in its prototype.
*
* @example findComponentByFields("renderTooltip", "shouldShowTooltip")
* @example findComponentByFields("renderTooltip", "shouldShowTooltip", Tooltip => Tooltip)
*
* @param code A list of fields to search each exported class component for
* @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
*/
export function findComponentByFields<T extends object = any>(...fields: PropsFilter | [...PropsFilter, (component: ModuleExports) => LazyComponentType<T>]) {
const parse = (typeof fields.at(-1) === "function" ? fields.pop() : m => m) as (component: ModuleExports) => LazyComponentType<T>;
const newFields = fields as PropsFilter;
const ComponentResult = findComponent<T>(filters.componentByFields(...newFields), parse, { isIndirect: true });
if (IS_REPORTER) {
webpackSearchHistory.push(["findComponentByCode", [ComponentResult, ...newFields]]);
}
return ComponentResult;
}
/**
* Find the first module exports or export that includes all the given props.
*