allow whitelist to be a function

Author: mattkrick

src/__tests__/createMiddleware-test.js

@@ -96,6 +96,20 @@ describe('createMiddleware', () => {
  engine.save.should.have.been.called;
  });
 
+ it('should allow whitelist function', () => {
+ const engine = { save: sinon.stub().resolves() };
+ const store = { getState: sinon.spy() };
+ const next = sinon.spy();
+ const action = { type: 'ALLOWED' };
+ const whitelistFn = (type) => {
+ return type === 'ALLOWED';
+ };
+
+ createMiddleware(engine, [], whitelistFn)(store)(next)(action);
+
+ engine.save.should.have.been.called;
+ });
+
  describeConsoleWarnInNonProduction(
  'should not process functions',
  () => {

src/createMiddleware.js

@@ -51,12 +51,20 @@ function isValidAction(action) {
  return false;
 }
 
+function handleWhitelist(action, actionWhitelist) {
+ if (Array.isArray(actionWhitelist)) {
+ // Don't filter if the whitelist is empty
+ return actionWhitelist.length === 0 ? true : actionWhitelist.indexOf(action.type) !== -1;
+ }
+ // actionWhitelist is a function that returns true or false
+ return actionWhitelist(action.type);
+}
 
 export default (engine, actionBlacklist = [], actionWhitelist = []) => {
  // Also don't save if we process our own actions
  const blacklistedActions = [...actionBlacklist, LOAD, SAVE];
 
- if (process.env.NODE_ENV !== 'production') {
+ if (process.env.NODE_ENV !== 'production' && Array.isArray(actionWhitelist)) {
  warnAboutConfusingFiltering(actionBlacklist, actionWhitelist);
  }
 
@@ -69,9 +77,7 @@ export default (engine, actionBlacklist = [], actionWhitelist = []) => {
  }
 
  const isOnBlacklist = blacklistedActions.indexOf(action.type) !== -1;
- const isOnWhitelist = actionWhitelist.length === 0
- ? true // Don't filter if the whitelist is empty
- : actionWhitelist.indexOf(action.type) !== -1;
+ const isOnWhitelist = handleWhitelist(action, actionWhitelist);
 
  // Skip blacklisted actions
  if (!isOnBlacklist && isOnWhitelist) {