42 lines
1.4 KiB
JavaScript
42 lines
1.4 KiB
JavaScript
'use strict';
|
|
|
|
var protocolHttp = require('@smithy/protocol-http');
|
|
|
|
function resolveHostHeaderConfig(input) {
|
|
return input;
|
|
}
|
|
const hostHeaderMiddleware = (options) => (next) => async (args) => {
|
|
if (!protocolHttp.HttpRequest.isInstance(args.request))
|
|
return next(args);
|
|
const { request } = args;
|
|
const { handlerProtocol = "" } = options.requestHandler.metadata || {};
|
|
if (handlerProtocol.indexOf("h2") >= 0 && !request.headers[":authority"]) {
|
|
delete request.headers["host"];
|
|
request.headers[":authority"] = request.hostname + (request.port ? ":" + request.port : "");
|
|
}
|
|
else if (!request.headers["host"]) {
|
|
let host = request.hostname;
|
|
if (request.port != null)
|
|
host += `:${request.port}`;
|
|
request.headers["host"] = host;
|
|
}
|
|
return next(args);
|
|
};
|
|
const hostHeaderMiddlewareOptions = {
|
|
name: "hostHeaderMiddleware",
|
|
step: "build",
|
|
priority: "low",
|
|
tags: ["HOST"],
|
|
override: true,
|
|
};
|
|
const getHostHeaderPlugin = (options) => ({
|
|
applyToStack: (clientStack) => {
|
|
clientStack.add(hostHeaderMiddleware(options), hostHeaderMiddlewareOptions);
|
|
},
|
|
});
|
|
|
|
exports.getHostHeaderPlugin = getHostHeaderPlugin;
|
|
exports.hostHeaderMiddleware = hostHeaderMiddleware;
|
|
exports.hostHeaderMiddlewareOptions = hostHeaderMiddlewareOptions;
|
|
exports.resolveHostHeaderConfig = resolveHostHeaderConfig;
|