Init App Check in firebase_manager.dart

mlcommons · May 6, 2024 · 6e29da8 · 6e29da8
1 parent 624162e
commit 6e29da8
Showing 1 changed file with 20 additions and 2 deletions.
diff --git a/flutter/lib/firebase/firebase_manager.dart b/flutter/lib/firebase/firebase_manager.dart
@@ -1,5 +1,8 @@
 import 'dart:convert';
 
+import 'package:flutter/foundation.dart';
+
+import 'package:firebase_app_check/firebase_app_check.dart';
 import 'package:firebase_auth/firebase_auth.dart';
 import 'package:firebase_core/firebase_core.dart';
 import 'package:firebase_storage/firebase_storage.dart';
@@ -32,14 +35,29 @@ class FirebaseManager {
     }
     final currentPlatform = DefaultFirebaseOptions.currentPlatform;
     final app = await Firebase.initializeApp(options: currentPlatform);
-    print('Firebase initialized using projectId: ${app.options.projectId}');
-
+    // _initAppCheck must run before any Firebase services, but after calling Firebase.initializeApp();
+    await _initAppCheck();
     await _initAuthentication();
     _initStorage();
     _isInitialized = true;
+    print('Firebase initialized using projectId: ${app.options.projectId}');
     return instance;
   }
 
+  Future<void> _initAppCheck() async {
+    if (kReleaseMode) {
+      await FirebaseAppCheck.instance.activate(
+        androidProvider: AndroidProvider.playIntegrity,
+        appleProvider: AppleProvider.deviceCheck,
+      );
+    } else {
+      await FirebaseAppCheck.instance.activate(
+        androidProvider: AndroidProvider.debug,
+        appleProvider: AppleProvider.debug,
+      );
+    }
+  }
+
   Future<void> _initAuthentication() async {
     FirebaseUIAuth.configureProviders(FirebaseAuthService.providers);
     _authService.firebaseAuth = FirebaseAuth.instance;