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-20087 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 John Walker.
  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() != "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 // consoleCommand(cmd);
 175
 176    /* Note, do not turn notifier back on here ... */
 177    if (selectFilesRadio->isChecked()) {
 178       setConsoleCurrent();
 179 //    new restorePage();
 180       closeStackPage();
 181    } else {
 182       closeStackPage();
 183       mainWin->resetFocus();
 184    }
 185    m_console->notify(true);
 186 }
 187
 188
 189 /*
 190  * Destroy the instace of the class
 191  */
 192 void prerestorePage::cancelButtonPushed()
 193 {
 194    mainWin->set_status(tr("Canceled"));
 195    this->hide();
 196    m_console->notify(true);
 197    closeStackPage();
 198 }
 199
 200
 201 /*
 202  * Handle updating the other widget with job defaults when the job combo is changed.
 203  */
 204 void prerestorePage::job_name_change(int index)
 205 {
 206    job_defaults job_defs;
 207
 208    (void)index;
 209    job_defs.job_name = jobCombo->currentText();
 210    if (m_console->get_job_defaults(job_defs)) {
 211       filesetCombo->setCurrentIndex(filesetCombo->findText(job_defs.fileset_name, Qt::MatchExactly));
 212       clientCombo->setCurrentIndex(clientCombo->findText(job_defs.client_name, Qt::MatchExactly));
 213       poolCombo->setCurrentIndex(poolCombo->findText(job_defs.pool_name, Qt::MatchExactly));
 214       storageCombo->setCurrentIndex(storageCombo->findText(job_defs.store_name, Qt::MatchExactly));
 215    }
 216 }
 217
 218 /*
 219  * Handle the change of enabled of input widgets when the recent checkbox state
 220  * is changed.
 221  */
 222 void prerestorePage::recentChanged(int state)
 223 {
 224    if ((state == Qt::Unchecked) && (selectJobRadio->isChecked())) {
 225       beforeDateTime->setEnabled(true);
 226    } else {
 227       beforeDateTime->setEnabled(false);
 228    }
 229 }
 230
 231
 232 /*
 233  * For when jobs list is to be used, return a list which is the needed items from
 234  * the job record
 235  */
 236 int prerestorePage::jobdefsFromJob(QStringList &fieldlist, QString &jobId)
 237 {
 238    QString job, client, fileset;
 239    QString query("");
 240    query = "SELECT DISTINCT Job.Name AS JobName, Client.Name AS Client,"
 241    " FileSet.FileSet AS FileSet, Job.EndTime AS JobEnd,"
 242    " Job.Type AS JobType"
 243    " From Job, Client, FileSet"
 244    " WHERE Job.FileSetId=FileSet.FileSetId AND Job.ClientId=Client.ClientId"
 245    " AND JobId=\'" + jobId + "\'";
 246    if (mainWin->m_sqlDebug) { Pmsg1(000, "query = %s\n", query.toUtf8().data()); }
 247    QStringList results;
 248    if (m_console->sql_cmd(query, results)) {
 249       QString field;
 250
 251       /* Iterate through the lines of results, there should only be one. */
 252       foreach (QString resultline, results) {
 253          fieldlist = resultline.split("\t");
 254       } /* foreach resultline */
 255    } /* if results from query */
 256    return results.count();
 257 }
 258
 259 /*
 260  * Function to handle when the jobidlist line edit input loses focus or is entered
 261  */
 262 void prerestorePage::jobIdEditFinished()
 263 {
 264    checkJobIdList();
 265 }
 266
 267 bool prerestorePage::checkJobIdList()
 268 {
 269    /* Need to check and make sure the text is a comma separated list of integers */
 270    QString line = jobIdEdit->text();
 271    if (line.contains(" ")) {
 272       QMessageBox::warning(this, "Bat",
 273          tr("There can be no spaces in the text for the joblist.\n"
 274          "Press OK to continue?"), QMessageBox::Ok );
 275       return false;
 276    }
 277    QStringList joblist = line.split(",", QString::SkipEmptyParts);
 278    bool allintokay = true, alljobok = true, allisjob = true;
 279    QString jobName(""), clientName("");
 280    foreach (QString job, joblist) {
 281       bool intok;
 282       job.toInt(&intok, 10);
 283       if (intok) {
 284          /* are the integers representing a list of jobs all with the same job
 285           * and client */
 286          QStringList fields;
 287          if (jobdefsFromJob(fields, job) == 1) {
 288             if (jobName == "")
 289                jobName = fields[0];
 290             else if (jobName != fields[0])
 291                alljobok = false;
 292             if (clientName == "")
 293                clientName = fields[1];
 294             else if (clientName != fields[1])
 295                alljobok = false;
 296          } else {
 297             allisjob = false;
 298          }
 299       } else {
 300          allintokay = false;
 301       }
 302    }
 303    if (!allintokay){
 304       QMessageBox::warning(this, "Bat",
 305          tr("The string is not a comma separated list of integers.\n"
 306          "Press OK to continue?"), QMessageBox::Ok );
 307       return false;
 308    }
 309    if (!allisjob){
 310       QMessageBox::warning(this, tr("Bat"),
 311          tr("At least one of the jobs is not a valid job of type \"Backup\".\n"
 312          "Press OK to continue?"), QMessageBox::Ok );
 313       return false;
 314    }
 315    if (!alljobok){
 316       QMessageBox::warning(this, "Bat",
 317          tr("All jobs in the list must be of the same jobName and same client.\n"
 318          "Press OK to continue?"), QMessageBox::Ok );
 319       return false;
 320    }
 321    return true;
 322 }
 323
 324 /*
 325  * Handle the change of enabled of input widgets when the job radio buttons
 326  * are changed.
 327  */
 328 void prerestorePage::jobRadioClicked(bool checked)
 329 {
 330    if (checked) {
 331       jobCombo->setEnabled(true);
 332       filesetCombo->setEnabled(true);
 333       clientCombo->setEnabled(true);
 334       poolCombo->setEnabled(true);
 335       storageCombo->setEnabled(true);
 336       recentCheckBox->setEnabled(true);
 337       if (!recentCheckBox->isChecked()) {
 338          beforeDateTime->setEnabled(true);
 339       }
 340       jobIdEdit->setEnabled(false);
 341       selectJobRadio->setChecked(true);
 342       selectJobIdsRadio->setChecked(false);
 343    } else {
 344       jobCombo->setEnabled(false);
 345       filesetCombo->setEnabled(false);
 346       clientCombo->setEnabled(false);
 347       poolCombo->setEnabled(false);
 348       storageCombo->setEnabled(false);
 349       recentCheckBox->setEnabled(false);
 350       beforeDateTime->setEnabled(false);
 351       jobIdEdit->setEnabled(true);
 352       selectJobRadio->setChecked(false);
 353       selectJobIdsRadio->setChecked(true);
 354    }
 355    Dmsg2(200, "jobRadio=%d jobidsRadio=%d\n", selectJobRadio->isChecked(),
 356          selectJobIdsRadio->isChecked());
 357 }
 358
 359 void prerestorePage::jobidsRadioClicked(bool checked)
 360 {
 361    if (checked) {
 362       jobCombo->setEnabled(false);
 363       filesetCombo->setEnabled(false);
 364       clientCombo->setEnabled(false);
 365       poolCombo->setEnabled(false);
 366       storageCombo->setEnabled(false);
 367       recentCheckBox->setEnabled(false);
 368       beforeDateTime->setEnabled(false);
 369       jobIdEdit->setEnabled(true);
 370       selectJobRadio->setChecked(false);
 371       selectJobIdsRadio->setChecked(true);
 372    } else {
 373       jobCombo->setEnabled(true);
 374       filesetCombo->setEnabled(true);
 375       clientCombo->setEnabled(true);
 376       poolCombo->setEnabled(true);
 377       storageCombo->setEnabled(true);
 378       recentCheckBox->setEnabled(true);
 379       if (!recentCheckBox->isChecked()) {
 380          beforeDateTime->setEnabled(true);
 381       }
 382       jobIdEdit->setEnabled(false);
 383       selectJobRadio->setChecked(true);
 384       selectJobIdsRadio->setChecked(false);
 385    }
 386    Dmsg2(200, "jobRadio=%d jobidsRadio=%d\n", selectJobRadio->isChecked(),
 387          selectJobIdsRadio->isChecked());
 388 }