recoil-toolkit/src/_core/tunnel/getStore.ts

recoil-toolkit

import { RecoilState } from 'recoil';
import { RecoilStore } from '../../types';

interface RecoilStores {
   [name: string]: RecoilStore | undefined;
}

export const recoilStores: RecoilStores = {};
export const DEFAULT_STORE = '$defaultStore$';

export function set<T>(recoilVal: RecoilState<T>, valOrUpdater: ((currVal: T) => T) | T) {
   if (recoilStores[DEFAULT_STORE]?.set)
      return recoilStores[DEFAULT_STORE]?.set(recoilVal, valOrUpdater);
   throw new Error('Recoil Toolkit Tunnel not ready');
}

const pendigGetStorePromises: {
   resolve: (value: RecoilStore) => void;
   reject: (reason?: unknown) => void;
   name: string;
   pending: boolean;
}[] = [];

export const flushGetStorePending = (name: string) => {
   pendigGetStorePromises.forEach(p => {
      if (p.pending && p.name === name) {
         // @ts-ignore
         if (recoilStores[name]) {
            // @ts-ignore
            p.resolve(recoilStores[name]);
         } else p.reject(new Error('no store found'));
         p.pending = false;
      }
   });
};

export function getRecoilStore(name: string = DEFAULT_STORE): Promise<RecoilStore> {
   return new Promise<RecoilStore>((resolve, reject) => {
      if (recoilStores[name]) {
         // @ts-ignore
         resolve(recoilStores[name]);
      } else {
         pendigGetStorePromises.push({
            resolve,
            reject,
            name,
            pending: true,
         });
      }
   });
}