PawletOS/lineageos_updater
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Updater: Allow to suspend A/B updates

Browse Source 
Change-Id: I0387fd491a07a2214e4331a2cfe25988e0016a61
This commit is contained in:
Luca Stefani
2018-07-10 19:54:20 +02:00
committed by Rashed Abdel-Tawab
parent 6a012ef541
commit 50d0fd5291
4 changed files with 127 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
res/values/strings.xml
View File

@@ -42,6 +42,7 @@
<string name="download_completed_notification">Download completed</string>
<string name="download_starting_notification">Starting download</string>
<string name="update_failed_notification">Update failed</string>
<string name="installation_suspended_notification">Installation suspended</string>
<string name="new_updates_found_title">New updates</string>
@@ -49,6 +50,7 @@
<string name="pause_button">Pause</string>
<string name="resume_button">Resume</string>
<string name="suspend_button">Suspend</string>
<string name="installing_update">Installing update package</string>
<string name="installing_update_error">Install error</string>

72
src/org/lineageos/updater/controller/ABUpdateInstaller.java
View File

@@ -43,6 +43,7 @@ class ABUpdateInstaller {
private static final String TAG = "ABUpdateInstaller";
private static final String PREF_INSTALLING_AB_ID = "installing_ab_id";
private static final String PREF_INSTALLING_SUSPENDED_AB_ID = "installing_suspended_ab_id";
private static ABUpdateInstaller sInstance = null;
@@ -53,6 +54,9 @@ class ABUpdateInstaller {
private UpdateEngine mUpdateEngine;
private boolean mBound;
private boolean mFinalizing;
private int mProgress;
private final UpdateEngineCallback mUpdateEngineCallback = new UpdateEngineCallback() {
@Override
@@ -71,10 +75,10 @@ class ABUpdateInstaller {
update.setStatus(UpdateStatus.INSTALLING);
mUpdaterController.notifyUpdateChange(mDownloadId);
}
int progress = Math.round(percent * 100);
mUpdaterController.getActualUpdate(mDownloadId).setInstallProgress(progress);
boolean finalizing = status == UpdateEngine.UpdateStatusConstants.FINALIZING;
mUpdaterController.getActualUpdate(mDownloadId).setFinalizing(finalizing);
mProgress = Math.round(percent * 100);
mUpdaterController.getActualUpdate(mDownloadId).setInstallProgress(mProgress);
mFinalizing = status == UpdateEngine.UpdateStatusConstants.FINALIZING;
mUpdaterController.getActualUpdate(mDownloadId).setFinalizing(mFinalizing);
mUpdaterController.notifyInstallProgress(mDownloadId);
}
break;
@@ -127,6 +131,16 @@ class ABUpdateInstaller {
TextUtils.equals(pref.getString(Constants.PREF_NEEDS_REBOOT_ID, null), downloadId);
}
static synchronized boolean isInstallingUpdateSuspended(Context context) {
SharedPreferences pref = PreferenceManager.getDefaultSharedPreferences(context);
return pref.getString(ABUpdateInstaller.PREF_INSTALLING_SUSPENDED_AB_ID, null) != null;
}
static synchronized boolean isInstallingUpdateSuspended(Context context, String downloadId) {
SharedPreferences pref = PreferenceManager.getDefaultSharedPreferences(context);
return downloadId.equals(pref.getString(ABUpdateInstaller.PREF_INSTALLING_SUSPENDED_AB_ID, null));
}
static synchronized boolean isWaitingForReboot(Context context, String downloadId) {
String waitingId = PreferenceManager.getDefaultSharedPreferences(context)
.getString(Constants.PREF_NEEDS_REBOOT_ID, null);
@@ -273,4 +287,54 @@ class ABUpdateInstaller {
public void setPerformanceMode(boolean enable) {
mUpdateEngine.setPerformanceMode(enable);
}
public boolean suspend() {
if (!isInstallingUpdate(mContext)) {
Log.e(TAG, "cancel: Not installing any update");
return false;
}
if (!mBound) {
Log.e(TAG, "Not connected to update engine");
return false;
}
mUpdateEngine.suspend();
mUpdaterController.getActualUpdate(mDownloadId)
.setStatus(UpdateStatus.INSTALLATION_SUSPENDED);
mUpdaterController.notifyUpdateChange(mDownloadId);
PreferenceManager.getDefaultSharedPreferences(mContext).edit()
.putString(PREF_INSTALLING_SUSPENDED_AB_ID, mDownloadId)
.apply();
return true;
}
public boolean resume() {
if (!isInstallingUpdateSuspended(mContext)) {
Log.e(TAG, "cancel: No update is suspended");
return false;
}
if (!mBound) {
Log.e(TAG, "Not connected to update engine");
return false;
}
mUpdateEngine.resume();
mUpdaterController.getActualUpdate(mDownloadId).setStatus(UpdateStatus.INSTALLING);
mUpdaterController.notifyUpdateChange(mDownloadId);
mUpdaterController.getActualUpdate(mDownloadId).setInstallProgress(mProgress);
mUpdaterController.getActualUpdate(mDownloadId).setFinalizing(mFinalizing);
mUpdaterController.notifyInstallProgress(mDownloadId);
PreferenceManager.getDefaultSharedPreferences(mContext).edit()
.remove(PREF_INSTALLING_SUSPENDED_AB_ID)
.apply();
return true;
}
}

55
src/org/lineageos/updater/controller/UpdaterService.java
View File

@@ -54,6 +54,9 @@ public class UpdaterService extends Service {
public static final String ACTION_INSTALL_UPDATE = "action_install_update";
public static final String ACTION_INSTALL_STOP = "action_install_stop";
public static final String ACTION_INSTALL_SUSPEND = "action_install_suspend";
public static final String ACTION_INSTALL_RESUME = "action_install_resume";
private static final String ONGOING_NOTIFICATION_CHANNEL =
"ongoing_notification_channel";
@@ -212,6 +215,20 @@ public class UpdaterService extends Service {
installer.reconnect();
installer.cancel();
}
} else if (ACTION_INSTALL_SUSPEND.equals(intent.getAction())) {
if (ABUpdateInstaller.isInstallingUpdate(this)) {
ABUpdateInstaller installer = ABUpdateInstaller.getInstance(this,
mUpdaterController);
installer.reconnect();
installer.suspend();
}
} else if (ACTION_INSTALL_RESUME.equals(intent.getAction())) {
if (ABUpdateInstaller.isInstallingUpdateSuspended(this)) {
ABUpdateInstaller installer = ABUpdateInstaller.getInstance(this,
mUpdaterController);
installer.reconnect();
installer.resume();
}
}
return ABUpdateInstaller.isInstallingUpdate(this) ? START_STICKY : START_NOT_STICKY;
}
@@ -355,6 +372,11 @@ public class UpdaterService extends Service {
getString(R.string.dialog_prepare_zip_message) :
getString(R.string.installing_update);
mNotificationStyle.bigText(text);
if (ABUpdateInstaller.isInstallingUpdate(this)) {
mNotificationBuilder.addAction(android.R.drawable.ic_media_pause,
getString(R.string.suspend_button),
getSuspendInstallationPendingIntent());
}
mNotificationBuilder.setTicker(text);
mNotificationBuilder.setOngoing(true);
mNotificationBuilder.setAutoCancel(false);
@@ -364,6 +386,7 @@ public class UpdaterService extends Service {
}
case INSTALLED: {
stopForeground(STOP_FOREGROUND_DETACH);
mNotificationBuilder.mActions.clear();
mNotificationBuilder.setStyle(null);
mNotificationBuilder.setSmallIcon(R.drawable.ic_system_update);
mNotificationBuilder.setProgress(0, 0, false);
@@ -398,6 +421,25 @@ public class UpdaterService extends Service {
tryStopSelf();
break;
}
case INSTALLATION_SUSPENDED: {
stopForeground(STOP_FOREGROUND_DETACH);
// In case we pause before the first progress update
mNotificationBuilder.setProgress(100, update.getProgress(), false);
mNotificationBuilder.mActions.clear();
String text = getString(R.string.installation_suspended_notification);
mNotificationStyle.bigText(text);
mNotificationBuilder.setStyle(mNotificationStyle);
mNotificationBuilder.setSmallIcon(R.drawable.ic_pause);
mNotificationBuilder.addAction(android.R.drawable.ic_media_play,
getString(R.string.resume_button),
getResumeInstallationPendingIntent());
mNotificationBuilder.setTicker(text);
mNotificationBuilder.setOngoing(true);
mNotificationBuilder.setAutoCancel(false);
mNotificationManager.notify(NOTIFICATION_ID, mNotificationBuilder.build());
tryStopSelf();
break;
}
}
}
@@ -463,4 +505,17 @@ public class UpdaterService extends Service {
return PendingIntent.getBroadcast(this, 0, intent, PendingIntent.FLAG_UPDATE_CURRENT);
}
private PendingIntent getSuspendInstallationPendingIntent() {
final Intent intent = new Intent(this, UpdaterService.class);
intent.setAction(ACTION_INSTALL_SUSPEND);
return PendingIntent.getService(this, 0, intent,
PendingIntent.FLAG_UPDATE_CURRENT);
}
private PendingIntent getResumeInstallationPendingIntent() {
final Intent intent = new Intent(this, UpdaterService.class);
intent.setAction(ACTION_INSTALL_RESUME);
return PendingIntent.getService(this, 0, intent,
PendingIntent.FLAG_UPDATE_CURRENT);
}
}

3
src/org/lineageos/updater/model/UpdateStatus.java
View File

@@ -29,7 +29,8 @@ public enum UpdateStatus {
INSTALLING,
INSTALLED,
INSTALLATION_FAILED,
INSTALLATION_CANCELLED;
INSTALLATION_CANCELLED,
INSTALLATION_SUSPENDED;
public static final class Persistent {
public static final int UNKNOWN = 0;