From eefcb3d75810d327fd70b53489cbb38551f4a5a0 Mon Sep 17 00:00:00 2001
From: althafvly <althafvly@gmail.com>
Date: Thu, 26 Jan 2023 19:24:30 +0530
Subject: [PATCH] Updater: Check for available space on install/update/resume

- Display required space in MB. GB is confusing because its a rounded figure.
- Minimum required space update file * 2
- For resume, subtract the downloaded file size.

Change-Id: Ia7beb22869374e5885100c0eb72e108275e9ada9
---
 res/values/strings.xml                        |  2 +-
 .../lineageos/updater/UpdatesListAdapter.java | 61 ++++++++++++++-----
 2 files changed, 48 insertions(+), 15 deletions(-)

diff --git a/res/values/strings.xml b/res/values/strings.xml
index 00b71278..28c92f44 100644
--- a/res/values/strings.xml
+++ b/res/values/strings.xml
@@ -62,7 +62,7 @@
     <string name="dialog_battery_low_title">Low battery</string>
     <string name="dialog_battery_low_message_pct">The battery level is too low, you need at least <xliff:g id="percent_discharging">%1$d</xliff:g>%% of the battery to continue, <xliff:g id="percent_charging">%2$d</xliff:g>%% if charging.</string>
     <string name="dialog_free_space_low_title">Free up space</string>
-    <string name="dialog_free_space_low_message_pct">The update cannot be completed because there is insufficient free space; at least <xliff:g id="needed_free_space">%1$s</xliff:g> of free space is required. Please clear your internal storage before proceeding.</string>
+    <string name="dialog_free_space_low_message_pct">There is insufficient space available to successfully complete the update. You must free up another <xliff:g id="needed_free_space">%1$s</xliff:g> of space in the internal storage before proceeding.</string>
 
     <string name="reboot">Reboot</string>
 
diff --git a/src/org/lineageos/updater/UpdatesListAdapter.java b/src/org/lineageos/updater/UpdatesListAdapter.java
index 04ad0028..0894f2aa 100644
--- a/src/org/lineageos/updater/UpdatesListAdapter.java
+++ b/src/org/lineageos/updater/UpdatesListAdapter.java
@@ -57,8 +57,10 @@ import org.lineageos.updater.misc.Utils;
 import org.lineageos.updater.model.UpdateInfo;
 import org.lineageos.updater.model.UpdateStatus;
 
+import java.io.File;
 import java.io.IOException;
 import java.text.DateFormat;
+import java.text.DecimalFormat;
 import java.text.NumberFormat;
 import java.util.List;
 
@@ -307,6 +309,32 @@ public class UpdatesListAdapter extends RecyclerView.Adapter<UpdatesListAdapter.
         notifyItemRangeChanged(position, getItemCount());
     }
 
+    private AlertDialog.Builder getSpaceDialog(UpdateInfo update) {
+        Resources resources = mActivity.getResources();
+        File updateFile = update.getFile();
+        long requiredSpace = update.getFileSize() * 2;
+        long availableFreeSpace = Utils.availableFreeSpace();
+
+        // Check for space depending on the downloaded file size on resume
+        if (updateFile != null && updateFile.exists() && updateFile.length() > 0 &&
+                updateFile.length() < update.getFileSize()) {
+            requiredSpace -= updateFile.length();
+        }
+
+        if (availableFreeSpace < requiredSpace) {
+            // Not enough space to download file
+            double spaceNeeded = (requiredSpace - availableFreeSpace) / (1024.0 * 1024.0);
+            String message = resources.getString(R.string.dialog_free_space_low_message_pct,
+                    new DecimalFormat("#MB").format(spaceNeeded));
+            return new AlertDialog.Builder(mActivity)
+                    .setTitle(R.string.dialog_free_space_low_title)
+                    .setMessage(message)
+                    .setPositiveButton(android.R.string.ok, null);
+        }
+
+        return null;
+    }
+
     private void startDownloadWithWarning(final String downloadId) {
         SharedPreferences preferences = PreferenceManager.getDefaultSharedPreferences(mActivity);
         boolean warn = preferences.getBoolean(Constants.PREF_MOBILE_DATA_WARNING, true);
@@ -344,7 +372,15 @@ public class UpdatesListAdapter extends RecyclerView.Adapter<UpdatesListAdapter.
             case DOWNLOAD:
                 button.setText(R.string.action_download);
                 button.setEnabled(enabled);
-                clickListener = enabled ? view -> startDownloadWithWarning(downloadId) : null;
+                clickListener = enabled ? view -> {
+                    AlertDialog.Builder freeSpaceDialog = getSpaceDialog(
+                            mUpdaterController.getUpdate(downloadId));
+                    if (freeSpaceDialog != null) {
+                        freeSpaceDialog.show();
+                    } else {
+                        startDownloadWithWarning(downloadId);
+                    }
+                } : null;
                 break;
             case PAUSE:
                 button.setText(R.string.action_pause);
@@ -360,7 +396,12 @@ public class UpdatesListAdapter extends RecyclerView.Adapter<UpdatesListAdapter.
                         update.getFile().length() == update.getFileSize();
                 clickListener = enabled ? view -> {
                     if (canInstall) {
-                        mUpdaterController.resumeDownload(downloadId);
+                        AlertDialog.Builder freeSpaceDialog = getSpaceDialog(update);
+                        if (freeSpaceDialog != null) {
+                            freeSpaceDialog.show();
+                        } else {
+                            mUpdaterController.resumeDownload(downloadId);
+                        }
                     } else {
                         mActivity.showSnackbar(R.string.snack_update_not_installable,
                                 Snackbar.LENGTH_LONG);
@@ -376,7 +417,10 @@ public class UpdatesListAdapter extends RecyclerView.Adapter<UpdatesListAdapter.
                 clickListener = enabled ? view -> {
                     if (canInstall) {
                         AlertDialog.Builder installDialog = getInstallDialog(downloadId);
-                        if (installDialog != null) {
+                        AlertDialog.Builder freeSpaceDialog = getSpaceDialog(update);
+                        if (freeSpaceDialog != null) {
+                            freeSpaceDialog.show();
+                        } else if (installDialog != null) {
                             installDialog.show();
                         }
                     } else {
@@ -472,17 +516,6 @@ public class UpdatesListAdapter extends RecyclerView.Adapter<UpdatesListAdapter.
             return null;
         }
 
-        long neededSpace = update.getFileSize() * 2;
-        final boolean hasFreeSpace = Utils.availableFreeSpace() >= neededSpace;
-        if (!hasFreeSpace) {
-            String message = resources.getString(R.string.dialog_free_space_low_message_pct,
-                    Utils.getFileSize(neededSpace));
-            return new AlertDialog.Builder(mActivity)
-                    .setTitle(R.string.dialog_free_space_low_title)
-                    .setMessage(message)
-                    .setPositiveButton(android.R.string.ok, null);
-        }
-
         String buildDate = StringGenerator.getDateLocalizedUTC(mActivity,
                 DateFormat.MEDIUM, update.getTimestamp());
         String buildInfoText = mActivity.getString(R.string.list_build_version_date_e,
-- 
GitLab