File indexing completed on 2024-04-14 05:34:14

 /*
     SPDX-FileCopyrightText: 2011 Vishesh Yadav <vishesh3y@gmail.com>
 
     SPDX-License-Identifier: GPL-2.0-or-later
 */
 
 #include "exportdialog.h"
 #include "fileviewhgpluginsettings.h"
 #include "commitinfowidget.h"
 #include "hgwrapper.h"
 
 #include <QCheckBox>
 #include <QGroupBox>
 #include <QVBoxLayout>
 #include <QGridLayout>
 #include <QListWidget>
 #include <QProcess>
 #include <QTextCodec>
 #include <QFileDialog>
 #include <KLocalizedString>
 #include <KMessageBox>
 
 HgExportDialog::HgExportDialog(QWidget *parent) :
     DialogBase(QDialogButtonBox::Ok | QDialogButtonBox::Cancel, parent)
 {
     // dialog properties
     this->setWindowTitle(i18nc("@title:window",
                 "<application>Hg</application> Export"));
     okButton()->setText(xi18nc("@action:button", "Export"));
 
     //
     setupUI();
     loadCommits();
 
     // Load saved settings
     FileViewHgPluginSettings *settings = FileViewHgPluginSettings::self();
     this->resize(QSize(settings->exportDialogWidth(),
                                settings->exportDialogHeight()));
 
     //
     connect(this, SIGNAL(finished(int)), this, SLOT(saveGeometry()));
 }
 
 void HgExportDialog::setupUI()
 {
     QGroupBox *mainGroup = new QGroupBox;
     QGridLayout *mainLayout = new QGridLayout;
     m_commitInfoWidget = new HgCommitInfoWidget;
 
     m_commitInfoWidget->setSelectionMode(QAbstractItemView::ExtendedSelection);
     mainLayout->addWidget(m_commitInfoWidget);
     mainGroup->setLayout(mainLayout);
 
     // options
     m_optionGroup = new QGroupBox(i18nc("@label:group", "Options"));
     m_optText = new QCheckBox(i18nc("@label", "Treat all files as text"));
     m_optGit = new QCheckBox(i18nc("@label", "Use Git extended diff format"));
     m_optNoDates = new QCheckBox(i18nc("@label", "Omit dates from diff headers"));
     QVBoxLayout *optionLayout = new QVBoxLayout;
     optionLayout->addWidget(m_optText);
     optionLayout->addWidget(m_optGit);
     optionLayout->addWidget(m_optNoDates);
     m_optionGroup->setLayout(optionLayout);
 
     //setup main dialog widget
     QVBoxLayout *lay = new QVBoxLayout;
     lay->addWidget(mainGroup);
     lay->addWidget(m_optionGroup);
     layout()->insertLayout(0, lay);
 }
 
 void HgExportDialog::loadCommits()
 {
     HgWrapper *hgWrapper = HgWrapper::instance();
 
     // update heads list
     QProcess process;
     process.setWorkingDirectory(hgWrapper->getBaseDir());
 
     QStringList args;
     args << QLatin1String("log");
     args << QLatin1String("--template");
     args << QLatin1String("{rev}\n{node|short}\n{branch}\n"
                           "{author}\n{desc|firstline}\n");
 
     process.start(QLatin1String("hg"), args);
     process.waitForFinished();
     m_commitInfoWidget->clear();
 
     const int FINAL = 5;
     char buffer[FINAL][1024];
     int count = 0;
     while (process.readLine(buffer[count], sizeof(buffer[count])) > 0) {
         if (count == FINAL - 1) {
             QString rev = QTextCodec::codecForLocale()->toUnicode(buffer[0]).trimmed();
             QString changeset = QTextCodec::codecForLocale()->toUnicode(buffer[1]).trimmed();
             QString branch = QTextCodec::codecForLocale()->toUnicode(buffer[2]).trimmed();
             QString author = QTextCodec::codecForLocale()->toUnicode(buffer[3]).trimmed();
             QString log = QTextCodec::codecForLocale()->toUnicode(buffer[4]).trimmed();
 
             QListWidgetItem *item = new QListWidgetItem;
             item->setData(Qt::DisplayRole, changeset);
             item->setData(Qt::UserRole + 1, rev);
             item->setData(Qt::UserRole + 2, branch);
             item->setData(Qt::UserRole + 3, author);
             item->setData(Qt::UserRole + 4, log);
             m_commitInfoWidget->addItem(item);
         }
         count = (count + 1)%FINAL;
     }
 }
 
 void HgExportDialog::saveGeometry()
 {
     FileViewHgPluginSettings *settings = FileViewHgPluginSettings::self();
     settings->setExportDialogHeight(this->height());
     settings->setExportDialogWidth(this->width());
     settings->save();
 }
 
 void HgExportDialog::done(int r)
 {
     if (r == QDialog::Accepted) {
         const QList<QListWidgetItem*> items = m_commitInfoWidget->selectedItems();
         if (items.empty()) {
             KMessageBox::error(this, i18nc("@message:error",
                      "Please select at least one changeset to be exported!"));
             return;
         }
 
         QStringList args;
         if (m_optText->checkState() == Qt::Checked) {
             args << QStringLiteral("--text");
         }
         if (m_optGit->checkState() == Qt::Checked) {
             args << QStringLiteral("--git");
         }
         if (m_optNoDates->checkState() == Qt::Checked) {
             args << QStringLiteral("--nodates");
         }
 
         args << QStringLiteral("-r");
         for (QListWidgetItem *item : items) {
             args << item->data(Qt::DisplayRole).toString();
         }
 
         QString directory = QFileDialog::getExistingDirectory(this);
         if (directory.isEmpty()) {
             return;
         }
         if (!directory.endsWith(QLatin1Char('/'))) {
             directory.append(QLatin1Char('/'));
         }
 
         args << QStringLiteral("--output");
         args << directory + QLatin1String("%b_%h.patch");
 
         HgWrapper *hgw = HgWrapper::instance();
         if (hgw->executeCommandTillFinished(QStringLiteral("export"), args)) {
             QDialog::done(r);
         }
         else {
             KMessageBox::error(this, hgw->readAllStandardError());
         }
     }
     else {
         QDialog::done(r);
     }
 }
 
 
 
 #include "moc_exportdialog.cpp"