2425ffc38e14caac4b805a95e9085b96a514a2c2
[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(qtrapids::PluginInterface* from, QWidget* widget)
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(qtrapids::PluginInterface* from, QWidget* widget)
136 {
137 #ifdef QTRAPIDS_DEBUG
138         qDebug() << "MainWindow::addPluginWidget():" << dlView_->currentItem();
139 #endif
140
141         int index = tabWidget_->addTab(widget, tr("Test"));
142         tabWidget_->setCurrentIndex(index);
143         //layout_->addWidget(widget);
144 }
145
146 void MainWindow::addToolbar(qtrapids::PluginInterface* from, QWidget* widget)
147 {
148 }
149
150 void MainWindow::addToolItem(qtrapids::PluginInterface* from, QWidget* widget)
151 {
152 }
153
154 void MainWindow::addMenu(qtrapids::PluginInterface* from, QWidget* widget)
155 {
156 }
157
158 void MainWindow::addMenuItem(qtrapids::PluginInterface* from, QWidget* widget)
159 {
160 }
161
162 //=========================== PRIVATE ================================
163
164 void MainWindow::LoadPlugins()
165 {
166         /// @todo get plugin directory from settings or go through multiple diectories
167         /// Now we only check the application directory
168         pluginsDir_ = QDir(qApp->applicationDirPath());
169         pluginsDir_.cd("plugins");
170         QStringList nameFilters;
171         nameFilters << "*.so";
172
173         foreach (QString fileName, pluginsDir_.entryList(nameFilters, QDir::Files)) {
174                 QPluginLoader pluginLoader(pluginsDir_.absoluteFilePath(fileName));
175
176                 if (!QLibrary::isLibrary(fileName)) {
177                         qDebug() << fileName << " not a library";
178                 }
179
180                 if (pluginLoader.load()) {
181                         qDebug() << "Plugin loaded: "  << fileName;
182                 } else {
183                         qDebug() << "Plugin load failed: " << pluginLoader.errorString();
184                 }
185
186                 QObject *baseInstance = pluginLoader.instance();
187                 if (!baseInstance) {
188                         qDebug() << "Base instance = NULL.";
189                 }
190
191                 qtrapids::PluginInterface *plugin = qobject_cast<qtrapids::PluginInterface*>(baseInstance);
192
193                 if (!plugin) {
194                         qDebug() << "Cast failed.";
195                 } else {
196                         plugin->initialize(this);
197                         pluginFileNames_ += fileName;
198                 }
199
200 //              QObject *plugin = pluginLoader.instance();
201 //              if (plugin) {
202 //                              populateMenus(plugin);
203 //                              pluginFileNames += fileName;
204 //              }
205
206         }
207
208         //pluginLoader_.setFileName("../libsearchplugin.so");
209
210
211
212
213 }
214
215 // =========================== SLOTS =================================
216 void MainWindow::on_openAction_clicked()
217 {
218         QFileDialog *dialog = new QFileDialog( this, "Open torrent file", QString(), tr("Torrent files (*.torrent)"));
219         dialog->setFileMode(QFileDialog::ExistingFile);
220         connect(dialog, SIGNAL(fileSelected(const QString&)), this, SLOT(on_torrentFileSelected(const QString&)));
221         dialog->show();
222
223 }
224
225 void MainWindow::on_removeAction_clicked()
226 {
227         qtrapids::QTorrentHandle handle = dlView_->removeSelected();
228         btSession_.removeTorrent(handle);
229 }
230
231 void MainWindow::on_quitAction_clicked()
232 {
233         close();
234 }
235
236 void MainWindow::on_preferencesAction_clicked()
237 {
238         if (!preferencesDialog_) {
239                 preferencesDialog_ = new PreferencesDialog(this);
240         }
241         preferencesDialog_->show();
242         preferencesDialog_->raise();
243         preferencesDialog_->activateWindow();
244 }
245
246 void MainWindow::on_aboutAction_clicked()
247 {
248         QMessageBox::about(this, tr("About QtRapids"), ABOUT_TEXT);
249 }
250
251
252 void MainWindow::on_aboutQtAction_clicked()
253 {
254         QMessageBox::aboutQt (this, tr("About Qt"));
255 }
256
257
258 void MainWindow::on_downloadItemSelectionChanged()
259 {
260 #ifdef QTRAPIDS_DEBUG
261         qDebug() << "MainWindow::on_seedItemSelectionChanged():" << dlView_->currentItem();
262 #endif
263         if (dlView_->currentItem() != NULL) {
264                 emit(itemSelected(true));
265         } else {
266                 emit(itemSelected(false));
267         }
268 }
269
270 void MainWindow::on_seedItemSelectionChanged()
271 {
272 #ifdef QTRAPIDS_DEBUG
273         qDebug() << "MainWindow::on_seedItemSelectionChanged():" << seedView_->currentItem();
274 #endif
275         if (seedView_->currentItem() != NULL) {
276                 emit(itemSelected(true));
277         } else {
278                 emit(itemSelected(false));
279         }
280 }
281
282 void MainWindow::handleToolBarAction(QAction* action)
283 {
284         if (action->text() == "Open") {
285                 on_openAction_clicked();
286         } else if (action->text() == "Remove") {
287                 on_removeAction_clicked();
288         }
289 }
290
291 void MainWindow::on_torrentFileSelected(const QString& file)
292 {
293 #ifdef QTRAPIDS_DEBUG
294         qDebug() << " MainWindow::on_torrentFileSelected(): " << file;
295 #endif
296         // Torrent filename empty, do nothing.
297         if (file == "") {
298                 return;
299         }
300
301         // Otherwise add torrent
302         // For params, see: http://www.rasterbar.com/products/libtorrent/manual.html#add-torrent
303         AddTorrentParams addParams;
304         boost::intrusive_ptr<libtorrent::torrent_info> tiTmp =
305             new libtorrent::torrent_info(boost::filesystem::path(file.toStdString()));
306         addParams.ti = tiTmp;
307         // save_path is the only mandatory parameter, rest are optional.
308         addParams.save_path = boost::filesystem::path(settings_.value("download/directory").toString().toStdString());
309         //addParams.storage_mode = libtorrent::storage_mode_allocate;
310         qtrapids::QTorrentHandle handle = btSession_.addTorrent(addParams);
311         dlView_->newItem(handle);
312 //      torrentHandles_.push_back(handlePtr);
313 #ifdef QTRAPIDS_DEBUG
314         qDebug() << "Is valid: " << handle.isValid();
315 #endif
316 }
317
318
319 void MainWindow::on_alert(std::auto_ptr<Alert> al)
320 {
321         if (al.get() != NULL) {
322 //              qDebug()
323 //                              << "MainWindow::on_torrentAlert(): "
324 //                              << QString::fromStdString(al->message());
325
326                 TorrentAlert *torrentAlert
327                 = dynamic_cast<TorrentAlert*> (al.get());
328
329                 if (torrentAlert) {
330                         qtrapids::QTorrentHandle torrentHandle = qtrapids::QTorrentHandle(torrentAlert->handle);
331                         dlView_->updateItem(qtrapids::QTorrentHandle(torrentAlert->handle));
332                 }
333
334         }
335 }
336
337 /*
338 bool MainWindow::IsNewTorrent(std::auto_ptr<qtrapids::QTorrentHandle> handlePtr)
339 {
340         for (unsigned i = 0; i < torrentHandles_.size(); ++i) {
341                 if (torrentHandles_.at(i).get() == handlePtr.get()) {
342                         return false;
343                 } else {
344                         return true;
345                 }
346         }
347 }
348 */