Merge pull request #42 from meierrap/add-android-image-support

Add Android support to add image as background overlay
killserver · Jun 2, 2024 · 1463f0b · 1463f0b
2 parents 3c5302e + ed0a991
commit 1463f0b
Show file tree

Hide file tree

Showing 2 changed files with 103 additions and 9 deletions.
diff --git a/android/src/main/java/com/killserver/screenshotprev/RNScreenshotPreventModule.java b/android/src/main/java/com/killserver/screenshotprev/RNScreenshotPreventModule.java
@@ -1,21 +1,33 @@
 
 package com.killserver.screenshotprev;
 
+import android.app.Activity;
+import android.graphics.Bitmap;
+import android.graphics.BitmapFactory;
+import android.graphics.Color;
+import android.view.ViewGroup;
 import android.view.WindowManager;
+import android.widget.RelativeLayout;
+import android.widget.ImageView;
 
-import android.app.Activity;
 import com.facebook.react.bridge.ReactApplicationContext;
+import com.facebook.react.bridge.LifecycleEventListener;
 import com.facebook.react.bridge.ReactContextBaseJavaModule;
 import com.facebook.react.bridge.ReactMethod;
-import com.facebook.react.bridge.Callback;
 
-public class RNScreenshotPreventModule extends ReactContextBaseJavaModule {
+import java.io.IOException;
+import java.net.URL;
+
+public class RNScreenshotPreventModule extends ReactContextBaseJavaModule implements LifecycleEventListener {
 
   private final ReactApplicationContext reactContext;
+  private RelativeLayout overlayLayout;
+  private boolean secureFlagWasSet;
 
   public RNScreenshotPreventModule(ReactApplicationContext reactContext) {
     super(reactContext);
     this.reactContext = reactContext;
+    this.reactContext.addLifecycleEventListener(this);
   }
 
   @Override
@@ -48,10 +60,13 @@ public void run() {
   }
 
   @ReactMethod
-  public void enableSecureView() {
+  public void enableSecureView(String imagePath) {
     if (this.reactContext.hasCurrentActivity()) {
       final Activity activity = this.reactContext.getCurrentActivity();
       if (activity != null) {
+        if (overlayLayout == null) {
+          createOverlay(activity, imagePath);
+        }
         activity.runOnUiThread(new Runnable() {
           @Override
           public void run() {
@@ -70,21 +85,99 @@ public void disableSecureView() {
         activity.runOnUiThread(new Runnable() {
           @Override
           public void run() {
+            if (overlayLayout != null) {
+              ViewGroup rootView = (ViewGroup) activity.getWindow().getDecorView().getRootView();
+              rootView.removeView(overlayLayout);
+              overlayLayout = null;
+            }
             activity.getWindow().clearFlags(WindowManager.LayoutParams.FLAG_SECURE);
           }
         });
       }
     }
   }
 
-  // Required for rn built in EventEmitter Calls.
-  @ReactMethod
-  public void addListener(String eventName) {
+  private void createOverlay(Activity activity, String imagePath) {
+    overlayLayout = new RelativeLayout(activity);
+    overlayLayout.setBackgroundColor(Color.parseColor("#FFFFFF"));
+
+    // Create an ImageView
+    ImageView imageView = new ImageView(activity);
+    RelativeLayout.LayoutParams imageParams = new RelativeLayout.LayoutParams(
+      RelativeLayout.LayoutParams.MATCH_PARENT,
+      RelativeLayout.LayoutParams.WRAP_CONTENT);
+    imageParams.addRule(RelativeLayout.CENTER_IN_PARENT, RelativeLayout.TRUE);
 
+    imageView.setLayoutParams(imageParams);
+
+    // Set image resource
+    Bitmap bitmap = decodeImageUrl(imagePath);
+
+    if (bitmap != null) {
+      int imageHeight = (int)(bitmap.getHeight() * ((float) activity.getResources().getDisplayMetrics().widthPixels / bitmap.getWidth()));
+      Bitmap scaledBitmap = Bitmap.createScaledBitmap(bitmap, activity.getResources().getDisplayMetrics().widthPixels, imageHeight, true);
+      imageView.setImageBitmap(scaledBitmap);
+    }
+
+    overlayLayout.addView(imageView);
   }
 
-  @ReactMethod
-  public void removeListeners(Integer count) {
+  @Override
+  public void onHostResume() {
+    Activity currentActivity = this.reactContext.getCurrentActivity();
+    if (currentActivity != null && overlayLayout != null) {
+      currentActivity.runOnUiThread(new Runnable() {
+        @Override
+        public void run() {
+          ViewGroup rootView = (ViewGroup) currentActivity.getWindow().getDecorView().getRootView();
+          rootView.removeView(overlayLayout);
+          if (secureFlagWasSet) {
+            currentActivity.getWindow().setFlags(WindowManager.LayoutParams.FLAG_SECURE, WindowManager.LayoutParams.FLAG_SECURE);
+            secureFlagWasSet = false;
+          }
+        }
+      });
+    }
+  }
+
+  @Override
+  public void onHostPause() {
+    Activity currentActivity = this.reactContext.getCurrentActivity();
+    if (currentActivity != null && overlayLayout != null) {
+      currentActivity.runOnUiThread(new Runnable() {
+        @Override
+        public void run() {
+          ViewGroup rootView = (ViewGroup) currentActivity.getWindow().getDecorView().getRootView();
+          RelativeLayout.LayoutParams layoutParams = new RelativeLayout.LayoutParams(
+            ViewGroup.LayoutParams.MATCH_PARENT,
+            ViewGroup.LayoutParams.MATCH_PARENT);
+          rootView.addView(overlayLayout, layoutParams);
+
+          int flags = currentActivity.getWindow().getAttributes().flags;
+          if ((flags & WindowManager.LayoutParams.FLAG_SECURE) != 0) {
+            currentActivity.getWindow().clearFlags(WindowManager.LayoutParams.FLAG_SECURE);
+            secureFlagWasSet = true;
+          } else {
+            secureFlagWasSet = false;
+          }
+        }
+      });
+    }
+  }
 
+  @Override
+  public void onHostDestroy() {
+    // Cleanup if needed
+  }
+
+  private Bitmap decodeImageUrl(String imagePath) {
+    try {
+      URL imageUrl = new URL(imagePath);
+      Bitmap bitmap = BitmapFactory.decodeStream(imageUrl.openConnection().getInputStream());
+      return bitmap;
+    } catch (IOException e) {
+      e.printStackTrace();
+      return null;
+    }
   }
 }
diff --git a/android/src/main/java/com/killserver/screenshotprev/RNScreenshotPreventPackage.java b/android/src/main/java/com/killserver/screenshotprev/RNScreenshotPreventPackage.java
@@ -10,6 +10,7 @@
 import com.facebook.react.bridge.ReactApplicationContext;
 import com.facebook.react.uimanager.ViewManager;
 import com.facebook.react.bridge.JavaScriptModule;
+
 public class RNScreenshotPreventPackage implements ReactPackage {
     @Override
     public List<NativeModule> createNativeModules(ReactApplicationContext reactContext) {