X-Git-Url: https://git.sur5r.net/?a=blobdiff_plain;f=bacula%2Fsrc%2Fqt-console%2Frestore%2Fbrestore.cpp;h=baa3a27479efcbde38d2c06b7b29e2539ea43611;hb=23f1411adf9a9afc8259ca284b89c5135bd2e426;hp=09615603bf2f416c0c53425dd2d35b08eab514ba;hpb=57808179fdc33f45e625d47ed8a1f98f7f832029;p=bacula%2Fbacula

diff --git a/bacula/src/qt-console/restore/brestore.cpp b/bacula/src/qt-console/restore/brestore.cpp
index 09615603bf..baa3a27479 100644
--- a/bacula/src/qt-console/restore/brestore.cpp
+++ b/bacula/src/qt-console/restore/brestore.cpp
@@ -1,7 +1,7 @@
 /*
    Bacula® - The Network Backup Solution
 
-   Copyright (C) 2007-2009 Free Software Foundation Europe e.V.
+   Copyright (C) 2007-2010 Free Software Foundation Europe e.V.
 
    The main author of Bacula is Kern Sibbald, with contributions from
    many others, a complete list can be found in the file AUTHORS.
@@ -47,13 +47,14 @@ bRestore::bRestore()
    QTreeWidgetItem* thisitem = mainWin->getFromHash(this);
    thisitem->setIcon(0, QIcon(QString::fromUtf8(":images/browse.png")));
    m_populated = false;
+   m_closeable = false;
    m_current = NULL;
    RestoreList->setAcceptDrops(true);
 }
 
+// Populate client table and job associated
 void bRestore::setClient()
 {
-   Pmsg0(000, "Repopulating client table\n");
    // Select the same client, don't touch
    if (m_client == ClientList->currentText()) {
       return;
@@ -85,19 +86,20 @@ void bRestore::setClient()
 
    QString job;
    QStringList results;
+   QStringList fieldlist;
    if (m_console->sql_cmd(jobQuery, results)) {
-      QStringList fieldlist;
-
       /* Iterate through the record returned from the query */
       foreach (QString resultline, results) {
+         //  0       1          2     3
+         // JobId, StartTime, Level, Name
          fieldlist = resultline.split("\t");
          job = fieldlist[1] + " " + fieldlist[3] + "(" + fieldlist[2] + ") " + fieldlist[0];
-         JobList->addItem(job, QVariant(fieldlist[0]));
+         JobList->addItem(job, QVariant(fieldlist[0])); // set also private value
       }
    }
 }
 
-
+// Compute job associated and update the job cache if needed
 void bRestore::setJob()
 {
    if (JobList->currentIndex() < 1) {
@@ -126,6 +128,7 @@ void bRestore::setJob()
       return;
    }
 
+   // TODO: Can take some time if the job contains many dirs
    m_jobids = results.at(0);
    cmd = ".bvfs_update jobid=" + m_jobids;
    m_console->dir_cmd(cmd, results);
@@ -138,6 +141,12 @@ void bRestore::setJob()
 
 extern int decode_stat(char *buf, struct stat *statp, int32_t *LinkFI);
 
+// refresh button with a filter or limit/offset change
+void bRestore::refreshView()
+{
+   displayFiles(m_pathid, m_path);
+}
+
 void bRestore::displayFiles(int64_t pathid, QString path)
 {
    QString arg;
@@ -145,17 +154,19 @@ void bRestore::displayFiles(int64_t pathid, QString path)
    QStringList fieldlist;
    struct stat statp;
    int32_t LinkFI;
-   int nb;
-   int row=0;
+   int nb = 0;
+   int row = 0;
    Freeze frz_lst(*FileList); /* disable updating*/
    Freeze frz_rev(*FileRevisions); /* disable updating*/
    FileList->clearContents();
    FileRevisions->clearContents();
    FileRevisions->setRowCount(0);
 
+   // If we provide pathid, use it (path can be altered by encoding conversion)
    if (pathid > 0) {
       arg = " pathid=" + QString().setNum(pathid);
 
+      // Choose .. update current path to parent dir
       if (path == "..") {
          if (m_path == "/") {
             m_path = "";
@@ -173,9 +184,20 @@ void bRestore::displayFiles(int64_t pathid, QString path)
       m_path = path;
       arg = " path=\"" + m_path + "\"";
    }
-   LocationEntry->setText(m_path);
 
-   QString q = ".bvfs_lsdir jobid=" + m_jobids + arg;
+   // If a filter is set, add it to the current query
+   if (FilterEntry->text() != "") {
+      QString tmp = FilterEntry->text();
+      tmp.replace("\"", ".");   // basic escape of "
+      arg += " pattern=\"" + tmp + "\"";
+   }
+
+   LocationEntry->setText(m_path);
+   QString offset = QString().setNum(Offset1Spin->value());
+   QString limit=QString().setNum(Offset2Spin->value() - Offset1Spin->value());
+   QString q = ".bvfs_lsdir jobid=" + m_jobids + arg 
+      + " limit=" + limit + " offset=" + offset ;
+   if (mainWin->m_miscDebug) qDebug() << q;
    if (m_console->dir_cmd(q, results)) {
       nb = results.size();
       FileList->setRowCount(nb);
@@ -188,13 +210,17 @@ void bRestore::displayFiles(int64_t pathid, QString path)
          item.setTextFld(col++, fieldlist.at(5)); // path
          decode_stat(fieldlist.at(4).toLocal8Bit().data(), 
                      &statp, &LinkFI);
+         item.setBytesFld(col++, QString().setNum(statp.st_size));
          item.setDateFld(col++, statp.st_mtime); // date
-         item.widget(1)->setData(Qt::UserRole, fieldlist.join("\t")); // keep info
+         fieldlist.replace(3, m_jobids);      // use current jobids selection
+         // keep original info on the first cel that is never empty
+         item.widget(1)->setData(Qt::UserRole, fieldlist.join("\t"));
       }
    }
 
    results.clear();
-   q = ".bvfs_lsfiles jobid=" + m_jobids + arg;
+   q = ".bvfs_lsfiles jobid=" + m_jobids + arg
+      + " limit=" + limit + " offset=" + offset ;
    if (m_console->dir_cmd(q, results)) {
       FileList->setRowCount(results.size() + nb);
       foreach (QString resultline, results) {
@@ -207,6 +233,7 @@ void bRestore::displayFiles(int64_t pathid, QString path)
                      &statp, &LinkFI);
          item.setBytesFld(col++, QString().setNum(statp.st_size));
          item.setDateFld(col++, statp.st_mtime);
+         // keep original info on the first cel that is never empty
          item.widget(1)->setData(Qt::UserRole, fieldlist.join("\t")); // keep info
       }
    }
@@ -226,6 +253,7 @@ void bRestore::PgSeltreeWidgetClicked()
    }
 }
 
+// Display all versions of a file for this client
 void bRestore::displayFileVersion(QString pathid, QString fnid, 
                                   QString client, QString filename)
 {
@@ -269,7 +297,9 @@ void bRestore::displayFileVersion(QString pathid, QString fnid,
          fieldlist.removeLast(); // volname
          fieldlist.removeLast(); // md5
          fieldlist << m_path + filename;
-         item.widget(1)->setData(Qt::UserRole, fieldlist.join("\t")); // keep info
+
+         // keep original info on the first cel that is never empty
+         item.widget(1)->setData(Qt::UserRole, fieldlist.join("\t"));
       }
    }
    FileRevisions->verticalHeader()->hide();
@@ -304,6 +334,18 @@ void bRestore::clearVersions(QTableWidgetItem *item)
    m_current = item ;
 }
 
+void bRestore::clearRestoreList()
+{
+   RestoreList->clearContents();
+   RestoreList->setRowCount(0);
+}
+
+void bRestore::runRestore()
+{
+   bRunRestore *r = new bRunRestore(this);
+   r->setVisible(true);
+}
+
 void bRestore::setupPage()
 {
    ClientList->addItem("Client list");
@@ -316,7 +358,9 @@ void bRestore::setupPage()
            this, SLOT(showInfoForFile(QTableWidgetItem *)));
    connect(LocationBp, SIGNAL(pressed()), this, SLOT(applyLocation()));
    connect(MergeChk, SIGNAL(clicked()), this, SLOT(setJob()));
-
+   connect(ClearBp, SIGNAL(clicked()), this, SLOT(clearRestoreList()));
+   connect(RestoreBp, SIGNAL(clicked()), this, SLOT(runRestore()));
+   connect(FilterBp, SIGNAL(clicked()), this, SLOT(refreshView()));
    m_populated = true;
 }
 
@@ -324,6 +368,7 @@ bRestore::~bRestore()
 {
 }
 
+// Drag & Drop handling, not so easy...
 void bRestoreTable::mousePressEvent(QMouseEvent *event)
 {
    QTableWidget::mousePressEvent(event);
@@ -352,7 +397,7 @@ void bRestoreTable::mouseMoveEvent(QMouseEvent *event)
    }
 
    QList<QTableWidgetItem *> lst = selectedItems();
-   qDebug() << this << " selectedItems: " << lst;
+   if (mainWin->m_miscDebug) qDebug() << this << " selectedItems: " << lst;
    if (lst.isEmpty()) {
       return;
    }
@@ -396,6 +441,7 @@ void bRestoreTable::dragMoveEvent(QDragMoveEvent *event)
    }
 }
 
+// When user releases the button
 void bRestoreTable::dropEvent(QDropEvent *event)
 {
    int col=1;
@@ -419,6 +465,7 @@ void bRestoreTable::dropEvent(QDropEvent *event)
       item.setDateFld(col++, statp.st_mtime); // date
       item.setNumericFld(col++, fields.at(3)); // jobid
       item.setNumericFld(col++, fields.at(2)); // fileid
+      // keep original info on the first cel that is never empty
       item.widget(1)->setData(Qt::UserRole, event->mimeData()->text());
       event->acceptProposedAction();
    } else {
@@ -426,3 +473,248 @@ void bRestoreTable::dropEvent(QDropEvent *event)
    }
 }
 
+// Use File Relocation bp
+void bRunRestore::UFRcb()
+{
+   if (UseFileRelocationChk->checkState() == Qt::Checked) {
+      WhereEntry->setEnabled(false);
+      UseRegexpChk->setEnabled(true);
+      if (UseRegexpChk->checkState() == Qt::Checked) {
+         AddSuffixEntry->setEnabled(false);
+         AddPrefixEntry->setEnabled(false);
+         StripPrefixEntry->setEnabled(false);
+         WhereRegexpEntry->setEnabled(true);
+      } else {
+         AddSuffixEntry->setEnabled(true);
+         AddPrefixEntry->setEnabled(true);
+         StripPrefixEntry->setEnabled(true);
+         WhereRegexpEntry->setEnabled(false);
+      }
+   } else {
+      WhereEntry->setEnabled(true);
+      AddSuffixEntry->setEnabled(false);
+      AddPrefixEntry->setEnabled(false);
+      StripPrefixEntry->setEnabled(false);
+      UseRegexpChk->setEnabled(false);
+      WhereRegexpEntry->setEnabled(false);
+   }
+}
+
+// Expert mode for file relocation
+void bRunRestore::useRegexp()
+{
+   if (UseRegexpChk->checkState() == Qt::Checked) {
+      AddSuffixEntry->setEnabled(false);
+      AddPrefixEntry->setEnabled(false);
+      StripPrefixEntry->setEnabled(false);
+      WhereRegexpEntry->setEnabled(true);
+   } else {
+      AddSuffixEntry->setEnabled(true);
+      AddPrefixEntry->setEnabled(true);
+      StripPrefixEntry->setEnabled(true);
+      WhereRegexpEntry->setEnabled(false);
+   }
+}
+
+// Display Form to run the restore job
+bRunRestore::bRunRestore(bRestore *parent)
+{
+   brestore = parent;
+   setupUi(this);
+   ClientCb->addItems(parent->console()->client_list);
+   int i = ClientCb->findText(parent->m_client);
+   if (i >= 0) {
+      ClientCb->setCurrentIndex(i);
+   }
+   StorageCb->addItem(QString(""));
+   RestoreCb->addItems(parent->console()->restore_list);
+   WhenEditor->setDateTime(QDateTime::currentDateTime());
+   StorageCb->addItems(parent->console()->storage_list);
+   connect(UseFileRelocationChk, SIGNAL(clicked()), this, SLOT(UFRcb()));
+   connect(UseRegexpChk, SIGNAL(clicked()), this, SLOT(useRegexp()));
+   connect(ActionBp, SIGNAL(accepted()), this, SLOT(computeRestore()));
+   // TODO: handle multiple restore job
+   struct job_defaults jd;
+   if (parent->console()->restore_list.size() > 0) {
+      jd.job_name = parent->console()->restore_list[0];
+      brestore->console()->get_job_defaults(jd);
+      WhereEntry->setText(jd.where);
+   }
+   computeVolumeList();
+}
+
+void bRestore::get_info_from_selection(QStringList &fileids, 
+                                       QStringList &jobids,
+                                       QStringList &dirids,
+                                       QStringList &findexes)
+{
+   struct stat statp;
+   int32_t LinkFI;
+   for (int i=0; i < RestoreList->rowCount(); i++) {
+      QTableWidgetItem *item = RestoreList->item(i, 1);
+      QString data = item->data(Qt::UserRole).toString();
+      QStringList lst = data.split("\t");
+      if (lst.at(1) != "0") {   // skip path
+         fileids << lst.at(2);
+         jobids << lst.at(3);
+         decode_stat(lst.at(4).toLocal8Bit().data(), 
+                     &statp, &LinkFI);
+         if (LinkFI) {
+            findexes << lst.at(3) + "," + QString().setNum(LinkFI);
+         }
+      } else {
+         dirids << lst.at(0);
+         jobids << lst.at(3).split(","); // Can have multiple jobids
+      }
+   }
+   fileids.removeDuplicates();
+   jobids.removeDuplicates();
+   dirids.removeDuplicates();
+   findexes.removeDuplicates();
+}
+
+// To compute volume list with directories, query is much slower
+void bRunRestore::computeVolumeList()
+{
+   brestore->get_info_from_selection(m_fileids, m_jobids, m_dirids, m_findexes);
+   if (m_fileids.size() == 0) {
+      return;
+   }
+
+   Freeze frz_lst(*TableMedia); /* disable updating*/
+   QString q =
+" SELECT DISTINCT VolumeName, Enabled, InChanger "
+ " FROM File, "
+  " ( " // -- Get all media from this job 
+     " SELECT MIN(FirstIndex) AS FirstIndex, MAX(LastIndex) AS LastIndex, "
+            " VolumeName, Enabled, Inchanger "
+       " FROM JobMedia JOIN Media USING (MediaId) "
+      " WHERE JobId IN (" + m_jobids.join(",") + ") "
+      " GROUP BY VolumeName,Enabled,InChanger "
+   " ) AS allmedia "
+  " WHERE File.FileId IN (" + m_fileids.join(",") + ") "
+   " AND File.FileIndex >= allmedia.FirstIndex "
+   " AND File.FileIndex <= allmedia.LastIndex ";
+   int row=0;
+   QStringList results;
+   if (brestore->console()->sql_cmd(q, results)) {
+      QStringList fieldlist;
+      TableMedia->setRowCount(results.size());
+      /* Iterate through the record returned from the query */
+      foreach (QString resultline, results) {
+         // 0        1          2 
+         //volname, enabled, inchanger
+         fieldlist = resultline.split("\t");
+         int col=0;
+         TableItemFormatter item(*TableMedia, row++);
+         item.setInChanger(col++, fieldlist.at(2));    // inchanger
+         item.setTextFld(col++, fieldlist.at(0));      // Volume
+      }
+   }
+   TableMedia->verticalHeader()->hide();
+   TableMedia->resizeColumnsToContents();
+   TableMedia->resizeRowsToContents();
+   TableMedia->setEditTriggers(QAbstractItemView::NoEditTriggers);
+}
+
+int64_t bRunRestore::runRestore(QString tablename)
+{
+   QString q;
+   QString tmp;
+
+   tmp = ClientCb->currentText();
+   if (tmp == "") {
+      return 0;
+   }
+   q = "restore client=" + tmp;
+
+   tmp = CommentEntry->text();
+   if (tmp != "") {
+      tmp.replace("\"", " ");
+      q += " comment=\"" + tmp + "\"";
+   }
+
+   tmp = StorageCb->currentText();
+   if (tmp != "") {
+      q += " storage=" + tmp;
+   }
+
+   if (UseFileRelocationChk->checkState() == Qt::Checked) {
+      if (UseRegexpChk->checkState() == Qt::Checked) {
+         tmp = WhereRegexpEntry->text();
+         if (tmp != "") {
+            tmp.replace("\"", "");
+            q += " regexwhere=\"" + tmp + "\"";
+         }
+      } else {
+         QStringList lst;
+         tmp = StripPrefixEntry->text();
+         if (tmp != "") {
+            tmp.replace("\"", "");
+            lst.append("!" + tmp + "!!i");
+         }
+         tmp = AddPrefixEntry->text();
+         if (tmp != "") {
+            tmp.replace("\"", "");
+            lst.append("!^!" + tmp + "!");
+         }
+         tmp = AddSuffixEntry->text();
+         if (tmp != "") {
+            tmp.replace("\"", "");
+            lst.append("!([^/])$!$1" + tmp + "!");
+         }
+         if (lst.size() > 0) {
+            q += " regexwhere=\"" + lst.join(",") + "\"";
+         }
+      }
+   } else {
+      tmp = WhereEntry->text();
+      if (tmp != "") {
+         tmp.replace("\"", "");
+         q += " where=\"" + tmp + "\"";
+      }
+   }
+
+//   q += " priority=" + tmp.setNum(PrioritySb->value());
+//   q += " job=\"" + RestoreCb->currentText() + "\"";
+   q += " file=\"?" + tablename + "\"";
+   q += " when=\"" + WhenEditor->dateTime().toString("yyyy-MM-dd hh:mm:ss") + "\"";
+   q += " done yes";
+   
+   if (mainWin->m_miscDebug) qDebug() << q;
+   QStringList results;
+   if (brestore->console()->dir_cmd(q, results)) {
+      foreach (QString resultline, results) {
+         QStringList fieldlist = resultline.split("=");
+         if (fieldlist.size() == 2) {
+            return fieldlist.at(1).toLongLong();
+         }
+      }
+   }
+   return 0;
+}
+
+void bRunRestore::computeRestore()
+{
+   QString q = ".bvfs_restore path=b2123 jobid=" + m_jobids.join(",");
+   if (m_fileids.size() > 0) {
+      q += " fileid=" + m_fileids.join(",");
+   }
+   if (m_dirids.size() > 0) {
+      q += " dirid=" + m_dirids.join(",");
+   }
+   if (m_findexes.size() > 0) {
+      q += " hardlink=" + m_findexes.join(",");
+   }
+   if (mainWin->m_miscDebug) qDebug() << q;
+
+   QStringList results;
+   if (brestore->console()->dir_cmd(q, results)) {
+      if (results.size() == 1 && results[0] == "OK") {
+         int64_t jobid = runRestore("b2123");
+         if (mainWin->m_miscDebug) qDebug() << "jobid=" << jobid;
+         q = ".bvfs_cleanup path=b2123";
+         brestore->console()->dir_cmd(q, results);
+      }
+   }   
+}