92 lines
3.6 KiB
TypeScript
92 lines
3.6 KiB
TypeScript
|
/*
|
||
|
|
* Vencord, a Discord client mod
|
||
|
|
* Copyright (c) 2024 Vendicated and contributors
|
||
|
|
* SPDX-License-Identifier: GPL-3.0-or-later
|
||
|
|
*/
|
||
|
|
|
||
|
|
export const proxyInnerGet = Symbol.for("vencord.proxyInner.get");
|
||
|
|
export const proxyInnerValue = Symbol.for("vencord.proxyInner.innerValue");
|
||
|
|
|
||
|
|
// Proxies demand that these properties be unmodified, so proxyInner
|
||
|
|
// will always return the function default for them.
|
||
|
|
const unconfigurable = ["arguments", "caller", "prototype"];
|
||
|
|
|
||
|
|
const handler: ProxyHandler<any> = {
|
||
|
|
...Object.fromEntries(Object.getOwnPropertyNames(Reflect).map(propName =>
|
||
|
|
[propName, (target: any, ...args: any[]) => Reflect[propName](target[proxyInnerGet](), ...args)]
|
||
|
|
)),
|
||
|
|
ownKeys: target => {
|
||
|
|
const keys = Reflect.ownKeys(target[proxyInnerGet]());
|
||
|
|
for (const key of unconfigurable) {
|
||
|
|
if (!keys.includes(key)) keys.push(key);
|
||
|
|
}
|
||
|
|
return keys;
|
||
|
|
},
|
||
|
|
getOwnPropertyDescriptor: (target, p) => {
|
||
|
|
if (typeof p === "string" && unconfigurable.includes(p))
|
||
|
|
return Reflect.getOwnPropertyDescriptor(target, p);
|
||
|
|
|
||
|
|
const descriptor = Reflect.getOwnPropertyDescriptor(target[proxyInnerGet](), p);
|
||
|
|
if (descriptor) Object.defineProperty(target, p, descriptor);
|
||
|
|
return descriptor;
|
||
|
|
}
|
||
|
|
};
|
||
|
|
|
||
|
|
/**
|
||
|
|
* A proxy which has an inner value that can be set later.
|
||
|
|
* When a property is accessed, the proxy looks for the property value in its inner value, and errors if it's not set.
|
||
|
|
* @param err The error to throw when the inner value is not set
|
||
|
|
* @returns A proxy which will act like the inner value when accessed
|
||
|
|
*/
|
||
|
|
export function proxyInner<T = any>(err = new Error("Proxy inner value is undefined, setInnerValue was never called."), isChild = false): [proxy: T, setInnerValue: (innerValue: T) => void] {
|
||
|
|
let isSameTick = true;
|
||
|
|
if (!isChild) setTimeout(() => isSameTick = false, 0);
|
||
|
|
|
||
|
|
const proxyDummy = Object.assign(function () { }, {
|
||
|
|
[proxyInnerGet]: function () {
|
||
|
|
if (proxyDummy[proxyInnerValue] == null) {
|
||
|
|
throw err;
|
||
|
|
}
|
||
|
|
|
||
|
|
return proxyDummy[proxyInnerValue];
|
||
|
|
},
|
||
|
|
[proxyInnerValue]: void 0 as T | undefined
|
||
|
|
});
|
||
|
|
|
||
|
|
const recursiveSetInnerValues = [] as Array<(innerValue: T) => void>;
|
||
|
|
|
||
|
|
function setInnerValue(innerValue: T) {
|
||
|
|
proxyDummy[proxyInnerValue] = innerValue;
|
||
|
|
recursiveSetInnerValues.forEach(setInnerValue => setInnerValue(innerValue));
|
||
|
|
}
|
||
|
|
|
||
|
|
return [new Proxy(proxyDummy, {
|
||
|
|
...handler,
|
||
|
|
get(target, p, receiver) {
|
||
|
|
if (p === proxyInnerValue) return target[proxyInnerValue];
|
||
|
|
if (p === proxyInnerGet) return target[proxyInnerGet];
|
||
|
|
|
||
|
|
// If we're still in the same tick, it means the proxy was immediately used.
|
||
|
|
// thus, we proxy the get access to make things like destructuring work as expected
|
||
|
|
// meow here will also be a proxy
|
||
|
|
// `const { meow } = findByProps("meow");`
|
||
|
|
if (!isChild && isSameTick) {
|
||
|
|
const [recursiveProxy, recursiveSetInnerValue] = proxyInner(err, true);
|
||
|
|
recursiveSetInnerValues.push((innerValue: T) => {
|
||
|
|
recursiveSetInnerValue(Reflect.get(innerValue as object, p, receiver));
|
||
|
|
});
|
||
|
|
|
||
|
|
return recursiveProxy;
|
||
|
|
}
|
||
|
|
|
||
|
|
const innerTarget = target[proxyInnerGet]();
|
||
|
|
if (typeof innerTarget === "object" || typeof innerTarget === "function") {
|
||
|
|
return Reflect.get(innerTarget, p, receiver);
|
||
|
|
}
|
||
|
|
|
||
|
|
throw new Error("proxyInner called on a primitive value");
|
||
|
|
|
||
|
|
}
|
||
|
|
}) as T, setInnerValue];
|
||
|
|
}
|