Added build script of Debian
[qtrapids] / src / gui / MainWindow.cpp
1 /***************************************************************************
2  *   Copyright (C) 2010 by Ixonos Plc   *
3  *                                                                         *
4  *   This program is free software; you can redistribute it and/or modify  *
5  *   it under the terms of the GNU General Public License as published by  *
6  *   the Free Software Foundation; version 2 of the License.               *
7  *                                                                         *
8  *   This program is distributed in the hope that it will be useful,       *
9  *   but WITHOUT ANY WARRANTY; without even the implied warranty of        *
10  *   MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the         *
11  *   GNU General Public License for more details.                          *
12  *                                                                         *
13  *   You should have received a copy of the GNU General Public License     *
14  *   along with this program; if not, write to the                         *
15  *   Free Software Foundation, Inc.,                                       *
16  *   59 Temple Place - Suite 330, Boston, MA  02111-1307, USA.             *
17  ***************************************************************************/
18
19
20 #include <QDebug>
21
22 #include <QMenuBar>
23 #include <QToolBar>
24 #include <QAction>
25 #include <QFileDialog>
26 #include <QMessageBox>
27 #include <QApplication>
28 #include <QPluginLoader>
29
30 #include "DownloadView.h"
31 #include "SeedView.h"
32 #include "PreferencesDialog.h"
33
34 #include "MainWindow.h"
35
36 const QString ABOUT_TEXT
37 = QString(QObject::trUtf8("QtRapids, a simple BitTorrent client based on"
38                           "\nQt and Libtorrent."
39                           "\n\nURL: http://qtrapids.garage.maemo.org/"
40                           "\n\nAuthors:\nLassi Väätämöinen, lassi.vaatamoinen@ixonos.com"
41                           "\nDenis Zalevskiy, denis.zalewsky@ixonos.com"
42                           "\n\nIxonos Plc, Finland\n"));
43
44 const QString PLUGINS_DIR = "plugins";
45
46 // Consturctor
47 MainWindow::MainWindow():
48                 QMainWindow(), // Superclass
49                 tabWidget_(NULL),
50                 dlView_(NULL),
51                 seedView_(NULL),
52                 searchWidget_(NULL),
53                 preferencesDialog_(NULL),
54                 settings_(),
55                 pluginDirs_(),
56 //      torrentHandles_(),
57                 btSession_()
58 {
59         // MENUBAR
60         QMenuBar *menuBar = new QMenuBar();
61         QMenu *tempMenu = NULL;
62
63         tempMenu = menuBar->addMenu(tr("&File"));
64         QAction *openAction = tempMenu->addAction(tr("&Open"));
65         QAction *removeAction = tempMenu->addAction(tr("&Remove"));
66         removeAction->setEnabled(false);
67         QAction *quitAction = tempMenu->addAction(tr("&Quit"));
68
69         tempMenu = menuBar->addMenu(tr("&Settings"));
70         QAction *preferencesAction = tempMenu->addAction(tr("&Preferences"));
71
72         tempMenu = menuBar->addMenu(tr("&Help"));
73         QAction *aboutAction = tempMenu->addAction(tr("&About"));
74         QAction *aboutQtAction = tempMenu->addAction(tr("About &Qt"));
75
76         setMenuBar(menuBar);
77         connect(openAction, SIGNAL(triggered()), this, SLOT(on_openAction_clicked()));
78         connect(removeAction, SIGNAL(triggered()), this, SLOT(on_removeAction_clicked()));
79         connect(this, SIGNAL(itemSelected(bool)), removeAction, SLOT(setEnabled(bool)));
80         connect(quitAction, SIGNAL(triggered()), this, SLOT(on_quitAction_clicked()));
81         connect(preferencesAction, SIGNAL(triggered()), this, SLOT(on_preferencesAction_clicked()));
82         connect(aboutAction, SIGNAL(triggered()), this, SLOT(on_aboutAction_clicked()));
83         connect(aboutQtAction, SIGNAL(triggered()), this, SLOT(on_aboutQtAction_clicked()));
84
85         // TABWIDGET (central widget)
86         tabWidget_ = new QTabWidget();
87         tabWidget_->setTabsClosable(true);
88
89         /// @todo Exception handling
90         dlView_ = new DownloadView(this);
91         seedView_ = new SeedView(this);
92         tabWidget_->addTab(dlView_, tr("Downloads"));
93         tabWidget_->addTab(seedView_, tr("Seeds"));
94         connect(dlView_, SIGNAL(itemSelectionChanged()), this,
95                 SLOT(on_downloadItemSelectionChanged()));
96         connect(seedView_, SIGNAL(itemSelectionChanged()), this,
97                 SLOT(on_seedItemSelectionChanged()));
98
99
100         // Tab widget as central widget.
101         setCentralWidget(tabWidget_);
102
103         // TOOLBAR
104         QToolBar *toolBar = new QToolBar();
105         toolBar->addAction(tr("Open"));
106         removeAction = toolBar->addAction(tr("Remove"));
107         removeAction->setEnabled(false);
108         addToolBar(Qt::TopToolBarArea, toolBar);
109
110         connect(this, SIGNAL(itemSelected(bool)), removeAction,
111                 SLOT(setEnabled(bool)));
112         connect(toolBar, SIGNAL(actionTriggered(QAction*)), this,
113                 SLOT(handleToolBarAction(QAction*)));
114         connect (tabWidget_, SIGNAL(tabCloseRequested(int)), this, SLOT(on_tabWidget_tabCloseRequested(int)));
115         
116         connect(&btSession_, SIGNAL(alert(std::auto_ptr<Alert>)),
117                 this, SLOT(on_alert(std::auto_ptr<Alert>)));
118
119         LoadPlugins();
120         RestoreSettings();
121 }
122
123
124 MainWindow::~MainWindow()
125 {
126 }
127
128 // ===================== Implements PluginInterface =========================
129 /// @todo add PluginInterface parameter to request plugin name 
130 bool MainWindow::setGui(QWidget* widget, PluginWidgetType type, qtrapids::PluginInterface* plugin)
131 {
132 #ifdef QTRAPIDS_DEBUG
133         qDebug() << "MainWindow::setGui():" << dlView_->currentItem();
134 #endif
135
136         if (plugin && plugin->identifier() == "SearchPlugin") {
137                 searchWidget_ = widget;
138         } else {
139                 return false;
140         }
141         
142         tabWidget_->addTab(widget, tr("Search"));
143         return true;
144 }
145
146 /// @todo Add PluginInterface parameter to check which plugin gives the widget, to handle appropriately
147 void MainWindow::addPluginWidget(QWidget* widget, PluginWidgetType type)
148 {
149 #ifdef QTRAPIDS_DEBUG
150         qDebug() << "MainWindow::addPluginWidget():" << dlView_->currentItem();
151 #endif
152
153         if (type == qtrapids::PluginHostInterface::TAB_PAGE) {
154                 int index = tabWidget_->addTab(widget, tr("Results"));
155                 tabWidget_->setCurrentIndex(index);
156                 //layout_->addWidget(widget);
157         }
158 }
159 void MainWindow::addToolbar(QWidget* widget, PluginWidgetType type)
160 {
161 }
162
163 void MainWindow::addToolItem(QWidget* widget, PluginWidgetType type)
164 {
165 }
166
167 void MainWindow::addMenu(QWidget* widget, PluginWidgetType type)
168 {
169 }
170
171 void MainWindow::addMenuItem(QWidget* widget, PluginWidgetType type)
172 {
173 }
174
175 bool MainWindow::eventRequest(QVariant param, PluginRequest req)
176 {
177         if (req == qtrapids::PluginHostInterface::OPEN_FILE) {
178                 QString sourceFile = param.toString();
179                 
180                 // Get the source files name from the full path:
181                 QFileInfo fInfo(sourceFile);
182                 QString targetFile = fInfo.fileName();
183                 targetFile = settings_.value("download/directory").toString() + "/" + targetFile;
184                 
185                 // Copy temoporary file to Downloads directory...
186                 if (!QFile::copy(sourceFile, targetFile)) {
187                         qDebug() << "File copying failed";
188                         return false;
189                 } else {
190                         // If copying was successful, remove the original temporary file.
191                         QFile::remove(sourceFile);
192                 }
193                 
194                 /// @todo Torrent bencoding validity should be checked before starting(?)
195                 // ...and start the torrent:
196                 on_torrentFileSelected(targetFile);
197         
198         } else if (req == qtrapids::PluginHostInterface::READ_BUFFER) {
199                 // Create torrent information from char* buffer and start.
200                 StartTorrentFromBufferData(param.toByteArray().constData(), param.toByteArray().size());
201         }
202         
203         return true;
204 }
205
206
207 //=========================== PRIVATE ================================
208
209 void MainWindow::LoadPlugins()
210 {
211         // Get plugin directories from 
212         QStringList pluginDirsTmp = settings_.value("plugins/path").toStringList();
213         QStringList nameFilters("*.so");
214         
215         /// @todo enable "application directory" for plugin search in development/debug mode only. In release version
216         /// search plugins directory under $HOME/.qtrapids or system library paths
217         pluginDirsTmp << qApp->applicationDirPath();
218         pluginDirsTmp.removeDuplicates();
219         
220         foreach (QString dir, pluginDirsTmp) {
221                 pluginDirs_.append(QDir(dir));
222         }
223         
224         foreach (QDir dir, pluginDirs_) {
225                 
226                 if (dir.cd(PLUGINS_DIR)) {
227                         
228                         foreach (QString fileName, dir.entryList(nameFilters, QDir::Files)) {
229                                 QPluginLoader pluginLoader(dir.absoluteFilePath(fileName));
230
231                                 // If plugin not loaded from another directory, then load
232                                 if (!pluginFileNames_.contains(fileName) && QLibrary::isLibrary(fileName)) {
233
234                                         if (pluginLoader.load()) {
235                                                 qDebug() << "Plugin loaded: "  << fileName;
236                                         } else {
237                                                 qWarning() << "Plugin load failed: " << pluginLoader.errorString();
238                                         }
239
240                                         QObject *baseInstance = pluginLoader.instance();
241                                         if (!baseInstance) {
242                                                 qDebug() << "Base instance = NULL.";
243                                         }
244
245                                         qtrapids::PluginInterface *plugin = qobject_cast<qtrapids::PluginInterface*>(baseInstance);
246
247                                         if (!plugin) {
248                                                 qDebug() << "Cast failed.";
249                                         } else {
250                                                 qtrapids::PluginInterface::Info info;
251                                                 info.directory = dir.path();
252                                                 qDebug() << dir.path();
253                                                 plugin->initialize(this, info);
254                                                 pluginFileNames_ += fileName;
255                                         }
256                                 } else {
257                                         qWarning() << "Plugin " 
258                                                 << fileName 
259                                                 << " already loaded from another directory, or not a valid library file";
260                                 }
261                         }
262                         
263                 } else {
264                         qWarning() << PLUGINS_DIR <<  "directory not accessible or does not exist in "  << dir.path();
265                 }
266         }
267 }
268
269
270 void MainWindow::RestoreSettings()
271 {
272         btSession_.setUploadRateLimit(settings_.value("network/uploadRate").toInt());
273         btSession_.setDownloadRateLimit(settings_.value("network/downloadRate").toInt());
274 }
275
276
277 // Opens torrent information from buffer data and adds torrent to session 
278 void MainWindow::StartTorrentFromBufferData(char const* data, int size)
279 {
280         // For params, see: http://www.rasterbar.com/products/libtorrent/manual.html#add-torrent
281         /// @todo Should typedef libtorrent::torrent_info to something
282         AddTorrentParams addParams;
283         boost::intrusive_ptr<libtorrent::torrent_info> tiTmp =
284             new libtorrent::torrent_info(data, size);
285         addParams.ti = tiTmp;
286         // save_path is the only mandatory parameter, rest are optional.
287         addParams.save_path = boost::filesystem::path(settings_.value("download/directory").toString().toStdString());
288         //addParams.storage_mode = libtorrent::storage_mode_allocate;
289         qtrapids::QTorrentHandle handle = btSession_.addTorrent(addParams);
290         dlView_->newItem(handle);
291 //      torrentHandles_.push_back(handlePtr);
292 #ifdef QTRAPIDS_DEBUG
293         qDebug() << "Is valid: " << handle.isValid();
294 #endif
295 }
296
297 // =========================== PRIVATE SLOTS =================================
298 void MainWindow::on_openAction_clicked()
299 {
300         QFileDialog *dialog = new QFileDialog( this, "Open torrent file", QString(), tr("Torrent files (*.torrent)"));
301         dialog->setFileMode(QFileDialog::ExistingFile);
302         connect(dialog, SIGNAL(fileSelected(const QString&)), this, SLOT(on_torrentFileSelected(const QString&)));
303         dialog->show();
304 }
305
306 void MainWindow::on_removeAction_clicked()
307 {
308         qtrapids::QTorrentHandle handle = dlView_->removeSelected();
309         btSession_.removeTorrent(handle);
310 }
311
312
313 void MainWindow::on_quitAction_clicked()
314 {
315         close();
316 }
317
318
319 void MainWindow::on_preferencesAction_clicked()
320 {
321         if (!preferencesDialog_) {
322                 preferencesDialog_ = new PreferencesDialog(this, NULL, &btSession_);
323         }
324
325         preferencesDialog_->show();
326         preferencesDialog_->raise();
327         preferencesDialog_->activateWindow();
328 }
329
330
331 void MainWindow::on_aboutAction_clicked()
332 {
333         QMessageBox::about(this, tr("About QtRapids"), ABOUT_TEXT);
334 }
335
336
337 void MainWindow::on_aboutQtAction_clicked()
338 {
339         QMessageBox::aboutQt (this, tr("About Qt"));
340 }
341
342
343 void MainWindow::on_tabWidget_tabCloseRequested(int index)
344 {
345         
346         int searchWidgetIndex = tabWidget_->indexOf(searchWidget_);
347         
348         // Allow closing other tabs than the first two
349         // TODO The first two may well be closable, just add "show tabs" action for these in the menu
350         if (index != 0 && index != 1 && index != searchWidgetIndex) {
351                 QWidget *remove = tabWidget_->widget(index);
352                 tabWidget_->removeTab(index);
353                 delete remove;
354                 remove = NULL;
355         }
356 }
357
358
359 void MainWindow::on_downloadItemSelectionChanged()
360 {
361 #ifdef QTRAPIDS_DEBUG
362         qDebug() << "MainWindow::on_seedItemSelectionChanged():" << dlView_->currentItem();
363 #endif
364         if (dlView_->currentItem() != NULL) {
365                 emit(itemSelected(true));
366         } else {
367                 emit(itemSelected(false));
368         }
369 }
370
371
372 void MainWindow::on_seedItemSelectionChanged()
373 {
374 #ifdef QTRAPIDS_DEBUG
375         qDebug() << "MainWindow::on_seedItemSelectionChanged():" << seedView_->currentItem();
376 #endif
377         if (seedView_->currentItem() != NULL) {
378                 emit(itemSelected(true));
379         } else {
380                 emit(itemSelected(false));
381         }
382 }
383
384
385 void MainWindow::handleToolBarAction(QAction* action)
386 {
387         if (action->text() == "Open") {
388                 on_openAction_clicked();
389         } else if (action->text() == "Remove") {
390                 on_removeAction_clicked();
391         }
392 }
393
394
395 void MainWindow::on_torrentFileSelected(const QString& file)
396 {
397 #ifdef QTRAPIDS_DEBUG
398         qDebug() << " MainWindow::on_torrentFileSelected(): " << file;
399 #endif
400         // Torrent filename empty, do nothing.
401         if (file == "") {
402                 return;
403         }
404
405         // Otherwise add torrent
406         // For params, see: http://www.rasterbar.com/products/libtorrent/manual.html#add-torrent
407         /// @todo Should typedef libtorrent::torrent_info to something
408         AddTorrentParams addParams;
409         boost::intrusive_ptr<libtorrent::torrent_info> tiTmp =
410             new libtorrent::torrent_info(boost::filesystem::path(file.toStdString()));
411         addParams.ti = tiTmp;
412         // save_path is the only mandatory parameter, rest are optional.
413         addParams.save_path = boost::filesystem::path(settings_.value("download/directory").toString().toStdString());
414         //addParams.storage_mode = libtorrent::storage_mode_allocate;
415         qtrapids::QTorrentHandle handle = btSession_.addTorrent(addParams);
416         dlView_->newItem(handle);
417 //      torrentHandles_.push_back(handlePtr);
418 #ifdef QTRAPIDS_DEBUG
419         qDebug() << "Is valid: " << handle.isValid();
420 #endif
421 }
422
423 void MainWindow::on_alert(std::auto_ptr<Alert> al)
424 {
425         if (al.get() != NULL) {
426 //              qDebug()
427 //                              << "MainWindow::on_torrentAlert(): "
428 //                              << QString::fromStdString(al->message());
429
430                 TorrentAlert *torrentAlert
431                 = dynamic_cast<TorrentAlert*> (al.get());
432
433                 if (torrentAlert) {
434                         qtrapids::QTorrentHandle torrentHandle = qtrapids::QTorrentHandle(torrentAlert->handle);
435                         dlView_->updateItem(qtrapids::QTorrentHandle(torrentAlert->handle));
436                 }
437
438         }
439 }
440
441 /*
442 bool MainWindow::IsNewTorrent(std::auto_ptr<qtrapids::QTorrentHandle> handlePtr)
443 {
444         for (unsigned i = 0; i < torrentHandles_.size(); ++i) {
445     if (torrentHandles_.at(i).get() == handlePtr.get()) {
446                         return false;
447                 } else {
448                         return true;
449                 }
450         }
451 }
452 */