feat: add refresh hooks api implementation

Author: orinokai

packages/runtime/src/templates/getHandler.ts

@@ -65,7 +65,7 @@ const makeHandler = (conf: NextConfig, app, pageRoot, staticManifest: Array<[str
   // We memoize this because it can be shared between requests, but don't instantiate it until
   // the first request because we need the host and port.
   let bridge: NodeBridge
-  const getBridge = (event: HandlerEvent): NodeBridge => {
+  const getBridge = (event: HandlerEvent, context: HandlerContext): NodeBridge => {
     if (bridge) {
       return bridge
     }
@@ -79,6 +79,7 @@ const makeHandler = (conf: NextConfig, app, pageRoot, staticManifest: Array<[str
       customServer: false,
       hostname: url.hostname,
       port,
+      netlifyRevalidateToken: context.clientContext?.custom?.odb_refresh_hooks,
     })
     const requestHandler = nextServer.getRequestHandler()
     const server = new Server(async (req, res) => {
@@ -115,7 +116,7 @@ const makeHandler = (conf: NextConfig, app, pageRoot, staticManifest: Array<[str
       process.env._NETLIFY_GRAPH_TOKEN = graphToken
     }
 
-    const { headers, ...result } = await getBridge(event).launcher(event, context)
+    const { headers, ...result } = await getBridge(event, context).launcher(event, context)
 
     // Convert all headers to multiValueHeaders
 

packages/runtime/src/templates/server.ts

@@ -1,19 +1,81 @@
-import { NodeRequestHandler } from 'next/dist/server/next-server'
+import https from 'https'
+
+import { NodeRequestHandler, Options } from 'next/dist/server/next-server'
 
 import { getNextServer, NextServerType } from './handlerUtils'
 
 const NextServer: NextServerType = getNextServer()
 
+interface NetlifyNextServerOptions extends Options {
+  netlifyRevalidateToken: string
+}
+
 class NetlifyNextServer extends NextServer {
+  private netlifyRevalidateToken?: string
+
+  public constructor(options: NetlifyNextServerOptions) {
+    super(options)
+    this.netlifyRevalidateToken = options.netlifyRevalidateToken
+  }
+
   public getRequestHandler(): NodeRequestHandler {
     const handler = super.getRequestHandler()
-    return (req, res, parsedUrl) => {
+    return async (req, res, parsedUrl) => {
       if (req.headers['x-prerender-revalidate']) {
-        console.log('Revalidate request')
+        if (this.netlifyRevalidateToken) {
+          try {
+            await this.netlifyRevalidate(req.url)
+            console.log('Revalidated', req.url)
+          } catch {
+            // TODO: status 500 error refreshing ODB cache
+            console.log('Error revalidating', req.url)
+          }
+        } else {
+          // TODO: status 400 refresh hooks not enabled for site in proxy
+          console.log('Missing revalidate token', req.url)
+        }
       }
       return handler(req, res, parsedUrl)
     }
   }
+
+  private netlifyRevalidate(url: string) {
+    const path = new URL(url).pathname
+    const domain = this.hostname
+    const siteId = process.env.SITE_ID
+
+    return new Promise((resolve, reject) => {
+      const body = JSON.stringify({ paths: [path], domain })
+
+      const req = https
+        .request(
+          {
+            hostname: 'api.netlify.com',
+            port: 443,
+            path: `/api/v1/sites/${siteId}/refresh_on_demand_builders`,
+            method: 'POST',
+            headers: {
+              'Content-Type': 'application/json',
+              'Content-Length': body.length,
+              Authorization: `Bearer ${this.netlifyRevalidateToken}`,
+            },
+          },
+          (res) => {
+            let data = ''
+            res.on('data', (chunk) => {
+              data += chunk
+            })
+            res.on('end', () => {
+              resolve(JSON.parse(data))
+            })
+          },
+        )
+        .on('error', reject)
+
+      req.write(body)
+      req.end()
+    })
+  }
 }
 
 export { NetlifyNextServer }