From 6534932b20050ee9c44d09f2a6840b1c39a014d2 Mon Sep 17 00:00:00 2001
From: Gabriele M <moto.falcon.git@gmail.com>
Date: Sat, 8 Jul 2017 01:07:59 +0200
Subject: [PATCH] Show "verify" for complete INCOMPLETE updates

INCOMPLETE is used for any update that isn't verified, even if the
file has been completed. This change is mostly needed to support
imported downloads, which we take as complete, but unverified.
---
 res/values/strings.xml                        |  1 +
 .../lineageos/updater/UpdatesListAdapter.java | 19 +++++++++++++++++--
 2 files changed, 18 insertions(+), 2 deletions(-)

diff --git a/res/values/strings.xml b/res/values/strings.xml
index cec618d9..f3abd16c 100644
--- a/res/values/strings.xml
+++ b/res/values/strings.xml
@@ -37,6 +37,7 @@
     <string name="download_button">Download</string>
     <string name="pause_button">Pause</string>
     <string name="resume_button">Resume</string>
+    <string name="verify_button">Verify</string>
     <string name="cancel_button">Delete</string>
     <string name="install_button">Install</string>
 
diff --git a/src/org/lineageos/updater/UpdatesListAdapter.java b/src/org/lineageos/updater/UpdatesListAdapter.java
index 3324ec7c..a6905343 100644
--- a/src/org/lineageos/updater/UpdatesListAdapter.java
+++ b/src/org/lineageos/updater/UpdatesListAdapter.java
@@ -43,6 +43,7 @@ public class UpdatesListAdapter extends RecyclerView.Adapter<UpdatesListAdapter.
         DOWNLOAD,
         PAUSE,
         RESUME,
+        VERIFY,
         CANCEL,
         INSTALL
     }
@@ -119,8 +120,12 @@ public class UpdatesListAdapter extends RecyclerView.Adapter<UpdatesListAdapter.
                     Utils.canInstall(update) && !mUpdaterController.isVerifyingUpdate();
             int persistentStatus = update.getPersistentStatus();
             if (persistentStatus == UpdateStatus.Persistent.INCOMPLETE) {
-                setButtonAction(viewHolder.mButton1, Action.RESUME, downloadId,
-                        enabled && update.getAvailableOnline());
+                if (update.getFile().length() == update.getFileSize()) {
+                    setButtonAction(viewHolder.mButton1, Action.VERIFY, downloadId, enabled);
+                } else {
+                    setButtonAction(viewHolder.mButton1, Action.RESUME, downloadId,
+                            enabled && update.getAvailableOnline());
+                }
                 setButtonAction(viewHolder.mButton2, Action.CANCEL, downloadId, enabled);
             } else if (persistentStatus == UpdateStatus.Persistent.VERIFIED) {
                 setButtonAction(viewHolder.mButton1, Action.INSTALL, downloadId, enabled);
@@ -185,6 +190,16 @@ public class UpdatesListAdapter extends RecyclerView.Adapter<UpdatesListAdapter.
                     }
                 });
                 break;
+            case VERIFY:
+                button.setText(R.string.verify_button);
+                button.setEnabled(enabled);
+                button.setOnClickListener(!enabled ? null : new View.OnClickListener() {
+                    @Override
+                    public void onClick(View view) {
+                        mUpdaterController.resumeDownload(downloadId);
+                    }
+                });
+                break;
             case CANCEL:
                 button.setText(R.string.cancel_button);
                 button.setEnabled(enabled);