refactor: add support user defined UA

Signed-off-by: Vladislav Polyakov <[email protected]>
connectrpc · Oct 19, 2024 · cc5907b · cc5907b
1 parent b4d1f11
commit cc5907b
Show file tree

Hide file tree

Showing 6 changed files with 87 additions and 4 deletions.
diff --git a/packages/connect/src/protocol-connect/request-header.spec.ts b/packages/connect/src/protocol-connect/request-header.spec.ts
@@ -54,6 +54,23 @@ describe("requestHeader", () => {
     expect(headers.get("Connect-Timeout-Ms")).toBe("10");
   });
 
+  it("should create request headers with user provided user-agent", () => {
+    const headers = requestHeader(
+      "unary",
+      true,
+      10,
+      { "User-Agent": "grpc-es/0.0.0" },
+      true,
+    );
+    expect(listHeaderKeys(headers)).toEqual([
+      "connect-protocol-version",
+      "connect-timeout-ms",
+      "content-type",
+      "user-agent",
+    ]);
+    expect(headers.get("User-Agent")).toBe("grpc-es/0.0.0");
+  });
+
   it("should exclude user-agent", () => {
     const headers = requestHeader("unary", true, undefined, undefined, false);
     expect(listHeaderKeys(headers)).toEqual([

diff --git a/packages/connect/src/protocol-connect/request-header.ts b/packages/connect/src/protocol-connect/request-header.ts
@@ -59,9 +59,14 @@ export function requestHeader(
         : contentTypeStreamJson,
   );
   result.set(headerProtocolVersion, protocolVersion);
-  if (setUserAgent) {
+
+  if (!result.has(headerUserAgent) && setUserAgent) {
+    // Note that we do not strictly comply with gRPC user agents.
+    // We use "connect-es/1.2.3" where gRPC would use "grpc-es/1.2.3".
+    // See https://github.com/grpc/grpc/blob/c462bb8d485fc1434ecfae438823ca8d14cf3154/doc/PROTOCOL-HTTP2.md#user-agents
     result.set(headerUserAgent, "CONNECT_ES_USER_AGENT");
   }
+
   return result;
 }
 

diff --git a/packages/connect/src/protocol-grpc-web/request-header.spec.ts b/packages/connect/src/protocol-grpc-web/request-header.spec.ts
@@ -66,6 +66,42 @@ describe("requestHeader", () => {
     expect(headers.get("Grpc-Timeout")).toBe("10m");
   });
 
+  it("should set user provided user-agent header", () => {
+    const headers = requestHeader(
+      true,
+      10,
+      { "User-Agent": "grpc-es/0.0.0" },
+      true,
+    );
+    expect(listHeaderKeys(headers)).toEqual([
+      "content-type",
+      "grpc-timeout",
+      "user-agent",
+      "x-grpc-web",
+      "x-user-agent",
+    ]);
+    expect(headers.get("User-Agent")).toBe("grpc-es/0.0.0");
+    expect(headers.get("X-User-Agent")).toBe("grpc-es/0.0.0");
+  });
+
+  it("should set user provided x-user-agent header", () => {
+    const headers = requestHeader(
+      true,
+      10,
+      { "X-User-Agent": "grpc-es/0.0.0" },
+      true,
+    );
+    expect(listHeaderKeys(headers)).toEqual([
+      "content-type",
+      "grpc-timeout",
+      "user-agent",
+      "x-grpc-web",
+      "x-user-agent",
+    ]);
+    expect(headers.get("User-Agent")).toBe("grpc-es/0.0.0");
+    expect(headers.get("X-User-Agent")).toBe("grpc-es/0.0.0");
+  });
+
   it("should create request headers with compression", () => {
     const compressionMock: Compression = {
       name: "gzip",

diff --git a/packages/connect/src/protocol-grpc-web/request-header.ts b/packages/connect/src/protocol-grpc-web/request-header.ts
@@ -46,9 +46,16 @@ export function requestHeader(
   // Note that we do not strictly comply with gRPC user agents.
   // We use "connect-es/1.2.3" where gRPC would use "grpc-es/1.2.3".
   // See https://github.com/grpc/grpc/blob/c462bb8d485fc1434ecfae438823ca8d14cf3154/doc/PROTOCOL-HTTP2.md#user-agents
-  result.set(headerXUserAgent, "CONNECT_ES_USER_AGENT");
+  let userAgent = "CONNECT_ES_USER_AGENT";
+  userAgent = result.has(headerUserAgent)
+    ? result.get(headerUserAgent)!
+    : result.has(headerXUserAgent)
+      ? result.get(headerXUserAgent)!
+      : userAgent;
+
+  result.set(headerXUserAgent, userAgent);
   if (setUserAgent) {
-    result.set(headerUserAgent, "CONNECT_ES_USER_AGENT");
+    result.set(headerUserAgent, userAgent);
   }
   if (timeoutMs !== undefined) {
     result.set(headerTimeout, `${timeoutMs}m`);

diff --git a/packages/connect/src/protocol-grpc/request-header.spec.ts b/packages/connect/src/protocol-grpc/request-header.spec.ts
@@ -48,6 +48,17 @@ describe("requestHeader", () => {
     expect(headers.get("Grpc-Timeout")).toBe("10m");
   });
 
+  it("should create request headers with user provided user-agent", () => {
+    const headers = requestHeader(true, 10, { "User-Agent": "grpc-es/0.0.0" });
+    expect(listHeaderKeys(headers)).toEqual([
+      "content-type",
+      "grpc-timeout",
+      "te",
+      "user-agent",
+    ]);
+    expect(headers.get("User-Agent")).toBe("grpc-es/0.0.0");
+  });
+
   it("should create request headers with compression", () => {
     const compressionMock: Compression = {
       name: "gzip",

diff --git a/packages/connect/src/protocol-grpc/request-header.ts b/packages/connect/src/protocol-grpc/request-header.ts
@@ -32,12 +32,19 @@ export function requestHeader(
   timeoutMs: number | undefined,
   userProvidedHeaders: HeadersInit | undefined,
 ): Headers {
-  const result = new Headers(userProvidedHeaders);
+  const result = new Headers(userProvidedHeaders ?? {});
   result.set(
     headerContentType,
     useBinaryFormat ? contentTypeProto : contentTypeJson,
   );
 
+  if (!result.has(headerUserAgent)) {
+    // Note that we do not strictly comply with gRPC user agents.
+    // We use "connect-es/1.2.3" where gRPC would use "grpc-es/1.2.3".
+    // See https://github.com/grpc/grpc/blob/c462bb8d485fc1434ecfae438823ca8d14cf3154/doc/PROTOCOL-HTTP2.md#user-agents
+    result.set(headerUserAgent, "CONNECT_ES_USER_AGENT");
+  }
+
   if (timeoutMs !== undefined) {
     result.set(headerTimeout, `${timeoutMs}m`);
   }