]> git.sur5r.net Git - bacula/bacula/blob - bacula/src/qt-console/medialist/medialist.cpp
After tables or trees are created, size the columns.
[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
45 MediaList::MediaList()
46 {
47    setupUi(this);
48    m_name = "Media";
49    pgInitialize();
50
51    /* mp_treeWidget, Storage Tree Tree Widget inherited from ui_medialist.h */
52    m_populated = false;
53    m_checkcurwidget = true;
54    m_closeable = false;
55    setTitle();
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    QString currentpool("");
70    QString resultline;
71    QStringList results;
72    const char *query = 
73       "SELECT Pool.Name AS Pool, Media.VolumeName AS Media, Media.MediaId AS Id, Media.VolStatus AS VolStatus,"
74       " Media.Enabled AS Enabled, Media.VolBytes AS Bytes, Media.VolFiles AS FileCount, Media.VolJobs AS JobCount,"
75       " Media.VolRetention AS VolumeRetention, Media.MediaType AS MediaType, Media.LastWritten AS LastWritten"
76       " FROM Media, Pool"
77       " WHERE Media.PoolId=Pool.PoolId"
78       " ORDER BY Pool";
79    QStringList headerlist = (QStringList()
80       << "Pool Name" << "Volume Name" << "Media Id" << "Volume Status" << "Enabled"
81       << "Volume Bytes" << "Volume Files" << "Volume Jobs" << "Volume Retention" 
82       << "Media Type" << "Last Written");
83
84    m_checkcurwidget = false;
85    mp_treeWidget->clear();
86    m_checkcurwidget = true;
87    mp_treeWidget->setColumnCount(headerlist.count());
88    topItem = new QTreeWidgetItem(mp_treeWidget);
89    topItem->setText(0, "Pools");
90    topItem->setData(0, Qt::UserRole, 0);
91    topItem->setExpanded(true);
92
93    mp_treeWidget->setHeaderLabels(headerlist);
94
95    /* FIXME Make this a user configurable loggin action and dont use printf */
96    //printf("MediaList query cmd : %s\n",query);
97    if (m_console->sql_cmd(query, results)) {
98       QString field;
99       QStringList fieldlist;
100
101       foreach (resultline, results) {
102          fieldlist = resultline.split("\t");
103          int index = 0;
104          /* Iterate through fields in the record */
105          foreach (field, fieldlist) {
106             field = field.trimmed();  /* strip leading & trailing spaces */
107             if (field != "") {
108                /* The first field is the pool name */
109                if (index == 0) {
110                   /* If new pool name, create new Pool item */
111                   if (currentpool != field) {
112                      currentpool = field;
113                      pooltreeitem = new QTreeWidgetItem(topItem);
114                      pooltreeitem->setText(0, field);
115                      pooltreeitem->setData(0, Qt::UserRole, 1);
116                      pooltreeitem->setExpanded(true);
117                   }
118                   mediatreeitem = new QTreeWidgetItem(pooltreeitem);
119                   mediatreeitem->setData(index, Qt::UserRole, 2);
120                } else {
121                   /* Put media fields under the pool tree item */
122                   mediatreeitem->setData(index, Qt::UserRole, 2);
123                   mediatreeitem->setText(index, field);
124                }
125             }
126             index++;
127          }
128       }
129    }
130    /* Resize the columns */
131    for(int cnter=0; cnter<headerlist.count(); cnter++) {
132       mp_treeWidget->resizeColumnToContents(cnter);
133    }
134 }
135
136 /*
137  * Called from the signal of the context sensitive menu!
138  */
139 void MediaList::editMedia()
140 {
141    MediaEdit* edit = new MediaEdit(m_console, m_currentlyselected);
142    edit->show();
143 }
144
145 /*
146  * Called from the signal of the context sensitive menu!
147  */
148 void MediaList::showJobs()
149 {
150    QString emptyclient("");
151    QTreeWidgetItem *parentItem = mainWin->getFromHash(this);
152    mainWin->createPageJobList(m_currentlyselected, emptyclient, parentItem);
153 }
154
155 /*
156  * When the treeWidgetItem in the page selector tree is singleclicked, Make sure
157  * The tree has been populated.
158  */
159 void MediaList::PgSeltreeWidgetClicked()
160 {
161    if (!m_populated) {
162       populateTree();
163       createContextMenu();
164       m_populated=true;
165    }
166 }
167
168 /*
169  * Added to set the context menu policy based on currently active treeWidgetItem
170  * signaled by currentItemChanged
171  */
172 void MediaList::treeItemChanged(QTreeWidgetItem *currentwidgetitem, QTreeWidgetItem *previouswidgetitem )
173 {
174    /* m_checkcurwidget checks to see if this is during a refresh, which will segfault */
175    if (m_checkcurwidget) {
176       /* The Previous item */
177       if (previouswidgetitem) { /* avoid a segfault if first time */
178          int treedepth = previouswidgetitem->data(0, Qt::UserRole).toInt();
179          if (treedepth == 2){
180             mp_treeWidget->removeAction(actionEditVolume);
181             mp_treeWidget->removeAction(actionListJobsOnVolume);
182          }
183       }
184
185       int treedepth = currentwidgetitem->data(0, Qt::UserRole).toInt();
186       if (treedepth == 2){
187          m_currentlyselected=currentwidgetitem->text(1);
188          mp_treeWidget->addAction(actionEditVolume);
189          mp_treeWidget->addAction(actionListJobsOnVolume);
190       }
191    }
192 }
193
194 /* 
195  * Setup a context menu 
196  * Made separate from populate so that it would not create context menu over and
197  * over as the tree is repopulated.
198  */
199 void MediaList::createContextMenu()
200 {
201    mp_treeWidget->setContextMenuPolicy(Qt::ActionsContextMenu);
202    mp_treeWidget->addAction(actionRefreshMediaList);
203    connect(actionEditVolume, SIGNAL(triggered()), this, SLOT(editMedia()));
204    connect(actionListJobsOnVolume, SIGNAL(triggered()), this, SLOT(showJobs()));
205    connect(mp_treeWidget, SIGNAL(
206            currentItemChanged(QTreeWidgetItem *, QTreeWidgetItem *)),
207            this, SLOT(treeItemChanged(QTreeWidgetItem *, QTreeWidgetItem *)));
208    /* connect to the action specific to this pages class */
209    connect(actionRefreshMediaList, SIGNAL(triggered()), this,
210                 SLOT(populateTree()));
211 }
212
213 /*
214  * Virtual function which is called when this page is visible on the stack
215  */
216 void MediaList::currentStackItem()
217 {
218    if(!m_populated) {
219       populateTree();
220       /* add context sensitive menu items specific to this classto the page
221        * selector tree. m_m_contextActions is QList of QActions, so this is 
222        * only done once with the first population. */
223       m_contextActions.append(actionRefreshMediaList);
224       /* Create the context menu for the medialist tree */
225       createContextMenu();
226       m_populated=true;
227    }
228 }