39 lines
1.3 KiB
TypeScript
39 lines
1.3 KiB
TypeScript
|
import fs from 'node:fs';
|
||
|
|
import type { AstroConfig, AstroIntegration } from 'astro';
|
||
|
|
const STATUS_CODE_PAGE_REGEXP = /\/[0-9]{3}\/?$/;
|
||
|
|
|
||
|
|
/** Construct sitemap.xml given a set of URLs */
|
||
|
|
function generateSitemap(pages: string[]) {
|
||
|
|
// TODO: find way to respect <link rel="canonical"> URLs here
|
||
|
|
// TODO: find way to exclude pages from sitemap
|
||
|
|
const urls = [...pages].filter((url) => !STATUS_CODE_PAGE_REGEXP.test(url));
|
||
|
|
urls.sort((a, b) => a.localeCompare(b, 'en', { numeric: true })); // sort alphabetically so sitemap is same each time
|
||
|
|
let sitemap = `<?xml version="1.0" encoding="UTF-8"?><urlset xmlns="http://www.sitemaps.org/schemas/sitemap/0.9">`;
|
||
|
|
for (const url of urls) {
|
||
|
|
sitemap += `<url><loc>${url}</loc></url>`;
|
||
|
|
}
|
||
|
|
sitemap += `</urlset>\n`;
|
||
|
|
return sitemap;
|
||
|
|
}
|
||
|
|
|
||
|
|
export default function createPlugin(): AstroIntegration {
|
||
|
|
let config: AstroConfig;
|
||
|
|
return {
|
||
|
|
name: '@astrojs/sitemap',
|
||
|
|
hooks: {
|
||
|
|
'astro:config:done': async ({ config: _config }) => {
|
||
|
|
config = _config;
|
||
|
|
},
|
||
|
|
'astro:build:done': async ({ pages, dir }) => {
|
||
|
|
const finalSiteUrl = config.buildOptions.site;
|
||
|
|
if (!finalSiteUrl) {
|
||
|
|
return;
|
||
|
|
}
|
||
|
|
const pageUrls = pages.map((p) => new URL(p.pathname, finalSiteUrl).href);
|
||
|
|
const sitemapContent = generateSitemap(pageUrls);
|
||
|
|
fs.writeFileSync(new URL('sitemap.xml', dir), sitemapContent);
|
||
|
|
},
|
||
|
|
},
|
||
|
|
};
|
||
|
|
}
|