Don't lose pages. Handle fragments in chapter URLs.
[dorian] / mainwindow.cpp
1 #include <QtGui>
2 #include <QtDebug>
3 #include <QDir>
4 #include <QApplication>
5 #include <QFileInfo>
6 #include <QStringList>
7
8 #ifdef Q_WS_MAEMO_5
9 #   include <QtMaemo5/QMaemo5InformationBox>
10 #   include <QtDBus>
11 #   include <mce/mode-names.h>
12 #   include <mce/dbus-names.h>
13 #endif // Q_WS_MAEMO_5
14
15 #include "bookview.h"
16 #include "book.h"
17 #include "library.h"
18 #include "infodialog.h"
19 #include "librarydialog.h"
20 #include "devtools.h"
21 #include "mainwindow.h"
22 #include "settingswindow.h"
23 #include "bookmarksdialog.h"
24 #include "settings.h"
25 #include "chaptersdialog.h"
26 #include "fullscreenwindow.h"
27 #include "trace.h"
28 #include "bookfinder.h"
29
30 #ifdef DORIAN_TEST_MODEL
31 #include "modeltest.h"
32 #endif
33
34 #ifdef Q_WS_MAC
35 #   define ICON_PREFIX ":/icons/mac/"
36 #else
37 #   define ICON_PREFIX ":/icons/"
38 #endif
39
40 MainWindow::MainWindow(QWidget *parent):
41     QMainWindow(parent), view(0), preventBlankingTimer(-1)
42 {
43     Trace t("MainWindow::MainWindow");
44 #ifdef Q_WS_MAEMO_5
45     setAttribute(Qt::WA_Maemo5StackedWindow, true);
46 #endif
47     setWindowTitle("Dorian");
48
49     // Central widget. Must be an intermediate, because the book view widget
50     // can be re-parented later
51     QFrame *central = new QFrame(this);
52     QVBoxLayout *layout = new QVBoxLayout(central);
53     layout->setMargin(0);
54     central->setLayout(layout);
55     setCentralWidget(central);
56
57     // Book view
58     view = new BookView(central);
59     view->show();
60     layout->addWidget(view);
61
62     // Tool bar
63     setUnifiedTitleAndToolBarOnMac(true);
64     settings = new QDialog(this);
65     toolBar = addToolBar("controls");
66     toolBar->setMovable(false);
67     toolBar->setFloatable(false);
68     toolBar->toggleViewAction()->setVisible(false);
69 #if defined(Q_WS_X11) && !defined(Q_WS_MAEMO_5)
70     toolBar->setIconSize(QSize(42, 42));
71 #endif
72
73     previousAction = addToolBarAction(view, SLOT(goPrevious()), "previous");
74     nextAction = addToolBarAction(view, SLOT(goNext()), "next");
75     chaptersAction = addToolBarAction(this, SLOT(showChapters()), "chapters");
76     bookmarksAction = addToolBarAction(this, SLOT(showBookmarks()), "bookmarks");
77
78 #ifdef Q_WS_MAEMO_5
79     infoAction = menuBar()->addAction(tr("Book details"));
80     connect(infoAction, SIGNAL(triggered()), this, SLOT(showInfo()));
81     libraryAction = menuBar()->addAction(tr("Library"));
82     connect(libraryAction, SIGNAL(triggered()), this, SLOT(showLibrary()));
83     settingsAction = menuBar()->addAction(tr("Settings"));
84     connect(settingsAction, SIGNAL(triggered()), this, SLOT(showSettings()));
85     devToolsAction = menuBar()->addAction(tr("Developer"));
86     connect(devToolsAction, SIGNAL(triggered()), this, SLOT(showDevTools()));
87 #else
88     infoAction = addToolBarAction(this, SLOT(showInfo()), "document-properties");
89     libraryAction = addToolBarAction(this, SLOT(showLibrary()),
90                                      "system-file-manager");
91     settingsAction = addToolBarAction(this, SLOT(showSettings()),
92                                       "preferences-system");
93     devToolsAction = addToolBarAction(this, SLOT(showDevTools()), "developer");
94 #endif // Q_WS_MAEMO_5
95
96     QFrame *frame = new QFrame(toolBar);
97     frame->setSizePolicy(QSizePolicy::Expanding, QSizePolicy::Fixed);
98     toolBar->addWidget(frame);
99
100     fullScreenAction = addToolBarAction(this, SLOT(showBig()), "view-fullscreen");
101
102     // Handle model changes
103     connect(Library::instance(), SIGNAL(nowReadingChanged()),
104             this, SLOT(onCurrentBookChanged()));
105
106     // Load book on command line, or load last read book, or load default book
107     Library *library = Library::instance();
108     if (QCoreApplication::arguments().size() == 2) {
109         QString path = QCoreApplication::arguments()[1];
110         library->add(path);
111         QModelIndex index = library->find(path);
112         if (index.isValid()) {
113             library->setNowReading(index);
114         }
115     }
116     else {
117         QModelIndex index = library->nowReading();
118         if (index.isValid()) {
119             library->setNowReading(index);
120         }
121         else {
122             if (!library->rowCount()) {
123                 library->add(":/books/2 B R 0 2 B.epub");
124             }
125             library->setNowReading(library->index(0));
126         }
127     }
128
129     // Handle settings changes
130     Settings *settings = Settings::instance();
131     connect(settings, SIGNAL(valueChanged(const QString &)),
132             this, SLOT(onSettingsChanged(const QString &)));
133     settings->setValue("orientation", settings->value("orientation"));
134     settings->setValue("lightson", settings->value("lightson"));
135
136     // Handle loading chapters
137     connect(view, SIGNAL(chapterLoadStart(int)),
138             this, SLOT(onChapterLoadStart()));
139     connect(view, SIGNAL(chapterLoadEnd(int)),
140             this, SLOT(onChapterLoadEnd(int)));
141
142     // Shadow window for full screen
143     fullScreenWindow = new FullScreenWindow(this);
144     connect(fullScreenWindow, SIGNAL(restore()), this, SLOT(showRegular()));
145
146     // Create thread for finding books in directories
147     bookFinder = new BookFinder();
148     connect(bookFinder, SIGNAL(add(const QString &)),
149             library, SLOT(add(const QString &)));
150     connect(bookFinder, SIGNAL(remove(const QString &)),
151             library, SLOT(remove(const QString &)));
152     bookFinder->moveToThread(&bookFinderThread);
153     bookFinderThread.start();
154
155 #if 0
156     bool ret = QMetaObject::invokeMethod(
157         bookFinder,
158         "find",
159         Q_ARG(QStringList, QStringList(QString("/Users/polster/Books"))),
160         Q_ARG(QStringList, library->bookPaths()));
161     t.trace(QString("Invoking BookFinder::find ") + (ret?"succeeded":"failed"));
162 #endif
163
164 #ifdef DORIAN_TEST_MODEL
165     (void)new ModelTest(Library::instance(), this);
166 #endif
167 }
168
169 MainWindow::~MainWindow()
170 {
171     bookFinderThread.quit();
172     bookFinderThread.wait();
173     delete bookFinder;
174 }
175
176 void MainWindow::onCurrentBookChanged()
177 {
178     setCurrentBook(Library::instance()->nowReading());
179 }
180
181 void MainWindow::showRegular()
182 {
183     Trace t("MainWindow::showRegular");
184     fullScreenWindow->hide();
185     fullScreenWindow->leaveChild();
186     view->setParent(centralWidget());
187     centralWidget()->layout()->addWidget(view);
188 }
189
190 void MainWindow::showBig()
191 {
192     Trace t("MainWindow::showBig");
193     centralWidget()->layout()->removeWidget(view);
194     fullScreenWindow->takeChild(view);
195     fullScreenWindow->showFullScreen();
196 }
197
198 void MainWindow::setCurrentBook(const QModelIndex &current)
199 {
200     mCurrent = current;
201     Book *book = Library::instance()->book(current);
202     view->setBook(book);
203     setWindowTitle(book? book->shortName(): tr("Dorian"));
204 }
205
206 QAction *MainWindow::addToolBarAction(const QObject *receiver,
207                                       const char *member,
208                                       const QString &name)
209 {
210     return toolBar->
211         addAction(QIcon(ICON_PREFIX + name + ".png"), "", receiver, member);
212 }
213
214 void MainWindow::showLibrary()
215 {
216     (new LibraryDialog(this))->show();
217 }
218
219 void MainWindow::showSettings()
220 {
221     (new SettingsWindow(this))->show();
222 }
223
224 void MainWindow::showInfo()
225 {
226     if (mCurrent.isValid()) {
227         (new InfoDialog(Library::instance()->book(mCurrent), this))->exec();
228     }
229 }
230
231 void MainWindow::showDevTools()
232 {
233     (new DevTools())->exec();
234 }
235
236 void MainWindow::showBookmarks()
237 {
238     Book *book = Library::instance()->book(mCurrent);
239     if (book) {
240         BookmarksDialog *bookmarks = new BookmarksDialog(book, this);
241         bookmarks->setWindowModality(Qt::WindowModal);
242         connect(bookmarks, SIGNAL(addBookmark()), this, SLOT(onAddBookmark()));
243         connect(bookmarks, SIGNAL(goToBookmark(int)),
244                 this, SLOT(onGoToBookmark(int)));
245         bookmarks->show();
246     }
247 }
248
249 void MainWindow::closeEvent(QCloseEvent *event)
250 {
251     Trace t("MainWindow::closeEvent");
252     view->setLastBookmark();
253     event->accept();
254 }
255
256 void MainWindow::onSettingsChanged(const QString &key)
257 {
258 #ifdef Q_WS_MAEMO_5
259     if (key == "orientation") {
260         QString value = Settings::instance()->value(key).toString();
261         Trace::trace(QString("MainWindow::onSettingsChanged: orientation %1").
262                      arg(value));
263         if (value == "portrait") {
264             setAttribute(Qt::WA_Maemo5LandscapeOrientation, false);
265             setAttribute(Qt::WA_Maemo5PortraitOrientation, true);
266         }
267         else {
268             setAttribute(Qt::WA_Maemo5PortraitOrientation, false);
269             setAttribute(Qt::WA_Maemo5LandscapeOrientation, true);
270         }
271     } else if (key == "lightson") {
272         bool enable = Settings::instance()->value(key, false).toBool();
273         Trace::trace(QString("MainWindow::onSettingsChanged: lightson: %1").
274                      arg(enable));
275         killTimer(preventBlankingTimer);
276         if (enable) {
277             preventBlankingTimer = startTimer(29 * 1000);
278         }
279     }
280 #else
281     Q_UNUSED(key);
282 #endif // Q_WS_MAEMO_5
283 }
284
285 void MainWindow::onChapterLoadStart()
286 {
287     Trace t("MainWindow::onChapterLoadStart");
288 #ifdef Q_WS_MAEMO_5
289     setAttribute(Qt::WA_Maemo5ShowProgressIndicator, true);
290 #endif
291 }
292
293 void MainWindow::onChapterLoadEnd(int index)
294 {
295     Trace t("MainWindow::onChapterLoadEnd");
296     bool enablePrevious = false;
297     bool enableNext = false;
298     Book *book = Library::instance()->book(mCurrent);
299     if (book) {
300         if (index > 0) {
301             enablePrevious = true;
302         }
303         if (index < (book->toc.size() - 1)) {
304             enableNext = true;
305         }
306     }
307 #ifdef Q_WS_MAEMO_5
308     setAttribute(Qt::WA_Maemo5ShowProgressIndicator, false);
309     previousAction->setIcon(QIcon(enablePrevious?
310         ":/icons/previous.png" : ":/icons/previous-disabled.png"));
311     nextAction->setIcon(QIcon(enableNext?
312         ":/icons/next.png": ":/icons/next-disabled.png"));
313 #endif // Q_WS_MAEMO_5
314     previousAction->setEnabled(enablePrevious);
315     nextAction->setEnabled(enableNext);
316 }
317
318 void MainWindow::onAddBookmark()
319 {
320     Trace t("MainWindow:onAddBookmark");
321     view->addBookmark();
322 }
323
324 void MainWindow::onGoToBookmark(int index)
325 {
326     Trace t("MainWindow::onGoToBookmark");
327     Book *book = Library::instance()->book(mCurrent);
328     view->goToBookmark(book->bookmarks()[index]);
329 }
330
331 void MainWindow::showChapters()
332 {
333     Book *book = Library::instance()->book(mCurrent);
334     if (book) {
335         ChaptersDialog *chapters = new ChaptersDialog(book, this);
336         chapters->setWindowModality(Qt::WindowModal);
337         connect(chapters, SIGNAL(goToChapter(int)),
338                 this, SLOT(onGoToChapter(int)));
339         chapters->show();
340     }
341 }
342
343 void MainWindow::onGoToChapter(int index)
344 {
345     Trace t("MainWindow::onGoToChapter");
346
347     Book *book = Library::instance()->book(mCurrent);
348     if (!book) {
349         t.trace("No current book?");
350         return;
351     }
352
353     QString id = book->chapters[index];
354     QString href = book->content[id].href;
355     QString baseRef(href);
356     QUrl url(QString("file://") + href);
357     if (url.hasFragment()) {
358         QString fragment = url.fragment();
359         baseRef.chop(fragment.length() + 1);
360     }
361
362     // Swipe through all content items to find the one matching the chapter href
363     // FIXME: Do we need to index content items by href, too?
364     QString contentKey;
365     bool found = false;
366     foreach (contentKey, book->content.keys()) {
367         if (contentKey == id) {
368             continue;
369         }
370         if (book->content[contentKey].href == baseRef) {
371             found = true;
372             t.trace(QString("Key for %1 is %2").arg(baseRef).arg(contentKey));
373             break;
374         }
375     }
376     if (!found) {
377         t.trace("Could not find key for " + baseRef);
378         return;
379     }
380
381     int tocIndex = book->toc.indexOf(contentKey);
382     if (tocIndex != -1) {
383         view->goToBookmark(Book::Bookmark(tocIndex, 0));
384     } else {
385         qCritical() << "Could not find toc index of chapter" << id;
386     }
387 }
388
389 void MainWindow::timerEvent(QTimerEvent *event)
390 {
391     if (event->timerId() == preventBlankingTimer) {
392 #ifdef Q_WS_MAEMO_5
393         QDBusInterface mce(MCE_SERVICE, MCE_REQUEST_PATH,
394                            MCE_REQUEST_IF, QDBusConnection::systemBus());
395         mce.call(MCE_PREVENT_BLANK_REQ);
396 #endif // Q_WS_MAEMO_5
397         Trace::trace("MainWindow::timerEvent: Prevent display blanking");
398     }
399 }