Allow TAR uploads to storage.

Author: annajowang

src/gcp/storage.ts

@@ -11,6 +11,12 @@ import { ensure } from "../ensureApiEnabled";
 import * as utils from "../utils";
 import { fieldMasks } from "./proto";
 
+/** Content Type **/
+export enum ContentType {
+  ZIP = "ZIP",
+  TAR = "TAR",
+}
+
 /** Bucket Interface */
 interface BucketResponse {
   kind: string;
@@ -252,6 +258,7 @@ export async function upload(
   };
 }
 
+
 /**
  * Uploads a zip file to the specified bucket using the firebasestorage api.
  */
@@ -260,21 +267,27 @@ export async function uploadObject(
   source: { file: string; stream: Readable },
   /** Bucket to upload to. */
   bucketName: string,
+  contentType?: ContentType,
 ): Promise<{
   bucket: string;
   object: string;
   generation: string | null;
 }> {
-  if (path.extname(source.file) !== ".zip") {
+  if (contentType == ContentType.TAR) {
+    if (path.extname(source.file) !== ".tar.gz") {
+      throw new FirebaseError(`Expected a file name ending in .tar.gz, got ${source.file}`);
+    }
+  } else if (path.extname(source.file) !== ".zip") {
     throw new FirebaseError(`Expected a file name ending in .zip, got ${source.file}`);
   }
+
   const localAPIClient = new Client({ urlPrefix: storageOrigin() });
   const location = `/${bucketName}/${path.basename(source.file)}`;
   const res = await localAPIClient.request({
     method: "PUT",
     path: location,
     headers: {
-      "Content-Type": "application/zip",
+      "Content-Type": contentType == ContentType.TAR ? "application/octet-stream" : "application/zip",
       "x-goog-content-length-range": "0,123289600",
     },
     body: source.stream,