Spaces:

aukaru
/

344

Paused

344 / src /proxy /middleware /request /mutators /finalize-signed-request.ts

Upload 236 files

5c5b371 verified over 1 year ago

1.2 kB

import { ProxyReqMutator } from "../index";

/**
 * For AWS/GCP/Azure/Google requests, the body is signed earlier in the request
 * pipeline, before the proxy middleware. This function just assigns the path
 * and headers to the proxy request.
 */
export const finalizeSignedRequest: ProxyReqMutator = (manager) => {
  const req = manager.request;
  if (!req.signedRequest) {
    throw new Error("Expected req.signedRequest to be set");
  }

  // The path depends on the selected model and the assigned key's region.
  manager.setPath(req.signedRequest.path);

  // Amazon doesn't want extra headers, so we need to remove all of them and
  // reassign only the ones specified in the signed request.
  const headers = req.signedRequest.headers;
  Object.keys(headers).forEach((key) => {
    manager.removeHeader(key);
  });
  Object.entries(req.signedRequest.headers).forEach(([key, value]) => {
    manager.setHeader(key, value);
  });
  const serialized =
    typeof req.signedRequest.body === "string"
      ? req.signedRequest.body
      : JSON.stringify(req.signedRequest.body);
  manager.setHeader("Content-Length", String(Buffer.byteLength(serialized)));
  manager.setBody(serialized);
};

	import { ProxyReqMutator } from "../index";

	/**
	* For AWS/GCP/Azure/Google requests, the body is signed earlier in the request
	* pipeline, before the proxy middleware. This function just assigns the path
	* and headers to the proxy request.
	*/
	export const finalizeSignedRequest: ProxyReqMutator = (manager) => {
	const req = manager.request;
	if (!req.signedRequest) {
	throw new Error("Expected req.signedRequest to be set");
	}

	// The path depends on the selected model and the assigned key's region.
	manager.setPath(req.signedRequest.path);

	// Amazon doesn't want extra headers, so we need to remove all of them and
	// reassign only the ones specified in the signed request.
	const headers = req.signedRequest.headers;
	Object.keys(headers).forEach((key) => {
	manager.removeHeader(key);
	});
	Object.entries(req.signedRequest.headers).forEach(([key, value]) => {
	manager.setHeader(key, value);
	});
	const serialized =
	typeof req.signedRequest.body === "string"
	? req.signedRequest.body
	: JSON.stringify(req.signedRequest.body);
	manager.setHeader("Content-Length", String(Buffer.byteLength(serialized)));
	manager.setBody(serialized);
	};