fix: create correct v4 signed url with cname (#868)

jkwlui · web-flow · commit ace3b5e3819c · 2019-09-30T15:05:05.000-07:00
diff --git a/src/file.ts b/src/file.ts
@@ -2587,7 +2587,8 @@ class File extends ServiceObject<File> {
     }
 
     const extensionHeaders = Object.assign({}, config.extensionHeaders);
-    extensionHeaders.host = 'storage.googleapis.com';
+    const fqdn = new url.URL(config.cname || STORAGE_DOWNLOAD_BASE_URL);
+    extensionHeaders.host = fqdn.host;
     if (config.method === 'POST') {
       extensionHeaders['x-goog-resumable'] = 'start';
     }
@@ -2625,7 +2626,7 @@ class File extends ServiceObject<File> {
 
       const canonicalRequest = [
         config.method,
-        config.resource,
+        config.cname ? `/${config.name}` : config.resource,
         canonicalQueryParams,
         extensionHeadersString,
         signedHeaders,
diff --git a/test/file.ts b/test/file.ts
@@ -3040,6 +3040,29 @@ describe('File', () => {
           }
         );
       });
+
+      it('should generate v4 signed url with provided cname', done => {
+        const host = 'http://www.example.com';
+
+        file.getSignedUrl(
+          {
+            action: 'read',
+            cname: host,
+            version: 'v4',
+            expires: Date.now() + 2000,
+          },
+          (err: Error, signedUrl: string) => {
+            assert.ifError(err);
+            const expected =
+              'http://www.example.com/file-name.png?' +
+              'X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=client-email' +
+              '%2F20190318%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20190318T000000Z&' +
+              'X-Goog-Expires=2&X-Goog-SignedHeaders=host&X-Goog-Signature=b228276adbab';
+            assert.strictEqual(signedUrl, expected);
+            done();
+          }
+        );
+      });
     });
 
     describe('promptSaveAs', () => {
