fix: exclude headers not included in script

src/codegen/codegen.ts

@@ -9,7 +9,7 @@ import { generateOptions } from './options'
 import { getContentTypeWithCharsetHeader } from '@/utils/headers'
 import { REQUIRED_IMPORTS } from '@/constants/imports'
 import { generateImportStatement } from './imports'
-import { cleanupRecording } from './codegen.utils'
+import { cleanupRecording, shouldIncludeHeaderInScript } from './codegen.utils'
 import { groupProxyData } from '@/utils/groups'
 
 interface GenerateScriptParams {
@@ -186,9 +186,8 @@ function generateSleep(timing: ThinkTime['timing']): string {
 }
 
 export function generateRequestParams(request: ProxyData['request']): string {
-  const headersToExclude = ['Cookie', 'User-Agent', 'Host', 'Content-Length']
   const headers = request.headers
-    .filter(([name]) => !headersToExclude.includes(name))
+    .filter(([name]) => shouldIncludeHeaderInScript(name))
     .map(([name, value]) => `'${name}': \`${value}\``)
     .join(',')
 

src/codegen/codegen.utils.ts

@@ -1,6 +1,9 @@
 import { flow } from 'lodash-es'
 import { ProxyData } from '@/types'
 import { getLocationHeader, getUpgradeHeader } from '@/utils/headers'
+import { canonicalHeaderKey } from '@/rules/utils'
+
+const HEADERS_TO_EXCLUDE = ['Cookie', 'User-Agent', 'Host', 'Content-Length']
 
 // TODO: find a well-maintained library for this
 export function stringify(value: unknown): string {
@@ -120,3 +123,7 @@ export const removeWebsocketRequests = (recording: ProxyData[]) => {
 export function cleanupRecording(recording: ProxyData[]) {
   return flow(removeWebsocketRequests, mergeRedirects)(recording)
 }
+
+export function shouldIncludeHeaderInScript(key: string) {
+  return !HEADERS_TO_EXCLUDE.includes(canonicalHeaderKey(key))
+}

src/views/Generator/RuleEditor/HeaderSelect.hooks.ts

@@ -2,6 +2,7 @@ import { Filter } from '@/types/rules'
 import { matchFilter } from '@/rules/utils'
 import { ProxyData } from '@/types'
 import { useMemo } from 'react'
+import { shouldIncludeHeaderInScript } from '@/codegen/codegen.utils'
 
 export function useHeaderOptions(
   recording: ProxyData[],
@@ -21,9 +22,16 @@ export function useHeaderOptions(
       new Set(headers.map((header) => header[0]))
     )
 
-    return uniqueHeaderNames.sort().map((headerName) => ({
+    const options = uniqueHeaderNames.sort().map((headerName) => ({
       value: headerName,
       label: headerName,
     }))
+
+    // Don't show headers that are excluded from the codegen
+    if (extractFrom === 'request') {
+      return options.filter(({ value }) => shouldIncludeHeaderInScript(value))
+    }
+
+    return options
   }, [recording, extractFrom, filter])
 }