@legendapp/state 2.2.0-next.7 → 2.2.0-next.71

package/src/persist-plugins/async-storage.ts

@@ -0,0 +1,102 @@
+import type {
+    Change,
+    ObservablePersistPlugin,
+    ObservablePersistenceConfigLocalGlobalOptions,
+    PersistMetadata,
+} from '@legendapp/state';
+import { internal, isArray, setAtPath } from '@legendapp/state';
+import type { AsyncStorageStatic } from '@react-native-async-storage/async-storage';
+
+const MetadataSuffix = '__m';
+
+let AsyncStorage: AsyncStorageStatic;
+
+const { safeParse, safeStringify } = internal;
+
+export class ObservablePersistAsyncStorage implements ObservablePersistPlugin {
+    private data: Record<string, any> = {};
+
+    // Init
+    public async initialize(config: ObservablePersistenceConfigLocalGlobalOptions) {
+        let tables: readonly string[] = [];
+        const storageConfig = config.asyncStorage;
+        if (storageConfig) {
+            AsyncStorage = storageConfig.AsyncStorage;
+            const { preload } = storageConfig;
+            try {
+                if (preload === true) {
+                    // If preloadAllKeys, load all keys and preload tables on startup
+                    tables = await AsyncStorage.getAllKeys();
+                } else if (isArray(preload)) {
+                    // If preloadKeys, preload load the tables on startup
+                    tables = preload;
+                }
+                if (tables) {
+                    const values = await AsyncStorage.multiGet(tables);
+
+                    values.forEach(([table, value]) => {
+                        this.data[table] = value ? safeParse(value) : undefined;
+                    });
+                }
+            } catch (e) {
+                console.error('[legend-state] ObservablePersistAsyncStorage failed to initialize', e);
+            }
+        } else {
+            console.error('[legend-state] Missing asyncStorage configuration');
+        }
+    }
+    public loadTable(table: string): void | Promise<void> {
+        if (this.data[table] === undefined) {
+            try {
+                return (async () => {
+                    const value = await AsyncStorage.getItem(table);
+                    this.data[table] = value ? safeParse(value) : undefined;
+                })();
+            } catch {
+                console.error('[legend-state] ObservablePersistLocalAsyncStorage failed to parse', table);
+            }
+        }
+    }
+    // Gets
+    public getTable(table: string, init: object) {
+        return this.data[table] ?? init ?? {};
+    }
+    public getMetadata(table: string): PersistMetadata {
+        return this.getTable(table + MetadataSuffix, {});
+    }
+    // Sets
+    public set(table: string, changes: Change[]): Promise<void> {
+        if (!this.data[table]) {
+            this.data[table] = {};
+        }
+
+        for (let i = 0; i < changes.length; i++) {
+            const { path, valueAtPath, pathTypes } = changes[i];
+            this.data[table] = setAtPath(this.data[table], path as string[], pathTypes, valueAtPath);
+        }
+        return this.save(table);
+    }
+    public setMetadata(table: string, metadata: PersistMetadata) {
+        return this.setValue(table + MetadataSuffix, metadata);
+    }
+    public async deleteTable(table: string) {
+        return AsyncStorage.removeItem(table);
+    }
+    public deleteMetadata(table: string) {
+        return this.deleteTable(table + MetadataSuffix);
+    }
+    // Private
+    private async setValue(table: string, value: any) {
+        this.data[table] = value;
+        await this.save(table);
+    }
+    private async save(table: string) {
+        const v = this.data[table];
+
+        if (v !== undefined && v !== null) {
+            return AsyncStorage.setItem(table, safeStringify(v));
+        } else {
+            return AsyncStorage.removeItem(table);
+        }
+    }
+}

package/src/persist-plugins/fetch.ts

@@ -0,0 +1,33 @@
+import { ObservablePersistRemoteFunctions, isString, type ObservablePersistRemoteSetParams } from '@legendapp/state';
+
+interface PersistFetchProps {
+    get: string | RequestInfo;
+    set?: string | RequestInfo;
+    getInit?: RequestInit;
+    setInit?: RequestInit;
+    valueType?: 'arrayBuffer' | 'blob' | 'formData' | 'json' | 'text';
+}
+
+export function persistPluginFetch({
+    get,
+    set,
+    getInit,
+    setInit,
+    valueType,
+}: PersistFetchProps): ObservablePersistRemoteFunctions {
+    const ret: ObservablePersistRemoteFunctions = {
+        get() {
+            return fetch(get, getInit).then((response) => response[valueType || 'json']());
+        },
+    };
+
+    if (set) {
+        ret.set = async ({ value }: ObservablePersistRemoteSetParams<any>) => {
+            const requestInfo = isString(set) ? ({ url: set } as RequestInfo) : set;
+            await fetch(Object.assign({ method: 'POST' }, requestInfo, { body: JSON.stringify(value) }), setInit);
+            // Return undefined to indicate no changes
+        };
+    }
+
+    return ret;
+}