Multiple app support for cloud_firestore

packages/cloud_firestore/CHANGELOG.md

@@ -1,3 +1,7 @@
+## 0.4.1
+
+* Support for multiple Firebase apps
+
 ## 0.4.0
 
 * **Breaking change**. Hide Firestore codec class from public API.

packages/cloud_firestore/android/src/main/java/io/flutter/plugins/firebase/cloudfirestore/CloudFirestorePlugin.java

@@ -13,6 +13,7 @@
 import com.google.android.gms.tasks.Task;
 import com.google.android.gms.tasks.TaskCompletionSource;
 import com.google.android.gms.tasks.Tasks;
+import com.google.firebase.FirebaseApp;
 import com.google.firebase.firestore.CollectionReference;
 import com.google.firebase.firestore.DocumentChange;
 import com.google.firebase.firestore.DocumentReference;
@@ -72,14 +73,19 @@ private CloudFirestorePlugin(MethodChannel channel) {
     this.channel = channel;
   }
 
+  private FirebaseFirestore getFirestore(Map<String, Object> arguments) {
+    String appName = (String) arguments.get("app");
+    return FirebaseFirestore.getInstance(FirebaseApp.getInstance(appName));
+  }
+
   private CollectionReference getCollectionReference(Map<String, Object> arguments) {
     String path = (String) arguments.get("path");
-    return FirebaseFirestore.getInstance().collection(path);
+    return getFirestore(arguments).collection(path);
   }
 
   private DocumentReference getDocumentReference(Map<String, Object> arguments) {
     String path = (String) arguments.get("path");
-    return FirebaseFirestore.getInstance().document(path);
+    return getFirestore(arguments).document(path);
   }
 
   private Map<String, Object> parseQuerySnapshot(QuerySnapshot querySnapshot) {
@@ -248,7 +254,7 @@ public void onMethodCall(MethodCall call, final Result result) {
           final Task<Map<String, Object>> transactionTCSTask = transactionTCS.getTask();
 
           final Map<String, Object> arguments = call.arguments();
-          FirebaseFirestore.getInstance()
+          getFirestore(arguments)
               .runTransaction(
                   new Transaction.Function<Void>() {
                     @Nullable
@@ -379,7 +385,8 @@ protected Void doInBackground(Void... voids) {
       case "WriteBatch#create":
         {
           int handle = nextBatchHandle++;
-          WriteBatch batch = FirebaseFirestore.getInstance().batch();
+          final Map<String, Object> arguments = call.arguments();
+          WriteBatch batch = getFirestore(arguments).batch();
           batches.put(handle, batch);
           result.success(handle);
           break;
@@ -573,6 +580,8 @@ protected void writeValue(ByteArrayOutputStream stream, Object value) {
       writeDouble(stream, ((GeoPoint) value).getLongitude());
     } else if (value instanceof DocumentReference) {
       stream.write(DOCUMENT_REFERENCE);
+      writeBytes(
+          stream, ((DocumentReference) value).getFirestore().getApp().getName().getBytes(UTF8));
       writeBytes(stream, ((DocumentReference) value).getPath().getBytes(UTF8));
     } else {
       super.writeValue(stream, value);
@@ -588,9 +597,13 @@ protected Object readValueOfType(byte type, ByteBuffer buffer) {
         readAlignment(buffer, 8);
         return new GeoPoint(buffer.getDouble(), buffer.getDouble());
       case DOCUMENT_REFERENCE:
-        final byte[] bytes = readBytes(buffer);
-        final String path = new String(bytes, UTF8);
-        return FirebaseFirestore.getInstance().document(path);
+        final byte[] appNameBytes = readBytes(buffer);
+        String appName = new String(appNameBytes, UTF8);
+        final FirebaseFirestore firestore =
+            FirebaseFirestore.getInstance(FirebaseApp.getInstance(appName));
+        final byte[] pathBytes = readBytes(buffer);
+        final String path = new String(pathBytes, UTF8);
+        return firestore.document(path);
       default:
         return super.readValueOfType(type, buffer);
     }

packages/cloud_firestore/example/lib/main.dart

@@ -5,17 +5,35 @@
 import 'dart:async';
 
 import 'package:flutter/material.dart';
+import 'package:firebase_core/firebase_core.dart';
 import 'package:cloud_firestore/cloud_firestore.dart';
 
-void main() {
-  runApp(new MaterialApp(title: 'Firestore Example', home: new MyHomePage()));
+Future main() async {
+  final FirebaseOptions options = const FirebaseOptions(
+    googleAppID: '1:79601577497:ios:5f2bcc6ba8cecddd',
+    gcmSenderID: '79601577497',
+    apiKey: 'AIzaSyArgmRGfB5kiQT6CunAOmKRVKEsxKmy6YI-G72PVU',
+    projectID: 'flutter-firestore',
+  );
+  FirebaseApp.configure(
+    name: 'test',
+    options: options,
+  );
+  final Firestore firestore = new Firestore(app: new FirebaseApp.named('test'));
+
+  runApp(new MaterialApp(
+      title: 'Firestore Example', home: new MyHomePage(firestore: firestore)));
 }
 
 class MessageList extends StatelessWidget {
+  MessageList({this.firestore});
+
+  final Firestore firestore;
+
   @override
   Widget build(BuildContext context) {
     return new StreamBuilder<QuerySnapshot>(
-      stream: Firestore.instance.collection('messages').snapshots,
+      stream: firestore.collection('messages').snapshots,
       builder: (BuildContext context, AsyncSnapshot<QuerySnapshot> snapshot) {
         if (!snapshot.hasData) return const Text('Loading...');
         final int messageCount = snapshot.data.documents.length;
@@ -35,10 +53,15 @@ class MessageList extends StatelessWidget {
 }
 
 class MyHomePage extends StatelessWidget {
-  CollectionReference get messages => Firestore.instance.collection('messages');
+  MyHomePage({this.firestore});
+  final Firestore firestore;
+  CollectionReference get messages => firestore.collection('messages');
 
   Future<Null> _addMessage() async {
-    messages.document().setData(<String, String>{'message': 'Hello world!'});
+    final DocumentReference document = messages.document();
+    document.setData(<String, dynamic>{
+      'message': 'Hello world!',
+    });
   }
 
   @override
@@ -47,7 +70,7 @@ class MyHomePage extends StatelessWidget {
       appBar: new AppBar(
         title: const Text('Firestore Example'),
       ),
-      body: new MessageList(),
+      body: new MessageList(firestore: firestore),
       floatingActionButton: new FloatingActionButton(
         onPressed: _addMessage,
         tooltip: 'Increment',

packages/cloud_firestore/example/pubspec.yaml

@@ -6,6 +6,7 @@ dependencies:
     sdk: flutter
   cloud_firestore:
     path: ../
+  firebase_core: "^0.2.0"
 
 flutter:
   uses-material-design: true

packages/cloud_firestore/ios/Classes/CloudFirestorePlugin.h

@@ -4,5 +4,5 @@
 
 #import <Flutter/Flutter.h>
 
-@interface FLTCloudFirestorePlugin : NSObject<FlutterPlugin>
+@interface FLTCloudFirestorePlugin : NSObject <FlutterPlugin>
 @end