feat: auth context support

.gitignore

@@ -5,4 +5,5 @@ lib
 .vscode
 *.tgz
 .tmp
-*.log
+*.log
+.idea

package.json

@@ -47,7 +47,7 @@
     "@types/mocha": "^5.2.7",
     "chai": "^4.2.0",
     "firebase-admin": "^12.0.0",
-    "firebase-functions": "^4.6.0",
+    "firebase-functions": "^4.9.0",
     "firebase-tools": "^8.9.2",
     "mocha": "^6.2.2",
     "prettier": "^1.19.1",
@@ -57,7 +57,7 @@
   },
   "peerDependencies": {
     "firebase-admin": "^8.0.0 || ^9.0.0 || ^10.0.0 || ^11.0.0 || ^12.0.0",
-    "firebase-functions": ">=4.3.0",
+    "firebase-functions": ">=4.9.0",
     "jest": ">=28.0.0"
   },
   "engines": {

src/cloudevent/mocks/firestore/firestore-on-document-created-with-auth-context.ts

@@ -0,0 +1,17 @@
+import { MockCloudEventAbstractFactory } from '../../types';
+import { CloudEvent, CloudFunction, firestore } from 'firebase-functions/v2';
+import { getEventType } from '../helpers';
+import { QueryDocumentSnapshot } from 'firebase-admin/firestore';
+import { getDocumentSnapshotCloudEventWithAuthContext } from './helpers';
+
+export const firestoreOnDocumentCreatedWithAuthContext: MockCloudEventAbstractFactory<firestore.FirestoreAuthEvent<
+  QueryDocumentSnapshot
+>> = {
+  generateMock: getDocumentSnapshotCloudEventWithAuthContext,
+  match(cloudFunction: CloudFunction<CloudEvent<unknown>>): boolean {
+    return (
+      getEventType(cloudFunction) ===
+      'google.cloud.firestore.document.v1.created.withAuthContext'
+    );
+  },
+};

src/cloudevent/mocks/firestore/firestore-on-document-deleted-with-auth-context.ts

@@ -0,0 +1,17 @@
+import { MockCloudEventAbstractFactory } from '../../types';
+import { CloudEvent, CloudFunction, firestore } from 'firebase-functions/v2';
+import { getEventType } from '../helpers';
+import { QueryDocumentSnapshot } from 'firebase-admin/firestore';
+import { getDocumentSnapshotCloudEventWithAuthContext } from './helpers';
+
+export const firestoreOnDocumentDeletedWithAuthContext: MockCloudEventAbstractFactory<firestore.FirestoreAuthEvent<
+  QueryDocumentSnapshot
+>> = {
+  generateMock: getDocumentSnapshotCloudEventWithAuthContext,
+  match(cloudFunction: CloudFunction<CloudEvent<unknown>>): boolean {
+    return (
+      getEventType(cloudFunction) ===
+      'google.cloud.firestore.document.v1.deleted.withAuthContext'
+    );
+  },
+};

src/cloudevent/mocks/firestore/firestore-on-document-updated-with-auth-context.ts

@@ -0,0 +1,22 @@
+import { MockCloudEventAbstractFactory } from '../../types';
+import {
+  Change,
+  CloudEvent,
+  CloudFunction,
+  firestore,
+} from 'firebase-functions/v2';
+import { getEventType } from '../helpers';
+import { QueryDocumentSnapshot } from 'firebase-admin/firestore';
+import { getDocumentSnapshotChangeCloudEventWithAuthContext } from './helpers';
+
+export const firestoreOnDocumentUpdatedWithAuthContext: MockCloudEventAbstractFactory<firestore.FirestoreAuthEvent<
+  Change<QueryDocumentSnapshot>
+>> = {
+  generateMock: getDocumentSnapshotChangeCloudEventWithAuthContext,
+  match(cloudFunction: CloudFunction<CloudEvent<unknown>>): boolean {
+    return (
+      getEventType(cloudFunction) ===
+      'google.cloud.firestore.document.v1.updated.withAuthContext'
+    );
+  },
+};

src/cloudevent/mocks/firestore/firestore-on-document-written-with-auth-context.ts

@@ -0,0 +1,22 @@
+import { MockCloudEventAbstractFactory } from '../../types';
+import {
+  Change,
+  CloudEvent,
+  CloudFunction,
+  firestore,
+} from 'firebase-functions/v2';
+import { getEventType } from '../helpers';
+import { DocumentSnapshot } from 'firebase-admin/firestore';
+import { getDocumentSnapshotChangeCloudEventWithAuthContext } from './helpers';
+
+export const firestoreOnDocumentWrittenWithAuthContext: MockCloudEventAbstractFactory<firestore.FirestoreAuthEvent<
+  Change<DocumentSnapshot>
+>> = {
+  generateMock: getDocumentSnapshotChangeCloudEventWithAuthContext,
+  match(cloudFunction: CloudFunction<CloudEvent<unknown>>): boolean {
+    return (
+      getEventType(cloudFunction) ===
+      'google.cloud.firestore.document.v1.written.withAuthContext'
+    );
+  },
+};

src/cloudevent/mocks/firestore/helpers.ts

@@ -47,6 +47,31 @@ export function getDocumentSnapshotCloudEvent(
   };
 }
 
+export function getDocumentSnapshotCloudEventWithAuthContext(
+  cloudFunction: CloudFunction<firestore.FirestoreAuthEvent<DocumentSnapshot>>,
+  cloudEventPartial?: DeepPartial<
+    firestore.FirestoreAuthEvent<DocumentSnapshot | object>
+  >
+) {
+  const eventWithoutAuthContext = getDocumentSnapshotCloudEvent(
+    cloudFunction,
+    cloudEventPartial
+  );
+  const authContext: { authId?: string; authType: firestore.AuthType } = {
+    authType: 'unknown',
+  };
+  if (cloudEventPartial?.authId) {
+    authContext.authId = cloudEventPartial.authId;
+  }
+  if (cloudEventPartial?.authType) {
+    authContext.authType = cloudEventPartial.authType;
+  }
+  return {
+    ...eventWithoutAuthContext,
+    ...authContext,
+  };
+}
+
 /** Creates a mock CloudEvent that contains a Change<DocumentSnapshot> as its data. */
 export function getDocumentSnapshotChangeCloudEvent(
   cloudFunction: CloudFunction<
@@ -82,6 +107,34 @@ export function getDocumentSnapshotChangeCloudEvent(
   };
 }
 
+export function getDocumentSnapshotChangeCloudEventWithAuthContext(
+  cloudFunction: CloudFunction<
+    firestore.FirestoreAuthEvent<Change<DocumentSnapshot>>
+  >,
+  cloudEventPartial?: DeepPartial<
+    firestore.FirestoreAuthEvent<Change<DocumentSnapshot> | ChangeLike>
+  >
+) {
+  const eventWithoutAuthContext = getDocumentSnapshotChangeCloudEvent(
+    cloudFunction,
+    cloudEventPartial
+  );
+  const authContext: { authId?: string; authType: firestore.AuthType } = {
+    authType: 'unknown',
+  };
+  if (cloudEventPartial?.authId) {
+    authContext.authId = cloudEventPartial.authId;
+  }
+  if (cloudEventPartial?.authType) {
+    authContext.authType = cloudEventPartial.authType;
+  }
+
+  return {
+    ...eventWithoutAuthContext,
+    ...authContext,
+  };
+}
+
 /** Finds or provides reasonable defaults for mock FirestoreEvent data. */
 function getFirestoreEventFields(
   cloudFunction: CloudFunction<

src/cloudevent/mocks/partials.ts

@@ -27,6 +27,10 @@ import {
 import { storageV1 } from './storage';
 import { remoteConfigOnConfigUpdated } from './remoteconfig/remote-config-on-config-updated';
 import { testLabOnTestMatrixCompleted } from './testlab/test-lab-on-test-matrix-completed';
+import { firestoreOnDocumentCreatedWithAuthContext } from './firestore/firestore-on-document-created-with-auth-context';
+import { firestoreOnDocumentDeletedWithAuthContext } from './firestore/firestore-on-document-deleted-with-auth-context';
+import { firestoreOnDocumentUpdatedWithAuthContext } from './firestore/firestore-on-document-updated-with-auth-context';
+import { firestoreOnDocumentWrittenWithAuthContext } from './firestore/firestore-on-document-written-with-auth-context';
 
 /**
  * Note: Ordering matters. Some MockEventPartials will match more generally
@@ -59,6 +63,10 @@ export const LIST_OF_MOCK_CLOUD_EVENT_PARTIALS: Array<MockCloudEventAbstractFact
   firestoreOnDocumentDeleted,
   firestoreOnDocumentUpdated,
   firestoreOnDocumentWritten,
+  firestoreOnDocumentCreatedWithAuthContext,
+  firestoreOnDocumentDeletedWithAuthContext,
+  firestoreOnDocumentUpdatedWithAuthContext,
+  firestoreOnDocumentWrittenWithAuthContext,
 
   // CustomEventPublished must be called last
   eventarcOnCustomEventPublished,