77 lines
1.8 KiB
TypeScript
77 lines
1.8 KiB
TypeScript
import type { CreatePreviewServer } from 'astro';
|
|
import http from 'http';
|
|
import { fileURLToPath } from 'url';
|
|
import { createServer } from './http-server.js';
|
|
import type { createExports } from './server';
|
|
|
|
const preview: CreatePreviewServer = async function ({
|
|
client,
|
|
serverEntrypoint,
|
|
host,
|
|
port,
|
|
base,
|
|
}) {
|
|
type ServerModule = ReturnType<typeof createExports>;
|
|
type MaybeServerModule = Partial<ServerModule>;
|
|
let ssrHandler: ServerModule['handler'];
|
|
try {
|
|
process.env.ASTRO_NODE_AUTOSTART = 'disabled';
|
|
const ssrModule: MaybeServerModule = await import(serverEntrypoint.toString());
|
|
if (typeof ssrModule.handler === 'function') {
|
|
ssrHandler = ssrModule.handler;
|
|
} else {
|
|
throw new Error(
|
|
`The server entrypoint doesn't have a handler. Are you sure this is the right file?`
|
|
);
|
|
}
|
|
} catch (err) {
|
|
if ((err as any).code === 'ERR_MODULE_NOT_FOUND') {
|
|
throw new Error(
|
|
`The server entrypoint ${fileURLToPath(
|
|
serverEntrypoint
|
|
)} does not exist. Have you ran a build yet?`
|
|
);
|
|
} else {
|
|
throw err;
|
|
}
|
|
}
|
|
|
|
const handler: http.RequestListener = (req, res) => {
|
|
ssrHandler(req, res, (ssrErr: any) => {
|
|
if (ssrErr) {
|
|
res.writeHead(500);
|
|
res.end(ssrErr.toString());
|
|
} else {
|
|
res.writeHead(404);
|
|
res.end();
|
|
}
|
|
});
|
|
};
|
|
|
|
const baseWithoutTrailingSlash: string = base.endsWith('/')
|
|
? base.slice(0, base.length - 1)
|
|
: base;
|
|
function removeBase(pathname: string): string {
|
|
if (pathname.startsWith(base)) {
|
|
return pathname.slice(baseWithoutTrailingSlash.length);
|
|
}
|
|
return pathname;
|
|
}
|
|
|
|
const server = createServer(
|
|
{
|
|
client,
|
|
port,
|
|
host,
|
|
removeBase,
|
|
},
|
|
handler
|
|
);
|
|
|
|
// eslint-disable-next-line no-console
|
|
console.log(`Preview server listening on http://${host}:${port}`);
|
|
|
|
return server;
|
|
};
|
|
|
|
export { preview as default };
|