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

   1 /*
   2    Bacula® - The Network Backup Solution
   3
   4    Copyright (C) 2007-2009 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  *  MediaList Class
  33  *
  34  *   Dirk Bartley, March 2007
  35  *
  36  */
  37
  38 #include "bat.h"
  39 #include <QAbstractEventDispatcher>
  40 #include <QMenu>
  41 #include <math.h>
  42 #include "medialist.h"
  43 #include "mediaedit/mediaedit.h"
  44 #include "mediainfo/mediainfo.h"
  45 #include "joblist/joblist.h"
  46 #include "relabel/relabel.h"
  47 #include "run/run.h"
  48 #include "util/fmtwidgetitem.h"
  49
  50 MediaList::MediaList()
  51 {
  52    setupUi(this);
  53    m_name = tr("Pools");
  54    pgInitialize();
  55    QTreeWidgetItem* thisitem = mainWin->getFromHash(this);
  56    thisitem->setIcon(0,QIcon(QString::fromUtf8(":images/cartridge.png")));
  57
  58    /* mp_treeWidget, Storage Tree Tree Widget inherited from ui_medialist.h */
  59    m_populated = false;
  60    m_checkcurwidget = true;
  61    m_closeable = false;
  62    /* add context sensitive menu items specific to this classto the page
  63     * selector tree. m_contextActions is QList of QActions */
  64    m_contextActions.append(actionRefreshMediaList);
  65 }
  66
  67 MediaList::~MediaList()
  68 {
  69    if (m_populated)
  70       writeExpandedSettings();
  71 }
  72
  73 /*
  74  * The main meat of the class!!  The function that querries the director and
  75  * creates the widgets with appropriate values.
  76  */
  77 void MediaList::populateTree()
  78 {
  79    if (m_populated)
  80       writeExpandedSettings();
  81    m_populated = true;
  82
  83    Freeze frz(*mp_treeWidget); /* disable updating*/
  84
  85    QStringList headerlist = (QStringList()
  86       << tr("Volume Name") << tr("Id") << tr("Status") << tr("Enabled") << tr("Bytes") << tr("Files")
  87       << tr("Jobs") << tr("Retention") << tr("Media Type") << tr("Slot") << tr("Use Duration")
  88       << tr("Max Jobs") << tr("Max Files") << tr("Max Bytes") << tr("Recycle")
  89       << tr("Last Written") << tr("First Written") << tr("Read Time")
  90       << tr("Write Time") << tr("Recycle Count") << tr("Recycle Pool"));
  91
  92    m_checkcurwidget = false;
  93    mp_treeWidget->clear();
  94    m_checkcurwidget = true;
  95    mp_treeWidget->setColumnCount(headerlist.count());
  96    m_topItem = new QTreeWidgetItem(mp_treeWidget);
  97    m_topItem->setText(0, tr("Pools"));
  98    m_topItem->setData(0, Qt::UserRole, 0);
  99    m_topItem->setExpanded(true);
 100
 101    mp_treeWidget->setHeaderLabels(headerlist);
 102
 103    QSettings settings(m_console->m_dir->name(), "bat");
 104    settings.beginGroup("MediaListTreeExpanded");
 105    QString query;
 106
 107    QTreeWidgetItem *pooltreeitem;
 108
 109    /* Comma separated list of pools first */
 110    bool first = true;
 111    QString pool_comsep("");
 112    foreach (QString pool_listItem, m_console->pool_list) {
 113       if (first) {
 114          pool_comsep += "'" + pool_listItem + "'";
 115          first = false;
 116       }
 117       else
 118          pool_comsep += ",'" + pool_listItem + "'";
 119    }
 120    /* Now use pool_comsep list to perform just one query */
 121    if (pool_comsep != "") {
 122       query =  "SELECT Pool.Name AS pul,"
 123          " Media.VolumeName AS Media, "
 124          " Media.MediaId AS Id, Media.VolStatus AS VolStatus,"
 125          " Media.Enabled AS Enabled, Media.VolBytes AS Bytes,"
 126          " Media.VolFiles AS FileCount, Media.VolJobs AS JobCount,"
 127          " Media.VolRetention AS VolumeRetention, Media.MediaType AS MediaType,"
 128          " Media.InChanger AS InChanger, Media.Slot AS Slot, "
 129          " Media.VolUseDuration AS UseDuration,"
 130          " Media.MaxVolJobs AS MaxJobs, Media.MaxVolFiles AS MaxFiles,"
 131          " Media.MaxVolBytes AS MaxBytes, Media.Recycle AS Recycle,"
 132          " Media.LastWritten AS LastWritten,"
 133          " Media.FirstWritten AS FirstWritten,"
 134          " (VolReadTime/1000000) AS ReadTime, (VolWriteTime/1000000) AS WriteTime,"
 135          " RecycleCount AS ReCyCount,"
 136          " RecPool.Name AS RecyclePool"
 137          " FROM Media"
 138          " JOIN Pool ON (Media.PoolId=Pool.PoolId)"
 139          " LEFT OUTER JOIN Pool AS RecPool ON (Media.RecyclePoolId=RecPool.PoolId)"
 140          " WHERE ";
 141       query += " Pool.Name IN (" + pool_comsep + ")";
 142       query += " ORDER BY Pool.Name, Media";
 143
 144       if (mainWin->m_sqlDebug) {
 145          Pmsg1(000, "MediaList query cmd : %s\n",query.toUtf8().data());
 146       }
 147       QStringList results;
 148       int counter = 0;
 149       if (m_console->sql_cmd(query, results)) {
 150          QStringList fieldlist;
 151          QString prev_pool("");
 152          QString this_pool("");
 153
 154          /* Iterate through the lines of results. */
 155          foreach (QString resultline, results) {
 156             fieldlist = resultline.split("\t");
 157             this_pool = fieldlist.takeFirst();
 158             if (prev_pool != this_pool) {
 159                prev_pool = this_pool;
 160                pooltreeitem = new QTreeWidgetItem(m_topItem);
 161                pooltreeitem->setText(0, this_pool);
 162                pooltreeitem->setData(0, Qt::UserRole, 1);
 163             }
 164             if(settings.contains(this_pool)) {
 165                pooltreeitem->setExpanded(settings.value(this_pool).toBool());
 166             } else {
 167                pooltreeitem->setExpanded(true);
 168             }
 169
 170             if (fieldlist.size() < 18)
 171                continue; // some fields missing, ignore row
 172
 173             int index = 0;
 174             TreeItemFormatter mediaitem(*pooltreeitem, 2);
 175
 176             /* Iterate through fields in the record */
 177             QStringListIterator fld(fieldlist);
 178
 179             /* volname */
 180             mediaitem.setTextFld(index++, fld.next());
 181
 182             /* id */
 183             mediaitem.setNumericFld(index++, fld.next());
 184
 185             /* status */
 186             mediaitem.setVolStatusFld(index++, fld.next());
 187
 188             /* enabled */
 189             mediaitem.setBoolFld(index++, fld.next());
 190
 191             /* bytes */
 192             mediaitem.setBytesFld(index++, fld.next());
 193
 194             /* files */
 195             mediaitem.setNumericFld(index++, fld.next());
 196
 197             /* jobs */
 198             mediaitem.setNumericFld(index++, fld.next());
 199
 200             /* retention */
 201             mediaitem.setDurationFld(index++, fld.next());
 202
 203             /* media type */
 204             mediaitem.setTextFld(index++, fld.next());
 205
 206             /* inchanger + slot */
 207             int inchanger = fld.next().toInt();
 208             if (inchanger) {
 209                mediaitem.setNumericFld(index++, fld.next());
 210             }
 211             else {
 212                /* volume not in changer, show blank slot */
 213                mediaitem.setNumericFld(index++, "");
 214                fld.next();
 215             }
 216
 217             /* use duration */
 218             mediaitem.setDurationFld(index++, fld.next());
 219
 220             /* max jobs */
 221             mediaitem.setNumericFld(index++, fld.next());
 222
 223             /* max files */
 224             mediaitem.setNumericFld(index++, fld.next());
 225
 226             /* max bytes */
 227             mediaitem.setBytesFld(index++, fld.next());
 228
 229             /* recycle */
 230             mediaitem.setBoolFld(index++, fld.next());
 231
 232             /* last written */
 233             mediaitem.setTextFld(index++, fld.next());
 234
 235             /* first written */
 236             mediaitem.setTextFld(index++, fld.next());
 237
 238             /* read time */
 239             mediaitem.setDurationFld(index++, fld.next());
 240
 241             /* write time */
 242             mediaitem.setDurationFld(index++, fld.next());
 243
 244             /* Recycle Count */
 245             mediaitem.setNumericFld(index++, fld.next());
 246
 247             /* recycle pool */
 248             mediaitem.setTextFld(index++, fld.next());
 249
 250          } /* foreach resultline */
 251          counter += 1;
 252       } /* if results from query */
 253    } /* foreach pool_listItem */
 254    settings.endGroup();
 255    /* Resize the columns */
 256    for(int cnter=0; cnter<headerlist.count(); cnter++) {
 257       mp_treeWidget->resizeColumnToContents(cnter);
 258    }
 259 }
 260
 261 /*
 262  * Called from the signal of the context sensitive menu!
 263  */
 264 void MediaList::editVolume()
 265 {
 266    MediaEdit* edit = new MediaEdit(mainWin->getFromHash(this), m_currentVolumeId);
 267    connect(edit, SIGNAL(destroyed()), this, SLOT(populateTree()));
 268 }
 269
 270 /*
 271  * Called from the signal of the context sensitive menu!
 272  */
 273 void MediaList::showJobs()
 274 {
 275    QTreeWidgetItem *parentItem = mainWin->getFromHash(this);
 276    mainWin->createPageJobList(m_currentVolumeName, "", "", "", parentItem);
 277 }
 278
 279 /*
 280  * Called from the signal of the context sensitive menu!
 281  */
 282 void MediaList::viewVolume()
 283 {
 284    QTreeWidgetItem *parentItem = mainWin->getFromHash(this);
 285    MediaInfo* view = new MediaInfo(parentItem, m_currentVolumeName);
 286    connect(view, SIGNAL(destroyed()), this, SLOT(populateTree()));
 287
 288 }
 289
 290 /*
 291  * When the treeWidgetItem in the page selector tree is singleclicked, Make sure
 292  * The tree has been populated.
 293  */
 294 void MediaList::PgSeltreeWidgetClicked()
 295 {
 296    if (!m_populated) {
 297       populateTree();
 298       createContextMenu();
 299    }
 300    if (!isOnceDocked()) {
 301       dockPage();
 302    }
 303 }
 304
 305 /*
 306  * Added to set the context menu policy based on currently active treeWidgetItem
 307  * signaled by currentItemChanged
 308  */
 309 void MediaList::treeItemChanged(QTreeWidgetItem *currentwidgetitem, QTreeWidgetItem *previouswidgetitem )
 310 {
 311    /* m_checkcurwidget checks to see if this is during a refresh, which will segfault */
 312    if (m_checkcurwidget) {
 313       /* The Previous item */
 314       if (previouswidgetitem) { /* avoid a segfault if first time */
 315          foreach(QAction* mediaAction, mp_treeWidget->actions()) {
 316             mp_treeWidget->removeAction(mediaAction);
 317          }
 318       }
 319
 320       int treedepth = currentwidgetitem->data(0, Qt::UserRole).toInt();
 321       m_currentVolumeName=currentwidgetitem->text(0);
 322       mp_treeWidget->addAction(actionRefreshMediaList);
 323       if (treedepth == 2){
 324          m_currentVolumeId=currentwidgetitem->text(1);
 325          mp_treeWidget->addAction(actionEditVolume);
 326          mp_treeWidget->addAction(actionListJobsOnVolume);
 327          mp_treeWidget->addAction(actionDeleteVolume);
 328          mp_treeWidget->addAction(actionPruneVolume);
 329          mp_treeWidget->addAction(actionPurgeVolume);
 330          mp_treeWidget->addAction(actionRelabelVolume);
 331          mp_treeWidget->addAction(actionVolumeFromPool);
 332       } else if (treedepth == 1) {
 333          mp_treeWidget->addAction(actionAllVolumesFromPool);
 334       }
 335    }
 336 }
 337
 338 /*
 339  * Setup a context menu
 340  * Made separate from populate so that it would not create context menu over and
 341  * over as the tree is repopulated.
 342  */
 343 void MediaList::createContextMenu()
 344 {
 345    mp_treeWidget->setContextMenuPolicy(Qt::ActionsContextMenu);
 346    connect(mp_treeWidget, SIGNAL(itemDoubleClicked(QTreeWidgetItem *, int)), this, SLOT(viewVolume()));
 347    connect(actionEditVolume, SIGNAL(triggered()), this, SLOT(editVolume()));
 348    connect(actionListJobsOnVolume, SIGNAL(triggered()), this, SLOT(showJobs()));
 349    connect(actionDeleteVolume, SIGNAL(triggered()), this, SLOT(deleteVolume()));
 350    connect(actionPurgeVolume, SIGNAL(triggered()), this, SLOT(purgeVolume()));
 351    connect(actionPruneVolume, SIGNAL(triggered()), this, SLOT(pruneVolume()));
 352    connect(actionRelabelVolume, SIGNAL(triggered()), this, SLOT(relabelVolume()));
 353    connect(mp_treeWidget, SIGNAL(
 354            currentItemChanged(QTreeWidgetItem *, QTreeWidgetItem *)),
 355            this, SLOT(treeItemChanged(QTreeWidgetItem *, QTreeWidgetItem *)));
 356    /* connect to the action specific to this pages class */
 357    connect(actionRefreshMediaList, SIGNAL(triggered()), this,
 358                 SLOT(populateTree()));
 359    connect(actionAllVolumes, SIGNAL(triggered()), this, SLOT(allVolumes()));
 360    connect(actionAllVolumesFromPool, SIGNAL(triggered()), this, SLOT(allVolumesFromPool()));
 361    connect(actionVolumeFromPool, SIGNAL(triggered()), this, SLOT(volumeFromPool()));
 362 }
 363
 364 /*
 365  * Virtual function which is called when this page is visible on the stack
 366  */
 367 void MediaList::currentStackItem()
 368 {
 369    if(!m_populated) {
 370       populateTree();
 371       /* Create the context menu for the medialist tree */
 372       createContextMenu();
 373    }
 374 }
 375
 376 /*
 377  * Called from the signal of the context sensitive menu to delete a volume!
 378  */
 379 void MediaList::deleteVolume()
 380 {
 381    if (QMessageBox::warning(this, "Bat",
 382       tr("Are you sure you want to delete??  !!!.\n"
 383 "This delete command is used to delete a Volume record and all associated catalog"
 384 " records that were created. This command operates only on the Catalog"
 385 " database and has no effect on the actual data written to a Volume. This"
 386 " command can be dangerous and we strongly recommend that you do not use"
 387 " it unless you know what you are doing.  All Jobs and all associated"
 388 " records (File and JobMedia) will be deleted from the catalog."
 389       "Press OK to proceed with delete operation.?"),
 390       QMessageBox::Ok | QMessageBox::Cancel)
 391       == QMessageBox::Cancel) { return; }
 392
 393    QString cmd("delete volume=");
 394    cmd += m_currentVolumeName;
 395    consoleCommand(cmd);
 396 }
 397
 398 /*
 399  * Called from the signal of the context sensitive menu to purge!
 400  */
 401 void MediaList::purgeVolume()
 402 {
 403    if (QMessageBox::warning(this, "Bat",
 404       tr("Are you sure you want to purge ??  !!!.\n"
 405 "The Purge command will delete associated Catalog database records from Jobs and"
 406 " Volumes without considering the retention period. Purge  works only on the"
 407 " Catalog database and does not affect data written to Volumes. This command can"
 408 " be dangerous because you can delete catalog records associated with current"
 409 " backups of files, and we recommend that you do not use it unless you know what"
 410 " you are doing.\n"
 411       "Press OK to proceed with the purge operation?"),
 412       QMessageBox::Ok | QMessageBox::Cancel)
 413       == QMessageBox::Cancel) { return; }
 414
 415    QString cmd("purge volume=");
 416    cmd += m_currentVolumeName;
 417    consoleCommand(cmd);
 418    populateTree();
 419 }
 420
 421 /*
 422  * Called from the signal of the context sensitive menu to prune!
 423  */
 424 void MediaList::pruneVolume()
 425 {
 426    new prunePage(m_currentVolumeName, "");
 427 }
 428
 429 /*
 430  * Called from the signal of the context sensitive menu to relabel!
 431  */
 432 void MediaList::relabelVolume()
 433 {
 434    setConsoleCurrent();
 435    new relabelDialog(m_console, m_currentVolumeName);
 436 }
 437
 438 /*
 439  * Called from the signal of the context sensitive menu to purge!
 440  */
 441 void MediaList::allVolumesFromPool()
 442 {
 443    QString cmd = "update volume AllFromPool=" + m_currentVolumeName;
 444    consoleCommand(cmd);
 445    populateTree();
 446 }
 447
 448 void MediaList::allVolumes()
 449 {
 450    QString cmd = "update volume allfrompools";
 451    consoleCommand(cmd);
 452    populateTree();
 453 }
 454
 455 /*
 456  * Called from the signal of the context sensitive menu to purge!
 457  */
 458 void MediaList::volumeFromPool()
 459 {
 460    QTreeWidgetItem *currentItem = mp_treeWidget->currentItem();
 461    QTreeWidgetItem *parent = currentItem->parent();
 462    QString pool = parent->text(0);
 463    QString cmd;
 464    cmd = "update volume=" + m_currentVolumeName + " frompool=" + pool;
 465    consoleCommand(cmd);
 466    populateTree();
 467 }
 468
 469 /*
 470  * Write settings to save expanded states of the pools
 471  */
 472 void MediaList::writeExpandedSettings()
 473 {
 474    if (m_topItem) {
 475       QSettings settings(m_console->m_dir->name(), "bat");
 476       settings.beginGroup("MediaListTreeExpanded");
 477       int childcount = m_topItem->childCount();
 478       for (int cnt=0; cnt<childcount; cnt++) {
 479          QTreeWidgetItem *poolitem = m_topItem->child(cnt);
 480          settings.setValue(poolitem->text(0), poolitem->isExpanded());
 481       }
 482       settings.endGroup();
 483    }
 484 }