one
Version:
One is a new React Framework that makes Vite serve both native and web.
192 lines (191 loc) • 5.81 kB
JavaScript
import * as React from "react";
import { Platform } from "react-native-web";
let currentLocation = typeof window !== "undefined" ? window.location.pathname + window.location.search : "";
let isBlocking = false;
let isProceeding = false;
let pendingNavigation = null;
const blockerCallbacks = /* @__PURE__ */new Map();
let listenersSetup = false;
function setupListeners() {
if (listenersSetup || typeof window === "undefined") return;
listenersSetup = true;
currentLocation = window.location.pathname + window.location.search;
window.addEventListener("popstate", () => {
if (isProceeding) return;
const nextLocation = window.location.pathname + window.location.search;
for (const [, callbacks] of blockerCallbacks) {
if (callbacks.shouldBlock()) {
isBlocking = true;
pendingNavigation = {
previousLocation: currentLocation,
nextLocation,
historyAction: "pop"
};
window.history.go(1);
callbacks.onBlock(pendingNavigation);
return;
}
}
currentLocation = nextLocation;
});
const originalPushState = window.history.pushState.bind(window.history);
const originalReplaceState = window.history.replaceState.bind(window.history);
window.history.pushState = function (state, title, url) {
if (isProceeding || !url) {
return originalPushState(state, title, url);
}
const nextLocation = typeof url === "string" ? url : url.toString();
for (const [, callbacks] of blockerCallbacks) {
if (callbacks.shouldBlock()) {
isBlocking = true;
pendingNavigation = {
previousLocation: currentLocation,
nextLocation,
historyAction: "push"
};
callbacks.onBlock(pendingNavigation);
return;
}
}
currentLocation = nextLocation;
return originalPushState(state, title, url);
};
window.history.replaceState = function (state, title, url) {
if (isProceeding || !url) {
return originalReplaceState(state, title, url);
}
const nextLocation = typeof url === "string" ? url : url.toString();
for (const [, callbacks] of blockerCallbacks) {
if (callbacks.shouldBlock()) {
isBlocking = true;
pendingNavigation = {
previousLocation: currentLocation,
nextLocation,
historyAction: "replace"
};
callbacks.onBlock(pendingNavigation);
return;
}
}
currentLocation = nextLocation;
return originalReplaceState(state, title, url);
};
window.addEventListener("beforeunload", event => {
for (const [, callbacks] of blockerCallbacks) {
if (callbacks.shouldBlock()) {
event.preventDefault();
event.returnValue = "";
return;
}
}
});
}
function useBlocker(shouldBlock) {
const [state, setState] = React.useState("unblocked");
const [blockedLocation, setBlockedLocation] = React.useState(null);
const idRef = React.useRef(null);
const shouldBlockRef = React.useRef(shouldBlock);
shouldBlockRef.current = shouldBlock;
React.useEffect(() => {
if (Platform.OS !== "web" || typeof window === "undefined") return;
setupListeners();
const id = Symbol("blocker");
idRef.current = id;
blockerCallbacks.set(id, {
shouldBlock: () => {
const block = shouldBlockRef.current;
if (typeof block === "function") {
return block({
currentLocation,
nextLocation: pendingNavigation?.nextLocation || "",
historyAction: pendingNavigation?.historyAction || "push"
});
}
return block;
},
onBlock: pending => {
setBlockedLocation(pending.nextLocation);
setState("blocked");
},
onProceed: () => {
setState("proceeding");
},
onReset: () => {
setState("unblocked");
setBlockedLocation(null);
}
});
return () => {
blockerCallbacks.delete(id);
};
}, []);
const reset = React.useCallback(() => {
isBlocking = false;
pendingNavigation = null;
setBlockedLocation(null);
setState("unblocked");
}, []);
const proceed = React.useCallback(() => {
if (!pendingNavigation) return;
setState("proceeding");
isProceeding = true;
const pending = pendingNavigation;
pendingNavigation = null;
isBlocking = false;
requestAnimationFrame(() => {
if (pending.historyAction === "pop") {
window.history.back();
} else if (pending.historyAction === "push") {
window.history.pushState(null, "", pending.nextLocation);
} else {
window.history.replaceState(null, "", pending.nextLocation);
}
currentLocation = pending.nextLocation;
requestAnimationFrame(() => {
isProceeding = false;
setBlockedLocation(null);
setState("unblocked");
});
});
}, []);
if (state === "unblocked") {
return {
state: "unblocked"
};
}
if (state === "proceeding") {
return {
state: "proceeding",
location: blockedLocation
};
}
return {
state: "blocked",
reset,
proceed,
location: blockedLocation
};
}
function checkBlocker(nextLocation, historyAction = "push") {
if (Platform.OS !== "web" || typeof window === "undefined") {
return false;
}
if (isProceeding) {
return false;
}
for (const [, callbacks] of blockerCallbacks) {
if (callbacks.shouldBlock()) {
isBlocking = true;
pendingNavigation = {
previousLocation: currentLocation,
nextLocation,
historyAction
};
callbacks.onBlock(pendingNavigation);
return true;
}
}
return false;
}
export { checkBlocker, useBlocker };
//# sourceMappingURL=useBlocker.mjs.map