provide-page/src/extractStates.js

Provides automatic server-side rendering and actions (regardless of whether or not client has JavaScript enabled) to React components. Use in conjunction with `provide-router`.

export default function extractStates(
  providerInstances, getStates, states = {}
) {
  states = extractServerStates(providerInstances, getStates, states);

  return {
    states,
    clientStates: extractClientStates(providerInstances, states)
  };
}

export function extractServerStates(
  providerInstances, getStates, states = {}
) {
  for (let providerKey in providerInstances) {
    let providerInstance = providerInstances[providerKey];

    if (!states[providerKey] || !providerInstance.store.key) {
      states[providerKey] = providerInstance.store.getState();
    }
  }

  return getStates ? getStates(states) : states;
}

export function extractClientStates(providerInstances, states) {
  const clientStates = {};

  for (let providerKey in states) {
    let providerInstance = providerInstances[providerKey];
    let clientStateKeys = providerInstance && providerInstance.clientStateKeys;
    let state = states[providerKey];

    if (typeof clientStateKeys === 'undefined') {
      clientStates[providerKey] = state;
    } else if (clientStateKeys && clientStateKeys.length && state) {
      clientStates[providerKey] = {};

      for (let stateKey of clientStateKeys) {
        clientStates[providerKey][stateKey] = state[stateKey];
      }
    }
  }

  return clientStates;
}