Spaces:
Runtime error
Runtime error
File size: 3,008 Bytes
db5855f |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 |
// @ts-check
import { existsSync, readFileSync } from 'fs';
import { join, resolve } from 'path';
import { generateNotebooksMetadataFile } from '../notebook-metadata/generate-notebooks-map.js';
import { createBuildChecksumFile } from './build-checksum.js';
import { NOTEBOOKS_METADATA_FILE_NAME, NOTEBOOKS_STATUS_FILE_NAME } from './constants.js';
import { fetchNotebooksStatusFile } from './fetch-notebooks-status.js';
/**
*
* @returns {import('vite').PluginOption}
*/
export const collectNotebooksFilesPlugin = () => {
/** @type {import('vite').ResolvedConfig} */
let config;
let distPath = '';
return {
name: 'generate-notebooks-map-file',
configResolved(resolvedConfig) {
config = resolvedConfig;
distPath = resolve(config.root, config.build.outDir);
},
async closeBundle() {
if (config.command === 'build') {
await generateNotebooksMetadataFile(distPath);
await fetchNotebooksStatusFile(distPath);
await createBuildChecksumFile(distPath);
}
},
async configureServer(devServer) {
const notebooksMapFileExists = existsSync(join(distPath, NOTEBOOKS_METADATA_FILE_NAME));
if (notebooksMapFileExists) {
console.info(
`"${NOTEBOOKS_METADATA_FILE_NAME}" file already exists and is served from "${distPath}" dist directory.`
);
} else {
await generateNotebooksMetadataFile(distPath);
}
const notebooksStatusFileExists = existsSync(join(distPath, NOTEBOOKS_STATUS_FILE_NAME));
if (notebooksStatusFileExists) {
console.info(
`"${NOTEBOOKS_STATUS_FILE_NAME}" file already exists and is served from "${distPath}" dist directory.`
);
} else {
console.info(`"${NOTEBOOKS_STATUS_FILE_NAME}" file is not found in "${distPath}" dist directory.\nFetching...`);
try {
await fetchNotebooksStatusFile(distPath);
} catch (error) {
console.warn(`Unable to fetch "${NOTEBOOKS_STATUS_FILE_NAME}".`);
console.warn(error);
// TODO Consider generating mock file
}
}
devServer.middlewares.use(...getFileMiddleware(NOTEBOOKS_METADATA_FILE_NAME, config.base, distPath));
devServer.middlewares.use(...getFileMiddleware(NOTEBOOKS_STATUS_FILE_NAME, config.base, distPath));
},
};
};
/**
* @param {string} fileName
* @param {string} urlBase
* @param {string} distPath
* @returns {[string, import('vite').Connect.NextHandleFunction]}
*/
function getFileMiddleware(fileName, urlBase, distPath) {
const route = `${urlBase}${fileName}`;
/** @type {import('vite').Connect.NextHandleFunction} */
const handler = (_, res) => {
const fileContent = readFileSync(join(distPath, fileName), {
encoding: 'utf8',
});
res.writeHead(200, { 'Content-Type': 'application/json' });
res.write(fileContent);
res.end();
};
return [route, handler];
}
|