From 7b02c5cda377c7cd9ebcf827ab368bf96e0534cf Mon Sep 17 00:00:00 2001
From: Henry Mao <1828968+calclavia@users.noreply.github.com>
Date: Mon, 7 Jul 2025 03:56:42 +0800
Subject: [PATCH] Retain URL search parameter

---
 src/client/auth.ts | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/src/client/auth.ts b/src/client/auth.ts
index 8839ecf0..495f62a4 100644
--- a/src/client/auth.ts
+++ b/src/client/auth.ts
@@ -263,6 +263,7 @@ export async function discoverOAuthProtectedResourceMetadata(
   } else {
     const issuer = new URL(serverUrl);
     const wellKnownPath = buildWellKnownPath('oauth-protected-resource', issuer.pathname);
+    wellKnownPath.search = issuer.search;
     url = new URL(wellKnownPath, issuer);
   }
 
@@ -365,6 +366,7 @@ export async function discoverOAuthMetadata(
   // Try path-aware discovery first (RFC 8414 compliant)
   const wellKnownPath = buildWellKnownPath('oauth-authorization-server', issuer.pathname);
   const pathAwareUrl = new URL(wellKnownPath, issuer);
+  pathAwareUrl.search = issuer.search;
   let response = await tryMetadataDiscovery(pathAwareUrl, protocolVersion);
 
   // If path-aware discovery fails with 404, try fallback to root discovery