fix: pass Koa response to setHeaders

README.md

@@ -67,9 +67,9 @@ app.use(async (ctx) => {
 
 ### setHeaders
 
-The function is called as `fn(res, path, stats)`, where the arguments are:
+The function is called as `fn(response, path, stats)`, where the arguments are:
 
-- `res`: the response object.
+- `response`: the Koa response object.
 - `path`: the resolved file path that is being sent.
 - `stats`: the stats object of the file that is being sent.
 

src/send.ts

@@ -113,7 +113,7 @@ export async function send(
   }
 
   // inject headers
-  setHeaders?.(ctx.res, filePath, stats);
+  setHeaders?.(ctx.response, filePath, stats);
 
   // stream
   ctx.set('Content-Length', stats.size.toString());

src/send.types.ts

@@ -2,7 +2,7 @@ import type {Stats} from 'node:fs';
 import type {ParameterizedContext} from 'koa';
 
 type SetHeaders = (
-  res: ParameterizedContext['res'],
+  response: ParameterizedContext['response'],
   path: string,
   stats: Stats,
 ) => void;

test/index.test.ts

@@ -765,18 +765,21 @@ describe('send(ctx, file)', () => {
 
       app.use(async (ctx) => {
         await send(ctx, testFilePath, {
-          setHeaders: function (res, path, stats) {
+          setHeaders: function (response, path, stats) {
             expect(path.slice(-normalizedTestFilePath.length)).toBe(
               normalizedTestFilePath
             );
             expect(stats.size).toBe(18);
-            expect(res).toBeTruthy();
+            expect(response).toBe(ctx.response);
 
             // these can be set
-            res.setHeader('Cache-Control', 'max-age=0,must-revalidate');
-            res.setHeader('Last-Modified', 'foo');
+            response.set('Cache-Control', 'max-age=0,must-revalidate');
+            response.set('Last-Modified', 'foo');
+            expect(response.get('Cache-Control')).toBe(
+              'max-age=0,must-revalidate'
+            );
             // this one can not
-            res.setHeader('Content-Length', 9000);
+            response.set('Content-Length', '9000');
           },
         });
       });