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