ryzom/tools/client/ryzom_installer/src/uninstalldialog.cpp

   1 // Ryzom - MMORPG Framework <http://dev.ryzom.com/projects/ryzom/>
   2 // Copyright (C) 2010  Winch Gate Property Limited
   3 //
   4 // This program is free software: you can redistribute it and/or modify
   5 // it under the terms of the GNU Affero General Public License as
   6 // published by the Free Software Foundation, either version 3 of the
   7 // License, or (at your option) any later version.
   8 //
   9 // This program is distributed in the hope that it will be useful,
  10 // but WITHOUT ANY WARRANTY; without even the implied warranty of
  11 // MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  12 // GNU Affero General Public License for more details.
  13 //
  14 // You should have received a copy of the GNU Affero General Public License
  15 // along with this program.  If not, see <http://www.gnu.org/licenses/>.
  16
  17 #include "stdpch.h"
  18 #include "uninstalldialog.h"
  19 #include "configfile.h"
  20 #include "utils.h"
  21
  22 #include "nel/misc/system_info.h"
  23 #include "nel/misc/common.h"
  24
  25 #ifdef DEBUG_NEW
  26         #define new DEBUG_NEW
  27 #endif
  28
  29 CUninstallDialog::CUninstallDialog(QWidget *parent):QDialog(parent), m_installerIndex(-1)
  30 {
  31         setWindowFlags(windowFlags() & ~Qt::WindowContextHelpButtonHint);
  32
  33         setupUi(this);
  34
  35         CConfigFile *config = CConfigFile::getInstance();
  36
  37         int serverCount = config->getServersCount();
  38
  39         QStandardItemModel *model = new QStandardItemModel(0, 2, this);
  40
  41         QStringList columns;
  42         columns << tr("Component");
  43         columns << tr("Size");
  44
  45         model->setHorizontalHeaderLabels(columns);
  46
  47         QStandardItem *item = NULL;
  48
  49         // clients
  50         for (int row = 0; row < serverCount; ++row)
  51         {
  52                 const CServer &server = config->getServer(row);
  53
  54                 if (QFile::exists(server.getDirectory()))
  55                 {
  56                         m_serversIndices[server.id] = model->rowCount();
  57
  58                         item = new QStandardItem(tr("Client for %1").arg(server.name));
  59                         item->setCheckable(true);
  60                         model->appendRow(item);
  61                 }
  62         }
  63
  64         int profilesCount = config->getProfilesCount();
  65
  66         // profiles
  67         for (int row = 0; row < profilesCount; ++row)
  68         {
  69                 const CProfile &profile = config->getProfile(row);
  70
  71                 m_profilesIndices[profile.id] = model->rowCount();
  72
  73                 item = new QStandardItem(tr("Profile #%1: %2").arg(profile.id).arg(profile.name));
  74                 item->setCheckable(true);
  75                 model->appendRow(item);
  76
  77         }
  78
  79         // installer
  80         m_installerIndex = model->rowCount();
  81
  82         item = new QStandardItem(tr("Installer"));
  83         item->setCheckable(true);
  84         model->appendRow(item);
  85
  86         // downloaded files
  87         m_downloadedFilesIndex = model->rowCount();
  88
  89         item = new QStandardItem(tr("Downloaded Files"));
  90         item->setCheckable(true);
  91         model->appendRow(item);
  92
  93         componentsTreeView->setModel(model);
  94         componentsTreeView->resizeColumnToContents(0);
  95
  96         // resize layout depending on content and constraints
  97         adjustSize();
  98
  99         // fix height because to left bitmap
 100         setFixedHeight(height());
 101
 102         // click signals
 103         connect(uninstallButton, SIGNAL(clicked()), SLOT(accept()));
 104         connect(cancelButton, SIGNAL(clicked()), SLOT(reject()));
 105         connect(model, SIGNAL(itemChanged(QStandardItem *)), SLOT(onItemChanged(QStandardItem *)));
 106
 107         // semi-hack to not update UI on another thread
 108         connect(this, SIGNAL(updateSize(int, QString)), SLOT(onUpdateSize(int, QString)));
 109         connect(this, SIGNAL(updateLayout()), SLOT(onUpdateLayout()));
 110 }
 111
 112 CUninstallDialog::~CUninstallDialog()
 113 {
 114 }
 115
 116 void CUninstallDialog::showEvent(QShowEvent *event)
 117 {
 118         QDialog::showEvent(event);
 119
 120         // update size of all components sizes in a thread to not block interface
 121         QtConcurrent::run(this, &CUninstallDialog::updateSizes);
 122 }
 123
 124 void CUninstallDialog::setSelectedComponents(const SComponents &components)
 125 {
 126         QStandardItemModel *model = qobject_cast<QStandardItemModel*>(componentsTreeView->model());
 127         if (model == NULL) return;
 128
 129         QStandardItem *item = NULL;
 130
 131         // servers
 132         IDIndicesMap::const_iterator it = m_serversIndices.begin(), iend = m_serversIndices.end();
 133
 134         while (it != iend)
 135         {
 136                 item = model->item(it.value());
 137
 138                 if (item) item->setCheckState(components.servers.indexOf(it.key()) > -1 ? Qt::Checked : Qt::Unchecked);
 139
 140                 ++it;
 141         }
 142
 143         // profiles
 144         it = m_profilesIndices.begin(), iend = m_profilesIndices.end();
 145
 146         while (it != iend)
 147         {
 148                 item = model->item(it.value());
 149
 150                 if (item) item->setCheckState(components.profiles.indexOf(it.key()) > -1 ? Qt::Checked : Qt::Unchecked);
 151
 152                 ++it;
 153         }
 154
 155         // installer
 156         item = model->item(m_installerIndex);
 157         if (item) item->setCheckState(components.installer ? Qt::Checked : Qt::Unchecked);
 158
 159         // downloaded files
 160         item = model->item(m_downloadedFilesIndex);
 161         if (item) item->setCheckState(components.downloadedFiles ? Qt::Checked : Qt::Unchecked);
 162 }
 163
 164 SComponents CUninstallDialog::getSelectedCompenents() const
 165 {
 166         SComponents res;
 167
 168         QStandardItemModel *model = qobject_cast<QStandardItemModel*>(componentsTreeView->model());
 169         if (model == NULL) return res;
 170
 171         QStandardItem *item = NULL;
 172
 173         // servers
 174         IDIndicesMap::const_iterator it = m_serversIndices.begin(), iend = m_serversIndices.end();
 175
 176         while (it != iend)
 177         {
 178                 item = model->item(it.value());
 179
 180                 if (item && item->checkState() == Qt::Checked) res.servers << it.key();
 181
 182                 ++it;
 183         }
 184
 185         // profiles
 186         it = m_profilesIndices.begin(), iend = m_profilesIndices.end();
 187
 188         while (it != iend)
 189         {
 190                 item = model->item(it.value());
 191
 192                 if (item && item->checkState() == Qt::Checked) res.profiles << it.key();
 193
 194                 ++it;
 195         }
 196
 197         // installer
 198         item = model->item(m_installerIndex);
 199         res.installer = item && item->checkState() == Qt::Checked;
 200
 201         // downloaded files
 202         item = model->item(m_downloadedFilesIndex);
 203         res.downloadedFiles = item && item->checkState() == Qt::Checked;
 204
 205         return res;
 206 }
 207
 208 void CUninstallDialog::accept()
 209 {
 210         QDialog::accept();
 211 }
 212
 213 void CUninstallDialog::onItemChanged(QStandardItem * /* item */)
 214 {
 215         updateButtons();
 216 }
 217
 218 void CUninstallDialog::onUpdateSize(int row, const QString &text)
 219 {
 220         QStandardItemModel *model = qobject_cast<QStandardItemModel*>(componentsTreeView->model());
 221         if (model == NULL) return;
 222
 223         // set size for a component
 224         QStandardItem *item = new QStandardItem(text);
 225         model->setItem(row, 1, item);
 226 }
 227
 228 void CUninstallDialog::onUpdateLayout()
 229 {
 230         componentsTreeView->resizeColumnToContents(1);
 231
 232         updateButtons();
 233 }
 234
 235 void CUninstallDialog::updateSizes()
 236 {
 237         CConfigFile *config = CConfigFile::getInstance();
 238
 239         // clients
 240         IDIndicesMap::const_iterator it = m_serversIndices.begin(), iend = m_serversIndices.end();
 241
 242         while(it != iend)
 243         {
 244                 const CServer &server = config->getServer(it.key());
 245
 246                 qint64 bytes = getDirectorySize(server.getDirectory(), true);
 247
 248                 emit updateSize(it.value(), qBytesToHumanReadable(bytes));
 249
 250                 ++it;
 251         }
 252
 253         // profiles
 254         it = m_profilesIndices.begin(), iend = m_profilesIndices.end();
 255
 256         while (it != iend)
 257         {
 258                 const CProfile &profile = config->getProfile(it.key());
 259
 260                 // wrong profile
 261                 if (profile.id.isEmpty()) continue;
 262
 263                 qint64 bytes = getDirectorySize(profile.getDirectory(), true);
 264
 265                 emit updateSize(it.value(), qBytesToHumanReadable(bytes));
 266
 267                 ++it;
 268         }
 269
 270         // downloaded files
 271         qint64 bytes = 0;
 272
 273         QDir dir(config->getInstallationDirectory());
 274
 275         QStringList filters;
 276
 277         filters << "*.log";
 278         filters << "*.7z";
 279         filters << "*.bnp";
 280         filters << "*.zip";
 281         filters << "*.part";
 282
 283         QFileInfoList downloadedFiles = dir.entryInfoList(filters, QDir::Files);
 284
 285         foreach(const QFileInfo &info, downloadedFiles)
 286         {
 287                 bytes += info.size();
 288         }
 289
 290         emit updateSize(m_downloadedFilesIndex, qBytesToHumanReadable(bytes));
 291
 292         emit updateLayout();
 293 }
 294
 295 void CUninstallDialog::updateButtons()
 296 {
 297         QStandardItemModel *model = qobject_cast<QStandardItemModel*>(componentsTreeView->model());
 298         if (model == NULL) return;
 299
 300         int checkedCount = 0;
 301
 302         for (int i = 0; i < model->rowCount(); ++i)
 303         {
 304                 if (model->item(i)->checkState() == Qt::Checked) ++checkedCount;
 305         }
 306
 307         // Uninstall button should be enabled only if at least one component is checked
 308         uninstallButton->setEnabled(checkedCount > 0);
 309 }