Vencord/src/webpack/webpack.ts

88 lines
2.7 KiB
TypeScript
Raw Normal View History

2022-09-05 13:54:02 -04:00
import type { WebpackInstance } from "discord-types/other";
export let wreq: WebpackInstance;
export let cache: WebpackInstance["c"];
2022-08-31 14:47:07 -04:00
export type FilterFn = (mod: any) => boolean;
export const filters = {
byProps: (props: string[]): FilterFn =>
props.length === 1
? m => m[props[0]] !== void 0
: m => props.every(p => m[p] !== void 0),
byDisplayName: (deezNuts: string): FilterFn => m => m.default?.displayName === deezNuts
};
export const subscriptions = new Map<FilterFn, CallbackFn>();
export const listeners = new Set<CallbackFn>();
export type CallbackFn = (mod: any) => void;
export function _initWebpack(instance: typeof window.webpackChunkdiscord_app) {
2022-09-05 13:54:02 -04:00
if (cache !== void 0) throw "no.";
2022-08-31 14:47:07 -04:00
2022-09-05 13:54:02 -04:00
wreq = instance.push([[Symbol()], {}, (r) => r]);
cache = wreq.c;
2022-08-31 14:47:07 -04:00
instance.pop();
}
export function find(filter: FilterFn, getDefault = true) {
if (typeof filter !== "function")
throw new Error("Invalid filter. Expected a function got", filter);
2022-09-05 13:54:02 -04:00
for (const key in cache) {
const mod = cache[key];
2022-08-31 14:47:07 -04:00
if (mod?.exports && filter(mod.exports))
return mod.exports;
if (mod?.exports?.default && filter(mod.exports.default))
return getDefault ? mod.exports.default : mod.exports;
}
return null;
}
export function findAll(filter: FilterFn, getDefault = true) {
if (typeof filter !== "function") throw new Error("Invalid filter. Expected a function got", filter);
const ret = [] as any[];
2022-09-05 13:54:02 -04:00
for (const key in cache) {
const mod = cache[key];
2022-08-31 14:47:07 -04:00
if (mod?.exports && filter(mod.exports)) ret.push(mod.exports);
if (mod?.exports?.default && filter(mod.exports.default)) ret.push(getDefault ? mod.exports.default : mod.exports);
}
return ret;
}
export function findByProps(...props: string[]) {
return find(filters.byProps(props));
}
export function findAllByProps(...props: string[]) {
return findAll(filters.byProps(props));
}
export function findByDisplayName(deezNuts: string) {
return find(filters.byDisplayName(deezNuts));
}
export function waitFor(filter: string | string[] | FilterFn, callback: CallbackFn) {
if (typeof filter === "string") filter = filters.byProps([filter]);
else if (Array.isArray(filter)) filter = filters.byProps(filter);
else if (typeof filter !== "function") throw new Error("filter must be a string, string[] or function, got", filter);
const existing = find(filter!);
if (existing) return void callback(existing);
subscriptions.set(filter, callback);
}
export function addListener(callback: CallbackFn) {
listeners.add(callback);
}
export function removeListener(callback: CallbackFn) {
listeners.delete(callback);
}