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