git.sur5r.net Git - bacula/bacula/blob - bacula/src/qt-console/restore/prerestore.cpp

   1 /*
   2    Bacula® - The Network Backup Solution
   3
   4    Copyright (C) 2007-2008 Free Software Foundation Europe e.V.
   5
   6    The main author of Bacula is Kern Sibbald, with contributions from
   7    many others, a complete list can be found in the file AUTHORS.
   8    This program is Free Software; you can redistribute it and/or
   9    modify it under the terms of version two of the GNU General Public
  10    License as published by the Free Software Foundation and included
  11    in the file LICENSE.
  12
  13    This program is distributed in the hope that it will be useful, but
  14    WITHOUT ANY WARRANTY; without even the implied warranty of
  15    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
  16    General Public License for more details.
  17
  18    You should have received a copy of the GNU General Public License
  19    along with this program; if not, write to the Free Software
  20    Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA
  21    02110-1301, USA.
  22
  23    Bacula® is a registered trademark of Kern Sibbald.
  24    The licensor of Bacula is the Free Software Foundation Europe
  25    (FSFE), Fiduciary Program, Sumatrastrasse 25, 8006 Zürich,
  26    Switzerland, email:ftf@fsfeurope.org.
  27 */
  28
  29 /*
  30  *   Version $Id$
  31  *
  32  *  preRestore -> dialog put up to determine the restore type
  33  *
  34  *   Kern Sibbald, February MMVII
  35  *
  36  */
  37
  38 #include "bat.h"
  39 #include "restore.h"
  40
  41 /* Constructor to have job id list default in */
  42 prerestorePage::prerestorePage(QString &data, unsigned int datatype)
  43 {
  44    m_dataIn = data;
  45    m_dataInType = datatype;
  46    buildPage();
  47 }
  48
  49 /* Basic Constructor */
  50 prerestorePage::prerestorePage()
  51 {
  52    m_dataIn = "";
  53    m_dataInType = R_NONE;
  54    buildPage();
  55 }
  56
  57 /*
  58  * This is really the constructor
  59  */
  60 void prerestorePage::buildPage()
  61 {
  62    m_name = tr("Restore");
  63    setupUi(this);
  64    pgInitialize();
  65    m_console->notify(false);
  66    m_closeable = true;
  67    QTreeWidgetItem* thisitem = mainWin->getFromHash(this);
  68    thisitem->setIcon(0,QIcon(QString::fromUtf8(":images/restore.png")));
  69
  70    if (!m_console->preventInUseConnect())
  71        return;
  72
  73    jobCombo->addItems(m_console->job_list);
  74    filesetCombo->addItems(m_console->fileset_list);
  75    clientCombo->addItems(m_console->client_list);
  76    poolCombo->addItem(tr("Any"));
  77    poolCombo->addItems(m_console->pool_list);
  78    storageCombo->addItems(m_console->storage_list);
  79    /* current or before . .  Start out with current checked */
  80    recentCheckBox->setCheckState(Qt::Checked);
  81    beforeDateTime->setDisplayFormat(mainWin->m_dtformat);
  82    beforeDateTime->setDateTime(QDateTime::currentDateTime());
  83    beforeDateTime->setEnabled(false);
  84    selectFilesRadio->setChecked(true);
  85    if (m_dataInType == R_NONE) {
  86       selectJobRadio->setChecked(true);
  87       selectJobIdsRadio->setChecked(false);
  88       jobIdEdit->setText(tr("Comma separted list of Job Ids"));
  89       jobIdEdit->setEnabled(false);
  90    } else if (m_dataInType == R_JOBIDLIST) {
  91       selectJobIdsRadio->setChecked(true);
  92       selectJobRadio->setChecked(false);
  93       jobIdEdit->setText(m_dataIn);
  94       jobRadioClicked(false);
  95       QStringList fieldlist;
  96       if (jobdefsFromJob(fieldlist, m_dataIn) == 1) {
  97          filesetCombo->setCurrentIndex(filesetCombo->findText(fieldlist[2], Qt::MatchExactly));
  98          clientCombo->setCurrentIndex(clientCombo->findText(fieldlist[1], Qt::MatchExactly));
  99          jobCombo->setCurrentIndex(jobCombo->findText(fieldlist[0], Qt::MatchExactly));
 100       }
 101    } else if (m_dataInType == R_JOBDATETIME) {
 102       selectJobRadio->setChecked(true);
 103       selectJobIdsRadio->setChecked(false);
 104       jobIdEdit->setText(tr("Comma separted list of Job Ids"));
 105       jobIdEdit->setEnabled(false);
 106       recentCheckBox->setCheckState(Qt::Unchecked);
 107       jobRadioClicked(true);
 108       QStringList fieldlist;
 109       if (jobdefsFromJob(fieldlist, m_dataIn) == 1) {
 110          filesetCombo->setCurrentIndex(filesetCombo->findText(fieldlist[2], Qt::MatchExactly));
 111          clientCombo->setCurrentIndex(clientCombo->findText(fieldlist[1], Qt::MatchExactly));
 112          jobCombo->setCurrentIndex(jobCombo->findText(fieldlist[0], Qt::MatchExactly));
 113          beforeDateTime->setDateTime(QDateTime::fromString(fieldlist[3], mainWin->m_dtformat));
 114      }
 115    }
 116    job_name_change(0);
 117    connect(jobCombo, SIGNAL(currentIndexChanged(int)), this, SLOT(job_name_change(int)));
 118    connect(okButton, SIGNAL(pressed()), this, SLOT(okButtonPushed()));
 119    connect(cancelButton, SIGNAL(pressed()), this, SLOT(cancelButtonPushed()));
 120    connect(recentCheckBox, SIGNAL(stateChanged(int)), this, SLOT(recentChanged(int)));
 121    connect(selectJobRadio, SIGNAL(clicked(bool)), this, SLOT(jobRadioClicked(bool)));
 122    connect(selectJobIdsRadio, SIGNAL(clicked(bool)), this, SLOT(jobidsRadioClicked(bool)));
 123    connect(jobIdEdit, SIGNAL(editingFinished()), this, SLOT(jobIdEditFinished()));
 124
 125    dockPage();
 126    setCurrent();
 127    this->show();
 128 }
 129
 130
 131 /*
 132  * Check to make sure all is ok then start either the select window or the restore
 133  * run window
 134  */
 135 void prerestorePage::okButtonPushed()
 136 {
 137    if (!selectJobRadio->isChecked()) {
 138       if (!checkJobIdList())
 139          return;
 140    }
 141    QString cmd;
 142
 143    this->hide();
 144
 145    cmd = QString("restore");
 146    cmd += " fileset=\"" + filesetCombo->currentText() + "\"";
 147    cmd += " client=\"" + clientCombo->currentText() + "\"";
 148    if (selectJobRadio->isChecked()) {
 149       if (poolCombo->currentText() != tr("Any") ){
 150          cmd += " pool=\"" + poolCombo->currentText() + "\"";
 151       }
 152       cmd += " storage=\"" + storageCombo->currentText() + "\"";
 153       if (recentCheckBox->checkState() == Qt::Checked) {
 154          cmd += " current";
 155       } else {
 156          QDateTime stamp = beforeDateTime->dateTime();
 157          QString before = stamp.toString(mainWin->m_dtformat);
 158          cmd += " before=\"" + before + "\"";
 159       }
 160    } else {
 161       cmd += " jobid=\"" + jobIdEdit->text() + "\"";
 162    }
 163    if (selectFilesRadio->isChecked()) {
 164       if (!selectJobIdsRadio->isChecked())
 165          cmd += " select";
 166    } else {
 167       cmd += " all done";
 168    }
 169
 170    if (mainWin->m_commandDebug) {
 171       Pmsg1(000, "preRestore command \'%s\'\n", cmd.toUtf8().data());
 172    }
 173    m_console->write_dir(cmd.toUtf8().data());
 174
 175    /* Note, do not turn notifier back on here ... */
 176    if (selectFilesRadio->isChecked()) {
 177       setConsoleCurrent();
 178       closeStackPage();
 179    } else {
 180       closeStackPage();
 181       mainWin->resetFocus();
 182    }
 183    m_console->notify(true);
 184 }
 185
 186
 187 /*
 188  * Destroy the instace of the class
 189  */
 190 void prerestorePage::cancelButtonPushed()
 191 {
 192    mainWin->set_status(tr("Canceled"));
 193    this->hide();
 194    m_console->notify(true);
 195    closeStackPage();
 196 }
 197
 198
 199 /*
 200  * Handle updating the other widget with job defaults when the job combo is changed.
 201  */
 202 void prerestorePage::job_name_change(int index)
 203 {
 204    job_defaults job_defs;
 205
 206    (void)index;
 207    job_defs.job_name = jobCombo->currentText();
 208    if (m_console->get_job_defaults(job_defs)) {
 209       filesetCombo->setCurrentIndex(filesetCombo->findText(job_defs.fileset_name, Qt::MatchExactly));
 210       clientCombo->setCurrentIndex(clientCombo->findText(job_defs.client_name, Qt::MatchExactly));
 211       poolCombo->setCurrentIndex(poolCombo->findText(tr("Any"), Qt::MatchExactly));
 212       storageCombo->setCurrentIndex(storageCombo->findText(job_defs.store_name, Qt::MatchExactly));
 213    }
 214 }
 215
 216 /*
 217  * Handle the change of enabled of input widgets when the recent checkbox state
 218  * is changed.
 219  */
 220 void prerestorePage::recentChanged(int state)
 221 {
 222    if ((state == Qt::Unchecked) && (selectJobRadio->isChecked())) {
 223       beforeDateTime->setEnabled(true);
 224    } else {
 225       beforeDateTime->setEnabled(false);
 226    }
 227 }
 228
 229
 230 /*
 231  * For when jobs list is to be used, return a list which is the needed items from
 232  * the job record
 233  */
 234 int prerestorePage::jobdefsFromJob(QStringList &fieldlist, QString &jobId)
 235 {
 236    QString job, client, fileset;
 237    QString query("");
 238    query = "SELECT DISTINCT Job.Name AS JobName, Client.Name AS Client,"
 239    " FileSet.FileSet AS FileSet, Job.EndTime AS JobEnd,"
 240    " Job.Type AS JobType"
 241    " From Job, Client, FileSet"
 242    " WHERE Job.FileSetId=FileSet.FileSetId AND Job.ClientId=Client.ClientId"
 243    " AND JobId=\'" + jobId + "\'";
 244    if (mainWin->m_sqlDebug) { Pmsg1(000, "query = %s\n", query.toUtf8().data()); }
 245    QStringList results;
 246    if (m_console->sql_cmd(query, results)) {
 247       QString field;
 248
 249       /* Iterate through the lines of results, there should only be one. */
 250       foreach (QString resultline, results) {
 251          fieldlist = resultline.split("\t");
 252       } /* foreach resultline */
 253    } /* if results from query */
 254    return results.count();
 255 }
 256
 257 /*
 258  * Function to handle when the jobidlist line edit input loses focus or is entered
 259  */
 260 void prerestorePage::jobIdEditFinished()
 261 {
 262    checkJobIdList();
 263 }
 264
 265 bool prerestorePage::checkJobIdList()
 266 {
 267    /* Need to check and make sure the text is a comma separated list of integers */
 268    QString line = jobIdEdit->text();
 269    if (line.contains(" ")) {
 270       QMessageBox::warning(this, "Bat",
 271          tr("There can be no spaces in the text for the joblist.\n"
 272          "Press OK to continue?"), QMessageBox::Ok );
 273       return false;
 274    }
 275    QStringList joblist = line.split(",", QString::SkipEmptyParts);
 276    bool allintokay = true, alljobok = true, allisjob = true;
 277    QString jobName(""), clientName("");
 278    foreach (QString job, joblist) {
 279       bool intok;
 280       job.toInt(&intok, 10);
 281       if (intok) {
 282          /* are the integers representing a list of jobs all with the same job
 283           * and client */
 284          QStringList fields;
 285          if (jobdefsFromJob(fields, job) == 1) {
 286             if (jobName == "")
 287                jobName = fields[0];
 288             else if (jobName != fields[0])
 289                alljobok = false;
 290             if (clientName == "")
 291                clientName = fields[1];
 292             else if (clientName != fields[1])
 293                alljobok = false;
 294          } else {
 295             allisjob = false;
 296          }
 297       } else {
 298          allintokay = false;
 299       }
 300    }
 301    if (!allintokay){
 302       QMessageBox::warning(this, "Bat",
 303          tr("The string is not a comma separated list of integers.\n"
 304          "Press OK to continue?"), QMessageBox::Ok );
 305       return false;
 306    }
 307    if (!allisjob){
 308       QMessageBox::warning(this, tr("Bat"),
 309          tr("At least one of the jobs is not a valid job of type \"Backup\".\n"
 310          "Press OK to continue?"), QMessageBox::Ok );
 311       return false;
 312    }
 313    if (!alljobok){
 314       QMessageBox::warning(this, "Bat",
 315          tr("All jobs in the list must be of the same jobName and same client.\n"
 316          "Press OK to continue?"), QMessageBox::Ok );
 317       return false;
 318    }
 319    return true;
 320 }
 321
 322 /*
 323  * Handle the change of enabled of input widgets when the job radio buttons
 324  * are changed.
 325  */
 326 void prerestorePage::jobRadioClicked(bool checked)
 327 {
 328    if (checked) {
 329       jobCombo->setEnabled(true);
 330       filesetCombo->setEnabled(true);
 331       clientCombo->setEnabled(true);
 332       poolCombo->setEnabled(true);
 333       storageCombo->setEnabled(true);
 334       recentCheckBox->setEnabled(true);
 335       if (!recentCheckBox->isChecked()) {
 336          beforeDateTime->setEnabled(true);
 337       }
 338       jobIdEdit->setEnabled(false);
 339       selectJobRadio->setChecked(true);
 340       selectJobIdsRadio->setChecked(false);
 341    } else {
 342       jobCombo->setEnabled(false);
 343       filesetCombo->setEnabled(false);
 344       clientCombo->setEnabled(false);
 345       poolCombo->setEnabled(false);
 346       storageCombo->setEnabled(false);
 347       recentCheckBox->setEnabled(false);
 348       beforeDateTime->setEnabled(false);
 349       jobIdEdit->setEnabled(true);
 350       selectJobRadio->setChecked(false);
 351       selectJobIdsRadio->setChecked(true);
 352    }
 353    Dmsg2(200, "jobRadio=%d jobidsRadio=%d\n", selectJobRadio->isChecked(),
 354          selectJobIdsRadio->isChecked());
 355 }
 356
 357 void prerestorePage::jobidsRadioClicked(bool checked)
 358 {
 359    if (checked) {
 360       jobCombo->setEnabled(false);
 361       filesetCombo->setEnabled(false);
 362       clientCombo->setEnabled(false);
 363       poolCombo->setEnabled(false);
 364       storageCombo->setEnabled(false);
 365       recentCheckBox->setEnabled(false);
 366       beforeDateTime->setEnabled(false);
 367       jobIdEdit->setEnabled(true);
 368       selectJobRadio->setChecked(false);
 369       selectJobIdsRadio->setChecked(true);
 370    } else {
 371       jobCombo->setEnabled(true);
 372       filesetCombo->setEnabled(true);
 373       clientCombo->setEnabled(true);
 374       poolCombo->setEnabled(true);
 375       storageCombo->setEnabled(true);
 376       recentCheckBox->setEnabled(true);
 377       if (!recentCheckBox->isChecked()) {
 378          beforeDateTime->setEnabled(true);
 379       }
 380       jobIdEdit->setEnabled(false);
 381       selectJobRadio->setChecked(true);
 382       selectJobIdsRadio->setChecked(false);
 383    }
 384    Dmsg2(200, "jobRadio=%d jobidsRadio=%d\n", selectJobRadio->isChecked(),
 385          selectJobIdsRadio->isChecked());
 386 }