UNPKG

next

Version:

The React Framework

128 lines (127 loc) • 5.92 kB
import { InvariantError } from '../../shared/lib/invariant-error'; import { serializeUseCacheCacheStore, parseUseCacheCacheStore } from './cache-store'; /** * Serializes a resume data cache into a JSON string for storage or * transmission. Handles 'use cache' values, fetch responses, and encrypted * bound args for inline server functions. * * @param resumeDataCache - The immutable cache to serialize * @returns A Promise that resolves to the serialized cache as a JSON string, or * 'null' if empty */ export async function stringifyResumeDataCache(resumeDataCache, isCacheComponentsEnabled) { if (process.env.NEXT_RUNTIME === 'edge') { throw new InvariantError('`stringifyResumeDataCache` should not be called in edge runtime.'); } if (resumeDataCache.fetch.size === 0 && resumeDataCache.cache.size === 0) { return 'null'; } const json = { store: { fetch: Object.fromEntries(resumeDataCache.fetch), cache: Object.fromEntries((await serializeUseCacheCacheStore(resumeDataCache.cache.entries(), isCacheComponentsEnabled)).filter((entry)=>entry !== null)), encryptedBoundArgs: Object.fromEntries(resumeDataCache.encryptedBoundArgs) } }; return JSON.stringify(json); } export function deflateResumeDataCache(serializedResumeDataCache) { if (process.env.NEXT_RUNTIME === 'edge') { throw new InvariantError('`deflateResumeDataCache` should not be called in edge runtime.'); } else { if (serializedResumeDataCache === 'null') { return serializedResumeDataCache; } // Compress the JSON string using zlib. As the data is already in memory, // we use the synchronous deflateSync function. const { deflateSync } = require('node:zlib'); return deflateSync(serializedResumeDataCache).toString('base64'); } } /** * Creates a new empty mutable resume data cache for pre-rendering. * Initializes fresh Map instances for both the 'use cache' and fetch caches. * Used at the start of pre-rendering to begin collecting cached values. * * @returns A new empty PrerenderResumeDataCache instance */ export function createPrerenderResumeDataCache(source) { if (source) { return { mutable: true, cache: new Map(source.cache), fetch: new Map(source.fetch), encryptedBoundArgs: new Map(source.encryptedBoundArgs), decryptedBoundArgs: new Map(source.decryptedBoundArgs), imageResponses: new Map(source.imageResponses) }; } else { return { mutable: true, cache: new Map(), fetch: new Map(), encryptedBoundArgs: new Map(), decryptedBoundArgs: new Map(), imageResponses: new Map() }; } } export function createRenderResumeDataCache(resumeDataCacheOrPersistedCache, maxPostponedStateSizeBytes, disableResumeDataCacheCompression = false) { if (process.env.NEXT_RUNTIME === 'edge') { throw new InvariantError('`createRenderResumeDataCache` should not be called in edge runtime.'); } else { if (typeof resumeDataCacheOrPersistedCache !== 'string') { // If the cache is already read-only, return it directly. Otherwise we // perform a type change by overriding the discriminator — the underlying // Map references are still shared, but callers should treat the result // as immutable. if (!resumeDataCacheOrPersistedCache.mutable) { return resumeDataCacheOrPersistedCache; } return { ...resumeDataCacheOrPersistedCache, mutable: false }; } if (resumeDataCacheOrPersistedCache === 'null') { return { mutable: false, cache: new Map(), fetch: new Map(), encryptedBoundArgs: new Map(), decryptedBoundArgs: new Map(), imageResponses: new Map() }; } let serializedResumeDataCache = resumeDataCacheOrPersistedCache; if (!disableResumeDataCacheCompression) { // This should be a compressed string. Let's decompress it using zlib. // As the data we already want to decompress is in memory, we use the // synchronous inflateSync function. const { inflateSync } = require('node:zlib'); // Limit decompressed size to prevent zipbomb attacks. This is 5x the // configured maxPostponedStateSize, allowing reasonable compression // ratios while preventing extreme decompression bombs. // Default is 500MB (5x the default 100MB compressed limit). const maxDecompressedSize = maxPostponedStateSizeBytes ? maxPostponedStateSizeBytes * 5 : 500 * 1024 * 1024; try { serializedResumeDataCache = inflateSync(Buffer.from(serializedResumeDataCache, 'base64'), { maxOutputLength: maxDecompressedSize }).toString('utf-8'); } catch (err) { if (err instanceof RangeError && err.code === 'ERR_BUFFER_TOO_LARGE') { throw new Error(`Decompressed resume data cache exceeded ${maxDecompressedSize} byte limit`); } throw err; } } const json = JSON.parse(serializedResumeDataCache); return { mutable: false, cache: parseUseCacheCacheStore(Object.entries(json.store.cache)), fetch: new Map(Object.entries(json.store.fetch)), encryptedBoundArgs: new Map(Object.entries(json.store.encryptedBoundArgs)), decryptedBoundArgs: new Map(), imageResponses: new Map() }; } } //# sourceMappingURL=resume-data-cache.js.map