wisesplit/lib/parseAtomToJSON.ts
2022-11-06 18:23:09 -08:00

36 lines
1.1 KiB
TypeScript

import { PrimitiveAtom, Getter } from "jotai";
export const parseAtomToJSON = <T extends PrimitiveAtom<any>, K extends any>(atomToParse: T, get: Getter): K => {
const getAtomValue = get(atomToParse);
if (getAtomValue instanceof Array) {
return parseAtomArrayToJSON(getAtomValue, get) as K;
}
if (getAtomValue instanceof Object) {
return parseAtomObjectToJSON(atomToParse as PrimitiveAtom<Object>, get) as K;
}
return getAtomValue as K;
}
const parseAtomObjectToJSON = <T extends PrimitiveAtom<Object>>(atomObject: T, get: Getter) => {
const parsed = {} as Record<string, unknown>;
const getAtomObject = get(atomObject) as Record<string, PrimitiveAtom<unknown>>;
const itemKeys = Object.keys(getAtomObject);
itemKeys.forEach((key) => {
const atomValueToParse = getAtomObject[key];
parsed[key] = parseAtomToJSON(atomValueToParse, get);
})
return parsed;
}
const parseAtomArrayToJSON = <T extends PrimitiveAtom<unknown>[]>(atomArray: T, get: Getter) => {
return atomArray.map((atomElement) => parseAtomToJSON(atomElement, get));
}