one
Version:
One is a new React Framework that makes Vite serve both native and web.
218 lines (217 loc) • 7.92 kB
JavaScript
import { useCallback, useSyncExternalStore } from "react";
import { registerDevtoolsFunction } from "./devtools/registry.mjs";
import { useParams, usePathname } from "./hooks.mjs";
import { preloadedLoaderData, preloadingLoader } from "./router/router.mjs";
import { getLoaderPath } from "./utils/cleanUrl.mjs";
import { dynamicImport } from "./utils/dynamicImport.mjs";
import { weakKey } from "./utils/weakKey.mjs";
import { useServerContext } from "./vite/one-server-only.mjs";
const loaderTimingHistory = [],
MAX_TIMING_HISTORY = 50;
function recordLoaderTiming(entry) {
process.env.NODE_ENV === "development" && (loaderTimingHistory.unshift(entry), loaderTimingHistory.length > MAX_TIMING_HISTORY && loaderTimingHistory.pop(), typeof window < "u" && (window.dispatchEvent(new CustomEvent("one-loader-timing", {
detail: entry
})), entry.error && window.dispatchEvent(new CustomEvent("one-error", {
detail: {
error: {
message: entry.error,
name: "LoaderError"
},
route: {
pathname: entry.path
},
timestamp: Date.now(),
type: "loader"
}
}))));
}
function getLoaderTimingHistory() {
return loaderTimingHistory;
}
registerDevtoolsFunction("getLoaderTimingHistory", getLoaderTimingHistory);
const loaderState = {},
subscribers = /* @__PURE__ */new Set();
function updateState(path, updates) {
loaderState[path] = {
...loaderState[path],
...updates
}, subscribers.forEach(callback => {
callback();
});
}
function subscribe(callback) {
return subscribers.add(callback), () => subscribers.delete(callback);
}
function getLoaderState(path, preloadedData2) {
return loaderState[path] || (loaderState[path] = {
data: preloadedData2,
error: void 0,
promise: void 0,
state: "idle",
hasLoadedOnce: !!preloadedData2
}), loaderState[path];
}
async function refetchLoader(pathname2) {
const startTime2 = performance.now();
updateState(pathname2, {
state: "loading",
error: null
});
try {
const cacheBust = `${Date.now()}`,
loaderJSUrl2 = getLoaderPath(pathname2, !0, cacheBust),
moduleLoadStart2 = performance.now(),
module2 = await dynamicImport(loaderJSUrl2),
moduleLoadTime2 = performance.now() - moduleLoadStart2,
executionStart2 = performance.now(),
result2 = await module2.loader(),
executionTime2 = performance.now() - executionStart2,
totalTime2 = performance.now() - startTime2;
updateState(pathname2, {
data: result2,
state: "idle",
timestamp: Date.now(),
hasLoadedOnce: !0
}), recordLoaderTiming({
path: pathname2,
startTime: startTime2,
moduleLoadTime: moduleLoadTime2,
executionTime: executionTime2,
totalTime: totalTime2,
source: "refetch"
});
} catch (err) {
const totalTime2 = performance.now() - startTime2;
throw updateState(pathname2, {
error: err,
state: "idle"
}), recordLoaderTiming({
path: pathname2,
startTime: startTime2,
totalTime: totalTime2,
error: err instanceof Error ? err.message : String(err),
source: "refetch"
}), err;
}
}
process.env.NODE_ENV === "development" && typeof window < "u" && (window.__oneRefetchLoader = refetchLoader);
function useLoaderState(loader) {
const {
loaderProps: loaderPropsFromServerContext,
loaderData: loaderDataFromServerContext
} = useServerContext() || {},
params = useParams(),
pathname = usePathname(),
currentPath = pathname.replace(/\/index$/, "").replace(/\/$/, "") || "/";
if (typeof window > "u" && loader) return {
data: useAsyncFn(loader, loaderPropsFromServerContext || {
path: pathname,
params
}),
refetch: async () => {},
state: "idle"
};
const serverContextPath = loaderPropsFromServerContext?.path,
preloadedData = serverContextPath === currentPath ? loaderDataFromServerContext : void 0,
loaderStateEntry = useSyncExternalStore(subscribe, () => getLoaderState(currentPath, preloadedData), () => getLoaderState(currentPath, preloadedData)),
refetch = useCallback(() => refetchLoader(currentPath), [currentPath]);
if (!loader) return {
refetch,
state: loaderStateEntry.state
};
if (!loaderStateEntry.data && !loaderStateEntry.promise && !loaderStateEntry.hasLoadedOnce && loader) {
const resolvedPreloadData = preloadedLoaderData[currentPath];
if (resolvedPreloadData !== void 0) delete preloadedLoaderData[currentPath], delete preloadingLoader[currentPath], loaderStateEntry.data = resolvedPreloadData, loaderStateEntry.hasLoadedOnce = !0;else if (preloadingLoader[currentPath]) {
const promise2 = preloadingLoader[currentPath].then(val => {
delete preloadingLoader[currentPath], delete preloadedLoaderData[currentPath], updateState(currentPath, {
data: val,
hasLoadedOnce: !0,
promise: void 0
});
}).catch(err => {
console.error("Error running loader()", err), delete preloadingLoader[currentPath], updateState(currentPath, {
error: err,
promise: void 0
});
});
loaderStateEntry.promise = promise2;
} else {
const loadData = async () => {
const startTime = performance.now();
try {
const loaderJSUrl = getLoaderPath(currentPath, !0),
moduleLoadStart = performance.now(),
module = await dynamicImport(loaderJSUrl),
moduleLoadTime = performance.now() - moduleLoadStart,
executionStart = performance.now(),
result = await module.loader(),
executionTime = performance.now() - executionStart,
totalTime = performance.now() - startTime;
updateState(currentPath, {
data: result,
hasLoadedOnce: !0,
promise: void 0
}), recordLoaderTiming({
path: currentPath,
startTime,
moduleLoadTime,
executionTime,
totalTime,
source: "initial"
});
} catch (err) {
const totalTime2 = performance.now() - startTime;
updateState(currentPath, {
error: err,
promise: void 0
}), recordLoaderTiming({
path: currentPath,
startTime,
totalTime: totalTime2,
error: err instanceof Error ? err.message : String(err),
source: "initial"
});
}
},
promise = loadData();
loaderStateEntry.promise = promise;
}
}
if (loader) {
if (loaderStateEntry.error && !loaderStateEntry.hasLoadedOnce) throw loaderStateEntry.error;
if (loaderStateEntry.data === void 0 && loaderStateEntry.promise && !loaderStateEntry.hasLoadedOnce) throw loaderStateEntry.promise;
return {
data: loaderStateEntry.data,
refetch,
state: loaderStateEntry.state
};
} else return {
refetch,
state: loaderStateEntry.state
};
}
function useLoader(loader2) {
const {
data: data2
} = useLoaderState(loader2);
return data2;
}
const results = /* @__PURE__ */new Map(),
started = /* @__PURE__ */new Map();
function useAsyncFn(val, props) {
const key = (val ? weakKey(val) : "") + JSON.stringify(props);
if (val && !started.get(key)) {
started.set(key, !0);
let next = val(props);
next instanceof Promise && (next = next.then(final => {
results.set(key, final);
}).catch(err => {
console.error("Error running loader()", err), results.set(key, void 0);
})), results.set(key, next);
}
const current = results.get(key);
if (current instanceof Promise) throw current;
return current;
}
export { getLoaderTimingHistory, refetchLoader, useLoader, useLoaderState };
//# sourceMappingURL=useLoader.mjs.map