redva-loading/src/index.js

Auto loading plugin for redva.

const SHOW = '@@DVA_LOADING/SHOW';
const HIDE = '@@DVA_LOADING/HIDE';
const NAMESPACE = 'loading';

function createLoading(opts = {}) {
  const namespace = opts.namespace || NAMESPACE;

  const { only = [], except = [] } = opts;
  if (only.length > 0 && except.length > 0) {
    throw Error(
      'It is ambiguous to configurate `only` and `except` items at the same time.'
    );
  }

  const initialState = {
    global: false,
    models: {},
    actions: {},
  };

  const extraReducers = {
    [namespace](state = initialState, { type, payload }) {
      const { namespace, actionType } = payload || {};
      let ret;
      switch (type) {
        case SHOW:
          ret = {
            ...state,
            global: true,
            models: { ...state.models, [namespace]: true },
            actions: { ...state.actions, [actionType]: true },
          };
          break;
        case HIDE: // eslint-disable-line
          const actions = { ...state.actions, [actionType]: false };
          const models = {
            ...state.models,
            [namespace]: Object.keys(actions).some(actionType => {
              const _namespace = actionType.split('/')[0];
              if (_namespace !== namespace) return false;
              return actions[actionType];
            }),
          };
          const global = Object.keys(models).some(namespace => {
            return models[namespace];
          });
          ret = {
            ...state,
            global,
            models,
            actions,
          };
          break;
        default:
          ret = state;
          break;
      }
      return ret;
    },
  };

  function onAction(effect, model, actionType) {
    const { namespace } = model;
    if (
      (only.length === 0 && except.length === 0) ||
      (only.length > 0 && only.indexOf(actionType) !== -1) ||
      (except.length > 0 && except.indexOf(actionType) === -1)
    ) {
      return async function(action, { dispatch }) {
        try{
          await dispatch({ type: SHOW, payload: { namespace, actionType } });
          const result = await effect(...arguments);
          await dispatch({ type: HIDE, payload: { namespace, actionType } });
          return result;
        }catch(e){
          await dispatch({ type: HIDE, payload: { namespace, actionType } });
          throw e;
        }
      };
    } else {
      return effect;
    }
  }

  return {
    extraReducers,
    onAction,
  };
}

export default createLoading;