Added logic to decompress request body if it has gzip encoding header (…

…#2297)
Azure · Sep 25, 2023 · 68e1254 · 68e1254
1 parent d86cb77
commit 68e1254
Show file tree

Hide file tree

Showing 3 changed files with 41 additions and 4 deletions.
diff --git a/src/components/operations/operation-details/ko/runtime/operation-console.ts b/src/components/operations/operation-details/ko/runtime/operation-console.ts
@@ -538,9 +538,18 @@ export class OperationConsole {
                     : consoleOperation.name;
 
                 saveAs(blob, fileName);
-            }
-            else {
-                const responseBody = response.body.toString();
+            } else {
+                let responseBody: string;
+                if(this.useCorsProxy()) {
+                    const contentEncodingHeader = response.headers.find(x => x.name === KnownHttpHeaders.ContentEncoding.toLowerCase());
+                    if (contentEncodingHeader?.value === "gzip") {
+                        responseBody = await this.decompressBody(response.body);
+                    } else {
+                        responseBody = response.body.toString();
+                    }
+                } else {
+                    responseBody = response.body.toString();
+                }
 
                 if (responseContentType && Utils.isJsonContentType(responseContentType)) {
                     this.responseBody(Utils.formatJson(responseBody));
@@ -572,6 +581,31 @@ export class OperationConsole {
 
     private ws: WebsocketClient;
 
+    private async decompressBody(body: Buffer): Promise<string> {
+        const ds = new DecompressionStream("gzip");
+        const dsWriter = ds.writable.getWriter();
+        dsWriter.write(body);
+        dsWriter.close();
+        const output: Uint8Array[] = [];
+        const reader = ds.readable.getReader();
+        let totalSize = 0;
+
+        while (true) {
+            const { value, done } = await reader.read();
+            if (done) break;
+            output.push(value);
+            totalSize += value.byteLength;
+        }
+        const concatenated = new Uint8Array(totalSize);
+        let offset = 0;
+        for (const array of output) {
+            concatenated.set(array, offset);
+            offset += array.byteLength;
+        }
+        const decoder = new TextDecoder("utf-8");
+        return decoder.decode(concatenated);
+    }
+
     public async wsConnect(): Promise<void> {
         const operation = this.consoleOperation();
         const templateParameters = operation.templateParameters();

diff --git a/src/models/knownHttpHeaders.ts b/src/models/knownHttpHeaders.ts
@@ -2,6 +2,7 @@ export enum KnownHttpHeaders {
     Authorization = "Authorization",
     CacheControl = "Cache-Control",
     ContentType = "Content-Type",
+    ContentEncoding = "Content-Encoding",
     OcpApimSubscriptionKey = "Ocp-Apim-Subscription-Key",
     OcpApimTrace = "Ocp-Apim-Trace",
     OcpApimTraceLocation = "Ocp-Apim-Trace-Location",

diff --git a/src/modules.d.ts b/src/modules.d.ts
@@ -16,4 +16,6 @@ declare module "*.txt" {
 declare module "*.raw" {
     const content: string;
     export default content;
-}
+}
+
+declare let DecompressionStream: any;