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) 2000-2007 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 plus additions
  11    that are listed 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: medialist.cpp 4230 2007-02-21 20:07:37Z kerns $
  31  *
  32  *  MediaList Class
  33  *
  34  *   Dirk Bartley, March 2007
  35  *
  36  */
  37
  38 #include <QAbstractEventDispatcher>
  39 #include <QMenu>
  40 #include "bat.h"
  41 #include "medialist.h"
  42 #include "mediaedit/mediaedit.h"
  43 #include "joblist/joblist.h"
  44 #include "relabel/relabel.h"
  45
  46 MediaList::MediaList()
  47 {
  48    setupUi(this);
  49    m_name = "Media";
  50    pgInitialize();
  51
  52    /* mp_treeWidget, Storage Tree Tree Widget inherited from ui_medialist.h */
  53    m_populated = false;
  54    m_checkcurwidget = true;
  55    m_closeable = false;
  56 }
  57
  58 MediaList::~MediaList()
  59 {
  60 }
  61
  62 /*
  63  * The main meat of the class!!  The function that querries the director and
  64  * creates the widgets with appropriate values.
  65  */
  66 void MediaList::populateTree()
  67 {
  68    QTreeWidgetItem *mediatreeitem, *pooltreeitem, *topItem;
  69
  70    if (!m_console->preventInUseConnect())
  71        return;
  72
  73    QStringList headerlist = (QStringList()
  74       << "Volume Name" << "Id" << "Status" << "Enabled"
  75       << "Bytes" << "Files" << "Jobs" << "Retention"
  76       << "Media Type" << "Slot" << "Last Written");
  77    int statusIndex = headerlist.indexOf("Status");
  78
  79    m_checkcurwidget = false;
  80    mp_treeWidget->clear();
  81    m_checkcurwidget = true;
  82    mp_treeWidget->setColumnCount(headerlist.count());
  83    topItem = new QTreeWidgetItem(mp_treeWidget);
  84    topItem->setText(0, "Pools");
  85    topItem->setData(0, Qt::UserRole, 0);
  86    topItem->setExpanded(true);
  87
  88    mp_treeWidget->setHeaderLabels(headerlist);
  89
  90    QString query;
  91
  92    foreach (QString pool_listItem, m_console->pool_list) {
  93       pooltreeitem = new QTreeWidgetItem(topItem);
  94       pooltreeitem->setText(0, pool_listItem);
  95       pooltreeitem->setData(0, Qt::UserRole, 1);
  96       pooltreeitem->setExpanded(true);
  97
  98       query =  "SELECT Media.VolumeName AS Media, "
  99          " Media.MediaId AS Id, Media.VolStatus AS VolStatus,"
 100          " Media.Enabled AS Enabled, Media.VolBytes AS Bytes,"
 101          " Media.VolFiles AS FileCount, Media.VolJobs AS JobCount,"
 102          " Media.VolRetention AS VolumeRetention, Media.MediaType AS MediaType,"
 103          " Media.Slot AS Slot, Media.LastWritten AS LastWritten"
 104          " FROM Media, Pool"
 105          " WHERE Media.PoolId=Pool.PoolId";
 106       query += " AND Pool.Name='" + pool_listItem + "'";
 107       query += " ORDER BY Media";
 108
 109       if (mainWin->m_sqlDebug) {
 110          Pmsg1(000, "MediaList query cmd : %s\n",query.toUtf8().data());
 111       }
 112       QStringList results;
 113       if (m_console->sql_cmd(query, results)) {
 114          QString field;
 115          QStringList fieldlist;
 116
 117          /* Iterate through the lines of results. */
 118          foreach (QString resultline, results) {
 119             fieldlist = resultline.split("\t");
 120             int index = 0;
 121             mediatreeitem = new QTreeWidgetItem(pooltreeitem);
 122
 123             /* Iterate through fields in the record */
 124             foreach (field, fieldlist) {
 125                field = field.trimmed();  /* strip leading & trailing spaces */
 126                if (field != "") {
 127                   mediatreeitem->setData(index, Qt::UserRole, 2);
 128                   mediatreeitem->setData(index, Qt::UserRole, 2);
 129                   mediatreeitem->setText(index, field);
 130                   if (index == statusIndex) {
 131                      setStatusColor(mediatreeitem, field, index);
 132                   }
 133                }
 134                index++;
 135             } /* foreach field */
 136          } /* foreach resultline */
 137       } /* if results from query */
 138    } /* foreach pool_listItem */
 139    /* Resize the columns */
 140    for(int cnter=0; cnter<headerlist.count(); cnter++) {
 141       mp_treeWidget->resizeColumnToContents(cnter);
 142    }
 143 }
 144
 145 void MediaList::setStatusColor(QTreeWidgetItem *item, QString &field, int &index)
 146 {
 147    if (field == "Append" ) {
 148       item->setBackground(index, Qt::green);
 149    } else if (field == "Error") {
 150       item->setBackground(index, Qt::red);
 151    } else if ((field == "Used") || ("Full")){
 152       item->setBackground(index, Qt::yellow);
 153    }
 154 }
 155
 156 /*
 157  * Called from the signal of the context sensitive menu!
 158  */
 159 void MediaList::editVolume()
 160 {
 161    MediaEdit* edit = new MediaEdit(m_console, m_currentVolumeId);
 162    connect(edit, SIGNAL(destroyed()), this, SLOT(populateTree()));
 163 }
 164
 165 /*
 166  * Called from the signal of the context sensitive menu!
 167  */
 168 void MediaList::showJobs()
 169 {
 170    QString emptyclient("");
 171    QTreeWidgetItem *parentItem = mainWin->getFromHash(this);
 172    mainWin->createPageJobList(m_currentVolumeName, emptyclient, parentItem);
 173 }
 174
 175 /*
 176  * When the treeWidgetItem in the page selector tree is singleclicked, Make sure
 177  * The tree has been populated.
 178  */
 179 void MediaList::PgSeltreeWidgetClicked()
 180 {
 181    if (!m_populated) {
 182       populateTree();
 183       createContextMenu();
 184       m_populated=true;
 185    }
 186 }
 187
 188 /*
 189  * Added to set the context menu policy based on currently active treeWidgetItem
 190  * signaled by currentItemChanged
 191  */
 192 void MediaList::treeItemChanged(QTreeWidgetItem *currentwidgetitem, QTreeWidgetItem *previouswidgetitem )
 193 {
 194    /* m_checkcurwidget checks to see if this is during a refresh, which will segfault */
 195    if (m_checkcurwidget) {
 196       /* The Previous item */
 197       if (previouswidgetitem) { /* avoid a segfault if first time */
 198          int treedepth = previouswidgetitem->data(0, Qt::UserRole).toInt();
 199          if (treedepth == 2){
 200             mp_treeWidget->removeAction(actionEditVolume);
 201             mp_treeWidget->removeAction(actionListJobsOnVolume);
 202             mp_treeWidget->removeAction(actionDeleteVolume);
 203             mp_treeWidget->removeAction(actionPurgeVolume);
 204             mp_treeWidget->removeAction(actionRelabelVolume);
 205          }
 206       }
 207
 208       int treedepth = currentwidgetitem->data(0, Qt::UserRole).toInt();
 209       if (treedepth == 2){
 210          m_currentVolumeName=currentwidgetitem->text(0);
 211          m_currentVolumeId=currentwidgetitem->text(1);
 212          mp_treeWidget->addAction(actionEditVolume);
 213          mp_treeWidget->addAction(actionListJobsOnVolume);
 214          mp_treeWidget->addAction(actionDeleteVolume);
 215          mp_treeWidget->addAction(actionPurgeVolume);
 216          mp_treeWidget->addAction(actionRelabelVolume);
 217       }
 218    }
 219 }
 220
 221 /*
 222  * Setup a context menu
 223  * Made separate from populate so that it would not create context menu over and
 224  * over as the tree is repopulated.
 225  */
 226 void MediaList::createContextMenu()
 227 {
 228    mp_treeWidget->setContextMenuPolicy(Qt::ActionsContextMenu);
 229    mp_treeWidget->addAction(actionRefreshMediaList);
 230    connect(actionEditVolume, SIGNAL(triggered()), this, SLOT(editVolume()));
 231    connect(actionListJobsOnVolume, SIGNAL(triggered()), this, SLOT(showJobs()));
 232    connect(actionDeleteVolume, SIGNAL(triggered()), this, SLOT(deleteVolume()));
 233    connect(actionPurgeVolume, SIGNAL(triggered()), this, SLOT(purgeVolume()));
 234    connect(actionRelabelVolume, SIGNAL(triggered()), this, SLOT(relabelVolume()));
 235    connect(mp_treeWidget, SIGNAL(
 236            currentItemChanged(QTreeWidgetItem *, QTreeWidgetItem *)),
 237            this, SLOT(treeItemChanged(QTreeWidgetItem *, QTreeWidgetItem *)));
 238    /* connect to the action specific to this pages class */
 239    connect(actionRefreshMediaList, SIGNAL(triggered()), this,
 240                 SLOT(populateTree()));
 241 }
 242
 243 /*
 244  * Virtual function which is called when this page is visible on the stack
 245  */
 246 void MediaList::currentStackItem()
 247 {
 248    if(!m_populated) {
 249       populateTree();
 250       /* add context sensitive menu items specific to this classto the page
 251        * selector tree. m_m_contextActions is QList of QActions, so this is
 252        * only done once with the first population. */
 253       m_contextActions.append(actionRefreshMediaList);
 254       /* Create the context menu for the medialist tree */
 255       createContextMenu();
 256       m_populated=true;
 257    }
 258 }
 259
 260 /*
 261  * Called from the signal of the context sensitive menu!
 262  */
 263 void MediaList::deleteVolume()
 264 {
 265    if (QMessageBox::warning(this, tr("Bat"),
 266       tr("Are you sure you want to delete??  !!!.\n"
 267 "This delete command is used to delete a Volume record and all associated catalog"
 268 " records that were created. This command operates only on the Catalog"
 269 " database and has no effect on the actual data written to a Volume. This"
 270 " command can be dangerous and we strongly recommend that you do not use"
 271 " it unless you know what you are doing.  All Jobs and all associated"
 272 " records (File and JobMedia) will be deleted from the catalog."
 273       "Press OK to proceed with delete operation.?"),
 274       QMessageBox::Ok | QMessageBox::Cancel)
 275       == QMessageBox::Cancel) { return; }
 276
 277    QString cmd("delete volume=");
 278    cmd += m_currentVolumeName;
 279    consoleCommand(cmd);
 280 }
 281 /*
 282  * Called from the signal of the context sensitive menu!
 283  */
 284 void MediaList::purgeVolume()
 285 {
 286    if (QMessageBox::warning(this, tr("Bat"),
 287       tr("Are you sure you want to purge ??  !!!.\n"
 288 "The Purge command will delete associated Catalog database records from Jobs and"
 289 " Volumes without considering the retention period. Purge  works only on the"
 290 " Catalog database and does not affect data written to Volumes. This command can"
 291 " be dangerous because you can delete catalog records associated with current"
 292 " backups of files, and we recommend that you do not use it unless you know what"
 293 " you are doing.\n"
 294       "Press OK to proceed with the purge operation?"),
 295       QMessageBox::Ok | QMessageBox::Cancel)
 296       == QMessageBox::Cancel) { return; }
 297
 298    QString cmd("purge volume=");
 299    cmd += m_currentVolumeName;
 300    consoleCommand(cmd);
 301    populateTree();
 302 }
 303 /*
 304  * Called from the signal of the context sensitive menu!
 305  */
 306 void MediaList::relabelVolume()
 307 {
 308    setConsoleCurrent();
 309    new relabelDialog(m_console, m_currentVolumeName);
 310 }