fix: pass injector to pendingUntilEvent pipe

angular · Dec 19, 2024 · e7e0866 · e7e0866
1 parent 7364b44
commit e7e0866
Show file tree

Hide file tree

Showing 8 changed files with 44 additions and 31 deletions.
diff --git a/src/compat/database/database.ts b/src/compat/database/database.ts
@@ -1,4 +1,4 @@
-import { Inject, Injectable, InjectionToken, NgZone, Optional, PLATFORM_ID, inject } from '@angular/core';
+import { EnvironmentInjector, Inject, Injectable, InjectionToken, NgZone, Optional, PLATFORM_ID, inject } from '@angular/core';
 import { ɵAngularFireSchedulers } from '@angular/fire';
 import { AppCheckInstances } from '@angular/fire/app-check';
 import { FIREBASE_APP_NAME, FIREBASE_OPTIONS, ɵcacheInstance, ɵfirebaseAppFactory } from '@angular/fire/compat';
@@ -31,6 +31,7 @@ export const USE_EMULATOR = new InjectionToken<UseEmulatorArguments>('angularfir
 })
 export class AngularFireDatabase {
   public readonly database: firebase.database.Database;
+  private readonly injector = inject(EnvironmentInjector);
 
   constructor(
     @Inject(FIREBASE_OPTIONS) options: FirebaseOptions,
@@ -73,12 +74,12 @@ export class AngularFireDatabase {
     if (queryFn) {
       query = queryFn(ref);
     }
-    return createListReference<T>(query, this);
+    return createListReference<T>(query, this, this.injector);
   }
 
   object<T>(pathOrRef: PathReference): AngularFireObject<T> {
     const ref = inject(NgZone).runOutsideAngular(() => getRef(this.database, pathOrRef));
-    return createObjectReference<T>(ref, this);
+    return createObjectReference<T>(ref, this, this.injector);
   }
 
   createPushId() {

diff --git a/src/compat/database/list/create-reference.ts b/src/compat/database/list/create-reference.ts
@@ -1,4 +1,4 @@
-import { NgZone, inject } from '@angular/core';
+import { Injector, NgZone, inject } from '@angular/core';
 import { pendingUntilEvent } from '@angular/core/rxjs-interop';
 import type { Observable } from 'rxjs';
 import { map } from 'rxjs/operators';
@@ -10,7 +10,7 @@ import { createRemoveMethod } from './remove';
 import { snapshotChanges } from './snapshot-changes';
 import { stateChanges } from './state-changes';
 
-export function createListReference<T= any>(query: DatabaseQuery, afDatabase: AngularFireDatabase): AngularFireList<T> {
+export function createListReference<T= any>(query: DatabaseQuery, afDatabase: AngularFireDatabase, injector?: Injector): AngularFireList<T> {
   const outsideAngularScheduler = afDatabase.schedulers.outsideAngular;
   const refInZone = inject(NgZone).run(() => query.ref);
   return {
@@ -20,13 +20,13 @@ export function createListReference<T= any>(query: DatabaseQuery, afDatabase: An
     push: (data: T) => refInZone.push(data),
     remove: createRemoveMethod(refInZone),
     snapshotChanges(events?: ChildEvent[]) {
-      return snapshotChanges<T>(query, events, outsideAngularScheduler).pipe(pendingUntilEvent());
+      return snapshotChanges<T>(query, events, outsideAngularScheduler).pipe(pendingUntilEvent(injector));
     },
     stateChanges(events?: ChildEvent[]) {
-      return stateChanges<T>(query, events, outsideAngularScheduler).pipe(pendingUntilEvent());
+      return stateChanges<T>(query, events, outsideAngularScheduler).pipe(pendingUntilEvent(injector));
     },
     auditTrail(events?: ChildEvent[]) {
-      return auditTrail<T>(query, events, outsideAngularScheduler).pipe(pendingUntilEvent());
+      return auditTrail<T>(query, events, outsideAngularScheduler).pipe(pendingUntilEvent(injector));
     },
     valueChanges<K extends string>(events?: ChildEvent[], options?: {idField?: K}): Observable<(T & Record<string, string>)[]> {
       const snapshotChanges$ = snapshotChanges<T>(query, events, outsideAngularScheduler);
@@ -43,7 +43,7 @@ export function createListReference<T= any>(query: DatabaseQuery, afDatabase: An
             return a.payload.val() as T & Record<string, string>
           }
         })),
-        pendingUntilEvent()
+        pendingUntilEvent(injector)
       );
     }
   };

diff --git a/src/compat/database/object/create-reference.ts b/src/compat/database/object/create-reference.ts
@@ -1,15 +1,16 @@
+import { Injector } from '@angular/core';
 import { pendingUntilEvent } from '@angular/core/rxjs-interop';
 import { map } from 'rxjs/operators';
 import { AngularFireDatabase } from '../database';
 import { AngularFireObject, DatabaseQuery } from '../interfaces';
 import { createObjectSnapshotChanges } from './snapshot-changes';
 
-export function createObjectReference<T= any>(query: DatabaseQuery, afDatabase: AngularFireDatabase): AngularFireObject<T> {
+export function createObjectReference<T= any>(query: DatabaseQuery, afDatabase: AngularFireDatabase, injector?: Injector): AngularFireObject<T> {
   return {
     query,
     snapshotChanges<T>() {
       return createObjectSnapshotChanges<T>(query, afDatabase.schedulers.outsideAngular)().pipe(
-        pendingUntilEvent()
+        pendingUntilEvent(injector)
       );
     },
     update(data: Partial<T>) { return query.ref.update(data as any) as Promise<void>; },
@@ -18,7 +19,7 @@ export function createObjectReference<T= any>(query: DatabaseQuery, afDatabase:
     valueChanges<T>() {
       const snapshotChanges$ = createObjectSnapshotChanges(query, afDatabase.schedulers.outsideAngular)();
       return snapshotChanges$.pipe(
-        pendingUntilEvent(),
+        pendingUntilEvent(injector),
         map(action => action.payload.exists() ? action.payload.val() as T : null)
       );
     },

diff --git a/src/compat/firestore/collection-group/collection-group.ts b/src/compat/firestore/collection-group/collection-group.ts
@@ -1,3 +1,4 @@
+import { EnvironmentInjector, inject } from '@angular/core';
 import { pendingUntilEvent } from '@angular/core/rxjs-interop';
 import firebase from 'firebase/compat/app';
 import { Observable, from } from 'rxjs';
@@ -27,6 +28,8 @@ import { fromCollectionRef } from '../observable/fromRef';
  * fakeStock.valueChanges().subscribe(value => console.log(value));
  */
 export class AngularFirestoreCollectionGroup<T = DocumentData> {
+  private readonly injector = inject(EnvironmentInjector);
+
   /**
    * The constructor takes in a CollectionGroupQuery to provide wrapper methods
    * for data operations and data streaming.
@@ -43,14 +46,14 @@ export class AngularFirestoreCollectionGroup<T = DocumentData> {
   stateChanges(events?: DocumentChangeType[]): Observable<DocumentChangeAction<T>[]> {
     if (!events || events.length === 0) {
       return docChanges<T>(this.query, this.afs.schedulers.outsideAngular).pipe(
-        pendingUntilEvent()
+        pendingUntilEvent(this.injector)
       );
     }
     return docChanges<T>(this.query, this.afs.schedulers.outsideAngular)
       .pipe(
         map(actions => actions.filter(change => events.indexOf(change.type) > -1)),
         filter(changes =>  changes.length > 0),
-        pendingUntilEvent()
+        pendingUntilEvent(this.injector)
       );
   }
 
@@ -70,7 +73,7 @@ export class AngularFirestoreCollectionGroup<T = DocumentData> {
     const validatedEvents = validateEventsArray(events);
     const scheduledSortedChanges$ = sortedChanges<T>(this.query, validatedEvents, this.afs.schedulers.outsideAngular);
     return scheduledSortedChanges$.pipe(
-      pendingUntilEvent()
+      pendingUntilEvent(this.injector)
     );
   }
 
@@ -98,7 +101,7 @@ export class AngularFirestoreCollectionGroup<T = DocumentData> {
             return a.data();
           }
         })),
-        pendingUntilEvent()
+        pendingUntilEvent(this.injector)
       );
   }
 
@@ -107,7 +110,7 @@ export class AngularFirestoreCollectionGroup<T = DocumentData> {
    */
   get(options?: firebase.firestore.GetOptions) {
     return from(this.query.get(options)).pipe(
-      pendingUntilEvent()
+      pendingUntilEvent(this.injector)
     );
   }
 

diff --git a/src/compat/firestore/collection/collection.ts b/src/compat/firestore/collection/collection.ts
@@ -1,3 +1,4 @@
+import { EnvironmentInjector, inject } from '@angular/core';
 import { pendingUntilEvent } from '@angular/core/rxjs-interop';
 import firebase from 'firebase/compat/app';
 import { Observable, from } from 'rxjs';
@@ -41,6 +42,8 @@ export function validateEventsArray(events?: DocumentChangeType[]) {
  * fakeStock.valueChanges().subscribe(value => console.log(value));
  */
 export class AngularFirestoreCollection<T = DocumentData> {
+  private readonly injector = inject(EnvironmentInjector);
+
   /**
    * The constructor takes in a CollectionReference and Query to provide wrapper methods
    * for data operations and data streaming.
@@ -74,7 +77,7 @@ export class AngularFirestoreCollection<T = DocumentData> {
       pairwise(),
       filter(([prior, current]: DocumentChangeTuple<T>) => current.length > 0 || !prior),
       map(([, current]) => current),
-      pendingUntilEvent()
+      pendingUntilEvent(this.injector)
     );
   }
 
@@ -94,7 +97,7 @@ export class AngularFirestoreCollection<T = DocumentData> {
     const validatedEvents = validateEventsArray(events);
     const scheduledSortedChanges$ = sortedChanges<T>(this.query, validatedEvents, this.afs.schedulers.outsideAngular);
     return scheduledSortedChanges$.pipe(
-      pendingUntilEvent()
+      pendingUntilEvent(this.injector)
     );
   }
 
@@ -121,7 +124,7 @@ export class AngularFirestoreCollection<T = DocumentData> {
             return a.data();
           }
         })),
-        pendingUntilEvent()
+        pendingUntilEvent(this.injector)
       );
   }
 
@@ -130,7 +133,7 @@ export class AngularFirestoreCollection<T = DocumentData> {
    */
   get(options?: firebase.firestore.GetOptions) {
     return from(this.query.get(options)).pipe(
-      pendingUntilEvent(),
+      pendingUntilEvent(this.injector)
     );
   }
 

diff --git a/src/compat/firestore/document/document.ts b/src/compat/firestore/document/document.ts
@@ -1,3 +1,4 @@
+import { EnvironmentInjector, inject } from '@angular/core';
 import { pendingUntilEvent } from '@angular/core/rxjs-interop';
 import firebase from 'firebase/compat/app';
 import { Observable, from } from 'rxjs';
@@ -30,6 +31,7 @@ import { fromDocRef } from '../observable/fromRef';
  * Observable.from(fakeStock).subscribe(value => console.log(value));
  */
 export class AngularFirestoreDocument<T = DocumentData> {
+  private readonly injector = inject(EnvironmentInjector);
 
   /**
    * The constructor takes in a DocumentReference to provide wrapper methods
@@ -74,7 +76,7 @@ export class AngularFirestoreDocument<T = DocumentData> {
   snapshotChanges(): Observable<Action<DocumentSnapshot<T>>> {
     const scheduledFromDocRef$ = fromDocRef<T>(this.ref, this.afs.schedulers.outsideAngular);
     return scheduledFromDocRef$.pipe(
-      pendingUntilEvent()
+      pendingUntilEvent(this.injector)
     );
   }
 
@@ -102,7 +104,7 @@ export class AngularFirestoreDocument<T = DocumentData> {
    */
   get(options?: firebase.firestore.GetOptions) {
     return from(this.ref.get(options)).pipe(
-      pendingUntilEvent(),
+      pendingUntilEvent(this.injector)
     );
   }
 }
diff --git a/src/compat/storage/ref.ts b/src/compat/storage/ref.ts
@@ -1,3 +1,4 @@
+import { Injector } from '@angular/core';
 import { pendingUntilEvent } from '@angular/core/rxjs-interop';
 import { observeOutsideAngular } from '@angular/fire';
 import { Observable, from, of } from 'rxjs';
@@ -22,21 +23,22 @@ export interface AngularFireStorageReference {
  * creates observable methods from promise based methods.
  */
 export function createStorageRef(
-  ref: Reference
+  ref: Reference,
+  injector?: Injector
 ): AngularFireStorageReference {
   return {
     getDownloadURL: () => of(undefined).pipe(
       observeOutsideAngular,
       switchMap(() => ref.getDownloadURL()),
-      pendingUntilEvent()
+      pendingUntilEvent(injector)
     ),
     getMetadata: () => of(undefined).pipe(
       observeOutsideAngular,
       switchMap(() => ref.getMetadata()),
-      pendingUntilEvent()
+      pendingUntilEvent(injector)
     ),
     delete: () => from(ref.delete()),
-    child: (path: string) => createStorageRef(ref.child(path)),
+    child: (path: string) => createStorageRef(ref.child(path), injector),
     updateMetadata: (meta: SettableMetadata) => from(ref.updateMetadata(meta)),
     put: (data: any, metadata?: UploadMetadata) => {
       const task = ref.put(data, metadata);

diff --git a/src/compat/storage/storage.ts b/src/compat/storage/storage.ts
@@ -1,4 +1,4 @@
-import { Inject, Injectable, InjectionToken, NgZone, Optional, PLATFORM_ID } from '@angular/core';
+import { EnvironmentInjector, Inject, Injectable, InjectionToken, NgZone, Optional, PLATFORM_ID, inject } from '@angular/core';
 import { ɵAngularFireSchedulers } from '@angular/fire';
 import { AppCheckInstances } from '@angular/fire/app-check';
 import { FIREBASE_APP_NAME, FIREBASE_OPTIONS, ɵcacheInstance, ɵfirebaseAppFactory } from '@angular/fire/compat';
@@ -27,6 +27,7 @@ export const USE_EMULATOR = new InjectionToken<UseEmulatorArguments>('angularfir
 })
 export class AngularFireStorage {
   public readonly storage: firebase.storage.Storage;
+  private readonly injector = inject(EnvironmentInjector);
 
   constructor(
     @Inject(FIREBASE_OPTIONS) options: FirebaseOptions,
@@ -59,16 +60,16 @@ export class AngularFireStorage {
   }
 
   ref(path: string) {
-    return createStorageRef(this.storage.ref(path));
+    return createStorageRef(this.storage.ref(path), this.injector);
   }
 
   refFromURL(path: string) {
-    return createStorageRef(this.storage.refFromURL(path));
+    return createStorageRef(this.storage.refFromURL(path), this.injector);
   }
 
   upload(path: string, data: any, metadata?: UploadMetadata) {
     const storageRef = this.storage.ref(path);
-    const ref = createStorageRef(storageRef);
+    const ref = createStorageRef(storageRef, this.injector);
     return ref.put(data, metadata);
   }