UNPKG

killa

Version:

State management for Vanilla and React

68 lines (67 loc) 2.46 kB
import clone from "just-clone"; import { deepEquals, isObject } from "killa/deep-equals"; import { SYMBOL_STORE, SYMBOL_SUBSCRIBER } from "killa/constants"; function createStore(initializer = Object.assign({}), options) { let state; const subscribers = /* @__PURE__ */ new Set(); const compare = typeof (options == null ? void 0 : options.compare) === "function" ? options.compare : deepEquals; const getState = () => clone(state); const setState = (fn, force = false) => { const newState = fn(getState()) || {}; if (!compare(state, newState)) { const prevState = state; state = force ? Object.assign(newState) : Object.assign(getState(), clone(newState)); subscribers.forEach((subscriber) => { const _prevState = clone(prevState); const _newState = getState(); if (subscriber.$$subscriber && subscriber.$$selector) { const selectorState = subscriber.$$selectorState; const nextselectorState = subscriber.$$selector(state); if (!compare(selectorState, nextselectorState)) { subscriber.$$selectorState = nextselectorState; subscriber(_newState, _prevState); } return; } subscriber(_newState, _prevState); }); } }; const initialState = typeof initializer === "function" ? initializer(getState, setState) : initializer; const subscribe = (subscriber, selector) => { if (typeof selector === "function") { subscriber.$$subscriber = SYMBOL_SUBSCRIBER; subscriber.$$selectorState = selector(state); subscriber.$$selector = selector; } subscribers.add(subscriber); return () => subscribers.delete(subscriber); }; if (!isObject(initialState)) throw new Error("Store must be an object."); state = clone(initialState); const resetState = (state2 = null) => { const newState = state2 && isObject(state2) ? state2 : clone(initialState); store.setState(() => newState, true); }; const destroy = () => subscribers.clear(); const store = { $$store: SYMBOL_STORE, getState, setState, subscribe, getServerState: () => initialState, resetState, destroy }; if ((options == null ? void 0 : options.use) && Array.isArray(options.use)) { options.use.forEach((middleware) => middleware(store)); } if (process.env.NODE_ENV !== "test") { return Object.freeze(store); } return store; } export { createStore };