From e73620b18fd1419c4a9ce0c85ffe0adf4c3619a7 Mon Sep 17 00:00:00 2001
From: Dave Kerr <dave.kerr@orum.io>
Date: Fri, 16 May 2025 14:27:08 -0400
Subject: [PATCH] add user-agent to oauth request as well

---
 src/utils/security.ts | 14 +++++++++-----
 1 file changed, 9 insertions(+), 5 deletions(-)

diff --git a/src/utils/security.ts b/src/utils/security.ts
index 66caace..c9d0617 100644
--- a/src/utils/security.ts
+++ b/src/utils/security.ts
@@ -163,15 +163,19 @@ async function acquireOAuth2Token(schemeName: string, scheme: any): Promise<stri
         }
         
         console.error(\`Requesting OAuth2 token from \${tokenUrl}\`);
+
+        const headers: Record<string, string> = {
+            'Content-Type': 'application/x-www-form-urlencoded',
+            'Authorization': \`Basic \${Buffer.from(\`\${clientId}:\${clientSecret}\`).toString('base64')}\`
+        }
+
+        ${cliOptions?.userAgent ? `    headers['User-Agent'] = '${cliOptions?.userAgent}';` : ''}
         
         // Make the token request
         const response = await axios({
             method: 'POST',
             url: tokenUrl,
-            headers: {
-                'Content-Type': 'application/x-www-form-urlencoded',
-                'Authorization': \`Basic \${Buffer.from(\`\${clientId}:\${clientSecret}\`).toString('base64')}\`
-            },
+            headers: headers,
             data: formData.toString()
         });
         
@@ -437,7 +441,7 @@ async function executeApiTool(
     const headers: Record<string, string> = { 'Accept': 'application/json' };
     let requestBodyData: any = undefined;
 
-${cliOptions?.userAgent ? `headers['User-Agent'] = options.userAgent;` : ''}
+${cliOptions?.userAgent ? `    headers['User-Agent'] = '${cliOptions?.userAgent}';` : ''}
 
     // Apply parameters to the URL path, query, or headers
     definition.executionParameters.forEach((param) => {