Merge branch 'main' into link-checker-yaml

Author: vrushankportkey

src/handlers/handlerUtils.ts

@@ -43,8 +43,34 @@ export function constructRequest(
   method: string,
   forwardHeaders: string[],
   requestHeaders: Record<string, string>,
-  fn: endpointStrings
+  fn: endpointStrings,
+  c: Context
 ) {
+  let proxyHeaders: Record<string, string> = {};
+  // Handle proxy headers
+  if (fn === 'proxy') {
+    const poweredByHeadersPattern = `x-${POWERED_BY}-`;
+    const headersToAvoidForCloudflare = ['expect'];
+    const headersToIgnore = [
+      ...(env(c).CUSTOM_HEADERS_TO_IGNORE ?? []),
+      ...headersToAvoidForCloudflare,
+    ];
+    headersToIgnore.push('content-length');
+    Object.keys(requestHeaders).forEach((key: string) => {
+      if (
+        !headersToIgnore.includes(key) &&
+        !key.startsWith(poweredByHeadersPattern)
+      ) {
+        proxyHeaders[key] = requestHeaders[key];
+      }
+    });
+    // Remove brotli from accept-encoding because cloudflare has problems with it
+    if (proxyHeaders['accept-encoding']?.includes('br'))
+      proxyHeaders['accept-encoding'] = proxyHeaders[
+        'accept-encoding'
+      ]?.replace('br', '');
+  }
+
   let baseHeaders: any = {
     'content-type': 'application/json',
   };
@@ -69,7 +95,7 @@ export function constructRequest(
     ...baseHeaders,
     ...headers,
     ...forwardHeadersMap,
-    ...(fn === 'proxy' ? requestHeaders : {}),
+    ...(fn === 'proxy' && proxyHeaders),
   };
 
   let fetchOptions: RequestInit = {
@@ -269,7 +295,8 @@ export async function tryPost(
     method,
     forwardHeaders,
     requestHeaders,
-    fn
+    fn,
+    c
   );
 
   const headerContentType = headers[HEADER_KEYS.CONTENT_TYPE];
@@ -759,6 +786,8 @@ export function constructConfigFromRequestHeaders(
       requestHeaders[`x-${POWERED_BY}-amzn-sagemaker-inference-component`],
     amznSagemakerSessionId:
       requestHeaders[`x-${POWERED_BY}-amzn-sagemaker-session-id`],
+    amznSagemakerModelName:
+      requestHeaders[`x-${POWERED_BY}-amzn-sagemaker-model-name`],
   };
 
   const workersAiConfig = {

src/handlers/proxyHandler.ts

@@ -1,11 +1,10 @@
 import { Context } from 'hono';
-import { CONTENT_TYPES, POWERED_BY } from '../globals';
+import { CONTENT_TYPES } from '../globals';
 import {
   constructConfigFromRequestHeaders,
   tryTargetsRecursively,
 } from './handlerUtils';
 import { RouterError } from '../errors/RouterError';
-import { env } from 'hono/adapter';
 
 async function getRequestData(request: Request, contentType: string) {
   let finalRequest: any;
@@ -24,34 +23,6 @@ async function getRequestData(request: Request, contentType: string) {
   return finalRequest;
 }
 
-function headersToSend(
-  headersObj: Record<string, string>,
-  customHeadersToIgnore: Array<string>
-): Record<string, string> {
-  let final: Record<string, string> = {};
-  const poweredByHeadersPattern = `x-${POWERED_BY}-`;
-  const headersToAvoidForCloudflare = ['expect'];
-  const headersToAvoid = [
-    ...customHeadersToIgnore,
-    ...headersToAvoidForCloudflare,
-  ];
-  headersToAvoid.push('content-length');
-  Object.keys(headersObj).forEach((key: string) => {
-    if (
-      !headersToAvoid.includes(key) &&
-      !key.startsWith(poweredByHeadersPattern)
-    ) {
-      final[key] = headersObj[key];
-    }
-  });
-
-  // Remove brotli from accept-encoding because cloudflare has problems with it
-  if (final['accept-encoding']?.includes('br'))
-    final['accept-encoding'] = final['accept-encoding']?.replace('br', '');
-
-  return final;
-}
-
 export async function proxyHandler(c: Context): Promise<Response> {
   try {
     let requestHeaders = Object.fromEntries(c.req.raw.headers);
@@ -65,7 +36,7 @@ export async function proxyHandler(c: Context): Promise<Response> {
       c,
       camelCaseConfig,
       request,
-      headersToSend(requestHeaders, env(c).CUSTOM_HEADERS_TO_IGNORE ?? []),
+      requestHeaders,
       'proxy',
       c.req.method,
       'config'