Support WeChat login in WebKitWebViewUIImplementation

Author: louischan-oursky

packages/authgear-react-native/android/src/main/java/com/authgear/reactnative/AuthgearReactNativeModule.java

@@ -17,12 +17,15 @@
 import java.security.SecureRandom;
 import java.security.Signature;
 import java.security.interfaces.RSAPublicKey;
+import java.util.Objects;
 import java.util.UUID;
 
 import android.app.Activity;
+import android.content.BroadcastReceiver;
 import android.content.Context;
 import android.content.ContentResolver;
 import android.content.Intent;
+import android.content.IntentFilter;
 import android.content.SharedPreferences;
 import android.content.pm.PackageInfo;
 import android.graphics.Color;
@@ -46,13 +49,15 @@
 import com.facebook.react.bridge.ReadableMap;
 import com.facebook.react.bridge.WritableArray;
 import com.facebook.react.bridge.WritableMap;
+import com.facebook.react.modules.core.DeviceEventManagerModule;
 
 import com.google.crypto.tink.shaded.protobuf.InvalidProtocolBufferException;
 
 import androidx.annotation.NonNull;
 import androidx.annotation.RequiresApi;
 import androidx.biometric.BiometricManager;
 import androidx.biometric.BiometricPrompt;
+import androidx.core.content.ContextCompat;
 import androidx.fragment.app.FragmentActivity;
 import androidx.security.crypto.MasterKey;
 import androidx.security.crypto.EncryptedSharedPreferences;
@@ -63,6 +68,7 @@ public class AuthgearReactNativeModule extends ReactContextBaseJavaModule implem
 
     private static class Handle {
         Promise mPromise;
+        BroadcastReceiver mBroadcastReceiver;
 
         Handle(Promise promise) {
             this.mPromise = promise;
@@ -786,16 +792,40 @@ public void openAuthorizeURLWithWebView(ReadableMap options, Promise promise) {
 
             Context ctx = currentActivity;
 
+            String invocationID = options.getString("invocationID");
             Uri url = Uri.parse(options.getString("url"));
             Uri redirectURI = Uri.parse(options.getString("redirectURI"));
             Integer actionBarBackgroundColor = this.readColorInt(options, "actionBarBackgroundColor");
             Integer actionBarButtonTintColor = this.readColorInt(options, "actionBarButtonTintColor");
+            String wechatRedirectURIString = options.getString("androidWechatRedirectURI");
 
             WebKitWebViewActivity.Options webViewOptions = new WebKitWebViewActivity.Options();
             webViewOptions.url = url;
             webViewOptions.redirectURI = redirectURI;
             webViewOptions.actionBarBackgroundColor = actionBarBackgroundColor;
             webViewOptions.actionBarButtonTintColor = actionBarButtonTintColor;
+            if (wechatRedirectURIString != null) {
+                String intentAction = String.format("com.authgear.reactnative.callback.%s", invocationID);
+                handle.mBroadcastReceiver = new BroadcastReceiver() {
+                    @Override
+                    public void onReceive(Context context, Intent intent) {
+                        if (Objects.equals(intent.getAction(), intentAction)) {
+                            String uriString = intent.getExtras().getString("uri");
+                            WritableMap map = Arguments.createMap();
+                            map.putString("invocationID", invocationID);
+                            map.putString("url", uriString);
+                            AuthgearReactNativeModule.this.sendEvent("authgear-react-native", map);
+                        }
+                    }
+                };
+
+                IntentFilter intentFilter = new IntentFilter(intentAction);
+
+                webViewOptions.wechatRedirectURI = Uri.parse(wechatRedirectURIString);
+                webViewOptions.wechatRedirectURIIntentAction = intentAction;
+
+                ContextCompat.registerReceiver(ctx.getApplicationContext(), handle.mBroadcastReceiver, intentFilter, ContextCompat.RECEIVER_NOT_EXPORTED);
+            }
 
             Intent intent = WebKitWebViewActivity.createIntent(ctx, webViewOptions);
             currentActivity.startActivityForResult(intent, requestCode);
@@ -820,6 +850,10 @@ private Integer readColorInt(ReadableMap map, String key) {
         return null;
     }
 
+    private void sendEvent(String name, ReadableMap map) {
+        this.reactContext.getJSModule(DeviceEventManagerModule.RCTDeviceEventEmitter.class).emit(name, map);
+    }
+
     @ReactMethod
     public void openAuthorizeURL(String urlString, String callbackURL, boolean shareSessionWithSystemBrowser, Promise promise) {
         final Handle handle = new Handle(promise);
@@ -981,6 +1015,9 @@ public void onActivityResult(Activity activity, int requestCode, int resultCode,
         if (startActivityHandle != null) {
             final int tag = startActivityHandle.tag;
             final Promise promise = startActivityHandle.value.mPromise;
+            if (startActivityHandle.value.mBroadcastReceiver != null) {
+                activity.getApplicationContext().unregisterReceiver(startActivityHandle.value.mBroadcastReceiver);
+            }
             switch (tag) {
                 case ACTIVITY_PROMISE_TAG_CODE_AUTHORIZATION:
                     if (resultCode == Activity.RESULT_CANCELED) {

packages/authgear-react-native/android/src/main/java/com/authgear/reactnative/WebKitWebViewActivity.java

@@ -49,6 +49,9 @@ public static class Options {
         public Integer actionBarBackgroundColor;
         public Integer actionBarButtonTintColor;
 
+        public Uri wechatRedirectURI;
+        public String wechatRedirectURIIntentAction;
+
         public Options() {}
 
         private Options(Bundle bundle) {
@@ -60,18 +63,34 @@ private Options(Bundle bundle) {
             if (bundle.containsKey("actionBarButtonTintColor")) {
                 this.actionBarButtonTintColor = bundle.getInt("actionBarButtonTintColor");
             }
+            if (bundle.containsKey("wechatRedirectURI")) {
+                this.wechatRedirectURI = bundle.getParcelable("wechatRedirectURI");
+            }
+            if (bundle.containsKey("wechatRedirectURIIntentAction")) {
+                this.wechatRedirectURIIntentAction = bundle.getString("wechatRedirectURIIntentAction");
+            }
         }
 
         public Bundle toBundle() {
             Bundle bundle = new Bundle();
+
             bundle.putParcelable("url", this.url);
             bundle.putParcelable("redirectURI", this.redirectURI);
+
             if (this.actionBarBackgroundColor != null) {
                 bundle.putInt("actionBarBackgroundColor", this.actionBarBackgroundColor);
             }
             if (this.actionBarButtonTintColor != null) {
                 bundle.putInt("actionBarButtonTintColor", this.actionBarButtonTintColor);
             }
+
+            if (this.wechatRedirectURI != null) {
+                bundle.putParcelable("wechatRedirectURI", this.wechatRedirectURI);
+            }
+            if (this.wechatRedirectURIIntentAction != null) {
+                bundle.putString("wechatRedirectURIIntentAction", this.wechatRedirectURIIntentAction);
+            }
+
             return bundle;
         }
     }
@@ -122,6 +141,9 @@ private boolean shouldOverrideUrlLoading(Uri uri) {
             if (this.checkRedirectURI(uri)) {
                 return true;
             }
+            if (this.checkWechatRedirectURI(uri)) {
+                return true;
+            }
             return false;
         }
 
@@ -137,6 +159,26 @@ private boolean checkRedirectURI(Uri uri) {
             return false;
         }
 
+        private boolean checkWechatRedirectURI(Uri uri) {
+            Options options = this.activity.getOptions();
+            Uri wechatRedirectURI = options.wechatRedirectURI;
+            String wechatRedirectURIIntentAction = options.wechatRedirectURIIntentAction;
+            if (wechatRedirectURI == null || wechatRedirectURIIntentAction == null) {
+                return false;
+            }
+
+            Uri withoutQuery = this.removeQueryAndFragment(uri);
+            if (withoutQuery.toString().equals(wechatRedirectURI.toString())) {
+                Intent intent = new Intent(wechatRedirectURIIntentAction);
+                intent.setPackage(this.activity.getApplicationContext().getPackageName());
+                intent.putExtra("uri", uri.toString());
+                this.activity.getApplicationContext().sendBroadcast(intent);
+                return true;
+            }
+
+            return false;
+        }
+
         private Uri removeQueryAndFragment(Uri uri) {
             return uri.buildUpon().query(null).fragment(null).build();
         }