From d3ccf82ae1bd0f7958d9b0af5ca3fc2354ae62e2 Mon Sep 17 00:00:00 2001
From: Jerry Yan <792602257@qq.com>
Date: Sat, 14 Mar 2020 18:41:30 +0800
Subject: [PATCH] =?UTF-8?q?=E6=B7=BB=E5=8A=A0=E9=83=A8=E5=88=86=E5=A4=84?=
 =?UTF-8?q?=E7=90=86?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 .../RN/A/VersionUpgrade/UpgradeModule.java    | 22 ++++++++++++++-----
 1 file changed, 16 insertions(+), 6 deletions(-)

diff --git a/android/src/main/java/top/jerryyan/RN/A/VersionUpgrade/UpgradeModule.java b/android/src/main/java/top/jerryyan/RN/A/VersionUpgrade/UpgradeModule.java
index 63f2ee8..d9a7f77 100644
--- a/android/src/main/java/top/jerryyan/RN/A/VersionUpgrade/UpgradeModule.java
+++ b/android/src/main/java/top/jerryyan/RN/A/VersionUpgrade/UpgradeModule.java
@@ -158,11 +158,18 @@ public class UpgradeModule extends ReactContextBaseJavaModule {
                             dialog.setProgress((int)((1000.0f*current)/total));
                         }
                         fileOutputStream.flush();
-                        Message message = new Message();
+                        final Message message = new Message();
                         message.what = 0;
                         message.obj = file.toURI().toString();
                         handler.sendMessage(message);
-                        dialog.dismiss();
+                        dialog.setButton(ProgressDialog.BUTTON_POSITIVE, "更新",
+                                new DialogInterface.OnClickListener() {
+                                    @Override
+                                    public void onClick(DialogInterface dialog, int which) {
+                                        handler.sendMessage(message);
+                                    }
+                                }
+                        );
                     } catch (IOException e) {
                         Log.e("REQUEST FAILED", "AAA", e);
                     } finally {
@@ -257,10 +264,13 @@ final class DoInstallHandler extends Handler {
 
     @Override
     public void handleMessage(Message message) {
-        if(!module._hasPermissionToInstall()) module._requestInstallPermission();
-        if(message.what == 0){
-            String fileUri = (String)message.obj;
-            module.installPackage(fileUri);
+        if(!module._hasPermissionToInstall()) {
+            module._requestInstallPermission();
+        }else{
+            if(message.what == 0){
+                String fileUri = (String)message.obj;
+                module.installPackage(fileUri);
+            }
         }
     }
 }
\ No newline at end of file