shouldUpdateImmutable

shouldUpdateImmutable.js

import { is, isImmutable } from 'immutable';
import { shouldUpdate, shallowEqual } from 'recompose';

const shouldUpdateImmutable = shouldUpdate((props, nextProps) => {
  const keys = Object.keys(props);
  const nextKeys = Object.keys(nextProps);

  if (keys.length !== nextKeys.length) {
    return true;
  }

  for (let i = 0; i < keys.length; i += 1) {
    const key = keys[i];
    if (!hasOwnProperty.call(nextProps, key)) {
      return true;
    }

    const prop = props[key];
    const nextProp = nextProps[key];

    if (isImmutable(prop) && !is(prop, nextProp)) {
      return true;
    }

    if (!shallowEqual(prop, nextProp)) {
      return true;
    }
  }

  return false;
});

export default shouldUpdateImmutable;