feat: add version header

storacha · Oct 19, 2022 · fcbf1ac · fcbf1ac
1 parent 979b13b
commit fcbf1ac
Show file tree

Hide file tree

Showing 2 changed files with 14 additions and 3 deletions.
diff --git a/src/index.js b/src/index.js
@@ -19,7 +19,8 @@ import {
   withCarCids,
   withUnsupportedFeaturesHandler,
   withMemoryBudget,
-  withResponseMemoryRelease
+  withResponseMemoryRelease,
+  withVersionHeader
 } from './middleware.js'
 
 /**
@@ -36,6 +37,7 @@ export default {
     const middleware = composeMiddleware(
       withCdnCache,
       withCorsHeaders,
+      withVersionHeader,
       withContentDispositionHeader,
       withErrorHandler,
       withUnsupportedFeaturesHandler,

diff --git a/src/middleware.js b/src/middleware.js
@@ -98,8 +98,6 @@ export function withResponseMemoryRelease (handler) {
     const body = response.body
     if (!body) return response
 
-    console.log('adding response memory release transform...')
-
     return new Response(
       body.pipeThrough(new TransformStream({
         transform (chunk, controller) {
@@ -144,3 +142,14 @@ export function withDagula (handler) {
     return handler(request, env, { ...ctx, dagula })
   }
 }
+
+/**
+ * @type {import('@web3-storage/gateway-lib').Middleware<import('@web3-storage/gateway-lib').Context>}
+ */
+export function withVersionHeader (handler) {
+  return async (request, env, ctx) => {
+    const response = await handler(request, env, ctx)
+    response.headers.set('x-freeway-version', '1.3.0')
+    return response
+  }
+}