UNPKG

next

Version:

The React Framework

113 lines (112 loc) 4.87 kB
import * as path from 'node:path'; import * as fs from 'node:fs/promises'; import { normalizeAppPath } from '../../../shared/lib/router/utils/app-paths'; import { ensureLeadingSlash } from '../../../shared/lib/page-path/ensure-leading-slash'; import { getSegmentParam } from '../../../shared/lib/router/utils/get-segment-param'; const rootParamsLoader = async function() { const { appDir, pageExtensions } = this.getOptions(); const allRootParams = await collectRootParamsFromFileSystem({ appDir, pageExtensions }); // invalidate the result whenever a file/directory is added/removed inside the app dir or its subdirectories, // because that might mean that a root layout has been moved. this.addContextDependency(appDir); // If there's no root params, there's nothing to generate. if (allRootParams.size === 0) { return 'export {}'; } // Generate a getter for each root param we found. const sortedRootParamNames = Array.from(allRootParams).sort(); const content = [ `import { getRootParam } from 'next/dist/server/request/root-params';`, ...sortedRootParamNames.map((paramName)=>{ return `export function ${paramName}() { return getRootParam('${paramName}'); }`; }) ].join('\n'); return content; }; export default rootParamsLoader; async function collectRootParamsFromFileSystem(opts) { return collectRootParams({ appDir: opts.appDir, rootLayoutFilePaths: await findRootLayouts(opts) }); } function collectRootParams({ rootLayoutFilePaths, appDir }) { const allRootParams = new Set(); for (const rootLayoutFilePath of rootLayoutFilePaths){ const params = getParamsFromLayoutFilePath({ appDir, layoutFilePath: rootLayoutFilePath }); for (const param of params){ allRootParams.add(param); } } return allRootParams; } async function findRootLayouts({ appDir, pageExtensions }) { const layoutFilenameRegex = new RegExp(`^layout\\.(?:${pageExtensions.join('|')})$`); async function visit(directory) { let dir; try { dir = await fs.readdir(directory, { withFileTypes: true }); } catch (err) { // If the directory was removed before we managed to read it, just ignore it. if (err && typeof err === 'object' && 'code' in err && err.code === 'ENOENT') { return []; } throw err; } const subdirectories = []; for (const entry of dir){ if (entry.isDirectory()) { // Directories that start with an underscore are excluded from routing, so we shouldn't look for layouts inside. if (entry.name[0] === '_') { continue; } // Parallel routes cannot occur above a layout, so they can't contain a root layout. if (entry.name[0] === '@') { continue; } const absolutePathname = path.join(directory, entry.name); subdirectories.push(absolutePathname); } else if (entry.isFile()) { if (layoutFilenameRegex.test(entry.name)) { // We found a root layout, so we're not going to recurse into subdirectories, // meaning that we can skip the rest of the entries. // Note that we don't need to track any of the subdirectories as dependencies -- // changes in the subdirectories will only become relevant if this root layout is (re)moved, // in which case the loader will re-run, traverse deeper (because it no longer stops at this root layout) // and then track those directories as needed. const rootLayoutPath = path.join(directory, entry.name); return [ rootLayoutPath ]; } } } if (subdirectories.length === 0) { return []; } const subdirectoryRootLayouts = await Promise.all(subdirectories.map((subdirectory)=>visit(subdirectory))); return subdirectoryRootLayouts.flat(1); } return visit(appDir); } function getParamsFromLayoutFilePath({ appDir, layoutFilePath }) { const rootLayoutPath = normalizeAppPath(ensureLeadingSlash(path.dirname(path.relative(appDir, layoutFilePath)))); const segments = rootLayoutPath.split('/'); const paramNames = []; for (const segment of segments){ const param = getSegmentParam(segment); if (param !== null) { paramNames.push(param.paramName); } } return paramNames; } //# sourceMappingURL=next-root-params-loader.js.map