summaryrefslogtreecommitdiffstats
path: root/rbutil/rbutilqt/gui/backupdialog.cpp
blob: f12c47b5705e6ff9bb587c93adb2a6484f2e2d09 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
/***************************************************************************
 *             __________               __   ___.
 *   Open      \______   \ ____   ____ |  | _\_ |__   _______  ___
 *   Source     |       _//  _ \_/ ___\|  |/ /| __ \ /  _ \  \/  /
 *   Jukebox    |    |   (  <_> )  \___|    < | \_\ (  <_> > <  <
 *   Firmware   |____|_  /\____/ \___  >__|_ \|___  /\____/__/\_ \
 *                     \/            \/     \/    \/            \/
 *
 *   Copyright (C) 2012 by Dominik Riebeling
 *
 * All files in this archive are subject to the GNU General Public License.
 * See the file COPYING in the source tree root for full license agreement.
 *
 * This software is distributed on an "AS IS" basis, WITHOUT WARRANTY OF ANY
 * KIND, either express or implied.
 *
 ****************************************************************************/

#include <QThread>
#include <QDialog>
#include <QMessageBox>
#include <QFileDialog>
#include "backupdialog.h"
#include "ui_backupdialogfrm.h"
#include "rbsettings.h"
#include "progressloggergui.h"
#include "ziputil.h"
#include "rockboxinfo.h"
#include "Logger.h"

class BackupSizeThread : public QThread
{
    public:
        void run(void);
        void setPath(QString p) { m_path = p; }
        qint64 currentSize(void) { return m_currentSize; }

    private:
        QString m_path;
        qint64 m_currentSize;
};


void BackupSizeThread::run(void)
{
    LOG_INFO() << "Thread started, calculating" << m_path;
    m_currentSize = 0;

    QDirIterator it(m_path, QDirIterator::Subdirectories);
    while(it.hasNext()) {
        m_currentSize += QFileInfo(it.next()).size();
    }
    LOG_INFO() << "Thread done, sum:" << m_currentSize;
}


BackupDialog::BackupDialog(QWidget* parent) : QDialog(parent)
{
    ui.setupUi(this);

    m_thread = new BackupSizeThread();
    connect(m_thread, SIGNAL(finished()), this, SLOT(updateSizeInfo()));
    connect(m_thread, SIGNAL(terminated()), this, SLOT(updateSizeInfo()));

    connect(ui.buttonCancel, SIGNAL(clicked()), this, SLOT(close()));
    connect(ui.buttonCancel, SIGNAL(clicked()), m_thread, SLOT(quit()));
    connect(ui.buttonChange, SIGNAL(clicked()), this, SLOT(changeBackupPath()));
    connect(ui.buttonBackup, SIGNAL(clicked()), this, SLOT(backup()));

    ui.backupSize->setText(tr("Installation size: calculating ..."));
    m_mountpoint = RbSettings::value(RbSettings::Mountpoint).toString();

    m_backupName = RbSettings::value(RbSettings::BackupPath).toString();
    if(m_backupName.isEmpty()) {
        m_backupName = m_mountpoint;
    }
    RockboxInfo info(m_mountpoint);
    m_backupName += "/.backup/rockbox-backup-" + info.version() + ".zip";
    ui.backupLocation->setText(QDir::toNativeSeparators(m_backupName));

    m_thread->setPath(m_mountpoint + "/.rockbox");
    m_thread->start();
}


void BackupDialog::changeBackupPath(void)
{
    QString backupString = QFileDialog::getSaveFileName(this,
        tr("Select Backup Filename"), m_backupName, "*.zip");
    // only update if a filename was entered, ignore if cancelled
    if(!backupString.isEmpty()) {
        m_backupName = backupString;
        ui.backupLocation->setText(QDir::toNativeSeparators(m_backupName));
        RbSettings::setValue(RbSettings::BackupPath, QFileInfo(m_backupName).absolutePath());
    }
}


void BackupDialog::updateSizeInfo(void)
{
    double size = m_thread->currentSize() / (1024 * 1024);
    QString unit = "MiB";

    if(size > 1024) {
        size /= 1024;
        unit = "GiB";
    }

    ui.backupSize->setText(tr("Installation size: %L1 %2").arg(size, 0, 'g', 4).arg(unit));
}


void BackupDialog::backup(void)
{
    if(QFileInfo(m_backupName).isFile()) {
        if(QMessageBox::warning(this, tr("File exists"),
                tr("The selected backup file already exists. Overwrite?"),
                QMessageBox::Yes | QMessageBox::No) == QMessageBox::No) {
            return;
        }
    }
    m_logger = new ProgressLoggerGui(this);
    connect(m_logger, SIGNAL(closed()), this, SLOT(close()));
    m_logger->show();
    m_logger->addItem(tr("Starting backup ..."),LOGINFO);
    QCoreApplication::processEvents();

    // create dir, if it doesnt exist
    QFileInfo backupFile(m_backupName);
    if(!QDir(backupFile.path()).exists())
    {
        QDir a;
        a.mkpath(backupFile.path());
    }

    // create backup
    ZipUtil zip(this);
    connect(&zip, SIGNAL(logProgress(int, int)), m_logger, SLOT(setProgress(int, int)));
    connect(&zip, SIGNAL(logItem(QString, int)), m_logger, SLOT(addItem(QString, int)));
    zip.open(m_backupName, QuaZip::mdCreate);

    QString mp = m_mountpoint + "/.rockbox";
    if(zip.appendDirToArchive(mp, m_mountpoint)) {
        m_logger->addItem(tr("Backup successful."), LOGINFO);
    }
    else {
        m_logger->addItem(tr("Backup failed!"), LOGERROR);
    }
    zip.close();
    m_logger->setFinished();
}