71 lines
2.2 KiB
JavaScript
71 lines
2.2 KiB
JavaScript
import { mapValues, map, get } from 'lodash-es';
|
|
const middlewareFor = (type, middleware) => (api) => (next) => (action) => {
|
|
if (type !== '*' && action.type !== type) return next(action);
|
|
return middleware(api)(next)(action);
|
|
};
|
|
const sliceMw = (slice, mw) => (api) => {
|
|
const getSliceState = () => get(api.getState(), slice);
|
|
return mw(
|
|
Object.assign(Object.assign({}, api), { getState: getSliceState }),
|
|
);
|
|
};
|
|
|
|
export const augmentGetState = (getState, selectors) =>
|
|
Object.assign(
|
|
getState,
|
|
mapValues(selectors, (selector) => {
|
|
return (...args) => {
|
|
let result = selector(api.getState());
|
|
if (typeof result === 'function') return result(...args);
|
|
return result;
|
|
};
|
|
}),
|
|
);
|
|
|
|
export function augmentMiddlewareApi(api, actions, selectors) {
|
|
const getState = augmentGetState(() => api.getState(), selectors);
|
|
const dispatch = (action) => api.dispatch(action);
|
|
Object.assign(
|
|
dispatch,
|
|
mapValues(actions, (action) => {
|
|
return (...args) => api.dispatch(action(...args));
|
|
}),
|
|
);
|
|
return Object.assign(Object.assign({}, api), {
|
|
getState,
|
|
dispatch,
|
|
actions,
|
|
selectors,
|
|
});
|
|
}
|
|
export const effectToMiddleware = (effect, actions, selectors) => {
|
|
let mw = effect;
|
|
let action = '*';
|
|
if (Array.isArray(effect)) {
|
|
action = effect[0];
|
|
mw = effect[1];
|
|
mw = middlewareFor(action, mw);
|
|
}
|
|
return (api) => mw(augmentMiddlewareApi(api, actions, selectors));
|
|
};
|
|
const composeMw = (mws) => (api) => (original_next) =>
|
|
mws.reduceRight((next, mw) => mw(api)(next), original_next);
|
|
export function buildMiddleware(
|
|
effects = [],
|
|
actions = {},
|
|
selectors = {},
|
|
sub = {},
|
|
wrapper = undefined,
|
|
dux = undefined,
|
|
) {
|
|
let inner = map(sub, ({ middleware }, slice) =>
|
|
slice !== '*' && middleware ? sliceMw(slice, middleware) : undefined,
|
|
).filter((x) => x);
|
|
const local = effects.map((effect) =>
|
|
effectToMiddleware(effect, actions, selectors),
|
|
);
|
|
let mws = [...local, ...inner];
|
|
if (wrapper) mws = wrapper(mws, dux);
|
|
return composeMw(mws);
|
|
}
|