revenge-mod/Revenge
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

[Storage] Add wrapSync promise accessor

Browse source
This commit is contained in:
redstonekasi 2023-01-30 14:58:06 +01:00 committed by Beef
parent 44652c4002
commit 48edf97b9a
1 changed files with 20 additions and 5 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

25
src/lib/storage.ts
View file

@ -5,6 +5,7 @@ import createEmitter from "./emitter";
const MMKVManager = RN.NativeModules.MMKVManager as MMKVManager;
const emitterSymbol = Symbol("emitter accessor");
const syncAwaitSymbol = Symbol("wrapSync promise accessor");
export function createProxy(target: any = {}): { proxy: any, emitter: Emitter } {
const emitter = createEmitter();
@ -91,11 +92,25 @@ export async function createStorage<T>(storeName: string): Promise<Awaited<T>> {
export function wrapSync<T extends Promise<any>>(store: T): Awaited<T> {
let awaited: any = undefined;
store.then((v) => (awaited = v));
return new Proxy(
{} as Awaited<T>,
Object.fromEntries(Object.getOwnPropertyNames(Reflect)
const awaitQueue: (() => void)[] = [];
const awaitInit = (cb: () => void) => (awaited ? cb() : awaitQueue.push(cb));
store.then((v) => {
awaited = v;
awaitQueue.forEach((cb) => cb());
});
return new Proxy({} as Awaited<T>, {
...Object.fromEntries(Object.getOwnPropertyNames(Reflect)
// @ts-expect-error
.map((k) => [k, (t: T, ...a: any[]) => Reflect[k](awaited ?? t, ...a)])),
);
get(target, prop, recv) {
if (prop === syncAwaitSymbol)
return awaitInit;
return Reflect.get(awaited ?? target, prop, recv);
}
});
}
export const awaitSyncWrapper = (store: any) => new Promise<void>((res) => store[syncAwaitSymbol](res));