File indexing completed on 2024-04-21 09:36:42

 /*******************************************************************************
  * Copyright (C) 2008-2013 Konstantinos Smanis <konstantinos.smanis@gmail.com> *
  *                                                                             *
  * This program is free software: you can redistribute it and/or modify it     *
  * under the terms of the GNU General Public License as published by the Free  *
  * Software Foundation, either version 3 of the License, or (at your option)   *
  * any later version.                                                          *
  *                                                                             *
  * This program is distributed in the hope that it will be useful, but WITHOUT *
  * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or       *
  * FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License for    *
  * more details.                                                               *
  *                                                                             *
  * You should have received a copy of the GNU General Public License along     *
  * with this program. If not, see <http://www.gnu.org/licenses/>.              *
  *******************************************************************************/
 
 //Own
 #include "installDlg.h"
 
 //Qt
 #include <QFile>
 #include <QRadioButton>
 #include <QPushButton>
 #include <QIcon>
 #include <QProgressDialog>
 
 //KDE
 #include <KFormat>
 #include <KLocalizedString>
 #include <KMessageBox>
 #include <KAuth/Action>
 #include <KAuth/ExecuteJob>
 using namespace KAuth;
 #include <Solid/Device>
 #include <Solid/StorageAccess>
 #include <Solid/StorageVolume>
 
 //Ui
 #include "ui_installDlg.h"
 
 InstallDialog::InstallDialog(QWidget *parent) : QDialog(parent)
 {
     QWidget *widget = new QWidget(this);
     ui = new Ui::InstallDialog;
     ui->setupUi(widget);
     ui->gridLayout->setContentsMargins(0, 0, 0, 0);
 
     auto *buttonBox = new QDialogButtonBox(QDialogButtonBox::Ok | QDialogButtonBox::Cancel);
     connect(buttonBox, &QDialogButtonBox::accepted, this, &InstallDialog::slotAccepted);
     connect(buttonBox, &QDialogButtonBox::rejected, this, &QDialog::reject);
     auto *okButton = buttonBox->button(QDialogButtonBox::Ok);
     okButton->setEnabled(false);
 
     QVBoxLayout *mainLayout = new QVBoxLayout;
     setLayout(mainLayout);
     mainLayout->addWidget(widget);
     mainLayout->addWidget(buttonBox);
 
     setWindowTitle(i18nc("@title:window", "Install/Recover Bootloader"));
     setWindowIcon(QIcon::fromTheme(QStringLiteral("system-software-update")));
     if (parent) {
         resize(parent->size());
     }
 
     ui->treeWidget_recover->headerItem()->setText(0, QString());
     ui->treeWidget_recover->header()->setSectionResizeMode(QHeaderView::Stretch);
     ui->treeWidget_recover->header()->setSectionResizeMode(0, QHeaderView::ResizeToContents);
     const auto devices = Solid::Device::listFromType(Solid::DeviceInterface::StorageAccess);
     for (const Solid::Device &device : devices) {
         if (!device.is<Solid::StorageAccess>() || !device.is<Solid::StorageVolume>()) {
             continue;
         }
         const Solid::StorageAccess *partition = device.as<Solid::StorageAccess>();
         if (!partition) {
             continue;
         }
         const Solid::StorageVolume *volume = device.as<Solid::StorageVolume>();
         if (!volume || volume->usage() != Solid::StorageVolume::FileSystem) {
             continue;
         }
 
         QString uuidDir = QStringLiteral("/dev/disk/by-uuid/"), uuid = volume->uuid(), name;
         name = (QFile::exists((name = uuidDir + uuid)) || QFile::exists((name = uuidDir + uuid.toLower())) || QFile::exists((name = uuidDir + uuid.toUpper())) ? QFile::symLinkTarget(name) : QString());
         QTreeWidgetItem *item = new QTreeWidgetItem(ui->treeWidget_recover, QStringList() << QString() << name << partition->filePath() << volume->label() << volume->fsType() << KFormat().formatByteSize(volume->size()));
         item->setIcon(1, QIcon::fromTheme(device.icon()));
         item->setTextAlignment(5, Qt::AlignRight | Qt::AlignVCenter);
         ui->treeWidget_recover->addTopLevelItem(item);
         QRadioButton *radio = new QRadioButton(ui->treeWidget_recover);
         connect(radio, &QRadioButton::clicked, okButton, &QWidget::setEnabled);
         ui->treeWidget_recover->setItemWidget(item, 0, radio);
     }
 }
 InstallDialog::~InstallDialog()
 {
     delete ui;
 }
 
 void InstallDialog::slotAccepted()
 {
     Action installAction(QStringLiteral("org.kde.kcontrol.kcmgrub2.install"));
     installAction.setHelperId(QStringLiteral("org.kde.kcontrol.kcmgrub2"));
     for (int i = 0; i < ui->treeWidget_recover->topLevelItemCount(); i++) {
         QRadioButton *radio = qobject_cast<QRadioButton *>(ui->treeWidget_recover->itemWidget(ui->treeWidget_recover->topLevelItem(i), 0));
         if (radio && radio->isChecked()) {
             installAction.addArgument(QStringLiteral("partition"), ui->treeWidget_recover->topLevelItem(i)->text(1));
             installAction.addArgument(QStringLiteral("mountPoint"), ui->treeWidget_recover->topLevelItem(i)->text(2));
             installAction.addArgument(QStringLiteral("mbrInstall"), !ui->checkBox_partition->isChecked());
             break;
         }
     }
     if (installAction.arguments().value(QStringLiteral("partition")).toString().isEmpty()) {
         KMessageBox::error(this, i18nc("@info", "Sorry, you have to select a partition with a proper name!"));
         return;
     }
     installAction.setParentWidget(this);
 
     KAuth::ExecuteJob *installJob = installAction.execute(KAuth::Action::AuthorizeOnlyMode);
     if (!installJob->exec()) {
         return;
     }
 
     QProgressDialog progressDlg(this);
     progressDlg.setWindowTitle(i18nc("@title:window", "Installing"));
     progressDlg.setLabelText(i18nc("@info:progress", "Installing GRUB..."));
     progressDlg.setCancelButton(nullptr);
     progressDlg.setModal(true);
     progressDlg.setMinimum(0);
     progressDlg.setMaximum(0);
     progressDlg.show();
     installJob = installAction.execute();
     connect(installJob, &KJob::finished, &progressDlg, &QWidget::hide);
 
     if (installJob->exec()) {
         QDialog *dialog = new QDialog(this);
         dialog->setWindowTitle(i18nc("@title:window", "Information"));
         dialog->setModal(true);
         dialog->setAttribute(Qt::WA_DeleteOnClose);
 
         QPushButton *detailsButton = new QPushButton;
         detailsButton->setObjectName(QStringLiteral("detailsButton"));
         detailsButton->setText(QApplication::translate("KMessageBox", "&Details") + QStringLiteral(" >>"));
         detailsButton->setIcon(QIcon::fromTheme(QStringLiteral("help-about")));
 
         QDialogButtonBox *buttonBox = new QDialogButtonBox(dialog);
         buttonBox->addButton(detailsButton, QDialogButtonBox::HelpRole);
         buttonBox->addButton(QDialogButtonBox::Ok);
         buttonBox->button(QDialogButtonBox::Ok)->setFocus();
 
         KMessageBox::createKMessageBox(dialog, buttonBox, QMessageBox::Information, i18nc("@info", "Successfully installed GRUB."),
                                        QStringList(), QString(), nullptr, KMessageBox::Notify,
                                        QString::fromUtf8(installJob->data().value(QStringLiteral("output")).toByteArray())); // krazy:exclude=qclasses
     } else {
         KMessageBox::detailedError(this, i18nc("@info", "Failed to install GRUB."), installJob->errorText());
     }
 
     accept();
 }