Fix download of not fully downloaded media

Author: Dev4Mod

app/src/main/java/com/wmods/wppenhacer/xposed/core/components/FMessageWpp.java

@@ -4,6 +4,7 @@
 import androidx.annotation.Nullable;
 
 import com.wmods.wppenhacer.xposed.core.WppCore;
+import com.wmods.wppenhacer.xposed.core.db.MessageStore;
 import com.wmods.wppenhacer.xposed.core.devkit.Unobfuscator;
 import com.wmods.wppenhacer.xposed.utils.ReflectionUtils;
 
@@ -163,8 +164,12 @@ public File getMediaFile() {
                 if (field.getType().isPrimitive()) continue;
                 var fileField = ReflectionUtils.getFieldByType(field.getType(), File.class);
                 if (fileField != null) {
-                    var mediaFile = ReflectionUtils.getObjectField(field, fmessage);
-                    return (File) fileField.get(mediaFile);
+                    var mediaObject = ReflectionUtils.getObjectField(field, fmessage);
+                    var mediaFile = (File) fileField.get(mediaObject);
+                    if (mediaFile != null) return mediaFile;
+                    var filePath = MessageStore.getInstance().getMediaFromID(getRowId());
+                    if (filePath == null) return null;
+                    return new File(filePath);
                 }
             }
         } catch (Exception e) {

app/src/main/java/com/wmods/wppenhacer/xposed/features/media/DownloadViewOnce.java

@@ -73,6 +73,10 @@ protected void afterHookedMethod(MethodHookParam param) throws Throwable {
                     item.setOnMenuItemClickListener(item1 -> {
                         try {
                             var file = fMessage.getMediaFile();
+                            if (file == null) {
+                                Utils.showToast(Utils.getApplication().getString(ResId.string.download_not_available), 1);
+                                return true;
+                            }
                             downloadFile(fMessage.getKey().remoteJid, file);
                         } catch (Exception e) {
                             Utils.showToast(e.getMessage(), Toast.LENGTH_LONG);
@@ -99,6 +103,10 @@ protected void afterHookedMethod(MethodHookParam param) throws Throwable {
                                     var fmessageObj = WppCore.getFMessageFromKey(keyMessageObj);
                                     var fmessage = new FMessageWpp(fmessageObj);
                                     var file = fmessage.getMediaFile();
+                                    if (file == null) {
+                                        Utils.showToast(Utils.getApplication().getString(ResId.string.download_not_available), 1);
+                                        return;
+                                    }
                                     var userJid = fmessage.getKey().remoteJid;
                                     try {
                                         downloadFile(userJid, file);

app/src/main/java/com/wmods/wppenhacer/xposed/features/media/StatusDownload.java

@@ -25,7 +25,6 @@
 import java.util.Collections;
 
 import de.robv.android.xposed.XSharedPreferences;
-import de.robv.android.xposed.XposedHelpers;
 
 public class StatusDownload extends Feature {
 
@@ -89,6 +88,10 @@ private void sharedStatus(FMessageWpp fMessageWpp) {
                 return;
             }
             var file = fMessageWpp.getMediaFile();
+            if (file == null) {
+                Utils.showToast(Utils.getApplication().getString(ResId.string.download_not_available), 1);
+                return;
+            }
             Intent intent = new Intent();
             var clazz = Unobfuscator.getClassByName("MediaComposerActivity", classLoader);
             intent.setClassName(Utils.getApplication().getPackageName(), clazz.getName());
@@ -104,6 +107,10 @@ private void sharedStatus(FMessageWpp fMessageWpp) {
     private void downloadFile(FMessageWpp fMessage) {
         try {
             var file = fMessage.getMediaFile();
+            if (file == null) {
+                Utils.showToast(Utils.getApplication().getString(ResId.string.download_not_available), 1);
+                return;
+            }
             var userJid = fMessage.getUserJid();
             var fileType = file.getName().substring(file.getName().lastIndexOf(".") + 1);
             var destination = getStatusDestination(file);

app/src/main/java/com/wmods/wppenhacer/xposed/utils/ResId.java

@@ -103,6 +103,7 @@ public static class string {
         public static int download_folder_permission;
         public static int no_contact_with_custom_privacy;
         public static int select_contacts;
+        public static int download_not_available;
     }
 
     public static class array {

app/src/main/res/values/strings.xml

@@ -415,4 +415,5 @@
     <string name="disable_the_proximity_sensor_sum">Disable Screen off on proximity sensor to the whole WhatsApp</string>
     <string name="disable_status_in_the_profile_photo">Disable Status in the profile photo</string>
     <string name="disable_status_in_the_profile_photo_sum">Disables the circle that appears in the profile photo of each contact when there is new status</string>
+    <string name="download_not_available">Download not available or media has not been fully downloaded</string>
 </resources>