@softchef/cdk-iot-device-management
Version:
IoT device management is composed of things, thing types, thing groups, jobs, files API services. The constructs can be used independently, that are based on full-managed service to create an API Gateway & Lambda function.
40 lines (39 loc) • 1.73 kB
JavaScript
import { __assign, __awaiter, __generator } from "tslib";
import { HttpRequest } from "@aws-sdk/protocol-http";
var CONTENT_LENGTH_HEADER = "content-length";
export function contentLengthMiddleware(bodyLengthChecker) {
var _this = this;
return function (next) {
return function (args) { return __awaiter(_this, void 0, void 0, function () {
var request, body, headers, length;
var _a;
return __generator(this, function (_b) {
request = args.request;
if (HttpRequest.isInstance(request)) {
body = request.body, headers = request.headers;
if (body &&
Object.keys(headers)
.map(function (str) { return str.toLowerCase(); })
.indexOf(CONTENT_LENGTH_HEADER) === -1) {
length = bodyLengthChecker(body);
if (length !== undefined) {
request.headers = __assign(__assign({}, request.headers), (_a = {}, _a[CONTENT_LENGTH_HEADER] = String(length), _a));
}
}
}
return [2, next(__assign(__assign({}, args), { request: request }))];
});
}); };
};
}
export var contentLengthMiddlewareOptions = {
step: "build",
tags: ["SET_CONTENT_LENGTH", "CONTENT_LENGTH"],
name: "contentLengthMiddleware",
override: true,
};
export var getContentLengthPlugin = function (options) { return ({
applyToStack: function (clientStack) {
clientStack.add(contentLengthMiddleware(options.bodyLengthChecker), contentLengthMiddlewareOptions);
},
}); };