Function sendStandardResponse

packages/standard-server-aws-lambda/src/response.ts:9–47

(
  responseStream: Stream.Writable,
  standardResponse: StandardResponse,
  options: SendStandardResponseOptions = {},
)

export interface SendStandardResponseOptions extends ToLambdaBodyOptions {}

export function sendStandardResponse(
  responseStream: Stream.Writable,
  standardResponse: StandardResponse,
  options: SendStandardResponseOptions = {},
): Promise<void> {
  return new Promise((resolve, reject) => {
    responseStream.once('error', reject)
    responseStream.once('close', resolve)

    const [body, standardHeaders] = toLambdaBody(standardResponse.body, standardResponse.headers, options)
    const [headers, setCookies] = toLambdaHeaders(standardHeaders)

    // awslambda is global aws lambda global object
    ;(globalThis as any).awslambda.HttpResponseStream.from(responseStream, {
      statusCode: standardResponse.status,
      headers,
      cookies: setCookies,
    })

    if (body === undefined) {
      responseStream.end()
    }
    else if (typeof body === 'string') {
      responseStream.write(body)
      responseStream.end() // awslambda will throw if we call .end with args
    }
    else {
      responseStream.once('close', () => {
        if (!body.closed) {
          body.destroy(responseStream.errored ?? undefined)
        }
      })

      body.once('error', error => responseStream.destroy(error))

      body.pipe(responseStream)
    }
  })
}