feat(afs): enable persistence option

Author: davideast

karma.conf.js

@@ -26,7 +26,7 @@ module.exports = function(config) {
 
       'karma-test-shim.js',
       'node_modules/firebase/firebase.js',
-      'node_modules/firestore/firestore.js',
+      'node_modules/firebase/firebase-firestore.js',
       'dist/packages-dist/bundles/core.umd.{js,map}',
       'dist/packages-dist/bundles/auth.umd.{js,map}',
       'dist/packages-dist/bundles/database.umd.{js,map}',

src/firestore/collection/collection.spec.ts

@@ -34,7 +34,7 @@ describe('AngularFirestoreCollection', () => {
     TestBed.configureTestingModule({
       imports: [
         AngularFireModule.initializeApp(COMMON_CONFIG),
-        AngularFirestoreModule
+        AngularFirestoreModule.enablePersistence()
       ]
     });
     inject([FirebaseApp, AngularFirestore], (_app: firebase.app.App, _afs: AngularFirestore) => {
@@ -102,7 +102,7 @@ describe('AngularFirestoreCollection', () => {
       });
     });
 
-    fit('should update order on queries', async (done) => {
+    it('should update order on queries', async (done) => {
       const ITEMS = 10;
       let count = 0;
       let firstIndex = 0;

src/firestore/document/document.spec.ts

@@ -21,7 +21,7 @@ describe('AngularFirestoreDocument', () => {
     TestBed.configureTestingModule({
       imports: [
         AngularFireModule.initializeApp(COMMON_CONFIG),
-        AngularFirestoreModule
+        AngularFirestoreModule.enablePersistence()
       ]
     });
     inject([FirebaseApp, AngularFirestore], (_app: firebase.app.App, _afs: AngularFirestore) => {

src/firestore/firestore.module.ts

@@ -1,23 +1,37 @@
-import { NgModule, NgZone } from '@angular/core';
+import { NgModule, InjectionToken } from '@angular/core';
 import { FirebaseApp, AngularFireModule } from 'angularfire2';
 import { AngularFirestore } from './firestore';
+import { from } from 'rxjs/observable/from';
 
-export function _getAngularFirestore(app: FirebaseApp) {
-  return new AngularFirestore(app);
+export const EnablePersistenceToken = new InjectionToken<boolean>('EnablePersistenceToken');
+
+export function _getAngularFirestore(app: FirebaseApp, enablePersistence: boolean) {
+  return new AngularFirestore(app, enablePersistence);
 }
 
 export const AngularFirestoreProvider = {
   provide: AngularFirestore,
   useFactory: _getAngularFirestore,
-  deps: [ FirebaseApp ]
+  deps: [ FirebaseApp, EnablePersistenceToken ]
 };
 
 export const FIRESTORE_PROVIDERS = [
   AngularFirestoreProvider,
+  { provide: EnablePersistenceToken, useValue: false },
 ];
 
 @NgModule({
   imports: [ AngularFireModule ],
   providers: [ FIRESTORE_PROVIDERS ]
 })
-export class AngularFirestoreModule { }
+export class AngularFirestoreModule {
+  static enablePersistence() {
+    return {
+      ngModule: AngularFireModule,
+      providers: [
+        { provide: EnablePersistenceToken, useValue: true },
+        AngularFirestoreProvider
+      ]
+    }
+  }
+}

src/firestore/firestore.spec.ts

@@ -25,7 +25,7 @@ describe('AngularFirestore', () => {
     TestBed.configureTestingModule({
       imports: [
         AngularFireModule.initializeApp(COMMON_CONFIG),
-        AngularFirestoreModule
+        AngularFirestoreModule.enablePersistence()
       ]
     });
     inject([FirebaseApp, AngularFirestore], (_app: firebase.app.App, _afs: AngularFirestore) => {
@@ -71,4 +71,11 @@ describe('AngularFirestore', () => {
     expect(quadWrapper).toThrowError();
   }); 
 
+  it('should enable persistence', (done) => {
+    const sub = afs.persistenceEnabled$.subscribe(isEnabled => {
+      expect(isEnabled).toBe(true);
+      done();
+    });
+  });
+
 });

src/firestore/firestore.ts

@@ -2,6 +2,7 @@ import * as firebase from 'firebase/app';
 import 'firebase/firestore';
 import { Observable } from 'rxjs/Observable';
 import { Subscriber } from 'rxjs/Subscriber';
+import { from } from 'rxjs/observable/from';
 import 'rxjs/add/operator/map';
 
 import { Injectable } from '@angular/core';
@@ -88,15 +89,20 @@ export function associateQuery(collectionRef: firebase.firestore.CollectionRefer
 @Injectable()
 export class AngularFirestore {
   public readonly firestore: firebase.firestore.Firestore;
+  public readonly persistenceEnabled$: Observable<boolean>;
 
   /**
    * Each Feature of AngularFire has a FirebaseApp injected. This way we
    * don't rely on the main Firebase App instance and we can create named
    * apps and use multiple apps.
    * @param app 
    */
-  constructor(public app: FirebaseApp) {
+  constructor(public app: FirebaseApp, shouldEnablePersistence) {
     this.firestore = app.firestore();
+
+    this.persistenceEnabled$ = shouldEnablePersistence ? 
+      from(app.firestore().enablePersistence().then(() => true)) :
+      from(new Promise((res, rej) => { res(false); }));
   }
 
   /**

tools/build.js

@@ -41,6 +41,7 @@ const GLOBALS = {
   'rxjs/add/operator/delay': 'Rx.Observable',
   'rxjs/add/operator/debounce': 'Rx.Observable',
   'rxjs/observable/fromEvent': 'Rx.Observable',
+  'rxjs/observable/from': 'Rx.Observable',
   'rxjs/operator': 'Rx.Observable.prototype',
   '@angular/core': 'ng.core',
   '@angular/compiler': 'ng.compiler',