85 lines
2.4 KiB
TypeScript
85 lines
2.4 KiB
TypeScript
import omit from 'lodash.omit';
|
|
import { AtomEffect } from 'recoil';
|
|
import { isDefined } from 'twenty-shared/utils';
|
|
import { z } from 'zod';
|
|
import { cookieStorage } from '~/utils/cookie-storage';
|
|
|
|
export const localStorageEffect =
|
|
<T>(key?: string): AtomEffect<T> =>
|
|
({ setSelf, onSet, node }) => {
|
|
const savedValue = localStorage.getItem(key ?? node.key);
|
|
if (savedValue != null) {
|
|
setSelf(JSON.parse(savedValue));
|
|
}
|
|
|
|
onSet((newValue, _, isReset) => {
|
|
isReset
|
|
? localStorage.removeItem(key ?? node.key)
|
|
: localStorage.setItem(key ?? node.key, JSON.stringify(newValue));
|
|
});
|
|
};
|
|
|
|
const customCookieAttributeZodSchema = z.object({
|
|
cookieAttributes: z.object({
|
|
expires: z.union([z.number(), z.instanceof(Date)]).optional(),
|
|
path: z.string().optional(),
|
|
domain: z.string().optional(),
|
|
secure: z.boolean().optional(),
|
|
}),
|
|
});
|
|
|
|
export const isCustomCookiesAttributesValue = (
|
|
value: unknown,
|
|
): value is { cookieAttributes: Cookies.CookieAttributes } =>
|
|
customCookieAttributeZodSchema.safeParse(value).success;
|
|
|
|
export const cookieStorageEffect =
|
|
<T>(
|
|
key: string,
|
|
attributes?: Cookies.CookieAttributes,
|
|
hooks?: {
|
|
validateInitFn?: (payload: T) => boolean;
|
|
},
|
|
): AtomEffect<T | null> =>
|
|
({ setSelf, onSet }) => {
|
|
const savedValue = cookieStorage.getItem(key);
|
|
if (
|
|
isDefined(savedValue) &&
|
|
savedValue.length !== 0 &&
|
|
(!isDefined(hooks?.validateInitFn) ||
|
|
hooks.validateInitFn(JSON.parse(savedValue)))
|
|
) {
|
|
setSelf(JSON.parse(savedValue));
|
|
}
|
|
|
|
const defaultAttributes = {
|
|
expires: new Date(Date.now() + 1000 * 60 * 60 * 24 * 7),
|
|
...(attributes ?? {}),
|
|
};
|
|
|
|
onSet((newValue, _, isReset) => {
|
|
const cookieAttributes = {
|
|
...defaultAttributes,
|
|
...(isCustomCookiesAttributesValue(newValue)
|
|
? newValue.cookieAttributes
|
|
: {}),
|
|
};
|
|
if (
|
|
!newValue ||
|
|
(Object.keys(newValue).length === 1 &&
|
|
isCustomCookiesAttributesValue(newValue))
|
|
) {
|
|
cookieStorage.removeItem(key, cookieAttributes);
|
|
return;
|
|
}
|
|
|
|
isReset
|
|
? cookieStorage.removeItem(key, cookieAttributes)
|
|
: cookieStorage.setItem(
|
|
key,
|
|
JSON.stringify(omit(newValue, ['cookieAttributes'])),
|
|
cookieAttributes,
|
|
);
|
|
});
|
|
};
|