@wordpress/data
Version:
Data module for WordPress.
231 lines (230 loc) • 6.46 kB
JavaScript
// packages/data/src/registry.js
import deprecated from "@wordpress/deprecated";
import createReduxStore from "./redux-store";
import coreDataStore from "./store";
import { createEmitter } from "./utils/emitter";
import { lock, unlock } from "./lock-unlock";
function getStoreName(storeNameOrDescriptor) {
return typeof storeNameOrDescriptor === "string" ? storeNameOrDescriptor : storeNameOrDescriptor.name;
}
function createRegistry(storeConfigs = {}, parent = null) {
const stores = {};
const emitter = createEmitter();
let listeningStores = null;
function globalListener() {
emitter.emit();
}
const subscribe = (listener, storeNameOrDescriptor) => {
if (!storeNameOrDescriptor) {
return emitter.subscribe(listener);
}
const storeName = getStoreName(storeNameOrDescriptor);
const store = stores[storeName];
if (store) {
return store.subscribe(listener);
}
if (!parent) {
return emitter.subscribe(listener);
}
return parent.subscribe(listener, storeNameOrDescriptor);
};
function select(storeNameOrDescriptor) {
const storeName = getStoreName(storeNameOrDescriptor);
listeningStores?.add(storeName);
const store = stores[storeName];
if (store) {
return store.getSelectors();
}
return parent?.select(storeName);
}
function __unstableMarkListeningStores(callback, ref) {
listeningStores = /* @__PURE__ */ new Set();
try {
return callback.call(this);
} finally {
ref.current = Array.from(listeningStores);
listeningStores = null;
}
}
function resolveSelect(storeNameOrDescriptor) {
const storeName = getStoreName(storeNameOrDescriptor);
listeningStores?.add(storeName);
const store = stores[storeName];
if (store) {
return store.getResolveSelectors();
}
return parent && parent.resolveSelect(storeName);
}
function suspendSelect(storeNameOrDescriptor) {
const storeName = getStoreName(storeNameOrDescriptor);
listeningStores?.add(storeName);
const store = stores[storeName];
if (store) {
return store.getSuspendSelectors();
}
return parent && parent.suspendSelect(storeName);
}
function dispatch(storeNameOrDescriptor) {
const storeName = getStoreName(storeNameOrDescriptor);
const store = stores[storeName];
if (store) {
return store.getActions();
}
return parent && parent.dispatch(storeName);
}
function withPlugins(attributes) {
return Object.fromEntries(
Object.entries(attributes).map(([key, attribute]) => {
if (typeof attribute !== "function") {
return [key, attribute];
}
return [
key,
function() {
return registry[key].apply(null, arguments);
}
];
})
);
}
function registerStoreInstance(name, createStore) {
if (stores[name]) {
console.error('Store "' + name + '" is already registered.');
return stores[name];
}
const store = createStore();
if (typeof store.getSelectors !== "function") {
throw new TypeError("store.getSelectors must be a function");
}
if (typeof store.getActions !== "function") {
throw new TypeError("store.getActions must be a function");
}
if (typeof store.subscribe !== "function") {
throw new TypeError("store.subscribe must be a function");
}
store.emitter = createEmitter();
const currentSubscribe = store.subscribe;
store.subscribe = (listener) => {
const unsubscribeFromEmitter = store.emitter.subscribe(listener);
const unsubscribeFromStore = currentSubscribe(() => {
if (store.emitter.isPaused) {
store.emitter.emit();
return;
}
listener();
});
return () => {
unsubscribeFromStore?.();
unsubscribeFromEmitter?.();
};
};
stores[name] = store;
store.subscribe(globalListener);
if (parent) {
try {
unlock(store.store).registerPrivateActions(
unlock(parent).privateActionsOf(name)
);
unlock(store.store).registerPrivateSelectors(
unlock(parent).privateSelectorsOf(name)
);
} catch (e) {
}
}
return store;
}
function register(store) {
registerStoreInstance(
store.name,
() => store.instantiate(registry)
);
}
function registerGenericStore(name, store) {
deprecated("wp.data.registerGenericStore", {
since: "5.9",
alternative: "wp.data.register( storeDescriptor )"
});
registerStoreInstance(name, () => store);
}
function registerStore(storeName, options) {
if (!options.reducer) {
throw new TypeError("Must specify store reducer");
}
const store = registerStoreInstance(
storeName,
() => createReduxStore(storeName, options).instantiate(registry)
);
return store.store;
}
function batch(callback) {
if (emitter.isPaused) {
callback();
return;
}
emitter.pause();
Object.values(stores).forEach((store) => store.emitter.pause());
try {
callback();
} finally {
emitter.resume();
Object.values(stores).forEach(
(store) => store.emitter.resume()
);
}
}
let registry = {
batch,
stores,
namespaces: stores,
// TODO: Deprecate/remove this.
subscribe,
select,
resolveSelect,
suspendSelect,
dispatch,
use,
register,
registerGenericStore,
registerStore,
__unstableMarkListeningStores
};
function use(plugin, options) {
if (!plugin) {
return;
}
registry = {
...registry,
...plugin(registry, options)
};
return registry;
}
registry.register(coreDataStore);
for (const [name, config] of Object.entries(storeConfigs)) {
registry.register(createReduxStore(name, config));
}
if (parent) {
parent.subscribe(globalListener);
}
const registryWithPlugins = withPlugins(registry);
lock(registryWithPlugins, {
privateActionsOf: (name) => {
try {
return unlock(stores[name].store).privateActions;
} catch (e) {
return {};
}
},
privateSelectorsOf: (name) => {
try {
return unlock(stores[name].store).privateSelectors;
} catch (e) {
return {};
}
}
});
return registryWithPlugins;
}
export {
createRegistry
};
//# sourceMappingURL=registry.js.map