Merge pull request #1891 from EnsembleUI/firebase-custom-token-auth

Firebase custom token auth

Author: Umair-Manzoor-47

modules/auth/lib/auth_module.dart

@@ -8,6 +8,7 @@ import 'package:ensemble_auth/connect/widget/connect_with_google.dart';
 import 'package:ensemble_auth/connect/widget/connect_with_microsoft.dart';
 import 'package:ensemble_auth/signin/auth_manager.dart';
 import 'package:ensemble_auth/signin/sign_in_anonymous.dart';
+import 'package:ensemble_auth/signin/sign_in_with_custom_token.dart';
 import 'package:ensemble_auth/signin/widget/sign_in_with_apple.dart';
 // import 'package:ensemble_auth/signin/widget/sign_in_with_auth0.dart';
 import 'package:ensemble_auth/signin/widget/sign_in_with_google.dart';
@@ -26,5 +27,6 @@ class AuthModuleImpl implements AuthModule {
     GetIt.I.registerSingleton<TokenManager>(TokenManagerImpl());
     GetIt.I.registerFactory<OAuthController>(() => OAuthControllerImpl());
     GetIt.I.registerFactory<SignInAnonymous>(() => SignInAnonymousImpl());
+    GetIt.I.registerFactory<SignInWithCustomToken>(() => SignInWithCustomTokenImpl());
   }
 }

modules/auth/lib/signin/auth_manager.dart

@@ -116,14 +116,8 @@ class AuthManager with UserAuthentication {
         print('Sign in anonymous failed');
         return null;
       }
-      Future<void> updateCurrentUser(BuildContext context, User newUser) async {
-        await StorageManager()
-            .writeToSystemStorage(UserAuthentication._idKey, newUser.uid);
-        await StorageManager()
-            .writeToSystemStorage(UserAuthentication._isAnonymous, true);
-      }
 
-      updateCurrentUser(context, user);
+      updateCurrentFirebaseUser(context, user);
 
       return user.isAnonymous;
     } catch (e) {
@@ -132,6 +126,37 @@ class AuthManager with UserAuthentication {
     }
   }
 
+  Future<String?> signInWithCustomToken(
+      BuildContext context,
+      {required String jwtToken}
+  ) async {
+    try {
+      customFirebaseApp ??= await _initializeFirebaseSignIn();
+      final _auth = FirebaseAuth.instanceFor(app: customFirebaseApp!);
+
+      UserCredential userCredential = await _auth.signInWithCustomToken(jwtToken);
+      User? user = userCredential.user;
+      if (user == null) {
+        print('Sign in with custom firebase auth token failed');
+        return null;
+      }
+      String? firebaseToken = await user.getIdToken();
+
+      updateCurrentFirebaseUser(context, user);
+      return firebaseToken;
+    } catch (e) {
+      print(e.toString());
+      return null;
+    }
+  }
+
+  Future<void> updateCurrentFirebaseUser(BuildContext context, User newUser) async {
+    await StorageManager()
+        .writeToSystemStorage(UserAuthentication._idKey, newUser.uid);
+    await StorageManager()
+        .writeToSystemStorage(UserAuthentication._isAnonymous, newUser.isAnonymous);
+  }
+
   Future<String?> _signInLocally(BuildContext context,
       {required AuthenticatedUser user, AuthToken? token}) async {
     // update the current user

modules/auth/lib/signin/sign_in_with_custom_token.dart

@@ -0,0 +1,26 @@
+import 'package:ensemble/framework/action.dart';
+import 'package:ensemble/framework/event.dart';
+import 'package:ensemble/framework/stub/auth_context_manager.dart';
+import 'package:ensemble/screen_controller.dart';
+import 'package:ensemble/widget/stub_widgets.dart';
+import 'package:ensemble_auth/signin/auth_manager.dart';
+import 'package:flutter/material.dart';
+
+class SignInWithCustomTokenImpl implements SignInWithCustomToken {
+  @override
+  Future<void> signInWithCustomToken(BuildContext context,
+      {required SignInWithCustomTokenAction action}) async {
+    final idToken = await AuthManager().signInWithCustomToken(context, jwtToken: action.jwtToken!);
+    if (idToken != null) {
+      if (action.onAuthenticated != null) {
+        AuthenticatedUser? currentUser = AuthManager().getCurrentUser();
+        ScreenController().executeAction(context, action.onAuthenticated!,
+            event: EnsembleEvent(null, data: {'user': currentUser, 'idToken': idToken}));
+      }
+    } else {
+      if (action.onError != null) {
+        ScreenController().executeAction(context, action.onError!);
+      }
+    }
+  }
+}

modules/ensemble/lib/framework/action.dart

@@ -930,6 +930,28 @@ class SignInAnonymousAction extends EnsembleAction {
   }
 }
 
+class SignInWithCustomTokenAction extends EnsembleAction {
+  final EnsembleAction? onAuthenticated;
+  final EnsembleAction? onError;
+  final String? jwtToken;
+
+  SignInWithCustomTokenAction({
+    super.initiator,
+    super.inputs,
+    required this.onAuthenticated,
+    required this.onError,
+    this.jwtToken,
+  });
+
+  factory SignInWithCustomTokenAction.fromYaml({Map? payload}) {
+    return SignInWithCustomTokenAction(
+      onAuthenticated: EnsembleAction.from(payload?['onAuthenticated']),
+      onError: EnsembleAction.from(payload?['onError']),
+      jwtToken: payload?['token'],
+    );
+  }
+}
+
 class ClearKeychain extends EnsembleAction {
   ClearKeychain({
     required this.key,
@@ -1038,6 +1060,7 @@ enum ActionType {
   setLocale,
   clearLocale,
   signInAnonymous,
+  signInWithCustomToken,
   handleDeeplink,
   createDeeplink,
   verifySignIn,
@@ -1170,6 +1193,8 @@ abstract class EnsembleAction {
       return FileUploadAction.fromYaml(payload: payload);
     } else if (actionType == ActionType.signInAnonymous) {
       return SignInAnonymousAction.fromYaml(payload: payload);
+    } else if (actionType == ActionType.signInWithCustomToken) {
+      return SignInWithCustomTokenAction.fromYaml(payload: payload);
     } else if (actionType == ActionType.pickFiles) {
       return FilePickerAction.fromYaml(payload: payload);
     } else if (actionType == ActionType.openUrl) {

modules/ensemble/lib/module/auth_module.dart

@@ -17,6 +17,7 @@ class AuthModuleStub implements AuthModule {
     GetIt.I.registerSingleton<TokenManager>(TokenManagerStub());
     GetIt.I.registerFactory<OAuthController>(() => OAuthControllerStub());
     GetIt.I.registerFactory<SignInAnonymous>(() => SignInAnonymousStub());
+    GetIt.I.registerFactory<SignInWithCustomToken>(() => SignInWithCustomTokenStub());
     // note that we don't inject AuthContextManagerStub(), since its presence
     // will prevent data_context to load
   }

modules/ensemble/lib/screen_controller.dart

@@ -405,6 +405,8 @@ class ScreenController {
       GetIt.I<FileManager>().pickFiles(context, action, scopeManager);
     } else if (action is SignInAnonymousAction) {
       GetIt.I<SignInAnonymous>().signInAnonymously(context, action: action);
+    } else if (action is SignInWithCustomTokenAction) {
+      GetIt.I<SignInWithCustomToken>().signInWithCustomToken(context, action: action);
     } else if (action is WalletConnectAction) {
       //  TODO store session:  WalletConnectSession? session = await sessionStorage.getSession();
 

modules/ensemble/lib/widget/stub_widgets.dart

@@ -68,3 +68,17 @@ class SignInAnonymousStub implements SignInAnonymous {
         "Auth is not enabled. Please review the Ensemble documentation.");
   }
 }
+
+abstract class SignInWithCustomToken {
+  Future<void> signInWithCustomToken(BuildContext context,
+      {required SignInWithCustomTokenAction action});
+}
+
+class SignInWithCustomTokenStub implements SignInWithCustomToken {
+  @override
+  Future<void> signInWithCustomToken(BuildContext context,
+      {required SignInWithCustomTokenAction action}) {
+    throw ConfigError(
+        "Auth is not enabled. Please review the Ensemble documentation.");
+  }
+}