| 1 | import{normalizeHttpResponse}from"@middy/util";import Accept from"@hapi/accept";const defaults={serializers:[],defaultContentType:undefined};const httpResponseSerializerMiddleware=(opts={})=>{const{serializers,defaultContentType}={...defaults,...opts};const httpResponseSerializerMiddlewareAfter=async request=>{normalizeHttpResponse(request);if(request.response.headers["Content-Type"])return;let types;if(request.event.requiredContentType){types=[request.event.requiredContentType]}else{const acceptHeader=request.event.headers.Accept??request.event.headers.accept;types=[...(acceptHeader&&Accept.mediaTypes(acceptHeader))??[],request.event.preferredContentType,defaultContentType]}for(const type of types){let breakTypes;for(const s of serializers){if(!s.regex.test(type)){continue}request.response.headers["Content-Type"]=type;const result=s.serializer(request.response);if(typeof result==="object"&&"body"in result){request.response=result}else{request.response.body=result}breakTypes=true;break}if(breakTypes)break}};const httpResponseSerializerMiddlewareOnError=async request=>{if(request.response===undefined)return;return httpResponseSerializerMiddlewareAfter(request)};return{after:httpResponseSerializerMiddlewareAfter,onError:httpResponseSerializerMiddlewareOnError}};export default httpResponseSerializerMiddleware
|