← Back to branch summary

~ci-train-bot/ubuntu-system-settings/ubuntu-system-settings-ubuntu-yakkety-landing-027

« back to all changes in this revision

Viewing changes to plugins/system-update/update_manager.cpp

  • Committer: Bileto Bot
  • Author(s): jonas-drange
  • Date: 2016-08-17 11:18:05 UTC
  • mfrom: (1631.6.245 updates-rewrite)
  • Revision ID: ci-train-bot@canonical.com-20160817111805-moyvik9wloy79ul3
rewrite the system update panel

Show diffs side-by-side

added added

removed removed

Lines of Context:
plugins/system-update/update_manager.cpp
1
 
/*
2
 
 * Copyright (C) 2014 Canonical Ltd
3
 
 *
4
 
 * This program is free software: you can redistribute it and/or modify
5
 
 * it under the terms of the GNU General Public License version 3 as
6
 
 * published by the Free Software Foundation.
7
 
 *
8
 
 * This program is distributed in the hope that it will be useful,
9
 
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
10
 
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
11
 
 * GNU General Public License for more details.
12
 
 *
13
 
 * You should have received a copy of the GNU General Public License
14
 
 * along with this program.  If not, see <http://www.gnu.org/licenses/>.
15
 
 *
16
 
 * Authors:
17
 
 * Diego Sarmentero <diego.sarmentero@canonical.com>
18
 
 *
19
 
*/
20
 
 
21
 
 
22
 
#include <QDBusInterface>
23
 
#include <QJsonDocument>
24
 
#include <QJsonObject>
25
 
#include <QJsonArray>
26
 
#include <QJsonValue>
27
 
#include <QProcessEnvironment>
28
 
#include <QString>
29
 
#include <QStringList>
30
 
#include <ssoservice.h>
31
 
 
32
 
#include "update_manager.h"
33
 
#include "network.h"
34
 
 
35
 
using namespace UbuntuOne;
36
 
 
37
 
#define CLICK_COMMAND "click"
38
 
#ifdef TESTS
39
 
    #define CHECK_CREDENTIALS "IGNORE_CREDENTIALS"
40
 
#else
41
 
    #define CHECK_CREDENTIALS "CHECK_CREDENTIALS"
42
 
#endif
43
 
 
44
 
const QString TEST_APP = "com.ubuntu.developer.testclick";
45
 
 
46
 
namespace UpdatePlugin {
47
 
 
48
 
UpdateManager *UpdateManager::m_instance = 0;
49
 
 
50
 
UpdateManager *UpdateManager::instance()
51
 
{
52
 
    if (!m_instance)
53
 
        m_instance = new UpdateManager;
54
 
 
55
 
    return m_instance;
56
 
}
57
 
 
58
 
UpdateManager::UpdateManager(QObject *parent):
59
 
    QObject(parent),
60
 
    m_systemCheckingUpdate(false),
61
 
    m_clickCheckingUpdate(false),
62
 
    m_checkingUpdates(0),
63
 
    m_downloadMode(-1)
64
 
{
65
 
    // SSO SERVICE
66
 
    QObject::connect(&m_service, SIGNAL(credentialsFound(const Token&)),
67
 
                     this, SLOT(handleCredentialsFound(Token)));
68
 
    QObject::connect(&m_service, SIGNAL(credentialsNotFound()),
69
 
                     this, SIGNAL(credentialsNotFound()));
70
 
    QObject::connect(&m_service, SIGNAL(credentialsNotFound()),
71
 
                  this, SLOT(clickUpdateNotAvailable()));
72
 
    QObject::connect(&m_service, SIGNAL(credentialsDeleted()),
73
 
                     this, SIGNAL(credentialsDeleted()));
74
 
    QObject::connect(&m_service, SIGNAL(credentialsDeleted()),
75
 
                  this, SLOT(clickUpdateNotAvailable()));
76
 
    // PROCESS
77
 
    QObject::connect(&m_process, SIGNAL(finished(int)),
78
 
                  this, SLOT(processOutput()));
79
 
    // NETWORK
80
 
    QObject::connect(&m_network, SIGNAL(updatesFound()),
81
 
                  this, SLOT(processUpdates()));
82
 
    QObject::connect(&m_network, SIGNAL(updatesNotFound()),
83
 
                  this, SLOT(clickUpdateNotAvailable()));
84
 
    QObject::connect(&m_network, SIGNAL(errorOccurred()),
85
 
                  this, SIGNAL(errorFound()));
86
 
    QObject::connect(&m_network, SIGNAL(networkError()),
87
 
                  this, SIGNAL(networkError()));
88
 
    QObject::connect(&m_network, SIGNAL(serverError()),
89
 
                  this, SIGNAL(serverError()));
90
 
    QObject::connect(&m_network, SIGNAL(credentialError()),
91
 
                     this, SLOT(handleCredentialsFailed()));
92
 
    QObject::connect(&m_network,
93
 
                     SIGNAL(clickTokenObtained(Update*, const QString&)),
94
 
                     this, SLOT(clickTokenReceived(Update*, const QString&)));
95
 
    // SYSTEM UPDATE
96
 
    QObject::connect(&m_systemUpdate, SIGNAL(updateAvailable(const QString&, Update*)),
97
 
                  this, SLOT(registerSystemUpdate(const QString&, Update*)));
98
 
    QObject::connect(&m_systemUpdate, SIGNAL(updateNotFound()),
99
 
                  this, SLOT(systemUpdateNotAvailable()));
100
 
    QObject::connect(&m_systemUpdate, SIGNAL(downloadModeChanged()),
101
 
                  SIGNAL(downloadModeChanged()));
102
 
    QObject::connect(&m_systemUpdate, SIGNAL(updateDownloaded()),
103
 
                  SLOT(updateDownloaded()));
104
 
    QObject::connect(&m_systemUpdate, SIGNAL(updateProcessFailed(const QString&)),
105
 
                  SIGNAL(updateProcessFailed(QString)));
106
 
    QObject::connect(&m_systemUpdate, SIGNAL(updateFailed(int, QString)),
107
 
                  SLOT(updateFailed(int, QString)));
108
 
    QObject::connect(&m_systemUpdate, SIGNAL(updatePaused(int)),
109
 
                  SLOT(systemUpdatePaused(int)));
110
 
    QObject::connect(&m_systemUpdate, SIGNAL(downloadStarted()),
111
 
                  SLOT(systemUpdateDownloadStarted()));
112
 
    QObject::connect(&m_systemUpdate, SIGNAL(updateProgress(int, double)),
113
 
                  SLOT(systemUpdateProgress(int, double)));
114
 
    QObject::connect(&m_systemUpdate, SIGNAL(rebooting(bool)),
115
 
                  SIGNAL(rebooting(bool)));
116
 
}
117
 
 
118
 
UpdateManager::~UpdateManager()
119
 
{
120
 
}
121
 
 
122
 
void UpdateManager::clickUpdateNotAvailable()
123
 
{
124
 
    m_clickCheckingUpdate = false;
125
 
    reportCheckState();
126
 
    updateNotAvailable();
127
 
}
128
 
 
129
 
void UpdateManager::systemUpdateNotAvailable()
130
 
{
131
 
    m_systemCheckingUpdate = false;
132
 
    reportCheckState();
133
 
    updateNotAvailable();
134
 
}
135
 
 
136
 
void UpdateManager::updateNotAvailable()
137
 
{
138
 
    m_checkingUpdates--;
139
 
    if (m_checkingUpdates == 0 && m_model.count() == 0) {
140
 
        Q_EMIT updatesNotFound();
141
 
    }
142
 
}
143
 
 
144
 
void UpdateManager::reportCheckState()
145
 
{
146
 
    if (!m_clickCheckingUpdate && !m_systemCheckingUpdate) {
147
 
        Q_EMIT checkFinished();
148
 
    }
149
 
}
150
 
 
151
 
void UpdateManager::checkUpdates()
152
 
{
153
 
    m_systemCheckingUpdate = true;
154
 
    m_clickCheckingUpdate = true;
155
 
    m_checkingUpdates = 2;
156
 
    m_model.clear();
157
 
    m_apps.clear();
158
 
    Q_EMIT modelChanged();
159
 
    bool enabled = enableAutopilotMode();
160
 
 
161
 
    // If we're in testing mode, always consider updates for TEST_APP
162
 
    if (enabled) {
163
 
        Update *app = new Update();
164
 
        app->initializeApplication(TEST_APP, QString("Test App"), QString("1.0"));
165
 
        m_apps.insert(app->getPackageName(), app);
166
 
    }
167
 
 
168
 
    if (getCheckForCredentials()) {
169
 
        m_systemUpdate.checkForUpdate();
170
 
        m_service.getCredentials();
171
 
    } else if (enabled) {
172
 
        systemUpdateNotAvailable();
173
 
        Token token("", "", "", "");
174
 
        handleCredentialsFound(token);
175
 
    } else {
176
 
        systemUpdateNotAvailable();
177
 
        clickUpdateNotAvailable();
178
 
    }
179
 
}
180
 
 
181
 
void UpdateManager::handleCredentialsFound(Token token)
182
 
{
183
 
    m_network.setUbuntuOneToken(token);
184
 
    QStringList args("list");
185
 
    args << "--manifest";
186
 
    QString command = getClickCommand();
187
 
    m_process.start(command, args);
188
 
}
189
 
 
190
 
void UpdateManager::handleCredentialsFailed()
191
 
{
192
 
    m_service.invalidateCredentials();
193
 
}
194
 
 
195
 
QString UpdateManager::getClickCommand()
196
 
{
197
 
    QProcessEnvironment environment = QProcessEnvironment::systemEnvironment();
198
 
    QString command = environment.value("CLICK_COMMAND", QString(CLICK_COMMAND));
199
 
    return command;
200
 
}
201
 
 
202
 
bool UpdateManager::getCheckForCredentials()
203
 
{
204
 
    QProcessEnvironment environment = QProcessEnvironment::systemEnvironment();
205
 
    QString value = environment.value("IGNORE_CREDENTIALS", QString(CHECK_CREDENTIALS));
206
 
    return value == "CHECK_CREDENTIALS";
207
 
}
208
 
 
209
 
bool UpdateManager::enableAutopilotMode()
210
 
{
211
 
    QProcessEnvironment environment = QProcessEnvironment::systemEnvironment();
212
 
    QString value = environment.value("AUTOPILOT_ENABLED", QString("AUTOPILOT_DISABLED"));
213
 
    return value == "AUTOPILOT_ENABLED";
214
 
}
215
 
 
216
 
void UpdateManager::processOutput()
217
 
{
218
 
    QString output(m_process.readAllStandardOutput());
219
 
 
220
 
    QJsonDocument document = QJsonDocument::fromJson(output.toUtf8());
221
 
 
222
 
    QJsonArray array = document.array();
223
 
 
224
 
    int i;
225
 
    for (i = 0; i < array.size(); i++) {
226
 
        QJsonObject object = array.at(i).toObject();
227
 
        QString name = object.value("name").toString();
228
 
        QString title = object.value("title").toString();
229
 
        QString version = object.value("version").toString();
230
 
        Update *app = new Update();
231
 
        app->initializeApplication(name, title, version);
232
 
        m_apps[app->getPackageName()] = app;
233
 
    }
234
 
 
235
 
    m_network.checkForNewVersions(m_apps);
236
 
}
237
 
 
238
 
void UpdateManager::processUpdates()
239
 
{
240
 
    m_clickCheckingUpdate = false;
241
 
    bool updateAvailable = false;
242
 
    foreach (QString id, m_apps.keys()) {
243
 
        if (m_model.contains(id))
244
 
            continue;
245
 
        Update *app = m_apps.value(id);
246
 
        QString packagename(UBUNTU_PACKAGE_NAME);
247
 
        if(app->getPackageName() != packagename && app->updateRequired()) {
248
 
            updateAvailable = true;
249
 
            m_model.append(QVariant::fromValue(app));
250
 
        }
251
 
    }
252
 
 
253
 
    if (updateAvailable) {
254
 
        Q_EMIT modelChanged();
255
 
        Q_EMIT updateAvailableFound(false);
256
 
    }
257
 
    reportCheckState();
258
 
}
259
 
 
260
 
void UpdateManager::registerSystemUpdate(const QString& packageName, Update *update)
261
 
{
262
 
    QString packagename(UBUNTU_PACKAGE_NAME);
263
 
    if (!m_apps.contains(packagename)) {
264
 
        m_apps[packageName] = update;
265
 
        m_model.insert(0, QVariant::fromValue(update));
266
 
        Q_EMIT modelChanged();
267
 
        Q_EMIT updateAvailableFound(update->updateState());
268
 
        if (update->getRemoteVersion() == m_latestDownload)
269
 
            updateDownloaded();
270
 
    }
271
 
    m_systemCheckingUpdate = false;
272
 
 
273
 
    reportCheckState();
274
 
}
275
 
 
276
 
void UpdateManager::updateDownloaded()
277
 
{
278
 
    QString packagename(UBUNTU_PACKAGE_NAME);
279
 
    if (m_apps.contains(packagename)) {
280
 
        Update *update = m_apps[packagename];
281
 
        update->setSelected(false);
282
 
        update->setUpdateState(false);
283
 
        update->setUpdateReady(true);
284
 
        m_latestDownload = update->getRemoteVersion();
285
 
        Q_EMIT systemUpdateDownloaded();
286
 
    }
287
 
}
288
 
void UpdateManager::updateFailed(int consecutiveFailureCount, QString lastReason)
289
 
{
290
 
    QString packagename(UBUNTU_PACKAGE_NAME);
291
 
    if (m_apps.contains(packagename)) {
292
 
        Update *update = m_apps[packagename];
293
 
        update->setSelected(false);
294
 
        update->setUpdateState(false);
295
 
        update->setDownloadProgress(0);
296
 
        Q_EMIT systemUpdateFailed(consecutiveFailureCount, lastReason);
297
 
    }
298
 
}
299
 
 
300
 
void UpdateManager::systemUpdatePaused(int value)
301
 
{
302
 
    QString packagename(UBUNTU_PACKAGE_NAME);
303
 
    if (m_apps.contains(packagename)) {
304
 
        Update *update = m_apps[packagename];
305
 
        update->setSelected(true);
306
 
        update->setUpdateState(false);
307
 
        update->setDownloadProgress(value);
308
 
        update->setStatus(Update::Paused);
309
 
    }
310
 
}
311
 
 
312
 
void UpdateManager::systemUpdateDownloadStarted()
313
 
{                 
314
 
    QString packagename(UBUNTU_PACKAGE_NAME);
315
 
    if (m_apps.contains(packagename)) {
316
 
        Update *update = m_apps[packagename];
317
 
        update->setSelected(true);
318
 
        update->setUpdateState(true);
319
 
        update->setStatus(Update::Downloading);
320
 
    }
321
 
} 
322
 
 
323
 
void UpdateManager::systemUpdateProgress(int value, double eta)
324
 
{
325
 
    Q_UNUSED(eta);
326
 
    QString packagename(UBUNTU_PACKAGE_NAME);
327
 
    if (m_apps.contains(packagename)) {
328
 
        Update *update = m_apps[packagename];
329
 
        update->setDownloadProgress(value);
330
 
    }
331
 
}
332
 
 
333
 
void UpdateManager::startDownload(const QString &packagename)
334
 
{
335
 
    m_apps[packagename]->setUpdateState(true);
336
 
    if (m_apps[packagename]->systemUpdate()) {
337
 
        m_systemUpdate.downloadUpdate();
338
 
    } else {
339
 
        downloadApp(m_apps[packagename]);
340
 
    }
341
 
}
342
 
 
343
 
void UpdateManager::forceAllowGSMDownload(const QString &packagename)
344
 
{
345
 
    if (!m_apps[packagename]->systemUpdate()) 
346
 
        return;
347
 
 
348
 
    m_apps[packagename]->setUpdateState(true);
349
 
        m_systemUpdate.forceAllowGSMDownload();
350
 
}
351
 
 
352
 
void UpdateManager::retryDownload(const QString &packagename)
353
 
{
354
 
    if (m_apps[packagename]->systemUpdate()) {
355
 
        Update *update = m_apps.take(packagename);
356
 
        m_systemUpdate.cancelUpdate();
357
 
        m_model.removeAt(0);
358
 
        update->deleteLater();
359
 
        Q_EMIT modelChanged();
360
 
        m_systemUpdate.checkForUpdate();
361
 
    } else {
362
 
        startDownload(packagename);
363
 
    }
364
 
}
365
 
 
366
 
void UpdateManager::pauseDownload(const QString &packagename)
367
 
{
368
 
    m_apps[packagename]->setUpdateState(false);
369
 
    m_systemUpdate.pauseDownload();
370
 
}
371
 
 
372
 
void UpdateManager::downloadApp(Update *app)
373
 
{
374
 
    m_network.getClickToken(app, app->downloadUrl());
375
 
}
376
 
 
377
 
void UpdateManager::clickTokenReceived(Update *app, const QString &clickToken)
378
 
{
379
 
    app->setError("");
380
 
    app->setClickToken(clickToken);
381
 
    app->setDownloadUrl(app->getClickUrl());
382
 
}
383
 
 
384
 
void UpdateManager::updateClickScope()
385
 
{
386
 
    // Refresh click scope
387
 
    QDBusMessage signal = QDBusMessage::createSignal(
388
 
                "/com/canonical/unity/scopes",
389
 
                "com.canonical.unity.scopes",
390
 
                "InvalidateResults");
391
 
    signal << "clickscope";
392
 
    QDBusConnection::sessionBus().send(signal);
393
 
 
394
 
    // When a click update is complete, emit modelChanged
395
 
    Q_EMIT modelChanged();
396
 
}
397
 
 
398
 
}