minh-custom-hooks-release
Version:
Đây là custom hooks được viết bởi hoanggminh2702, bao gồm:
122 lines • 6.57 kB
JavaScript
var __spreadArray = (this && this.__spreadArray) || function (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));
};
/* eslint-disable @typescript-eslint/no-explicit-any */
import { useCallback, useEffect, useRef, useState } from 'react';
import { defer } from 'rxjs';
import { fromPromise } from 'rxjs/internal/observable/innerFrom';
export default function useTaskEffectObservable(_a) {
var task = _a.task, observablePreHandler = _a.observablePreHandler, onNext = _a.onNext, onComplete = _a.onComplete, onError = _a.onError, onCancel = _a.onCancel, deps = _a.deps, _b = _a.enabled, enabled = _b === void 0 ? true : _b, loadingAtInit = _a.loadingAtInit;
var currentSubscription = useRef(null);
var _c = useState(), data = _c[0], setData = _c[1];
var _d = useState(), err = _d[0], setErr = _d[1];
var _e = useState(enabled && loadingAtInit ? "LOADING" /* EnumSubscriptionState.LOADING */ : "IDLE" /* EnumSubscriptionState.IDLE */), taskState = _e[0], setTaskState = _e[1];
var _f = useState(enabled && loadingAtInit ? "LOADING" /* EnumSubscriptionState.LOADING */ : "IDLE" /* EnumSubscriptionState.IDLE */), subscriptionState = _f[0], setSubscriptionState = _f[1];
var run = function () {
var _a;
(_a = currentSubscription.current) === null || _a === void 0 ? void 0 : _a.unsubscribe();
setTaskState("LOADING" /* EnumSubscriptionState.LOADING */);
setSubscriptionState("LOADING" /* EnumSubscriptionState.LOADING */);
if (task instanceof Promise) {
var toObservable = defer(function () {
return fromPromise(task);
});
if (typeof observablePreHandler === 'function') {
toObservable = observablePreHandler(toObservable);
}
currentSubscription.current = toObservable.subscribe({
next: function (v) {
setData(v);
setSubscriptionState("DATARECEIVED" /* EnumSubscriptionState.DATARECEIVED */);
onNext === null || onNext === void 0 ? void 0 : onNext(v);
},
error: function (err) {
setTaskState("FAILED" /* EnumSubscriptionState.FAILED */);
setSubscriptionState("FAILED" /* EnumSubscriptionState.FAILED */);
setErr(err);
onError === null || onError === void 0 ? void 0 : onError(err);
},
complete: function () {
setTaskState("SUCCESS" /* EnumSubscriptionState.SUCCESS */);
setSubscriptionState("SUCCESS" /* EnumSubscriptionState.SUCCESS */);
onComplete === null || onComplete === void 0 ? void 0 : onComplete();
},
});
}
else {
var source$ = defer(function () {
var generateObservable$ = task();
return generateObservable$ instanceof Promise ? fromPromise(generateObservable$) : generateObservable$;
});
if (typeof observablePreHandler === 'function') {
source$ = observablePreHandler(source$);
}
currentSubscription.current = source$.subscribe({
next: function (v) {
setData(v);
setSubscriptionState("DATARECEIVED" /* EnumSubscriptionState.DATARECEIVED */);
onNext === null || onNext === void 0 ? void 0 : onNext(v);
},
error: function (err) {
setErr(err);
setTaskState("FAILED" /* EnumSubscriptionState.FAILED */);
setSubscriptionState("FAILED" /* EnumSubscriptionState.FAILED */);
onError === null || onError === void 0 ? void 0 : onError(err);
},
complete: function () {
setTaskState("SUCCESS" /* EnumSubscriptionState.SUCCESS */);
setSubscriptionState("SUCCESS" /* EnumSubscriptionState.SUCCESS */);
onComplete === null || onComplete === void 0 ? void 0 : onComplete();
},
});
}
};
var cancel = useCallback(function () {
var _a;
(_a = currentSubscription.current) === null || _a === void 0 ? void 0 : _a.unsubscribe();
setTaskState("CANCELED" /* EnumSubscriptionState.CANCELED */);
setSubscriptionState("CANCELED" /* EnumSubscriptionState.CANCELED */);
onCancel === null || onCancel === void 0 ? void 0 : onCancel(data);
}, [data, onCancel]);
useEffect(function () {
if (enabled) {
run();
}
return function () {
var _a;
(_a = currentSubscription.current) === null || _a === void 0 ? void 0 : _a.unsubscribe();
};
}, __spreadArray(__spreadArray([], (deps !== null && deps !== void 0 ? deps : []), true), [enabled], false));
return {
// data state
data: data,
err: err,
// current state
state: taskState,
subscriptionState: subscriptionState,
// task state
isIdle: taskState === "IDLE" /* EnumSubscriptionState.IDLE */,
isLoading: taskState === "LOADING" /* EnumSubscriptionState.LOADING */,
isSuccess: taskState === "SUCCESS" /* EnumSubscriptionState.SUCCESS */,
isFailed: taskState === "FAILED" /* EnumSubscriptionState.FAILED */,
isCanceled: taskState === "CANCELED" /* EnumSubscriptionState.CANCELED */,
// subscription state
isSubscriptionIdle: taskState === "IDLE" /* EnumSubscriptionState.IDLE */,
isSubscriptionLoading: taskState === "LOADING" /* EnumSubscriptionState.LOADING */,
isSubscriptionDataReceived: taskState === "DATARECEIVED" /* EnumSubscriptionState.DATARECEIVED */,
isSubscriptionSuccess: taskState === "SUCCESS" /* EnumSubscriptionState.SUCCESS */,
isSubscriptionFailed: taskState === "FAILED" /* EnumSubscriptionState.FAILED */,
isSubscriptionCanceled: taskState === "CANCELED" /* EnumSubscriptionState.CANCELED */,
// action
refetch: run,
cancel: cancel,
subscription: currentSubscription.current,
};
}
//# sourceMappingURL=useTaskEffectObservable.js.map