- Search plugin interface changed: enum constants to define widget types to allow...
[qtrapids] / src / gui / MainWindow.cpp
1 /***************************************************************************
2  *   Copyright (C) 2009 by Lassi Väätämöinen   *
3  *   lassi.vaatamoinen@ixonos.com   *
4  *                                                                         *
5  *   This program is free software; you can redistribute it and/or modify  *
6  *   it under the terms of the GNU General Public License as published by  *
7  *   the Free Software Foundation; either version 2 of the License, or     *
8  *   (at your option) any later version.                                   *
9  *                                                                         *
10  *   This program is distributed in the hope that it will be useful,       *
11  *   but WITHOUT ANY WARRANTY; without even the implied warranty of        *
12  *   MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the         *
13  *   GNU General Public License for more details.                          *
14  *                                                                         *
15  *   You should have received a copy of the GNU General Public License     *
16  *   along with this program; if not, write to the                         *
17  *   Free Software Foundation, Inc.,                                       *
18  *   59 Temple Place - Suite 330, Boston, MA  02111-1307, USA.             *
19  ***************************************************************************/
20
21
22 #include <QDebug>
23
24 #include <QMenuBar>
25 #include <QToolBar>
26 #include <QAction>
27 #include <QFileDialog>
28 #include <QMessageBox>
29 #include <QApplication>
30
31
32 #include "DownloadView.h"
33 #include "SeedView.h"
34 #include "PreferencesDialog.h"
35
36 #include "MainWindow.h"
37
38 const QString ABOUT_TEXT
39 = QString(QObject::trUtf8("QtRapids, a simple BitTorrent client based on"
40                           "\nQt and Libtorrent."
41                           "\n\nURL: http://qtrapids.garage.maemo.org/"
42                           "\n\nAuthors:\nLassi Väätämöinen, lassi.vaatamoinen@ixonos.com"
43                           "\nDenis Zalevskiy, denis.zalewsky@ixonos.com"
44                           "\n\nIxonos Plc, Finland\n"));
45
46 // Consturctor
47 MainWindow::MainWindow():
48                 QMainWindow(), // Superclass
49                 tabWidget_(NULL),
50                 dlView_(NULL),
51                 seedView_(NULL),
52                 preferencesDialog_(NULL),
53                 settings_(),
54 //      torrentHandles_(),
55                 btSession_()
56 {
57         // MENUBAR
58         QMenuBar *menuBar = new QMenuBar();
59         QMenu *tempMenu = NULL;
60
61         tempMenu = menuBar->addMenu(tr("&File"));
62         QAction *openAction = tempMenu->addAction(tr("&Open"));
63         QAction *removeAction = tempMenu->addAction(tr("&Remove"));
64         removeAction->setEnabled(false);
65         QAction *quitAction = tempMenu->addAction(tr("&Quit"));
66
67         tempMenu = menuBar->addMenu(tr("&Settings"));
68         QAction *preferencesAction = tempMenu->addAction(tr("&Preferences"));
69
70         tempMenu = menuBar->addMenu(tr("&Help"));
71         QAction *aboutAction = tempMenu->addAction(tr("&About"));
72         QAction *aboutQtAction = tempMenu->addAction(tr("About &Qt"));
73
74         setMenuBar(menuBar);
75         connect(openAction, SIGNAL(triggered()), this, SLOT(on_openAction_clicked()));
76         connect(removeAction, SIGNAL(triggered()), this, SLOT(on_removeAction_clicked()));
77         connect(this, SIGNAL(itemSelected(bool)), removeAction, SLOT(setEnabled(bool)));
78         connect(quitAction, SIGNAL(triggered()), this, SLOT(on_quitAction_clicked()));
79         connect(preferencesAction, SIGNAL(triggered()), this, SLOT(on_preferencesAction_clicked()));
80         connect(aboutAction, SIGNAL(triggered()), this, SLOT(on_aboutAction_clicked()));
81         connect(aboutQtAction, SIGNAL(triggered()), this, SLOT(on_aboutQtAction_clicked()));
82
83         // TABWIDGET (central widget)
84         tabWidget_ = new QTabWidget();
85         tabWidget_->setTabsClosable(true);
86
87         /// @todo Exception handling
88         dlView_ = new DownloadView(this);
89         seedView_ = new SeedView(this);
90         tabWidget_->addTab(dlView_, tr("Downloads"));
91         tabWidget_->addTab(seedView_, tr("Seeds"));
92         connect(dlView_, SIGNAL(itemSelectionChanged()), this,
93                 SLOT(on_downloadItemSelectionChanged()));
94         connect(seedView_, SIGNAL(itemSelectionChanged()), this,
95                 SLOT(on_seedItemSelectionChanged()));
96
97
98         // Tab widget as central widget.
99         setCentralWidget(tabWidget_);
100
101         // TOOLBAR
102         QToolBar *toolBar = new QToolBar();
103         toolBar->addAction(tr("Open"));
104         removeAction = toolBar->addAction(tr("Remove"));
105         removeAction->setEnabled(false);
106         addToolBar(Qt::TopToolBarArea, toolBar);
107
108         connect(this, SIGNAL(itemSelected(bool)), removeAction,
109                 SLOT(setEnabled(bool)));
110         connect(toolBar, SIGNAL(actionTriggered(QAction*)), this,
111                 SLOT(handleToolBarAction(QAction*)));
112
113         connect(&btSession_, SIGNAL(alert(std::auto_ptr<Alert>)),
114                 this, SLOT(on_alert(std::auto_ptr<Alert>)));
115
116         LoadPlugins();
117 }
118
119
120 MainWindow::~MainWindow()
121 {
122 }
123
124 // ===================== Implements PluginInterface =========================
125 bool MainWindow::setGui(QWidget* widget, PluginWidgetType type)
126 {
127 #ifdef QTRAPIDS_DEBUG
128         qDebug() << "MainWindow::setGui():" << dlView_->currentItem();
129 #endif
130         tabWidget_->addTab(widget, tr("Search"));
131         return true;
132 }
133
134 /// @todo Add PluginInterface parameter check which plugin gives the widget to handle appropriately
135 void MainWindow::addPluginWidget(QWidget* widget, PluginWidgetType type)
136 {
137 #ifdef QTRAPIDS_DEBUG
138         qDebug() << "MainWindow::addPluginWidget():" << dlView_->currentItem();
139 #endif
140
141         if (type == qtrapids::PluginHostInterface::TAB_PAGE) {
142                 int index = tabWidget_->addTab(widget, tr("Test"));
143                 tabWidget_->setCurrentIndex(index);
144                 //layout_->addWidget(widget);
145         }
146 }
147 void MainWindow::addToolbar(QWidget* widget, PluginWidgetType type)
148 {
149 }
150
151 void MainWindow::addToolItem(QWidget* widget, PluginWidgetType type)
152 {
153 }
154
155 void MainWindow::addMenu(QWidget* widget, PluginWidgetType type)
156 {
157 }
158
159 void MainWindow::addMenuItem(QWidget* widget, PluginWidgetType type)
160 {
161 }
162
163 //=========================== PRIVATE ================================
164
165 void MainWindow::LoadPlugins()
166 {
167         /// @todo get plugin directory from settings or go through multiple diectories
168         /// Now we only check the application directory
169         pluginsDir_ = QDir(qApp->applicationDirPath());
170         pluginsDir_.cd("plugins");
171         QStringList nameFilters;
172         nameFilters << "*.so";
173
174         foreach (QString fileName, pluginsDir_.entryList(nameFilters, QDir::Files)) {
175                 QPluginLoader pluginLoader(pluginsDir_.absoluteFilePath(fileName));
176
177                 if (!QLibrary::isLibrary(fileName)) {
178                         qDebug() << fileName << " not a library";
179                 }
180
181                 if (pluginLoader.load()) {
182                         qDebug() << "Plugin loaded: "  << fileName;
183                 } else {
184                         qDebug() << "Plugin load failed: " << pluginLoader.errorString();
185                 }
186
187                 QObject *baseInstance = pluginLoader.instance();
188                 if (!baseInstance) {
189                         qDebug() << "Base instance = NULL.";
190                 }
191
192                 qtrapids::PluginInterface *plugin = qobject_cast<qtrapids::PluginInterface*>(baseInstance);
193
194                 if (!plugin) {
195                         qDebug() << "Cast failed.";
196                 } else {
197                         plugin->initialize(this);
198                         pluginFileNames_ += fileName;
199                 }
200         }
201 }
202
203 // =========================== SLOTS =================================
204 void MainWindow::on_openAction_clicked()
205 {
206         QFileDialog *dialog = new QFileDialog( this, "Open torrent file", QString(), tr("Torrent files (*.torrent)"));
207         dialog->setFileMode(QFileDialog::ExistingFile);
208         connect(dialog, SIGNAL(fileSelected(const QString&)), this, SLOT(on_torrentFileSelected(const QString&)));
209         dialog->show();
210
211 }
212
213 void MainWindow::on_removeAction_clicked()
214 {
215         qtrapids::QTorrentHandle handle = dlView_->removeSelected();
216         btSession_.removeTorrent(handle);
217 }
218
219 void MainWindow::on_quitAction_clicked()
220 {
221         close();
222 }
223
224 void MainWindow::on_preferencesAction_clicked()
225 {
226         if (!preferencesDialog_) {
227                 preferencesDialog_ = new PreferencesDialog(this);
228         }
229         preferencesDialog_->show();
230         preferencesDialog_->raise();
231         preferencesDialog_->activateWindow();
232 }
233
234 void MainWindow::on_aboutAction_clicked()
235 {
236         QMessageBox::about(this, tr("About QtRapids"), ABOUT_TEXT);
237 }
238
239
240 void MainWindow::on_aboutQtAction_clicked()
241 {
242         QMessageBox::aboutQt (this, tr("About Qt"));
243 }
244
245
246 void MainWindow::on_downloadItemSelectionChanged()
247 {
248 #ifdef QTRAPIDS_DEBUG
249         qDebug() << "MainWindow::on_seedItemSelectionChanged():" << dlView_->currentItem();
250 #endif
251         if (dlView_->currentItem() != NULL) {
252                 emit(itemSelected(true));
253         } else {
254                 emit(itemSelected(false));
255         }
256 }
257
258 void MainWindow::on_seedItemSelectionChanged()
259 {
260 #ifdef QTRAPIDS_DEBUG
261         qDebug() << "MainWindow::on_seedItemSelectionChanged():" << seedView_->currentItem();
262 #endif
263         if (seedView_->currentItem() != NULL) {
264                 emit(itemSelected(true));
265         } else {
266                 emit(itemSelected(false));
267         }
268 }
269
270 void MainWindow::handleToolBarAction(QAction* action)
271 {
272         if (action->text() == "Open") {
273                 on_openAction_clicked();
274         } else if (action->text() == "Remove") {
275                 on_removeAction_clicked();
276         }
277 }
278
279 void MainWindow::on_torrentFileSelected(const QString& file)
280 {
281 #ifdef QTRAPIDS_DEBUG
282         qDebug() << " MainWindow::on_torrentFileSelected(): " << file;
283 #endif
284         // Torrent filename empty, do nothing.
285         if (file == "") {
286                 return;
287         }
288
289         // Otherwise add torrent
290         // For params, see: http://www.rasterbar.com/products/libtorrent/manual.html#add-torrent
291         AddTorrentParams addParams;
292         boost::intrusive_ptr<libtorrent::torrent_info> tiTmp =
293             new libtorrent::torrent_info(boost::filesystem::path(file.toStdString()));
294         addParams.ti = tiTmp;
295         // save_path is the only mandatory parameter, rest are optional.
296         addParams.save_path = boost::filesystem::path(settings_.value("download/directory").toString().toStdString());
297         //addParams.storage_mode = libtorrent::storage_mode_allocate;
298         qtrapids::QTorrentHandle handle = btSession_.addTorrent(addParams);
299         dlView_->newItem(handle);
300 //      torrentHandles_.push_back(handlePtr);
301 #ifdef QTRAPIDS_DEBUG
302         qDebug() << "Is valid: " << handle.isValid();
303 #endif
304 }
305
306
307 void MainWindow::on_alert(std::auto_ptr<Alert> al)
308 {
309         if (al.get() != NULL) {
310 //              qDebug()
311 //                              << "MainWindow::on_torrentAlert(): "
312 //                              << QString::fromStdString(al->message());
313
314                 TorrentAlert *torrentAlert
315                 = dynamic_cast<TorrentAlert*> (al.get());
316
317                 if (torrentAlert) {
318                         qtrapids::QTorrentHandle torrentHandle = qtrapids::QTorrentHandle(torrentAlert->handle);
319                         dlView_->updateItem(qtrapids::QTorrentHandle(torrentAlert->handle));
320                 }
321
322         }
323 }
324
325 /*
326 bool MainWindow::IsNewTorrent(std::auto_ptr<qtrapids::QTorrentHandle> handlePtr)
327 {
328         for (unsigned i = 0; i < torrentHandles_.size(); ++i) {
329     if (torrentHandles_.at(i).get() == handlePtr.get()) {
330                         return false;
331                 } else {
332                         return true;
333                 }
334         }
335 }
336 */