bunext/src/functions/server/server-post-build-fn.ts

63 lines
1.7 KiB
TypeScript

import _ from "lodash";
import type { BundlerCTXMap, GlobalHMRControllerObject } from "../../types";
import grabPageComponent from "./web-pages/grab-page-component";
type Params = {
artifacts: BundlerCTXMap[];
};
export default async function serverPostBuildFn({ artifacts }: Params) {
if (!global.IS_FIRST_BUNDLE_READY) {
global.IS_FIRST_BUNDLE_READY = true;
}
if (!global.HMR_CONTROLLERS?.[0]) {
return;
}
for (let i = 0; i < global.HMR_CONTROLLERS.length; i++) {
const controller = global.HMR_CONTROLLERS[i];
const target_artifact = artifacts.find(
(a) => controller.target_map?.local_path == a.local_path,
);
const mock_req = new Request(controller.page_url);
const { serverRes } = await grabPageComponent({
req: mock_req,
});
const final_artifact: Omit<GlobalHMRControllerObject, "controller"> = {
..._.omit(controller, ["controller"]),
target_map: target_artifact,
};
if (!target_artifact) {
delete final_artifact.target_map;
}
if (serverRes) {
final_artifact.page_props = serverRes;
}
try {
let final_data: { [k: string]: any } = {};
if (global.ROOT_FILE_UPDATED) {
final_data = { reload: true };
} else {
final_data = final_artifact;
}
controller.controller.enqueue(
`event: update\ndata: ${JSON.stringify(final_data)}\n\n`,
);
global.ROOT_FILE_UPDATED = false;
} catch {
global.HMR_CONTROLLERS.splice(i, 1);
}
}
}