killa/dist/esm/core.mjs

State management for Vanilla and React

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
};