astro/packages/astro/test/test-adapter.js
2022-11-28 15:45:28 +08:00

74 lines
2 KiB
JavaScript

import { viteID } from '../dist/core/util.js';
/**
*
* @returns {import('../src/@types/astro').AstroIntegration}
*/
export default function ({ provideAddress = true, extendAdapter } = { provideAddress: true }) {
return {
name: 'my-ssr-adapter',
hooks: {
'astro:config:setup': ({ updateConfig }) => {
updateConfig({
vite: {
plugins: [
{
resolveId(id) {
if (id === '@my-ssr') {
return id;
} else if (id === 'astro/app') {
const viteId = viteID(new URL('../dist/core/app/index.js', import.meta.url));
return viteId;
}
},
load(id) {
if (id === '@my-ssr') {
return `
import { App } from 'astro/app';
import fs from 'fs';
class MyApp extends App {
#manifest = null;
constructor(manifest, streaming) {
super(manifest, streaming);
this.#manifest = manifest;
}
async render(request, routeData) {
const url = new URL(request.url);
if(this.#manifest.assets.has(url.pathname)) {
const filePath = new URL('../client/' + this.removeBase(url.pathname), import.meta.url);
const data = await fs.promises.readFile(filePath);
return new Response(data);
}
${provideAddress ? `request[Symbol.for('astro.clientAddress')] = '0.0.0.0';` : ''}
return super.render(request, routeData);
}
}
export function createExports(manifest) {
return {
manifest,
createApp: (streaming) => new MyApp(manifest, streaming)
};
}
`;
}
},
},
],
},
});
},
'astro:config:done': ({ setAdapter }) => {
setAdapter({
name: 'my-ssr-adapter',
serverEntrypoint: '@my-ssr',
exports: ['manifest', 'createApp'],
...extendAdapter,
});
},
},
};
}