neuro-store
Version:
230 lines (212 loc) • 9.14 kB
JavaScript
;
var react = require('react');
var jsxRuntime = require('react/jsx-runtime');
var immer = require('immer');
/******************************************************************************
Copyright (c) Microsoft Corporation.
Permission to use, copy, modify, and/or distribute this software for any
purpose with or without fee is hereby granted.
THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES WITH
REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF MERCHANTABILITY
AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY SPECIAL, DIRECT,
INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES WHATSOEVER RESULTING FROM
LOSS OF USE, DATA OR PROFITS, WHETHER IN AN ACTION OF CONTRACT, NEGLIGENCE OR
OTHER TORTIOUS ACTION, ARISING OUT OF OR IN CONNECTION WITH THE USE OR
PERFORMANCE OF THIS SOFTWARE.
***************************************************************************** */
/* global Reflect, Promise, SuppressedError, Symbol, Iterator */
function __awaiter(thisArg, _arguments, P, generator) {
function adopt(value) { return value instanceof P ? value : new P(function (resolve) { resolve(value); }); }
return new (P || (P = Promise))(function (resolve, reject) {
function fulfilled(value) { try { step(generator.next(value)); } catch (e) { reject(e); } }
function rejected(value) { try { step(generator["throw"](value)); } catch (e) { reject(e); } }
function step(result) { result.done ? resolve(result.value) : adopt(result.value).then(fulfilled, rejected); }
step((generator = generator.apply(thisArg, _arguments || [])).next());
});
}
function __generator(thisArg, body) {
var _ = { label: 0, sent: function() { if (t[0] & 1) throw t[1]; return t[1]; }, trys: [], ops: [] }, f, y, t, g = Object.create((typeof Iterator === "function" ? Iterator : Object).prototype);
return g.next = verb(0), g["throw"] = verb(1), g["return"] = verb(2), typeof Symbol === "function" && (g[Symbol.iterator] = function() { return this; }), g;
function verb(n) { return function (v) { return step([n, v]); }; }
function step(op) {
if (f) throw new TypeError("Generator is already executing.");
while (g && (g = 0, op[0] && (_ = 0)), _) try {
if (f = 1, y && (t = op[0] & 2 ? y["return"] : op[0] ? y["throw"] || ((t = y["return"]) && t.call(y), 0) : y.next) && !(t = t.call(y, op[1])).done) return t;
if (y = 0, t) op = [op[0] & 2, t.value];
switch (op[0]) {
case 0: case 1: t = op; break;
case 4: _.label++; return { value: op[1], done: false };
case 5: _.label++; y = op[1]; op = [0]; continue;
case 7: op = _.ops.pop(); _.trys.pop(); continue;
default:
if (!(t = _.trys, t = t.length > 0 && t[t.length - 1]) && (op[0] === 6 || op[0] === 2)) { _ = 0; continue; }
if (op[0] === 3 && (!t || (op[1] > t[0] && op[1] < t[3]))) { _.label = op[1]; break; }
if (op[0] === 6 && _.label < t[1]) { _.label = t[1]; t = op; break; }
if (t && _.label < t[2]) { _.label = t[2]; _.ops.push(op); break; }
if (t[2]) _.ops.pop();
_.trys.pop(); continue;
}
op = body.call(thisArg, _);
} catch (e) { op = [6, e]; y = 0; } finally { f = t = 0; }
if (op[0] & 5) throw op[1]; return { value: op[0] ? op[1] : void 0, done: true };
}
}
function __spreadArray(to, from, pack) {
if (pack || arguments.length === 2) for (var i = 0, l = from.length, ar; i < l; i++) {
if (ar || !(i in from)) {
if (!ar) ar = Array.prototype.slice.call(from, 0, i);
ar[i] = from[i];
}
}
return to.concat(ar || Array.prototype.slice.call(from));
}
typeof SuppressedError === "function" ? SuppressedError : function (error, suppressed, message) {
var e = new Error(message);
return e.name = "SuppressedError", e.error = error, e.suppressed = suppressed, e;
};
var StoreContext = react.createContext(undefined);
var thunkMiddleware = function (_a) {
var dispatch = _a.dispatch, getState = _a.getState;
return function (next) { return function (action) {
if (typeof action === "function") {
return action(dispatch, getState);
}
return next(action);
}; };
};
/* --- [StoreProvider] --- */
var StoreProvider = function (_a) {
var children = _a.children, store = _a.store;
var _b = react.useReducer(store.reducers, store.initialState), state = _b[0], baseDispatch = _b[1];
var coreDispatch = __spreadArray([thunkMiddleware], store.middlewares, true).reduceRight(function (next, mw) {
return mw({
dispatch: next,
getState: function () { return state; },
})(next);
}, baseDispatch);
// Add unwrap functionality
var dispatchWithPromise = Object.assign(coreDispatch, {
withPromise: function (action) {
try {
var result = coreDispatch(action);
if (result instanceof Promise) {
return result;
}
return Promise.resolve(result);
}
catch (err) {
return Promise.reject(err);
}
},
});
return (
// @ts-ignore
jsxRuntime.jsx(StoreContext.Provider, { value: { state: state, dispatch: dispatchWithPromise }, children: children }));
};
/* --- [useDispatch hook] --- */
var useDispatch = function () {
var context = react.useContext(StoreContext);
if (!context) {
throw new Error("useDispatch must be used within a StoreProvider");
}
return context.dispatch;
};
/* --- [useSelector hook] --- */
function useSelector(selector) {
var context = react.useContext(StoreContext);
if (!context) {
throw new Error("useSelector must be used within a StoreProvider");
}
return selector(context.state);
}
function createSlice(_a) {
var name = _a.name, initialState = _a.initialState, reducer = _a.reducer;
var immerReducer = Object.assign(immer.produce(reducer), { initialState: initialState });
return {
name: name,
initialState: initialState,
reducer: immerReducer,
};
}
/* --- [async Thunk] --- */
function asyncThunk(type, asyncFn) {
var _this = this;
var thunk = function (arg) { return function (dispatch) { return __awaiter(_this, void 0, void 0, function () {
var data, error_1;
return __generator(this, function (_a) {
switch (_a.label) {
case 0:
_a.trys.push([0, 2, , 3]);
dispatch({ type: "".concat(type, "/pending") });
return [4 /*yield*/, asyncFn(arg, { dispatch: dispatch })];
case 1:
data = _a.sent();
dispatch({ type: "".concat(type, "/fulfilled"), payload: data });
return [2 /*return*/, data];
case 2:
error_1 = _a.sent();
dispatch({ type: "".concat(type, "/rejected"), error: error_1 });
throw error_1;
case 3: return [2 /*return*/];
}
});
}); }; };
return Object.assign(thunk, {
pending: "".concat(type, "/pending"),
fulfilled: "".concat(type, "/fulfilled"),
rejected: "".concat(type, "/rejected"),
});
}
/** [----createStore----]
* @example
* // basic middleware example
* const middleware = ({ dispatch, getState }: any) => (next: any) =>(action: any) => {
* if (typeof action === "function") {
* return action(dispatch, getState);
* }
* return next(action);
* };
* @example
* // basic store
* const reducers = {
auth: authSlice.reducer,
};
type Reducers = typeof reducers;
type State = {
[K in keyof Reducers]: ReturnType<Reducers[K]>;
};
type Middleware = [];
const store = createStore<State, Reducers, Middleware>({
reducers: reducers,
middlewares: [],
});
*
*/
function createStore(store) {
function rootReducer(reducers) {
return function (state, action) {
var nextState = {};
for (var key in reducers) {
nextState[key] = reducers[key](state[key], action);
}
return nextState;
};
}
// Extract initial state from each reducer slice
var initialState = {};
for (var key in store.reducers) {
initialState[key] = store.reducers[key].initialState;
}
return {
initialState: initialState,
reducers: rootReducer(store.reducers),
middlewares: store.middlewares,
};
}
exports.StoreContext = StoreContext;
exports.StoreProvider = StoreProvider;
exports.asyncThunk = asyncThunk;
exports.createSlice = createSlice;
exports.createStore = createStore;
exports.useDispatch = useDispatch;
exports.useSelector = useSelector;