Allow TAR uploads to storage.

src/gcp/storage.ts

@@ -11,6 +11,12 @@
 import * as utils from "../utils";
 import { fieldMasks } from "./proto";
 
+/** Content Type **/
+export enum ContentType {
+  ZIP = "ZIP",
+  TAR = "TAR",
+}
+
 /** Bucket Interface */
 interface BucketResponse {
   kind: string;
@@ -195,7 +201,7 @@
   downloadTokens?: string;
 }
 
 export async function getDefaultBucket(projectId: string): Promise<string> {
   await ensure(projectId, firebaseStorageOrigin(), "storage", false);
   try {
     const localAPIClient = new Client({
@@ -211,9 +217,9 @@
         "Your project is being set up. Please wait a minute before deploying again.",
       );
     }
     return response.body.bucket.name.split("/").pop()!;
   } catch (err: any) {
     if (err?.status === 404) {
       throw new FirebaseError(
         `Firebase Storage has not been set up on project '${clc.bold(
           projectId,
@@ -225,8 +231,8 @@
   }
 }
 
 export async function upload(
   source: any,
   uploadUrl: string,
   extraHeaders?: Record<string, string>,
   ignoreQuotaProject?: boolean,
@@ -243,7 +249,7 @@
       "content-type": "application/zip",
       ...extraHeaders,
     },
     body: source.stream,
     skipLog: { resBody: true },
     ignoreQuotaProject,
   });
@@ -253,28 +259,39 @@
 }
 
 /**
- * Uploads a zip file to the specified bucket using the firebasestorage api.
+ * Uploads a zip or tar file to the specified bucket using the firebasestorage api.
  */
 export async function uploadObject(
   /** Source with file (name) to upload, and stream of file. */
   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") {
-    throw new FirebaseError(`Expected a file name ending in .zip, got ${source.file}`);
+  switch (contentType) {
+    case ContentType.TAR:
+      if (!source.file.endsWith(".tar.gz")) {
+        throw new FirebaseError(`Expected a file name ending in .tar.gz, got ${source.file}`);
+      }
+      break;
+    default:
+      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,
@@ -304,7 +321,7 @@
  * Deletes an object via Firebase Storage.
  * @param {string} location A Firebase Storage location, of the form "/v0/b/<bucket>/o/<object>"
  */
 export function deleteObject(location: string): Promise<any> {
   const localAPIClient = new Client({ urlPrefix: storageOrigin() });
   return localAPIClient.delete(location);
 }