get account info / user profile right after sign in

Author: anotherdev

firebase-auth-rest/core/src/main/java/com/anotherdev/firebase/auth/FirebaseUser.java

@@ -6,6 +6,8 @@
 import com.anotherdev.firebase.auth.provider.AuthCredential;
 import com.google.gson.JsonObject;
 
+import java.util.List;
+
 import io.reactivex.rxjava3.core.Completable;
 import io.reactivex.rxjava3.core.Single;
 
@@ -28,6 +30,9 @@ public interface FirebaseUser {
     @Nullable
     String getEmail();
 
+    @NonNull
+    List<UserInfo> getProviderData();
+
     boolean isAnonymous();
 
     @NonNull

firebase-auth-rest/core/src/main/java/com/anotherdev/firebase/auth/data/model/FirebaseUserImpl.java

@@ -10,7 +10,9 @@
 import com.anotherdev.firebase.auth.FirebaseUser;
 import com.anotherdev.firebase.auth.ImmutableUserProfileChangeRequest;
 import com.anotherdev.firebase.auth.SignInResponse;
+import com.anotherdev.firebase.auth.UserInfo;
 import com.anotherdev.firebase.auth.UserProfileChangeRequest;
+import com.anotherdev.firebase.auth.data.Data;
 import com.anotherdev.firebase.auth.provider.AuthCredential;
 import com.anotherdev.firebase.auth.provider.EmailAuthCredential;
 import com.anotherdev.firebase.auth.provider.IdpAuthCredential;
@@ -23,6 +25,8 @@
 import com.google.gson.JsonElement;
 import com.google.gson.JsonObject;
 
+import java.util.List;
+
 import io.reactivex.rxjava3.core.Completable;
 import io.reactivex.rxjava3.core.Single;
 import timber.log.Timber;
@@ -45,6 +49,7 @@ public class FirebaseUserImpl implements FirebaseUser {
 
     @Nullable
     transient FirebaseAuth auth;
+    transient Data data;
 
 
     FirebaseUserImpl() {
@@ -80,6 +85,13 @@ private FirebaseAuth getAuth() {
         return auth;
     }
 
+    public Data getData() {
+        if (data == null) {
+            data = Data.from(getAuth().getApp().getApplicationContext());
+        }
+        return data;
+    }
+
     @NonNull
     private RestAuthProvider getAuthInternal() throws ClassCastException {
         return (RestAuthProvider) getAuth();
@@ -135,6 +147,15 @@ public String getEmail() {
         return getAsString(userInfo, "email");
     }
 
+    @NonNull
+    @Override
+    public List<UserInfo> getProviderData() {
+        String uid = getUid();
+        return getData().getUserProfile(uid != null ? uid : "")
+                .get()
+                .providerUserInfo();
+    }
+
     @Override
     public boolean isAnonymous() {
         return firebaseAuthData.getIdentities().isEmpty();

firebase-auth-rest/core/src/main/java/com/anotherdev/firebase/auth/rest/RestAuthProvider.java

@@ -10,9 +10,12 @@
 import com.anotherdev.firebase.auth.SignInResponse;
 import com.anotherdev.firebase.auth.data.Data;
 import com.anotherdev.firebase.auth.data.model.FirebaseUserImpl;
+import com.anotherdev.firebase.auth.data.model.UserProfile;
 import com.anotherdev.firebase.auth.provider.IdpAuthCredential;
 import com.anotherdev.firebase.auth.rest.api.RestAuthApi;
 import com.anotherdev.firebase.auth.rest.api.model.ExchangeTokenRequest;
+import com.anotherdev.firebase.auth.rest.api.model.GetAccountInfoRequest;
+import com.anotherdev.firebase.auth.rest.api.model.GetAccountInfoResponse;
 import com.anotherdev.firebase.auth.rest.api.model.ImmutableSignInWithIdpRequest;
 import com.anotherdev.firebase.auth.rest.api.model.SendPasswordResetEmailRequest;
 import com.anotherdev.firebase.auth.rest.api.model.SendPasswordResetEmailResponse;
@@ -93,7 +96,8 @@ public Observable<FirebaseAuth> authStateChanges() {
     public Single<SignInResponse> signInAnonymously() {
         return RestAuthApi.auth()
                 .signInAnonymously(SignInAnonymouslyRequest.builder().build())
-                .map(this::saveCurrentUser);
+                .map(this::saveCurrentUser)
+                .map(this::getAccountInfo);
     }
 
     @NonNull
@@ -105,7 +109,8 @@ public Single<SignInResponse> createUserWithEmailAndPassword(@NonNull String ema
                 .build();
         return RestAuthApi.auth()
                 .createUserWithEmailAndPassword(request)
-                .map(this::saveCurrentUser);
+                .map(this::saveCurrentUser)
+                .map(this::getAccountInfo);
     }
 
     @NonNull
@@ -117,7 +122,8 @@ public Single<SignInResponse> signInWithEmailAndPassword(@NonNull String email,
                 .build();
         return RestAuthApi.auth()
                 .signInWithEmailAndPassword(request)
-                .map(this::saveCurrentUser);
+                .map(this::saveCurrentUser)
+                .map(this::getAccountInfo);
     }
 
     @NonNull
@@ -135,7 +141,8 @@ public Single<SignInResponse> signInWithCustomToken(@NonNull String customToken)
                 .build();
         return RestAuthApi.auth()
                 .signInWithCustomToken(request)
-                .map(this::saveCurrentUser);
+                .map(this::saveCurrentUser)
+                .map(this::getAccountInfo);
     }
 
     @NonNull
@@ -154,7 +161,8 @@ private Single<SignInResponse> performSignInWithCredential(ImmutableSignInWithId
                 .build();
         return RestAuthApi.auth()
                 .signInWithCredential(request)
-                .map(this::saveCurrentUser);
+                .map(this::saveCurrentUser)
+                .map(this::getAccountInfo);
     }
 
     @NonNull
@@ -245,4 +253,19 @@ private void saveCurrentUser(String idToken, String refreshToken) throws Firebas
             l.onIdTokenChanged(new InternalTokenResult(idToken));
         }
     }
+
+    private SignInResponse getAccountInfo(SignInResponse signInResponse) {
+        GetAccountInfoRequest request = GetAccountInfoRequest.builder()
+                .idToken(signInResponse.getIdToken())
+                .build();
+        GetAccountInfoResponse accountInfo = RestAuthApi.auth()
+                .getAccounts(request).blockingGet();
+
+        Data dataStore = Data.from(app.getApplicationContext());
+        for (UserProfile profile : accountInfo.getUsers()) {
+            final String uid = profile.getLocalId();
+            dataStore.getUserProfile(uid).set(profile);
+        }
+        return signInResponse;
+    }
 }