1- import type { NetlifyPluginOptions } from '@netlify/build'
2- import glob from 'fast-glob'
3- import { Buffer } from 'node:buffer'
41import { existsSync } from 'node:fs'
5- import { mkdir , readFile , writeFile } from 'node:fs/promises'
6- import { dirname , resolve } from 'node:path'
7- import { getPrerenderManifest } from '../config.js'
8- import { BLOB_DIR } from '../constants.js'
9-
10- export type CacheEntry = {
11- lastModified : number
12- value : CacheValue
13- }
14-
15- type CacheValue = PageCacheValue | RouteCacheValue | FetchCacheValue
16-
17- export type PageCacheValue = {
18- kind : 'PAGE'
19- html : string
20- pageData : string
21- headers ?: { [ k : string ] : string }
22- status ?: number
23- }
24-
25- type RouteCacheValue = {
26- kind : 'ROUTE'
27- body : string
28- headers : { [ k : string ] : string }
29- status : number
30- }
2+ import { readFile } from 'node:fs/promises'
3+ import { join } from 'node:path'
314
32- type FetchCacheValue = {
33- kind : 'FETCH'
34- data : {
35- headers : { [ k : string ] : string }
36- body : string
37- url : string
38- status ?: number
39- tags ?: string [ ]
40- }
41- }
42-
43- /**
44- * Write a cache entry to the blob upload directory using
45- * base64 keys to avoid collisions with directories
46- */
47- const writeCacheEntry = async ( key : string , value : CacheValue ) => {
48- const path = resolve ( BLOB_DIR , Buffer . from ( key ) . toString ( 'base64' ) )
49- const entry = JSON . stringify ( {
50- lastModified : Date . now ( ) ,
51- value,
52- } satisfies CacheEntry )
5+ import glob from 'fast-glob'
536
54- await mkdir ( dirname ( path ) , { recursive : true } )
55- await writeFile ( path , entry , 'utf-8' )
56- }
7+ import type {
8+ CacheValue ,
9+ FetchCacheValue ,
10+ PageCacheValue ,
11+ PluginContext ,
12+ RouteCacheValue ,
13+ } from '../plugin-context.js'
5714
5815/**
5916 * Normalize routes by stripping leading slashes and ensuring root path is index
@@ -87,15 +44,10 @@ const buildFetchCacheValue = async (path: string): Promise<FetchCacheValue> => (
8744/**
8845 * Upload prerendered content to the blob store
8946 */
90- export const copyPrerenderedContent = async ( {
91- constants : { PUBLISH_DIR } ,
92- utils : {
93- build : { failBuild } ,
94- } ,
95- } : Pick < NetlifyPluginOptions , 'constants' | 'utils' > ) => {
47+ export const copyPrerenderedContent = async ( ctx : PluginContext ) : Promise < void > => {
9648 try {
9749 // read prerendered content and build JSON key/values for the blob store
98- const manifest = await getPrerenderManifest ( { PUBLISH_DIR } )
50+ const manifest = await ctx . getPrerenderManifest ( )
9951
10052 await Promise . all (
10153 Object . entries ( manifest . routes ) . map ( async ( [ route , meta ] ) => {
@@ -104,62 +56,51 @@ export const copyPrerenderedContent = async ({
10456
10557 switch ( true ) {
10658 case meta . dataRoute ?. endsWith ( '.json' ) :
107- value = await buildPagesCacheValue ( resolve ( PUBLISH_DIR , 'server/pages' , key ) )
59+ value = await buildPagesCacheValue ( join ( ctx . publishDir , 'server/pages' , key ) )
10860 break
109-
11061 case meta . dataRoute ?. endsWith ( '.rsc' ) :
111- value = await buildAppCacheValue ( resolve ( PUBLISH_DIR , 'server/app' , key ) )
62+ value = await buildAppCacheValue ( join ( ctx . publishDir , 'server/app' , key ) )
11263 break
113-
11464 case meta . dataRoute === null :
115- value = await buildRouteCacheValue ( resolve ( PUBLISH_DIR , 'server/app' , key ) )
65+ value = await buildRouteCacheValue ( join ( ctx . publishDir , 'server/app' , key ) )
11666 break
117-
11867 default :
11968 throw new Error ( `Unrecognized content: ${ route } )
12069 }
12170
122- await writeCacheEntry ( key , value )
71+ await ctx . writeCacheEntry ( key , value )
12372 } ) ,
12473 )
12574
12675 // app router 404 pages are not in the prerender manifest
12776 // so we need to check for them manually
128- if ( existsSync ( resolve ( PUBLISH_DIR , `server/app/_not-found.html` ) ) ) {
77+ if ( existsSync ( join ( ctx . publishDir , `server/app/_not-found.html` ) ) ) {
12978 const key = '404'
130- const value = await buildAppCacheValue ( resolve ( PUBLISH_DIR , 'server/app/_not-found' ) )
131- await writeCacheEntry ( key , value )
79+ const value = await buildAppCacheValue ( join ( ctx . publishDir , 'server/app/_not-found' ) )
80+ await ctx . writeCacheEntry ( key , value )
13281 }
13382 } catch ( error ) {
134- failBuild (
135- 'Failed assembling prerendered content for upload' ,
136- error instanceof Error ? { error } : { } ,
137- )
83+ ctx . failBuild ( 'Failed assembling prerendered content for upload' , error )
13884 }
13985}
14086
14187/**
14288 * Upload fetch content to the blob store
14389 */
144- export const copyFetchContent = async ( {
145- constants : { PUBLISH_DIR } ,
146- utils : {
147- build : { failBuild } ,
148- } ,
149- } : Pick < NetlifyPluginOptions , 'constants' | 'utils' > ) => {
90+ export const copyFetchContent = async ( ctx : PluginContext ) : Promise < void > => {
15091 try {
15192 const paths = await glob ( [ '!(*.*)' ] , {
152- cwd : resolve ( PUBLISH_DIR , 'cache/fetch-cache' ) ,
93+ cwd : join ( ctx . publishDir , 'cache/fetch-cache' ) ,
15394 extglob : true ,
15495 } )
15596
15697 await Promise . all (
157- paths . map ( async ( key ) => {
158- const value = await buildFetchCacheValue ( resolve ( PUBLISH_DIR , 'cache/fetch-cache' , key ) )
159- await writeCacheEntry ( key , value )
98+ paths . map ( async ( key ) : Promise < void > => {
99+ const value = await buildFetchCacheValue ( join ( ctx . publishDir , 'cache/fetch-cache' , key ) )
100+ await ctx . writeCacheEntry ( key , value )
160101 } ) ,
161102 )
162103 } catch ( error ) {
163- failBuild ( 'Failed assembling fetch content for upload' , error instanceof Error ? { error } : { } )
104+ ctx . failBuild ( 'Failed assembling fetch content for upload' , error )
164105 }
165106}
0 commit comments