180 lines
7.1 KiB
JavaScript
Raw Normal View History

2025-09-05 14:59:21 +08:00
import { toLazyValue } from "./createDataLoader-BkaFsukJ.js";
import { ABORT_CONTROLLER_KEY, APP_KEY, IS_SSR_KEY, IS_USE_DATA_LOADER_KEY, LOADER_ENTRIES_KEY, LOADER_SET_KEY, NAVIGATION_RESULTS_KEY, PENDING_LOCATION_KEY, STAGED_NO_VALUE, assign, currentContext, getCurrentContext, isDataLoader, isSubsetOf, setCurrentContext, trackRoute, withLoaderContext } from "./utils-D_yE8zIx.js";
import { isNavigationFailure } from "vue-router";
import { effectScope, inject, shallowRef } from "vue";
//#region src/data-loaders/navigation-guard.ts
/**
* Key to inject the global loading state for loaders used in `useIsDataLoading`.
* @internal
*/
const IS_DATA_LOADING_KEY = Symbol();
/**
* TODO: export functions that allow preloading outside of a navigation guard
*/
/**
* Setups the different Navigation Guards to collect the data loaders from the route records and then to execute them.
* @internal used by the `DataLoaderPlugin`
* @see {@link DataLoaderPlugin}
*
* @param router - the router instance
* @returns
*/
function setupLoaderGuard({ router, app, effect: scope, isSSR, errors: globalErrors = [], selectNavigationResult = (results) => results[0].value }) {
if (router[LOADER_ENTRIES_KEY] != null) {
if (process.env.NODE_ENV !== "production") console.warn("[vue-router]: Data Loader was setup twice. Make sure to setup only once.");
return () => {};
}
if (process.env.NODE_ENV === "development" && !isSSR) console.warn("[vue-router]: Data Loader is experimental and subject to breaking changes in the future.");
router[LOADER_ENTRIES_KEY] = /* @__PURE__ */ new WeakMap();
router[APP_KEY] = app;
router[IS_SSR_KEY] = !!isSSR;
const isDataLoading = scope.run(() => shallowRef(false));
app.provide(IS_DATA_LOADING_KEY, isDataLoading);
const removeLoaderGuard = router.beforeEach((to) => {
if (router[PENDING_LOCATION_KEY]) router[PENDING_LOCATION_KEY].meta[ABORT_CONTROLLER_KEY]?.abort();
router[PENDING_LOCATION_KEY] = to;
to.meta[LOADER_SET_KEY] = /* @__PURE__ */ new Set();
to.meta[ABORT_CONTROLLER_KEY] = new AbortController();
to.meta[NAVIGATION_RESULTS_KEY] = [];
const lazyLoadingPromises = [];
for (const record of to.matched) if (!record.meta[LOADER_SET_KEY]) {
record.meta[LOADER_SET_KEY] = new Set(record.meta.loaders || []);
for (const componentName in record.components) {
const component = record.components[componentName];
const promise = (isAsyncModule(component) ? component() : Promise.resolve(component)).then((viewModule) => {
if (typeof viewModule === "function") return;
for (const exportName in viewModule) {
const exportValue = viewModule[exportName];
if (isDataLoader(exportValue)) record.meta[LOADER_SET_KEY].add(exportValue);
}
if (Array.isArray(viewModule.__loaders)) {
for (const loader of viewModule.__loaders) if (isDataLoader(loader)) record.meta[LOADER_SET_KEY].add(loader);
}
});
lazyLoadingPromises.push(promise);
}
}
return Promise.all(lazyLoadingPromises).then(() => {
for (const record of to.matched) for (const loader of record.meta[LOADER_SET_KEY]) to.meta[LOADER_SET_KEY].add(loader);
});
});
const removeDataLoaderGuard = router.beforeResolve((to, from) => {
const loaders = Array.from(to.meta[LOADER_SET_KEY]);
setCurrentContext([]);
isDataLoading.value = true;
return Promise.all(loaders.map((loader) => {
const { server, lazy, errors } = loader._.options;
if (!server && isSSR) return;
const ret = scope.run(() => app.runWithContext(() => loader._.load(to, router, from)));
return !isSSR && toLazyValue(lazy, to, from) ? void 0 : ret.catch((reason) => {
if (!errors) throw reason;
if (errors === true) {
if (Array.isArray(globalErrors) ? globalErrors.some((Err) => reason instanceof Err) : globalErrors(reason)) return;
} else if (Array.isArray(errors) ? errors.some((Err) => reason instanceof Err) : errors(reason)) return;
throw reason;
});
})).then(() => {
if (to.meta[NAVIGATION_RESULTS_KEY].length) return selectNavigationResult(to.meta[NAVIGATION_RESULTS_KEY]);
}).catch((error) => error instanceof NavigationResult ? error.value : Promise.reject(error)).finally(() => {
setCurrentContext([]);
isDataLoading.value = false;
});
});
const removeAfterEach = router.afterEach((to, from, failure) => {
if (failure) {
to.meta[ABORT_CONTROLLER_KEY]?.abort(failure);
if (isNavigationFailure(failure, 16)) for (const loader of to.meta[LOADER_SET_KEY]) {
const entry = loader._.getEntry(router);
entry.resetPending();
}
} else for (const loader of to.meta[LOADER_SET_KEY]) {
const { commit, lazy } = loader._.options;
if (commit === "after-load") {
const entry = loader._.getEntry(router);
if (entry && (!toLazyValue(lazy, to, from) || !entry.isLoading.value)) entry.commit(to);
}
}
if (router[PENDING_LOCATION_KEY] === to) router[PENDING_LOCATION_KEY] = null;
});
const removeOnError = router.onError((error, to) => {
to.meta[ABORT_CONTROLLER_KEY]?.abort(error);
if (router[PENDING_LOCATION_KEY] === to) router[PENDING_LOCATION_KEY] = null;
});
return () => {
delete router[LOADER_ENTRIES_KEY];
delete router[APP_KEY];
removeLoaderGuard();
removeDataLoaderGuard();
removeAfterEach();
removeOnError();
};
}
/**
* Allows differentiating lazy components from functional components and vue-class-component
* @internal
*
* @param component
*/
function isAsyncModule(asyncMod) {
return typeof asyncMod === "function" && !("displayName" in asyncMod) && !("props" in asyncMod) && !("emits" in asyncMod) && !("__vccOpts" in asyncMod);
}
/**
* Possible values to change the result of a navigation within a loader. Can be returned from a data loader and will
* appear in `selectNavigationResult`. If thrown, it will immediately cancel the navigation. It can only contain values
* that cancel the navigation.
*
* @example
* ```ts
* export const useUserData = defineLoader(async (to) => {
* const user = await fetchUser(to.params.id)
* if (!user) {
* return new NavigationResult('/404')
* }
* return user
* })
* ```
*/
var NavigationResult = class {
constructor(value) {
this.value = value;
}
};
/**
* Data Loader plugin to add data loading support to Vue Router.
*
* @example
* ```ts
* const router = createRouter({
* routes,
* history: createWebHistory(),
* })
*
* const app = createApp({})
* app.use(DataLoaderPlugin, { router })
* app.use(router)
* ```
*/
function DataLoaderPlugin(app, options) {
const effect = effectScope(true);
const removeGuards = setupLoaderGuard(assign({
app,
effect
}, options));
const { unmount } = app;
app.unmount = () => {
effect.stop();
removeGuards();
unmount.call(app);
};
}
/**
* Return a ref that reflects the global loading state of all loaders within a navigation.
* This state doesn't update if `refresh()` is manually called.
*/
function useIsDataLoading() {
return inject(IS_DATA_LOADING_KEY);
}
//#endregion
export { ABORT_CONTROLLER_KEY, APP_KEY, DataLoaderPlugin, IS_SSR_KEY, IS_USE_DATA_LOADER_KEY, LOADER_ENTRIES_KEY, LOADER_SET_KEY, NAVIGATION_RESULTS_KEY, NavigationResult, PENDING_LOCATION_KEY, STAGED_NO_VALUE, assign, currentContext, getCurrentContext, isSubsetOf, setCurrentContext, toLazyValue, trackRoute, useIsDataLoading, withLoaderContext };