]> git.sur5r.net Git - bacula/bacula/blob - bacula/src/qt-console/medialist/medialist.cpp
jobs: add actions and slots for performing commands.
[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 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 #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          mp_treeWidget->removeAction(actionEditVolume);
212          mp_treeWidget->removeAction(actionListJobsOnVolume);
213          mp_treeWidget->removeAction(actionDeleteVolume);
214          mp_treeWidget->removeAction(actionPruneVolume);
215          mp_treeWidget->removeAction(actionPurgeVolume);
216           mp_treeWidget->removeAction(actionRelabelVolume);
217          mp_treeWidget->removeAction(actionAllVolumesFromPool);
218       }
219
220       int treedepth = currentwidgetitem->data(0, Qt::UserRole).toInt();
221       m_currentVolumeName=currentwidgetitem->text(0);
222       if (treedepth == 2){
223          m_currentVolumeId=currentwidgetitem->text(1);
224          mp_treeWidget->addAction(actionEditVolume);
225          mp_treeWidget->addAction(actionListJobsOnVolume);
226          mp_treeWidget->addAction(actionDeleteVolume);
227          mp_treeWidget->addAction(actionPruneVolume);
228          mp_treeWidget->addAction(actionPurgeVolume);
229          mp_treeWidget->addAction(actionRelabelVolume);
230       } else if (treedepth == 1) {
231 /*  *******FIXME******
232  *  I can't seem to get "All volumes from pool" or "Volume from pool" to work
233  *  in one sentence command.   Works when you do it one step at a time vi console
234          mp_treeWidget->addAction(actionAllVolumesFromPool);
235 */
236       }
237    }
238 }
239
240 /* 
241  * Setup a context menu 
242  * Made separate from populate so that it would not create context menu over and
243  * over as the tree is repopulated.
244  */
245 void MediaList::createContextMenu()
246 {
247    mp_treeWidget->setContextMenuPolicy(Qt::ActionsContextMenu);
248    mp_treeWidget->addAction(actionRefreshMediaList);
249    connect(actionEditVolume, SIGNAL(triggered()), this, SLOT(editVolume()));
250    connect(actionListJobsOnVolume, SIGNAL(triggered()), this, SLOT(showJobs()));
251    connect(actionDeleteVolume, SIGNAL(triggered()), this, SLOT(deleteVolume()));
252    connect(actionPurgeVolume, SIGNAL(triggered()), this, SLOT(purgeVolume()));
253    connect(actionPruneVolume, SIGNAL(triggered()), this, SLOT(pruneVolume()));
254    connect(actionRelabelVolume, SIGNAL(triggered()), this, SLOT(relabelVolume()));
255    connect(mp_treeWidget, SIGNAL(
256            currentItemChanged(QTreeWidgetItem *, QTreeWidgetItem *)),
257            this, SLOT(treeItemChanged(QTreeWidgetItem *, QTreeWidgetItem *)));
258    /* connect to the action specific to this pages class */
259    connect(actionRefreshMediaList, SIGNAL(triggered()), this,
260                 SLOT(populateTree()));
261    connect(actionAllVolumesFromPool, SIGNAL(triggered()), this, SLOT(allVolumesFromPool()));
262 }
263
264 /*
265  * Virtual function which is called when this page is visible on the stack
266  */
267 void MediaList::currentStackItem()
268 {
269    if(!m_populated) {
270       populateTree();
271       /* Create the context menu for the medialist tree */
272       createContextMenu();
273       m_populated=true;
274    }
275 }
276
277 /*
278  * Called from the signal of the context sensitive menu to delete a volume!
279  */
280 void MediaList::deleteVolume()
281 {
282    if (QMessageBox::warning(this, tr("Bat"),
283       tr("Are you sure you want to delete??  !!!.\n"
284 "This delete command is used to delete a Volume record and all associated catalog"
285 " records that were created. This command operates only on the Catalog"
286 " database and has no effect on the actual data written to a Volume. This"
287 " command can be dangerous and we strongly recommend that you do not use"
288 " it unless you know what you are doing.  All Jobs and all associated"
289 " records (File and JobMedia) will be deleted from the catalog."
290       "Press OK to proceed with delete operation.?"),
291       QMessageBox::Ok | QMessageBox::Cancel)
292       == QMessageBox::Cancel) { return; }
293
294    QString cmd("delete volume=");
295    cmd += m_currentVolumeName;
296    consoleCommand(cmd);
297 }
298
299 /*
300  * Called from the signal of the context sensitive menu to purge!
301  */
302 void MediaList::purgeVolume()
303 {
304    if (QMessageBox::warning(this, tr("Bat"),
305       tr("Are you sure you want to purge ??  !!!.\n"
306 "The Purge command will delete associated Catalog database records from Jobs and"
307 " Volumes without considering the retention period. Purge  works only on the"
308 " Catalog database and does not affect data written to Volumes. This command can"
309 " be dangerous because you can delete catalog records associated with current"
310 " backups of files, and we recommend that you do not use it unless you know what"
311 " you are doing.\n"
312       "Press OK to proceed with the purge operation?"),
313       QMessageBox::Ok | QMessageBox::Cancel)
314       == QMessageBox::Cancel) { return; }
315
316    QString cmd("purge volume=");
317    cmd += m_currentVolumeName;
318    consoleCommand(cmd);
319    populateTree();
320 }
321
322 /*
323  * Called from the signal of the context sensitive menu to prune!
324  */
325 void MediaList::pruneVolume()
326 {
327    new prunePage(m_currentVolumeName, "");
328 }
329
330 /*
331  * Called from the signal of the context sensitive menu to relabel!
332  */
333 void MediaList::relabelVolume()
334 {
335    setConsoleCurrent();
336    new relabelDialog(m_console, m_currentVolumeName);
337 }
338
339 /*
340  * Called from the signal of the context sensitive menu to purge!
341  */
342 void MediaList::allVolumesFromPool()
343 {
344    QString cmd("update pool=");
345    cmd += m_currentVolumeName + " All Volumes From Pool";
346    consoleCommand(cmd);
347    populateTree();
348 }