84 lines
3.1 KiB
TypeScript
84 lines
3.1 KiB
TypeScript
import type { Atom, WritableAtom } from './atom.mjs';
|
|
type AnyValue = unknown;
|
|
type AnyError = unknown;
|
|
type AnyAtom = Atom<AnyValue>;
|
|
type OnUnmount = () => void;
|
|
/**
|
|
* State tracked for mounted atoms. An atom is considered "mounted" if it has a
|
|
* subscriber, or is a transitive dependency of another atom that has a
|
|
* subscriber.
|
|
*
|
|
* The mounted state of an atom is freed once it is no longer mounted.
|
|
*/
|
|
type Mounted = {
|
|
/** Set of listeners to notify when the atom value changes. */
|
|
readonly l: Set<() => void>;
|
|
/** Set of mounted atoms that the atom depends on. */
|
|
readonly d: Set<AnyAtom>;
|
|
/** Set of mounted atoms that depends on the atom. */
|
|
readonly t: Set<AnyAtom>;
|
|
/** Function to run when the atom is unmounted. */
|
|
u?: (batch: Batch) => void;
|
|
};
|
|
/**
|
|
* Mutable atom state,
|
|
* tracked for both mounted and unmounted atoms in a store.
|
|
*/
|
|
type AtomState<Value = AnyValue> = {
|
|
/**
|
|
* Map of atoms that the atom depends on.
|
|
* The map value is the epoch number of the dependency.
|
|
*/
|
|
readonly d: Map<AnyAtom, number>;
|
|
/**
|
|
* Set of atoms with pending promise that depend on the atom.
|
|
*
|
|
* This may cause memory leaks, but it's for the capability to continue promises
|
|
*/
|
|
readonly p: Set<AnyAtom>;
|
|
/** The epoch number of the atom. */
|
|
n: number;
|
|
/** Object to store mounted state of the atom. */
|
|
m?: Mounted;
|
|
/** Atom value */
|
|
v?: Value;
|
|
/** Atom error */
|
|
e?: AnyError;
|
|
/** Indicates that the atom value has been changed */
|
|
x?: true;
|
|
};
|
|
type Batch = Readonly<{
|
|
/** Atom dependents map */
|
|
D: Map<AnyAtom, Set<AnyAtom>>;
|
|
/** High priority functions */
|
|
H: Set<() => void>;
|
|
/** Medium priority functions */
|
|
M: Set<() => void>;
|
|
/** Low priority functions */
|
|
L: Set<() => void>;
|
|
}>;
|
|
type StoreArgs = readonly [
|
|
getAtomState: <Value>(atom: Atom<Value>) => AtomState<Value>,
|
|
atomRead: <Value>(atom: Atom<Value>, ...params: Parameters<Atom<Value>['read']>) => Value,
|
|
atomWrite: <Value, Args extends unknown[], Result>(atom: WritableAtom<Value, Args, Result>, ...params: Parameters<WritableAtom<Value, Args, Result>['write']>) => Result,
|
|
atomOnMount: <Value, Args extends unknown[], Result>(atom: WritableAtom<Value, Args, Result>, setAtom: (...args: Args) => Result) => OnUnmount | void
|
|
];
|
|
type DevStoreRev4 = {
|
|
dev4_get_internal_weak_map: () => {
|
|
get: (atom: AnyAtom) => AtomState | undefined;
|
|
};
|
|
dev4_get_mounted_atoms: () => Set<AnyAtom>;
|
|
dev4_restore_atoms: (values: Iterable<readonly [AnyAtom, AnyValue]>) => void;
|
|
};
|
|
type Store = {
|
|
get: <Value>(atom: Atom<Value>) => Value;
|
|
set: <Value, Args extends unknown[], Result>(atom: WritableAtom<Value, Args, Result>, ...args: Args) => Result;
|
|
sub: (atom: AnyAtom, listener: () => void) => () => void;
|
|
unstable_derive: (fn: (...args: StoreArgs) => StoreArgs) => Store;
|
|
};
|
|
export type INTERNAL_DevStoreRev4 = DevStoreRev4;
|
|
export type INTERNAL_PrdStore = Store;
|
|
type PrdOrDevStore = Store | (Store & DevStoreRev4);
|
|
export declare const createStore: () => PrdOrDevStore;
|
|
export declare const getDefaultStore: () => PrdOrDevStore;
|
|
export {};
|