From 925bdc37545c5efd3660f740c337219709c6f695 Mon Sep 17 00:00:00 2001
From: Cole Rogers <colerogers@users.noreply.github.com>
Date: Tue, 19 Oct 2021 12:12:37 -0400
Subject: [PATCH] Storage Emulator Enhancement (#3809)

* storage trigger works

* removing comment

* cleaning up request body code

* add types

* fixing comments

* removing wrapper, adding check on type field

* using Content-Type header

* linter

* nit
---
 src/emulator/functionsEmulator.ts      | 17 ++++++--
 src/emulator/storage/cloudFunctions.ts | 59 +++++++++++++++++++++-----
 2 files changed, 63 insertions(+), 13 deletions(-)

diff --git a/src/emulator/functionsEmulator.ts b/src/emulator/functionsEmulator.ts
index 4e3a82b2b09..6e54c3b2385 100644
--- a/src/emulator/functionsEmulator.ts
+++ b/src/emulator/functionsEmulator.ts
@@ -267,10 +267,21 @@ export class FunctionsEmulator implements EmulatorInstance {
     };
 
     const multicastHandler: express.RequestHandler = (req, res) => {
-      const reqBody = (req as RequestWithRawBody).rawBody;
-      const proto = JSON.parse(reqBody.toString());
-      const triggers = this.multicastTriggers[`${this.args.projectId}:${proto.eventType}`] || [];
       const projectId = req.params.project_id;
+      const reqBody = (req as RequestWithRawBody).rawBody;
+      let proto = JSON.parse(reqBody.toString());
+      let triggerKey: string;
+      if (req.headers["content-type"]?.includes("cloudevent")) {
+        triggerKey = `${this.args.projectId}:${proto.type}`;
+
+        if (EventUtils.isBinaryCloudEvent(req)) {
+          proto = EventUtils.extractBinaryCloudEventContext(req);
+          proto.data = req.body;
+        }
+      } else {
+        triggerKey = `${this.args.projectId}:${proto.eventType}`;
+      }
+      const triggers = this.multicastTriggers[triggerKey] || [];
 
       triggers.forEach((triggerId) => {
         this.workQueue.submit(() => {
diff --git a/src/emulator/storage/cloudFunctions.ts b/src/emulator/storage/cloudFunctions.ts
index 0074a950a4f..d921da51606 100644
--- a/src/emulator/storage/cloudFunctions.ts
+++ b/src/emulator/storage/cloudFunctions.ts
@@ -1,11 +1,17 @@
 import { EmulatorRegistry } from "../registry";
 import { EmulatorInfo, Emulators } from "../types";
-import * as request from "request";
 import { EmulatorLogger } from "../emulatorLogger";
 import { CloudStorageObjectMetadata, toSerializedDate } from "./metadata";
 import { Client } from "../../apiv2";
+import { StorageObjectData } from "@google/events/cloud/storage/v1/StorageObjectData";
 
 type StorageCloudFunctionAction = "finalize" | "metadataUpdate" | "delete" | "archive";
+const STORAGE_V2_ACTION_MAP: Record<StorageCloudFunctionAction, string> = {
+  finalize: "finalized",
+  metadataUpdate: "metadataUpdated",
+  delete: "deleted",
+  archive: "archived",
+};
 
 export class StorageCloudFunctions {
   private logger = EmulatorLogger.forEmulator(Emulators.STORAGE);
@@ -13,6 +19,7 @@ export class StorageCloudFunctions {
   private multicastOrigin = "";
   private multicastPath = "";
   private enabled = false;
+  private client?: Client;
 
   constructor(private projectId: string) {
     const functionsEmulator = EmulatorRegistry.get(Emulators.FUNCTIONS);
@@ -24,6 +31,7 @@ export class StorageCloudFunctions {
         this.functionsEmulatorInfo
       )}`;
       this.multicastPath = `/functions/projects/${projectId}/trigger_multicast`;
+      this.client = new Client({ urlPrefix: this.multicastOrigin, auth: false });
     }
   }
 
@@ -31,20 +39,32 @@ export class StorageCloudFunctions {
     action: StorageCloudFunctionAction,
     object: CloudStorageObjectMetadata
   ): Promise<void> {
-    if (!this.enabled) return;
-
-    const multicastEventBody = this.createEventRequestBody(action, object);
+    if (!this.enabled) {
+      return;
+    }
 
-    const c = new Client({ urlPrefix: this.multicastOrigin, auth: false });
-    let res;
+    const errStatus: Array<number> = [];
     let err: Error | undefined;
     try {
-      res = await c.post(this.multicastPath, multicastEventBody);
+      /** Legacy Google Events */
+      const eventBody = this.createLegacyEventRequestBody(action, object);
+      const eventRes = await this.client!.post(this.multicastPath, eventBody);
+      if (eventRes.status !== 200) {
+        errStatus.push(eventRes.status);
+      }
+      /** Modern CloudEvents */
+      const cloudEventBody = this.createCloudEventRequestBody(action, object);
+      const cloudEventRes = await this.client!.post(this.multicastPath, cloudEventBody, {
+        headers: { "Content-Type": "application/cloudevents+json; charset=UTF-8" },
+      });
+      if (cloudEventRes.status !== 200) {
+        errStatus.push(cloudEventRes.status);
+      }
     } catch (e) {
-      err = e;
+      err = e as Error;
     }
 
-    if (err || res?.status != 200) {
+    if (err || errStatus.length > 0) {
       this.logger.logLabeled(
         "WARN",
         "functions",
@@ -53,7 +73,8 @@ export class StorageCloudFunctions {
     }
   }
 
-  private createEventRequestBody(
+  /** Legacy Google Events type */
+  private createLegacyEventRequestBody(
     action: StorageCloudFunctionAction,
     objectMetadataPayload: ObjectMetadataPayload
   ): string {
@@ -70,6 +91,24 @@ export class StorageCloudFunctions {
       data: objectMetadataPayload,
     });
   }
+
+  /** Modern CloudEvents type */
+  private createCloudEventRequestBody(
+    action: StorageCloudFunctionAction,
+    objectMetadataPayload: ObjectMetadataPayload
+  ): string {
+    const ceAction = STORAGE_V2_ACTION_MAP[action];
+    if (!ceAction) {
+      throw new Error("Action is not definied as a CloudEvents action");
+    }
+    const data = (objectMetadataPayload as unknown) as StorageObjectData;
+    return JSON.stringify({
+      specVersion: 1,
+      type: `google.cloud.storage.object.v1.${ceAction}`,
+      source: `//storage.googleapis.com/projects/_/buckets/${objectMetadataPayload.bucket}/objects/${objectMetadataPayload.name}`,
+      data,
+    });
+  }
 }
 
 // From https://github.com/firebase/firebase-functions/blob/master/src/providers/storage.ts