git.sur5r.net Git - bacula/bacula/blob - bacula/src/qt-console/storage/storage.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  *  Storage Class
  33  *
  34  *   Dirk Bartley, March 2007
  35  *
  36  */
  37
  38 #include "bat.h"
  39 #include <QAbstractEventDispatcher>
  40 #include <QMenu>
  41 #include "storage.h"
  42 #include "label/label.h"
  43 #include "mount/mount.h"
  44 #include "status/storstat.h"
  45 #include "util/fmtwidgetitem.h"
  46
  47 Storage::Storage()
  48 {
  49    setupUi(this);
  50    pgInitialize(tr("Storage"));
  51    QTreeWidgetItem* thisitem = mainWin->getFromHash(this);
  52    thisitem->setIcon(0,QIcon(QString::fromUtf8(":images/package-x-generic.png")));
  53
  54    /* mp_treeWidget, Storage Tree Tree Widget inherited from ui_storage.h */
  55    m_populated = false;
  56    m_firstpopulation = true;
  57    m_checkcurwidget = true;
  58    m_closeable = false;
  59    m_currentStorage = "";
  60    /* add context sensitive menu items specific to this classto the page
  61     * selector tree. m_contextActions is QList of QActions */
  62    m_contextActions.append(actionRefreshStorage);
  63 }
  64
  65 Storage::~Storage()
  66 {
  67    if (m_populated)
  68       writeExpandedSettings();
  69 }
  70
  71 /*
  72  * The main meat of the class!!  The function that querries the director and
  73  * creates the widgets with appropriate values.
  74  */
  75 void Storage::populateTree()
  76 {
  77    if (m_populated)
  78       writeExpandedSettings();
  79    m_populated = true;
  80
  81    Freeze frz(*mp_treeWidget); /* disable updating */
  82
  83    m_checkcurwidget = false;
  84    mp_treeWidget->clear();
  85    m_checkcurwidget = true;
  86
  87    QStringList headerlist = (QStringList() << tr("Name") << tr("Id")
  88         << tr("Changer") << tr("Slot") << tr("Status") << tr("Enabled") << tr("Pool")
  89         << tr("Media Type") );
  90
  91    m_topItem = new QTreeWidgetItem(mp_treeWidget);
  92    m_topItem->setText(0, tr("Storage"));
  93    m_topItem->setData(0, Qt::UserRole, 0);
  94    m_topItem->setExpanded(true);
  95
  96    mp_treeWidget->setColumnCount(headerlist.count());
  97    mp_treeWidget->setHeaderLabels(headerlist);
  98
  99    QSettings settings(m_console->m_dir->name(), "bat");
 100    settings.beginGroup("StorageTreeExpanded");
 101
 102    bool first = true;
 103    QString storage_comsep("");
 104    QString storageName;
 105    foreach(storageName, m_console->storage_list){
 106       if (first) {
 107          storage_comsep += "'" + storageName + "'";
 108          first = false;
 109       }
 110       else
 111          storage_comsep += ",'" + storageName + "'";
 112    }
 113    if (storage_comsep != "") {
 114
 115       /* Set up query QString and header QStringList */
 116       QString query("SELECT"
 117                " Name AS StorageName,"
 118                " StorageId AS ID, AutoChanger AS Changer"
 119                " FROM Storage "
 120                " WHERE StorageId IN (SELECT MAX(StorageId) FROM Storage WHERE");
 121       query += " Name IN (" + storage_comsep + ")";
 122       query += " GROUP BY Name) ORDER BY Name";
 123
 124       QStringList results;
 125       /* This could be a log item */
 126       if (mainWin->m_sqlDebug) {
 127          Pmsg1(000, "Storage query cmd : %s\n",query.toUtf8().data());
 128       }
 129       if (m_console->sql_cmd(query, results)) {
 130
 131          QStringList fieldlist;
 132          foreach (QString resultline, results) {
 133             fieldlist = resultline.split("\t");
 134             storageName = fieldlist.takeFirst();
 135             if (m_firstpopulation) {
 136                settingsOpenStatus(storageName);
 137             }
 138             TreeItemFormatter storageItem(*m_topItem, 1);
 139             storageItem.setTextFld(0, storageName);
 140             if(settings.contains(storageName))
 141                storageItem.widget()->setExpanded(settings.value(storageName).toBool());
 142             else
 143                storageItem.widget()->setExpanded(true);
 144
 145             int index = 1;
 146             QStringListIterator fld(fieldlist);
 147
 148             /* storage id */
 149             storageItem.setNumericFld(index++, fld.next() );
 150
 151             /* changer */
 152             QString changer = fld.next();
 153             storageItem.setBoolFld(index++, changer);
 154
 155             if (changer == "1")
 156                mediaList(storageItem.widget(), fieldlist.first());
 157          }
 158       }
 159    }
 160    /* Resize the columns */
 161    for(int cnter=0; cnter<headerlist.size(); cnter++) {
 162       mp_treeWidget->resizeColumnToContents(cnter);
 163    }
 164    m_firstpopulation = false;
 165 }
 166
 167 /*
 168  * For autochangers    A query to show the tapes in the changer.
 169  */
 170 void Storage::mediaList(QTreeWidgetItem *parent, const QString &storageID)
 171 {
 172    QString query("SELECT Media.VolumeName AS Media, Media.Slot AS Slot,"
 173                  " Media.VolStatus AS VolStatus, Media.Enabled AS Enabled,"
 174                  " Pool.Name AS MediaPool, Media.MediaType AS MediaType"
 175                  " From Media"
 176                  " JOIN Pool ON (Media.PoolId=Pool.PoolId)"
 177                  " WHERE Media.StorageId='" + storageID + "'"
 178                  " AND Media.InChanger<>0"
 179                  " ORDER BY Media.Slot");
 180
 181    QStringList results;
 182    /* This could be a log item */
 183    if (mainWin->m_sqlDebug) {
 184       Pmsg1(000, "Storage query cmd : %s\n",query.toUtf8().data());
 185    }
 186    if (m_console->sql_cmd(query, results)) {
 187       QString resultline;
 188       QString field;
 189       QStringList fieldlist;
 190
 191       foreach (resultline, results) {
 192          fieldlist = resultline.split("\t");
 193          if (fieldlist.size() < 6)
 194              continue;
 195
 196          /* Iterate through fields in the record */
 197          QStringListIterator fld(fieldlist);
 198          int index = 0;
 199          TreeItemFormatter fmt(*parent, 2);
 200
 201          /* volname */
 202          fmt.setTextFld(index++, fld.next());
 203
 204          /* skip the next two columns, unused by media */
 205          index += 2;
 206
 207          /* slot */
 208          fmt.setNumericFld(index++, fld.next());
 209
 210          /* status */
 211          fmt.setVolStatusFld(index++, fld.next());
 212
 213          /* enabled */
 214          fmt.setBoolFld(index++, fld.next());
 215
 216          /* pool */
 217          fmt.setTextFld(index++, fld.next());
 218
 219          /* media type */
 220          fmt.setTextFld(index++, fld.next());
 221
 222       }
 223    }
 224 }
 225
 226 /*
 227  * When the treeWidgetItem in the page selector tree is singleclicked, Make sure
 228  * The tree has been populated.
 229  */
 230 void Storage::PgSeltreeWidgetClicked()
 231 {
 232    if(!m_populated) {
 233       populateTree();
 234       createContextMenu();
 235    }
 236    dockPage();
 237 }
 238
 239 /*
 240  * Added to set the context menu policy based on currently active treeWidgetItem
 241  * signaled by currentItemChanged
 242  */
 243 void Storage::treeItemChanged(QTreeWidgetItem *currentwidgetitem, QTreeWidgetItem *previouswidgetitem )
 244 {
 245    /* m_checkcurwidget checks to see if this is during a refresh, which will segfault */
 246    if (m_checkcurwidget) {
 247       /* The Previous item */
 248       if (previouswidgetitem) { /* avoid a segfault if first time */
 249          int treedepth = previouswidgetitem->data(0, Qt::UserRole).toInt();
 250          if (treedepth == 1){
 251             mp_treeWidget->removeAction(actionStatusStorageInConsole);
 252             mp_treeWidget->removeAction(actionStatusStorageWindow);
 253             mp_treeWidget->removeAction(actionLabelStorage);
 254             mp_treeWidget->removeAction(actionMountStorage);
 255             mp_treeWidget->removeAction(actionUnMountStorage);
 256             mp_treeWidget->removeAction(actionUpdateSlots);
 257             mp_treeWidget->removeAction(actionUpdateSlotsScan);
 258             mp_treeWidget->removeAction(actionRelease);
 259          }
 260       }
 261
 262       int treedepth = currentwidgetitem->data(0, Qt::UserRole).toInt();
 263       if (treedepth == 1){
 264          /* set a hold variable to the storage name in case the context sensitive
 265           * menu is used */
 266          m_currentStorage = currentwidgetitem->text(0);
 267          m_currentAutoChanger = currentwidgetitem->text(2) == tr("Yes");
 268          mp_treeWidget->addAction(actionStatusStorageInConsole);
 269          mp_treeWidget->addAction(actionStatusStorageWindow);
 270          mp_treeWidget->addAction(actionLabelStorage);
 271          mp_treeWidget->addAction(actionMountStorage);
 272          mp_treeWidget->addAction(actionUnMountStorage);
 273          mp_treeWidget->addAction(actionRelease);
 274          QString text;
 275          text = tr("Status Storage \"%1\"").arg(m_currentStorage);;
 276          actionStatusStorageInConsole->setText(text);
 277          text = tr("Status Storage \"%1\" in Window").arg(m_currentStorage);;
 278          actionStatusStorageWindow->setText(text);
 279          text = tr("Label media in Storage \"%1\"").arg(m_currentStorage);
 280          actionLabelStorage->setText(text);
 281          text = tr("Mount media in Storage \"%1\"").arg(m_currentStorage);
 282          actionMountStorage->setText(text);
 283          text = tr("\"UN\" Mount media in Storage \"%1\"").arg(m_currentStorage);
 284          text = tr("Release media in Storage \"%1\"").arg(m_currentStorage);
 285          actionRelease->setText(text);
 286          if (m_currentAutoChanger) {
 287             mp_treeWidget->addAction(actionUpdateSlots);
 288             mp_treeWidget->addAction(actionUpdateSlotsScan);
 289             text = tr("Barcode Scan media in Storage \"%1\"").arg(m_currentStorage);
 290             actionUpdateSlots->setText(text);
 291             text = tr("Read scan media in Storage \"%1\"").arg( m_currentStorage);
 292             actionUpdateSlotsScan->setText(text);
 293          }
 294       }
 295    }
 296 }
 297
 298 /*
 299  * Setup a context menu
 300  * Made separate from populate so that it would not create context menu over and
 301  * over as the tree is repopulated.
 302  */
 303 void Storage::createContextMenu()
 304 {
 305    mp_treeWidget->setContextMenuPolicy(Qt::ActionsContextMenu);
 306    mp_treeWidget->addAction(actionRefreshStorage);
 307    connect(mp_treeWidget, SIGNAL(
 308            currentItemChanged(QTreeWidgetItem *, QTreeWidgetItem *)),
 309            this, SLOT(treeItemChanged(QTreeWidgetItem *, QTreeWidgetItem *)));
 310    /* connect to the action specific to this pages class */
 311    connect(actionRefreshStorage, SIGNAL(triggered()), this,
 312                 SLOT(populateTree()));
 313    connect(actionStatusStorageInConsole, SIGNAL(triggered()), this,
 314                 SLOT(consoleStatusStorage()));
 315    connect(actionLabelStorage, SIGNAL(triggered()), this,
 316                 SLOT(consoleLabelStorage()));
 317    connect(actionMountStorage, SIGNAL(triggered()), this,
 318                 SLOT(consoleMountStorage()));
 319    connect(actionUnMountStorage, SIGNAL(triggered()), this,
 320                 SLOT(consoleUnMountStorage()));
 321    connect(actionUpdateSlots, SIGNAL(triggered()), this,
 322                 SLOT(consoleUpdateSlots()));
 323    connect(actionUpdateSlotsScan, SIGNAL(triggered()), this,
 324                 SLOT(consoleUpdateSlotsScan()));
 325    connect(actionRelease, SIGNAL(triggered()), this,
 326                 SLOT(consoleRelease()));
 327    connect(actionStatusStorageWindow, SIGNAL(triggered()), this,
 328                 SLOT(statusStorageWindow()));
 329 }
 330
 331 /*
 332  * Virtual function which is called when this page is visible on the stack
 333  */
 334 void Storage::currentStackItem()
 335 {
 336    if(!m_populated) {
 337       populateTree();
 338       /* Create the context menu for the storage tree */
 339       createContextMenu();
 340    }
 341 }
 342
 343 /*
 344  *  Functions to respond to local context sensitive menu sending console commands
 345  *  If I could figure out how to make these one function passing a string, Yaaaaaa
 346  */
 347 void Storage::consoleStatusStorage()
 348 {
 349    QString cmd("status storage=");
 350    cmd += m_currentStorage;
 351    consoleCommand(cmd);
 352 }
 353
 354 /* Label Media populating current storage by default */
 355 void Storage::consoleLabelStorage()
 356 {
 357    new labelPage(m_currentStorage);
 358 }
 359
 360 /* Mount currently selected storage */
 361 void Storage::consoleMountStorage()
 362 {
 363    if (m_currentAutoChanger == 0){
 364       /* no autochanger, just execute the command in the console */
 365       QString cmd("mount storage=");
 366       cmd += m_currentStorage;
 367       consoleCommand(cmd);
 368    } else {
 369       setConsoleCurrent();
 370       /* if this storage is an autochanger, lets ask for the slot */
 371       new mountDialog(m_console, m_currentStorage);
 372    }
 373 }
 374
 375 /* Unmount Currently selected storage */
 376 void Storage::consoleUnMountStorage()
 377 {
 378    QString cmd("umount storage=");
 379    cmd += m_currentStorage;
 380    consoleCommand(cmd);
 381 }
 382
 383 /* Update Slots */
 384 void Storage::consoleUpdateSlots()
 385 {
 386    QString cmd("update slots storage=");
 387    cmd += m_currentStorage;
 388    consoleCommand(cmd);
 389 }
 390
 391 /* Update Slots Scan*/
 392 void Storage::consoleUpdateSlotsScan()
 393 {
 394    QString cmd("update slots scan storage=");
 395    cmd += m_currentStorage;
 396    consoleCommand(cmd);
 397 }
 398
 399 /* Release a tape in the drive */
 400 void Storage::consoleRelease()
 401 {
 402    QString cmd("release storage=");
 403    cmd += m_currentStorage;
 404    consoleCommand(cmd);
 405 }
 406
 407 /*
 408  *  Open a status storage window
 409  */
 410 void Storage::statusStorageWindow()
 411 {
 412    /* if one exists, then just set it current */
 413    bool found = false;
 414    foreach(Pages *page, mainWin->m_pagehash) {
 415       if (mainWin->currentConsole() == page->console()) {
 416          if (page->name() == tr("Storage Status %1").arg(m_currentStorage)) {
 417             found = true;
 418             page->setCurrent();
 419          }
 420       }
 421    }
 422    if (!found) {
 423       QTreeWidgetItem *parentItem = mainWin->getFromHash(this);
 424       new StorStat(m_currentStorage, parentItem);
 425    }
 426 }
 427
 428 /*
 429  * Write settings to save expanded states of the pools
 430  */
 431 void Storage::writeExpandedSettings()
 432 {
 433    QSettings settings(m_console->m_dir->name(), "bat");
 434    settings.beginGroup("StorageTreeExpanded");
 435    int childcount = m_topItem->childCount();
 436    for (int cnt=0; cnt<childcount; cnt++) {
 437       QTreeWidgetItem *item = m_topItem->child(cnt);
 438       settings.setValue(item->text(0), item->isExpanded());
 439    }
 440    settings.endGroup();
 441 }
 442
 443 /*
 444  * If first time, then check to see if there were status pages open the last time closed
 445  * if so open
 446  */
 447 void Storage::settingsOpenStatus(QString &storage)
 448 {
 449    QSettings settings(m_console->m_dir->name(), "bat");
 450
 451    settings.beginGroup("OpenOnExit");
 452    QString toRead = "StorageStatus_" + storage;
 453    if (settings.value(toRead) == 1) {
 454       if (mainWin->m_sqlDebug) {
 455          Pmsg1(000, "Do open Storage Status window for : %s\n", storage.toUtf8().data());
 456       }
 457       new StorStat(storage, mainWin->getFromHash(this));
 458       setCurrent();
 459       mainWin->getFromHash(this)->setExpanded(true);
 460    } else {
 461       if (mainWin->m_sqlDebug) {
 462          Pmsg1(000, "Do NOT open Storage Status window for : %s\n", storage.toUtf8().data());
 463       }
 464    }
 465    settings.endGroup();
 466 }