Skip to content

Commit

Permalink
GH-1060 update tweaks
Browse files Browse the repository at this point in the history
* download to multimc folder hierarchy
* use rename, not copy
* keep backup after update
* clean previous backup before update
* it's not 'copy', it's 'replace'
  • Loading branch information
peterix committed Jun 9, 2015
1 parent 22c5ced commit 15b7c30
Show file tree
Hide file tree
Showing 6 changed files with 58 additions and 40 deletions.
3 changes: 2 additions & 1 deletion application/MainWindow.cpp
Original file line number Diff line number Diff line change
Expand Up @@ -988,7 +988,8 @@ void MainWindow::downloadUpdates(GoUpdate::Status status)
ProgressDialog updateDlg(this);
status.rootPath = MMC->rootPath;

GoUpdate::DownloadTask updateTask(status, &updateDlg);
auto dlPath = PathCombine(MMC->root(), "update", "XXXXXX");
GoUpdate::DownloadTask updateTask(status, dlPath, &updateDlg);
// If the task succeeds, install the updates.
if (updateDlg.exec(&updateTask))
{
Expand Down
78 changes: 45 additions & 33 deletions application/MultiMC.cpp
Original file line number Diff line number Diff line change
Expand Up @@ -662,49 +662,55 @@ void MultiMC::installUpdates(const QString updateFilesDir, GoUpdate::OperationLi
#error Unsupported operating system.
#endif

QString backupPath = PathCombine(root(), "update-backup");
QString trashPath = PathCombine(root(), "update-trash");
if(!ensureFolderPathExists(backupPath))
QString backupPath = PathCombine(root(), "update", "backup");

// clean up the backup folder. it should be empty before we start
if(!deletePath(backupPath))
{
qWarning() << "couldn't create folder" << backupPath;
return;
qWarning() << "couldn't remove previous backup folder" << backupPath;
}
if(!ensureFolderPathExists(trashPath))
// and it should exist.
if(!ensureFolderPathExists(backupPath))
{
qWarning() << "couldn't create folder" << trashPath;
qWarning() << "couldn't create folder" << backupPath;
return;
}

struct BackupEntry
{
QString orig;
QString backup;
};
enum Failure
{
Copy,
Replace,
Delete,
Start,
Nothing
} failedOperationType = Nothing;

QString failedFile;

QList <BackupEntry> backups;
QList <BackupEntry> trashcan;

// perform the update operations
for(auto op: operations)
{
switch(op.type)
{
case GoUpdate::Operation::OP_COPY:
// replace = move original out to backup, if it exists, move the new file in its place
case GoUpdate::Operation::OP_REPLACE:
{
QFileInfo replaced (PathCombine(root(), op.dest));
if(replaced.exists())
{
QString backupFilePath = PathCombine(backupPath, replaced.fileName());
QString backupName = op.dest;
backupName.replace('/', '_');
QString backupFilePath = PathCombine(backupPath, backupName);
if(!QFile::rename(replaced.absoluteFilePath(), backupFilePath))
{
qWarning() << "Couldn't rename:" << replaced.absoluteFilePath() << "to" << backupFilePath;
failedOperationType = Copy;
qWarning() << "Couldn't move:" << replaced.absoluteFilePath() << "to" << backupFilePath;
failedOperationType = Replace;
failedFile = op.dest;
goto FAILED;
}
Expand All @@ -713,17 +719,26 @@ void MultiMC::installUpdates(const QString updateFilesDir, GoUpdate::OperationLi
be.backup = backupFilePath;
backups.append(be);
}
// FIXME: use rename instead.
if(!QFile::copy(op.file, replaced.absoluteFilePath()))
// make sure the folder we are putting this into exists
if(!ensureFilePathExists(replaced.absoluteFilePath()))
{
qWarning() << "REPLACE: Couldn't create folder:" << replaced.absoluteFilePath();
failedOperationType = Replace;
failedFile = op.dest;
goto FAILED;
}
// now move the new file in
if(!QFile::rename(op.file, replaced.absoluteFilePath()))
{
qWarning() << "CPY: Couldn't copy:" << op.file << "to" << replaced.absoluteFilePath();
failedOperationType = Copy;
qWarning() << "REPLACE: Couldn't move:" << op.file << "to" << replaced.absoluteFilePath();
failedOperationType = Replace;
failedFile = op.dest;
goto FAILED;
}
QFile::setPermissions(replaced.absoluteFilePath(), unixModeToPermissions(op.mode));
}
break;
// delete = move original to backup
case GoUpdate::Operation::OP_DELETE:
{
QString trashFilePath = PathCombine(backupPath, op.file);
Expand All @@ -732,7 +747,7 @@ void MultiMC::installUpdates(const QString updateFilesDir, GoUpdate::OperationLi
{
if(!QFile::rename(origFilePath, trashFilePath))
{
qWarning() << "DEL: Couldn't move:" << op.file << "to" << trashFilePath;
qWarning() << "DELETE: Couldn't move:" << op.file << "to" << trashFilePath;
failedFile = op.file;
failedOperationType = Delete;
goto FAILED;
Expand All @@ -758,15 +773,6 @@ void MultiMC::installUpdates(const QString updateFilesDir, GoUpdate::OperationLi
failedOperationType = Start;
goto FAILED;
}
// now clean up the backed up stuff.
for(auto backup:backups)
{
QFile::remove(backup.backup);
}
for(auto backup:trashcan)
{
QFile::remove(backup.backup);
}
qApp->quit();
return;

Expand All @@ -787,13 +793,17 @@ void MultiMC::installUpdates(const QString updateFilesDir, GoUpdate::OperationLi
if(!QFile::rename(backup.backup, backup.orig))
{
revertOK = false;
qWarning() << "removing new" << backup.orig << "failed!";
qWarning() << "restoring" << backup.orig << "failed!";
}
}
for(auto backup:trashcan)
{
qWarning() << "restoring" << backup.orig << "from" << backup.backup;
revertOK &= QFile::rename(backup.backup, backup.orig);
if(!QFile::rename(backup.backup, backup.orig))
{
revertOK = false;
qWarning() << "restoring" << backup.orig << "failed!";
}
}
QString msg;
if(!revertOK)
Expand All @@ -804,11 +814,13 @@ void MultiMC::installUpdates(const QString updateFilesDir, GoUpdate::OperationLi
}
else switch (failedOperationType)
{
case Copy:
msg = tr("Couldn't replace file %1. Changes were reverted.\nSee the MultiMC log file for details.").arg(failedFile);
case Replace:
msg = tr("Couldn't replace file %1. Changes were reverted.\n"
"See the MultiMC log file for details.").arg(failedFile);
break;
case Delete:
msg = tr("Couldn't remove file %1. Changes were reverted.\nSee the MultiMC log file for details.").arg(failedFile);
msg = tr("Couldn't remove file %1. Changes were reverted.\n"
"See the MultiMC log file for details.").arg(failedFile);
break;
case Start:
msg = tr("The new version didn't start and the update was rolled back.");
Expand All @@ -817,7 +829,7 @@ void MultiMC::installUpdates(const QString updateFilesDir, GoUpdate::OperationLi
default:
return;
}
QMessageBox::critical(nullptr, tr("Update failed"), msg);
QMessageBox::critical(nullptr, tr("Update failed!"), msg);
}

void MultiMC::setIconTheme(const QString& name)
Expand Down
4 changes: 2 additions & 2 deletions logic/updater/DownloadTask.cpp
Original file line number Diff line number Diff line change
Expand Up @@ -27,8 +27,8 @@
namespace GoUpdate
{

DownloadTask::DownloadTask(Status status, QObject *parent)
: Task(parent)
DownloadTask::DownloadTask(Status status, QString target, QObject *parent)
: Task(parent), m_updateFilesDir(target)
{
m_status = status;

Expand Down
7 changes: 6 additions & 1 deletion logic/updater/DownloadTask.h
Original file line number Diff line number Diff line change
Expand Up @@ -30,7 +30,12 @@ class DownloadTask : public Task
Q_OBJECT

public:
explicit DownloadTask(Status status, QObject* parent = 0);
/**
* Create a download task
*
* target is a template - XXXXXX at the end will be replaced with a random generated string, ensuring uniqueness
*/
explicit DownloadTask(Status status, QString target, QObject* parent = 0);

/// Get the directory that will contain the update files.
QString updateFilesDir();
Expand Down
4 changes: 2 additions & 2 deletions logic/updater/GoUpdate.h
Original file line number Diff line number Diff line change
Expand Up @@ -68,7 +68,7 @@ struct Operation
{
static Operation CopyOp(QString fsource, QString fdest, int fmode=0644)
{
return Operation{OP_COPY, fsource, fdest, fmode};
return Operation{OP_REPLACE, fsource, fdest, fmode};
}
static Operation DeleteOp(QString file)
{
Expand All @@ -84,7 +84,7 @@ struct Operation
//! Specifies the type of operation that this is.
enum Type
{
OP_COPY,
OP_REPLACE,
OP_DELETE,
} type;

Expand Down
2 changes: 1 addition & 1 deletion tests/tst_DownloadTask.cpp
Original file line number Diff line number Diff line change
Expand Up @@ -40,7 +40,7 @@ QDebug operator<<(QDebug dbg, const Operation::Type &t)
{
switch (t)
{
case Operation::OP_COPY:
case Operation::OP_REPLACE:
dbg << "OP_COPY";
break;
case Operation::OP_DELETE:
Expand Down

0 comments on commit 15b7c30

Please sign in to comment.