fb2view.cpp 9.4 KB

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