52 lines
1.8 KiB
JavaScript
52 lines
1.8 KiB
JavaScript
import { HttpResponse } from "@smithy/core/protocols";
|
|
import { toStream } from "../to-stream/toStream";
|
|
const THROW_IF_EMPTY_BODY = {
|
|
CopyObjectCommand: true,
|
|
UploadPartCopyCommand: true,
|
|
CompleteMultipartUploadCommand: true,
|
|
};
|
|
export const throw200ExceptionsMiddleware = (config) => (next, context) => async (args) => {
|
|
const result = await next(args);
|
|
const { response } = result;
|
|
if (!HttpResponse.isInstance(response)) {
|
|
return result;
|
|
}
|
|
const { statusCode, body } = response;
|
|
if (statusCode < 200 || statusCode >= 300) {
|
|
return result;
|
|
}
|
|
const bodyBytes = await collectBody(body, config);
|
|
response.body = toStream(bodyBytes);
|
|
if (bodyBytes.length === 0 && THROW_IF_EMPTY_BODY[context.commandName]) {
|
|
const err = new Error("S3 aborted request");
|
|
err.$metadata = {
|
|
httpStatusCode: 503,
|
|
};
|
|
err.name = "InternalError";
|
|
throw err;
|
|
}
|
|
const bodyStringTail = config.utf8Encoder(bodyBytes.subarray(bodyBytes.length - 16));
|
|
if (bodyStringTail && bodyStringTail.endsWith("</Error>")) {
|
|
response.statusCode = 503;
|
|
}
|
|
return result;
|
|
};
|
|
const collectBody = (streamBody = new Uint8Array(), context) => {
|
|
if (streamBody instanceof Uint8Array) {
|
|
return Promise.resolve(streamBody);
|
|
}
|
|
return context.streamCollector(streamBody) || Promise.resolve(new Uint8Array());
|
|
};
|
|
export const throw200ExceptionsMiddlewareOptions = {
|
|
relation: "after",
|
|
toMiddleware: "deserializerMiddleware",
|
|
tags: ["THROW_200_EXCEPTIONS", "S3"],
|
|
name: "throw200ExceptionsMiddleware",
|
|
override: true,
|
|
};
|
|
export const getThrow200ExceptionsPlugin = (config) => ({
|
|
applyToStack: (clientStack) => {
|
|
clientStack.addRelativeTo(throw200ExceptionsMiddleware(config), throw200ExceptionsMiddlewareOptions);
|
|
},
|
|
});
|