git.sur5r.net Git - bacula/bacula/blob - bacula/src/qt-console/fileset/fileset.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 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$
  31  *
  32  *  FileSet Class
  33  *
  34  *   Dirk Bartley, March 2007
  35  *
  36  */
  37
  38 #include <QAbstractEventDispatcher>
  39 #include <QMenu>
  40 #include "bat.h"
  41 #include "fileset/fileset.h"
  42
  43 FileSet::FileSet()
  44 {
  45    setupUi(this);
  46    m_name = tr("FileSets");
  47    pgInitialize();
  48    QTreeWidgetItem* thisitem = mainWin->getFromHash(this);
  49    thisitem->setIcon(0,QIcon(QString::fromUtf8(":images/system-file-manager.png")));
  50
  51    /* mp_treeWidget, FileSet Tree Tree Widget inherited from ui_fileset.h */
  52    m_populated = false;
  53    m_checkcurwidget = true;
  54    m_closeable = false;
  55    readSettings();
  56    /* add context sensitive menu items specific to this classto the page
  57     * selector tree. m_contextActions is QList of QActions */
  58    m_contextActions.append(actionRefreshFileSet);
  59    dockPage();
  60 }
  61
  62 FileSet::~FileSet()
  63 {
  64    writeSettings();
  65 }
  66
  67 /*
  68  * The main meat of the class!!  The function that querries the director and
  69  * creates the widgets with appropriate values.
  70  */
  71 void FileSet::populateTree()
  72 {
  73    QTreeWidgetItem *filesetItem, *topItem;
  74
  75    if (!m_console->preventInUseConnect())
  76        return;
  77
  78    m_checkcurwidget = false;
  79    mp_treeWidget->clear();
  80    m_checkcurwidget = true;
  81
  82    QStringList headerlist = (QStringList() << tr("FileSet Name") << tr("FileSet Id")
  83        << tr("Create Time"));
  84
  85    topItem = new QTreeWidgetItem(mp_treeWidget);
  86    topItem->setText(0, tr("FileSet"));
  87    topItem->setData(0, Qt::UserRole, 0);
  88    topItem->setExpanded(true);
  89
  90    mp_treeWidget->setColumnCount(headerlist.count());
  91    mp_treeWidget->setHeaderLabels(headerlist);
  92
  93    foreach(QString filesetName, m_console->fileset_list) {
  94       filesetItem = new QTreeWidgetItem(topItem);
  95       filesetItem->setText(0, filesetName);
  96       filesetItem->setData(0, Qt::UserRole, 1);
  97       filesetItem->setExpanded(true);
  98
  99       /* Set up query QString and header QStringList */
 100       QString query("");
 101       query += "SELECT FileSet AS Name, FileSetId AS Id, CreateTime"
 102            " FROM FileSet"
 103            " WHERE ";
 104       query += " FileSet='" + filesetName + "'";
 105       query += " ORDER BY FileSet";
 106
 107       QStringList results;
 108       if (mainWin->m_sqlDebug) {
 109          Pmsg1(000, "FileSet query cmd : %s\n",query.toUtf8().data());
 110       }
 111       if (m_console->sql_cmd(query, results)) {
 112          int resultCount = results.count();
 113          if (resultCount == 1){
 114             QString resultline;
 115             QString field;
 116             QStringList fieldlist;
 117             /* there will only be one of these */
 118             foreach (resultline, results) {
 119                fieldlist = resultline.split("\t");
 120                int index = 0;
 121                /* Iterate through fields in the record */
 122                foreach (field, fieldlist) {
 123                   field = field.trimmed();  /* strip leading & trailing spaces */
 124                   filesetItem->setData(index, Qt::UserRole, 1);
 125                   /* Put media fields under the pool tree item */
 126                   filesetItem->setData(index, Qt::UserRole, 1);
 127                   filesetItem->setText(index, field);
 128                   index++;
 129                }
 130             }
 131          }
 132       }
 133    }
 134    /* Resize the columns */
 135    for (int cnter=0; cnter<headerlist.size(); cnter++) {
 136       mp_treeWidget->resizeColumnToContents(cnter);
 137    }
 138 }
 139
 140 /*
 141  * When the treeWidgetItem in the page selector tree is singleclicked, Make sure
 142  * The tree has been populated.
 143  */
 144 void FileSet::PgSeltreeWidgetClicked()
 145 {
 146    if (!m_populated) {
 147       populateTree();
 148       createContextMenu();
 149       m_populated = true;
 150    }
 151 }
 152
 153 /*
 154  * Added to set the context menu policy based on currently active treeWidgetItem
 155  * signaled by currentItemChanged
 156  */
 157 void FileSet::treeItemChanged(QTreeWidgetItem *currentwidgetitem,
 158                               QTreeWidgetItem *previouswidgetitem )
 159 {
 160    /* m_checkcurwidget checks to see if this is during a refresh, which will segfault */
 161    if (m_checkcurwidget) {
 162       /* The Previous item */
 163       if (previouswidgetitem) { /* avoid a segfault if first time */
 164          int treedepth = previouswidgetitem->data(0, Qt::UserRole).toInt();
 165          if (treedepth == 1) {
 166             mp_treeWidget->removeAction(actionStatusFileSetInConsole);
 167             mp_treeWidget->removeAction(actionShowJobs);
 168          }
 169       }
 170
 171       int treedepth = currentwidgetitem->data(0, Qt::UserRole).toInt();
 172       if (treedepth == 1){
 173          /* set a hold variable to the fileset name in case the context sensitive
 174           * menu is used */
 175          m_currentlyselected=currentwidgetitem->text(0);
 176          mp_treeWidget->addAction(actionStatusFileSetInConsole);
 177          mp_treeWidget->addAction(actionShowJobs);
 178       }
 179    }
 180 }
 181
 182 /*
 183  * Setup a context menu
 184  * Made separate from populate so that it would not create context menu over and
 185  * over as the tree is repopulated.
 186  */
 187 void FileSet::createContextMenu()
 188 {
 189    mp_treeWidget->setContextMenuPolicy(Qt::ActionsContextMenu);
 190    mp_treeWidget->addAction(actionRefreshFileSet);
 191    connect(mp_treeWidget, SIGNAL(
 192            currentItemChanged(QTreeWidgetItem *, QTreeWidgetItem *)),
 193            this, SLOT(treeItemChanged(QTreeWidgetItem *, QTreeWidgetItem *)));
 194    /* connect to the action specific to this pages class */
 195    connect(actionRefreshFileSet, SIGNAL(triggered()), this,
 196                 SLOT(populateTree()));
 197    connect(actionStatusFileSetInConsole, SIGNAL(triggered()), this,
 198                 SLOT(consoleStatusFileSet()));
 199    connect(actionShowJobs, SIGNAL(triggered()), this,
 200                 SLOT(showJobs()));
 201 }
 202
 203 /*
 204  * Function responding to actionListJobsofFileSet which calls mainwin function
 205  * to create a window of a list of jobs of this fileset.
 206  */
 207 void FileSet::consoleStatusFileSet()
 208 {
 209    QString cmd("status fileset=");
 210    cmd += m_currentlyselected;
 211    consoleCommand(cmd);
 212 }
 213
 214 /*
 215  * Virtual function which is called when this page is visible on the stack
 216  */
 217 void FileSet::currentStackItem()
 218 {
 219    if(!m_populated) {
 220       populateTree();
 221       /* Create the context menu for the fileset tree */
 222       createContextMenu();
 223       m_populated=true;
 224    }
 225 }
 226
 227 /*
 228  * Save user settings associated with this page
 229  */
 230 void FileSet::writeSettings()
 231 {
 232    QSettings settings(m_console->m_dir->name(), "bat");
 233    settings.beginGroup("FileSet");
 234    settings.setValue("geometry", saveGeometry());
 235    settings.endGroup();
 236 }
 237
 238 /*
 239  * Read and restore user settings associated with this page
 240  */
 241 void FileSet::readSettings()
 242 {
 243    QSettings settings(m_console->m_dir->name(), "bat");
 244    settings.beginGroup("FileSet");
 245    restoreGeometry(settings.value("geometry").toByteArray());
 246    settings.endGroup();
 247 }
 248
 249 /*
 250  * Create a JobList object pre-populating a fileset
 251  */
 252 void FileSet::showJobs()
 253 {
 254    QTreeWidgetItem *parentItem = mainWin->getFromHash(this);
 255    mainWin->createPageJobList("", "", "", m_currentlyselected, parentItem);
 256 }