react-sweet-state/lib/esm/enhancers/devtools.js

Global + local state combining the best of Redux and Context API

import defaults from '../defaults';

const connectDevTools = (storeState, config) => {
  const devTools = window.__REDUX_DEVTOOLS_EXTENSION__.connect(Object.assign({
    name: `Store ${storeState.key}`,
    serialize: true,
    trace: true
  }, config));

  devTools.init(storeState.getState());
  devTools.subscribe(message => {
    if (message.type === 'DISPATCH') {
      switch (message.payload.type) {
        case 'RESET':
          storeState.resetState();
          devTools.init(storeState.getState());
          return;

        case 'COMMIT':
          devTools.init(storeState.getState());
          return;

        case 'ROLLBACK':
          storeState.setState(JSON.parse(message.state));
          devTools.init(storeState.getState());
          return;

        case 'JUMP_TO_STATE':
        case 'JUMP_TO_ACTION':
          storeState.setState(JSON.parse(message.state));
          return;
      }
    } else if (message.type === 'ACTION') {
      let action = JSON.parse(message.payload);
      storeState.setState(action.payload);
    }
  });
  return devTools;
};

const withDevtools = createStoreState => function () {
  const storeState = createStoreState(...arguments);

  if (defaults.devtools && window && window.__REDUX_DEVTOOLS_EXTENSION__) {
    const origMutator = storeState.mutator;
    let devTools;

    const devtoolMutator = arg => {
      const result = origMutator(arg);

      try {
        if (!devTools) {
          const config = typeof defaults.devtools === 'function' ? defaults.devtools(storeState) : {};
          devTools = connectDevTools(storeState, config);
        }

        devTools.send({
          type: storeState.mutator.actionName,
          payload: arg
        }, storeState.getState(), {}, storeState.key);
      } catch (err) {
        /* ignore devtools errors */
      }

      return result;
    };

    storeState.mutator = devtoolMutator;
  }

  return storeState;
};

export default withDevtools;