1
0

fb2view.cpp 8.4 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324
  1. #include "fb2view.hpp"
  2. #include "fb2note.hpp"
  3. #include "fb2read.hpp"
  4. #include "fb2save.h"
  5. #include "fb2utils.h"
  6. #include "fb2xml2.h"
  7. #include <QAction>
  8. #include <QtDebug>
  9. #include <QFileDialog>
  10. #include <QNetworkRequest>
  11. #include <QToolTip>
  12. #include <QWebElement>
  13. #include <QWebFrame>
  14. #include <QWebPage>
  15. //---------------------------------------------------------------------------
  16. // Fb2NoteView
  17. //---------------------------------------------------------------------------
  18. class Fb2NoteView : public QWebView
  19. {
  20. public:
  21. explicit Fb2NoteView(QWidget* parent = 0) : QWebView(parent) {}
  22. void hint(const QWebElement element, const QRect &rect);
  23. protected:
  24. void paintEvent(QPaintEvent *event);
  25. };
  26. void Fb2NoteView::paintEvent(QPaintEvent *event)
  27. {
  28. QWebView::paintEvent(event);
  29. QPainter painter(this);
  30. painter.setPen(Qt::black);
  31. QSize size = geometry().size() - QSize(1, 1);
  32. painter.drawRect( QRect(QPoint(0, 0), size) );
  33. }
  34. void Fb2NoteView::hint(const QWebElement element, const QRect &rect)
  35. {
  36. QString html = element.toOuterXml();
  37. html.prepend(
  38. "<body bgcolor=lightyellow style='overflow:hidden;padding:0;margin:0;margin-top:2;'>"
  39. "<div class=body name=notes style='padding:0;margin:0;'>"
  40. );
  41. html.append("</div></body>");
  42. setGeometry(rect);
  43. setHtml(html);
  44. show();
  45. }
  46. //---------------------------------------------------------------------------
  47. // Fb2WebPage
  48. //---------------------------------------------------------------------------
  49. Fb2WebPage::Fb2WebPage(QObject *parent)
  50. : QWebPage(parent)
  51. {
  52. QWebSettings *s = settings();
  53. s->setAttribute(QWebSettings::AutoLoadImages, true);
  54. s->setAttribute(QWebSettings::DeveloperExtrasEnabled, true);
  55. s->setAttribute(QWebSettings::JavaEnabled, false);
  56. s->setAttribute(QWebSettings::JavascriptEnabled, true);
  57. s->setAttribute(QWebSettings::PrivateBrowsingEnabled, true);
  58. s->setAttribute(QWebSettings::PluginsEnabled, false);
  59. s->setAttribute(QWebSettings::ZoomTextOnly, true);
  60. s->setUserStyleSheetUrl(QUrl::fromLocalFile(":style.css"));
  61. }
  62. bool Fb2WebPage::acceptNavigationRequest(QWebFrame *frame, const QNetworkRequest &request, NavigationType type)
  63. {
  64. Q_UNUSED(frame);
  65. if (type == NavigationTypeLinkClicked) {
  66. qCritical() << request.url().fragment();
  67. return false;
  68. // QToolTip::showText(request.url().fragment());
  69. }
  70. return QWebPage::acceptNavigationRequest(frame, request, type);
  71. }
  72. //---------------------------------------------------------------------------
  73. // Fb2WebView
  74. //---------------------------------------------------------------------------
  75. Fb2WebView::Fb2WebView(QWidget *parent)
  76. : Fb2BaseWebView(parent)
  77. , m_noteView(0)
  78. , m_thread(0)
  79. {
  80. setPage(new Fb2WebPage(this));
  81. page()->setNetworkAccessManager(new Fb2NetworkAccessManager(*this));
  82. page()->setContentEditable(true);
  83. connect(page(), SIGNAL(contentsChanged()), this, SLOT(fixContents()));
  84. connect(page(), SIGNAL(linkHovered(QString,QString,QString)), this, SLOT(linkHovered(QString,QString,QString)));
  85. }
  86. Fb2WebView::~Fb2WebView()
  87. {
  88. FB2DELETE(m_noteView);
  89. }
  90. Fb2NoteView & Fb2WebView::noteView()
  91. {
  92. if (m_noteView) return *m_noteView;
  93. m_noteView = new Fb2NoteView(qobject_cast<QWidget*>(parent()));
  94. m_noteView->setPage(new Fb2WebPage(this));
  95. m_noteView->page()->setNetworkAccessManager(page()->networkAccessManager());
  96. m_noteView->page()->setContentEditable(false);
  97. m_noteView->setGeometry(QRect(100, 100, 400, 200));
  98. return *m_noteView;
  99. }
  100. QWebElement Fb2WebView::doc()
  101. {
  102. return page()->mainFrame()->documentElement();
  103. }
  104. QString Fb2WebView::toBodyXml()
  105. {
  106. QWebElement body = doc().findFirst("body");
  107. if (body.isNull()) return QString();
  108. return body.toOuterXml();
  109. }
  110. void Fb2WebView::fixContents()
  111. {
  112. foreach (QWebElement span, doc().findAll("span.apple-style-span[style]")) {
  113. span.removeAttribute("style");
  114. }
  115. }
  116. void Fb2WebView::mouseMoveEvent(QMouseEvent *event)
  117. {
  118. m_point = event->pos();
  119. QWebView::mouseMoveEvent(event);
  120. }
  121. void Fb2WebView::linkHovered(const QString &link, const QString &title, const QString &textContent)
  122. {
  123. Q_UNUSED(title);
  124. Q_UNUSED(textContent);
  125. const QString href = QUrl(link).fragment();
  126. if (href.isEmpty()) {
  127. if (m_noteView) m_noteView->hide();
  128. return;
  129. }
  130. const QString query = QString("DIV#%1").arg(href);
  131. const QWebElement element = doc().findFirst(query);
  132. if (element.isNull()) {
  133. if (m_noteView) m_noteView->hide();
  134. return;
  135. }
  136. QRect rect = geometry();
  137. QSize size = element.geometry().size() + QSize(2, 4);
  138. int center = rect.size().height() / 2;
  139. int h = size.height();
  140. if (h > center) size.setHeight(center - 10);
  141. int x = (rect.size().width() - size.width()) / 2;
  142. int y = m_point.y();
  143. if ( y > h ) y = y - h - 10; else y = y + 10;
  144. QPoint point = QPoint(x, y) + rect.topLeft();
  145. noteView().hint(element, QRect(point, size));
  146. }
  147. void Fb2WebView::load(const QString &filename, const QString &xml)
  148. {
  149. if (m_thread) return;
  150. m_thread = new Fb2ReadThread(this, filename, xml);
  151. m_thread->start();
  152. }
  153. bool Fb2WebView::save(QIODevice *device)
  154. {
  155. Fb2SaveHandler handler(*this, device);
  156. QXmlInputSource source;
  157. source.setData(toBodyXml());
  158. XML2::HtmlReader reader;
  159. reader.setContentHandler(&handler);
  160. reader.setErrorHandler(&handler);
  161. return reader.parse(source);
  162. }
  163. bool Fb2WebView::save(QByteArray *array, QList<int> *folds)
  164. {
  165. Fb2SaveHandler handler(*this, array, folds);
  166. QXmlInputSource source;
  167. source.setData(toBodyXml());
  168. XML2::HtmlReader reader;
  169. reader.setContentHandler(&handler);
  170. reader.setErrorHandler(&handler);
  171. return reader.parse(source);
  172. }
  173. bool Fb2WebView::save(QString *string)
  174. {
  175. // Use class QByteArray instead QString
  176. // to store information about encoding.
  177. QByteArray data;
  178. bool ok = save(&data);
  179. if (ok) *string = QString::fromUtf8(data.data());
  180. return ok;
  181. }
  182. QString Fb2WebView::temp(QString name)
  183. {
  184. return m_files.get(name).fileName();
  185. }
  186. void Fb2WebView::data(QString name, QByteArray data)
  187. {
  188. m_files.set(name, data);
  189. }
  190. void Fb2WebView::html(QString name, QString html)
  191. {
  192. setHtml(html, QUrl::fromLocalFile(name));
  193. if (m_thread) m_thread->deleteLater();
  194. m_thread = 0;
  195. }
  196. void Fb2WebView::zoomIn()
  197. {
  198. qreal zoom = zoomFactor();
  199. setZoomFactor(zoom * 1.1);
  200. }
  201. void Fb2WebView::zoomOut()
  202. {
  203. qreal zoom = zoomFactor();
  204. setZoomFactor(zoom * 0.9);
  205. }
  206. void Fb2WebView::zoomOrig()
  207. {
  208. setZoomFactor(1);
  209. }
  210. bool Fb2WebView::UndoEnabled()
  211. {
  212. return pageAction(QWebPage::Undo)->isEnabled();
  213. }
  214. bool Fb2WebView::RedoEnabled()
  215. {
  216. return pageAction(QWebPage::Redo)->isEnabled();
  217. }
  218. bool Fb2WebView::CutEnabled()
  219. {
  220. return pageAction(QWebPage::Cut)->isEnabled();
  221. }
  222. bool Fb2WebView::CopyEnabled()
  223. {
  224. return pageAction(QWebPage::Copy)->isEnabled();
  225. }
  226. bool Fb2WebView::BoldChecked()
  227. {
  228. return pageAction(QWebPage::ToggleBold)->isChecked();
  229. }
  230. bool Fb2WebView::ItalicChecked()
  231. {
  232. return pageAction(QWebPage::ToggleItalic)->isChecked();
  233. }
  234. bool Fb2WebView::StrikeChecked()
  235. {
  236. return pageAction(QWebPage::ToggleStrikethrough)->isChecked();
  237. }
  238. bool Fb2WebView::SubChecked()
  239. {
  240. return pageAction(QWebPage::ToggleSubscript)->isChecked();
  241. }
  242. bool Fb2WebView::SupChecked()
  243. {
  244. return pageAction(QWebPage::ToggleSuperscript)->isChecked();
  245. }
  246. void Fb2WebView::insertImage()
  247. {
  248. QString filters;
  249. filters += tr("Common Graphics (*.png *.jpg *.jpeg *.gif);;");
  250. filters += tr("Portable Network Graphics (PNG) (*.png);;");
  251. filters += tr("JPEG (*.jpg *.jpeg);;");
  252. filters += tr("Graphics Interchange Format (*.gif);;");
  253. filters += tr("All Files (*)");
  254. QString fn = QFileDialog::getOpenFileName(this, tr("Open image..."), QString(), filters);
  255. if (fn.isEmpty()) return;
  256. if (!QFile::exists(fn)) return;
  257. QUrl url = QUrl::fromLocalFile(fn);
  258. execCommand("insertImage", url.toString());
  259. }
  260. void Fb2WebView::insertNote()
  261. {
  262. Fb2NoteDlg dlg(*this);
  263. dlg.exec();
  264. }
  265. void Fb2WebView::insertLink()
  266. {
  267. }
  268. void Fb2WebView::execCommand(const QString &cmd, const QString &arg)
  269. {
  270. QString javascript = QString("document.execCommand(\"%1\",false,\"%2\")").arg(cmd).arg(arg);
  271. page()->mainFrame()->evaluateJavaScript(javascript);
  272. }
  273. QString Fb2WebView::status()
  274. {
  275. static const QString javascript = FB2::read(":/js/get_status.js");
  276. return page()->mainFrame()->evaluateJavaScript(javascript).toString();
  277. return QString();
  278. }