102 lines
3.4 KiB
TypeScript
102 lines
3.4 KiB
TypeScript
import { type Plugin } from "esbuild";
|
|
import type { PageFiles } from "../../../types";
|
|
import grabArtifactsFromBundledResults from "../grab-artifacts-from-bundled-result";
|
|
import { writeFileSync } from "fs";
|
|
import path from "path";
|
|
import grabDirNames from "../../../utils/grab-dir-names";
|
|
|
|
let build_start = 0;
|
|
let build_starts = 0;
|
|
const MAX_BUILD_STARTS = 2;
|
|
|
|
const { BUNX_TMP_DIR } = grabDirNames();
|
|
|
|
type Params = {
|
|
entryToPage: Map<
|
|
string,
|
|
PageFiles & {
|
|
tsx: string;
|
|
}
|
|
>;
|
|
post_build_fn?: (params: { artifacts: any[] }) => Promise<void> | void;
|
|
};
|
|
|
|
export default function ssrCTXArtifactTracker({
|
|
entryToPage,
|
|
post_build_fn,
|
|
}: Params) {
|
|
const artifactTracker: Plugin = {
|
|
name: "ssr-artifact-tracker",
|
|
setup(build) {
|
|
build.onStart(async () => {
|
|
build_starts++;
|
|
build_start = performance.now();
|
|
if (build_starts == MAX_BUILD_STARTS) {
|
|
global.BUNEXT_SSR_BUNDLER_CTX_DISPOSED = true;
|
|
await global.BUNEXT_SSR_BUNDLER_CTX?.dispose();
|
|
global.BUNEXT_SSR_BUNDLER_CTX = undefined;
|
|
}
|
|
});
|
|
|
|
build.onEnd(async (result) => {
|
|
if (result.errors.length > 0) {
|
|
global.BUNEXT_SSR_BUNDLER_CTX_DISPOSED = true;
|
|
try {
|
|
await global.BUNEXT_SSR_BUNDLER_CTX?.dispose();
|
|
} catch {}
|
|
global.BUNEXT_SSR_BUNDLER_CTX = undefined;
|
|
build_starts = 0;
|
|
for (const err of result.errors) {
|
|
console.error(
|
|
`SSR Build error: ${err.text}${err.location ? ` (${err.location.file}:${err.location.line}:${err.location.column})` : ""}`,
|
|
);
|
|
}
|
|
return;
|
|
}
|
|
|
|
const artifacts = grabArtifactsFromBundledResults({
|
|
result,
|
|
entryToPage,
|
|
virtual_match: `ssr-virtual`,
|
|
});
|
|
|
|
if (artifacts?.[0] && artifacts.length > 0) {
|
|
for (let i = 0; i < artifacts.length; i++) {
|
|
const artifact = artifacts[i];
|
|
if (
|
|
artifact?.local_path &&
|
|
global.BUNEXT_SSR_BUNDLER_CTX_MAP
|
|
) {
|
|
global.BUNEXT_SSR_BUNDLER_CTX_MAP[
|
|
artifact.local_path
|
|
] = artifact;
|
|
}
|
|
}
|
|
|
|
// post_build_fn?.({ artifacts });
|
|
|
|
// const elapsed = (performance.now() - build_start).toFixed(
|
|
// 0,
|
|
// );
|
|
// log.success(`SSR [Built] in ${elapsed}ms`);
|
|
}
|
|
|
|
try {
|
|
writeFileSync(
|
|
path.join(BUNX_TMP_DIR, "ctx-map.json"),
|
|
JSON.stringify(
|
|
global.BUNEXT_SSR_BUNDLER_CTX_MAP,
|
|
null,
|
|
4,
|
|
),
|
|
);
|
|
} catch (error) {}
|
|
|
|
global.BUNEXT_SSR_BUNDLER_CTX_DISPOSED = false;
|
|
});
|
|
},
|
|
};
|
|
|
|
return artifactTracker;
|
|
}
|