Feat/content type

dist/ResponseStream.d.ts

@@ -3,9 +3,11 @@ import { Stream } from 'stream';
 export declare class ResponseStream extends Stream.Writable {
     private response;
     _contentType?: string;
+    _isBase64Encoded?: boolean;
     constructor();
     _write(chunk: any, encoding: BufferEncoding, callback: (error?: Error | null) => void): void;
     getBufferedData(): Buffer;
     setContentType(contentType: string): void;
+    setIsBase64Encoded(isBase64Encoded: boolean): void;
 }
 //# sourceMappingURL=ResponseStream.d.ts.map

examples/src/binary-base64-handler.ts

@@ -0,0 +1,16 @@
+import { APIGatewayProxyEvent } from 'aws-lambda'
+import { ResponseStream, streamifyResponse } from 'lambda-stream'
+import { Readable } from 'stream'
+import { pipeline } from 'stream/promises'
+
+export const binaryBase64Handler = streamifyResponse(myHandler)
+
+async function myHandler(
+  event: APIGatewayProxyEvent,
+  responseStream: ResponseStream
+): Promise<void> {
+  const source = Readable.from(Buffer.from('hello world'))
+  responseStream.setContentType('binary/octet-stream')
+  responseStream.setIsBase64Encoded(true)
+  await pipeline(source, responseStream)
+}

src/ResponseStream.ts

@@ -3,6 +3,7 @@ import { Stream } from 'stream'
 export class ResponseStream extends Stream.Writable {
   private response: Buffer[]
   _contentType?: string
+  _isBase64Encoded?: boolean
 
   constructor() {
     super()
@@ -26,4 +27,8 @@ export class ResponseStream extends Stream.Writable {
   setContentType(contentType: string) {
     this._contentType = contentType
   }
+
+  setIsBase64Encoded(isBase64Encoded: boolean) {
+    this._isBase64Encoded = isBase64Encoded
+  }
 }

src/index.ts

@@ -23,8 +23,18 @@ export function streamifyResponse(handler: Function): Function {
       apply: async function (target, _, argList) {
         const responseStream: ResponseStream = patchArgs(argList)
         await target(...argList)
-        // Todo - honor content type
-        return responseStream.getBufferedData().toString()
+        return {
+          statusCode: 200,
+          headers: {
+            'content-type': responseStream._contentType || 'application/json',
+          },
+          ...(responseStream._isBase64Encoded
+            ? { isBase64Encoded: responseStream._isBase64Encoded }
+            : {}),
+          body: responseStream._isBase64Encoded
+            ? responseStream.getBufferedData().toString('base64')
+            : responseStream.getBufferedData().toString(),
+        }
       },
     })
   }

test/binary-pipeline.spec.ts

@@ -0,0 +1,9 @@
+import { binaryBase64Handler } from '../examples/src/binary-base64-handler'
+
+describe('simple handler', () => {
+  it('awaits and returns the response', async () => {
+    const resp = await binaryBase64Handler('hello')
+    const decodedMessage = Buffer.from(resp.body, 'base64').toString()
+    expect(decodedMessage).toEqual('hello world')
+  })
+})

test/pipeline.spec.ts

@@ -3,7 +3,7 @@ const { gzip } = require('../examples/example.js')
 describe('simple handler', () => {
   it('awaits and returns the response', async () => {
     const resp = await gzip('hello')
-    const parsed = JSON.parse(resp)
+    const parsed = JSON.parse(resp.body)
     expect(parsed).toMatchObject({ hello: 'world' })
   })
 })

test/simpleHandler.spec.ts

@@ -3,6 +3,6 @@ import { handler } from '../examples/src/handler'
 describe('simple handler', () => {
   it('awaits and returns the response', async () => {
     const resp = await handler('hello')
-    expect(resp).toEqual('Hello, world!')
+    expect(resp.body).toEqual('Hello, world!')
   })
 })