write-sitemap-chunk.ts — astro Source File

Source Code

import { createWriteStream, type WriteStream } from 'node:fs';
import { mkdir } from 'node:fs/promises';
import { normalize, resolve } from 'node:path';
import { pipeline, Readable } from 'node:stream';
import { promisify } from 'node:util';
import type { AstroConfig } from 'astro';
import { SitemapAndIndexStream, SitemapIndexStream, SitemapStream } from 'sitemap';
import replace from 'stream-replace-string';
import type { SitemapItem } from './index.js';

type WriteSitemapChunkConfig = {
	filenameBase: string;
	hostname: string;
	sitemapHostname?: string;
	sourceData: Record<string, SitemapItem[]>;
	destinationDir: string;
	customSitemaps?: string[];
	publicBasePath?: string;
	limit?: number;
	xslURL?: string;
	lastmod?: string;
	namespaces?: {
		news?: boolean;
		xhtml?: boolean;
		image?: boolean;
		video?: boolean;
	};
};

// adapted from sitemap.js/sitemap-simple
export async function writeSitemapChunk(
	{
		filenameBase,
		hostname,
		sitemapHostname = hostname,
		sourceData,
		destinationDir,
		limit = 50000,
		customSitemaps = [],
		publicBasePath = './',
		xslURL: xslUrl,
		lastmod,
		namespaces = { news: true, xhtml: true, image: true, video: true },
	}: WriteSitemapChunkConfig,
	astroConfig: AstroConfig,
) {
	await mkdir(destinationDir, { recursive: true });

	// Normalize publicBasePath
	let normalizedPublicBasePath = publicBasePath;
	if (!normalizedPublicBasePath.endsWith('/')) {
		normalizedPublicBasePath += '/';
	}

	// Array to collect all sitemap URLs for the index
	const sitemapUrls: Array<{ url: string; lastmod?: string }> = [];

	// Process each chunk separately
	for (const [chunkName, items] of Object.entries(sourceData)) {
		const sitemapAndIndexStream = new SitemapAndIndexStream({
// ... (68 more lines)