oleg
/
fb2edit
mirror of https://github.com/lintest/fb2edit.git


			
				
					
						
						
							123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689
							#include "fb2text.hpp"

#include <QVBoxLayout>
#include <QDockWidget>
#include <QFileDialog>
#include <QInputDialog>
#include <QMainWindow>
#include <QMenu>
#include <QToolBar>
#include <QWebInspector>
#include <QWebFrame>
#include <QWebPage>
#include <QtDebug>

#include "fb2dlgs.hpp"
#include "fb2note.hpp"
#include "fb2page.hpp"
#include "fb2save.hpp"
#include "fb2tree.hpp"
#include "fb2utils.h"

//---------------------------------------------------------------------------
//  FbTextAction
//---------------------------------------------------------------------------

FbTextAction::FbTextAction(const QString &text, QWebPage::WebAction action, FbTextEdit *parent)
    : QAction(text, parent)
    , m_action(action)
    , m_parent(parent)
{
}

FbTextAction::FbTextAction(const QIcon &icon, const QString &text, QWebPage::WebAction action, FbTextEdit* parent)
    : QAction(icon, text, parent)
    , m_action(action)
    , m_parent(parent)
{
}

QAction * FbTextAction::action()
{
    return m_parent->pageAction(m_action);
}

void FbTextAction::updateAction()
{
    if (QAction * act = action()) {
        if (isCheckable()) setChecked(act->isChecked());
        setEnabled(act->isEnabled());
    }
}

void FbTextAction::connectAction()
{
    if (QAction * act = action()) {
        connect(this, SIGNAL(triggered(bool)), act, SIGNAL(triggered(bool)));
        connect(act, SIGNAL(changed()), this, SLOT(updateAction()));
        if (isCheckable()) setChecked(act->isChecked());
        setEnabled(act->isEnabled());
    } else {
        if (isCheckable()) setChecked(false);
        setEnabled(false);
    }
}

void FbTextAction::disconnectAction()
{
    QAction * act = action();
    disconnect(act, 0, this, 0);
    disconnect(this, 0, act, 0);
}

//---------------------------------------------------------------------------
//  FbDockWidget
//---------------------------------------------------------------------------

FbDockWidget::FbDockWidget(const QString &title, QWidget *parent, Qt::WindowFlags flags)
    : QDockWidget(title, parent, flags)
{
    setFeatures(QDockWidget::DockWidgetClosable|QDockWidget::DockWidgetMovable|QDockWidget::DockWidgetFloatable);
    setAttribute(Qt::WA_DeleteOnClose);
}

//---------------------------------------------------------------------------
//  FbNoteView
//---------------------------------------------------------------------------

class FbNoteView : public QWebView
{
public:
    explicit FbNoteView(QWidget *parent, const QUrl &url);
    void hint(const QWebElement element, const QRect &rect);
protected:
    void paintEvent(QPaintEvent *event);
    const QUrl m_url;
};

FbNoteView::FbNoteView(QWidget *parent, const QUrl &url)
    : QWebView(parent)
    , m_url(url)
{
}

void FbNoteView::paintEvent(QPaintEvent *event)
{
    QWebView::paintEvent(event);
    QPainter painter(this);
    painter.setPen(Qt::black);
    QSize size = geometry().size() - QSize(1, 1);
    painter.drawRect( QRect(QPoint(0, 0), size) );
}

void FbNoteView::hint(const QWebElement element, const QRect &rect)
{
    QString html = element.toOuterXml();
    html.prepend(
        "<body bgcolor=lightyellow style='overflow:hidden;padding:0;margin:0;margin-top:2;'>"
        "<fb:body name=notes style='padding:0;margin:0;'>"
    );
    html.append("</fb:body></body>");
    setGeometry(rect);
    setHtml(html, m_url);
    show();
}

//---------------------------------------------------------------------------
//  FbTextBase
//---------------------------------------------------------------------------

void FbTextBase::addTools(QToolBar *tool)
{
    QAction *act;

    act = pageAction(QWebPage::Undo);
    act->setIcon(FbIcon("edit-undo"));
    act->setText(QObject::tr("&Undo"));
    act->setPriority(QAction::LowPriority);
    act->setShortcut(QKeySequence::Undo);
    tool->addAction(act);

    act = pageAction(QWebPage::Redo);
    act->setIcon(FbIcon("edit-redo"));
    act->setText(QObject::tr("&Redo"));
    act->setPriority(QAction::LowPriority);
    act->setShortcut(QKeySequence::Redo);
    tool->addAction(act);

    tool->addSeparator();

    act = pageAction(QWebPage::Cut);
    act->setIcon(FbIcon("edit-cut"));
    act->setText(QObject::tr("Cu&t"));
    act->setPriority(QAction::LowPriority);
    act->setShortcuts(QKeySequence::Cut);
    act->setStatusTip(QObject::tr("Cut the current selection's contents to the clipboard"));
    tool->addAction(act);

    act = pageAction(QWebPage::Copy);
    act->setIcon(FbIcon("edit-copy"));
    act->setText(QObject::tr("&Copy"));
    act->setPriority(QAction::LowPriority);
    act->setShortcuts(QKeySequence::Copy);
    act->setStatusTip(QObject::tr("Copy the current selection's contents to the clipboard"));
    tool->addAction(act);

    act = pageAction(QWebPage::Paste);
    act->setIcon(FbIcon("edit-paste"));
    act->setText(QObject::tr("&Paste"));
    act->setPriority(QAction::LowPriority);
    act->setShortcuts(QKeySequence::Paste);
    act->setStatusTip(QObject::tr("Paste the clipboard's contents into the current selection"));
    tool->addAction(act);

    tool->addSeparator();

    act = pageAction(QWebPage::RemoveFormat);
    act->setText(QObject::tr("Clear format"));
    act->setIcon(FbIcon("edit-clear"));

    act = pageAction(QWebPage::ToggleBold);
    act->setIcon(FbIcon("format-text-bold"));
    act->setText(QObject::tr("&Bold"));
    tool->addAction(act);

    act = pageAction(QWebPage::ToggleItalic);
    act->setIcon(FbIcon("format-text-italic"));
    act->setText(QObject::tr("&Italic"));
    tool->addAction(act);

    act = pageAction(QWebPage::ToggleStrikethrough);
    act->setIcon(FbIcon("format-text-strikethrough"));
    act->setText(QObject::tr("&Strikethrough"));
    tool->addAction(act);

    act = pageAction(QWebPage::ToggleSuperscript);
    act->setIcon(FbIcon("format-text-superscript"));
    act->setText(QObject::tr("Su&perscript"));
    tool->addAction(act);

    act = pageAction(QWebPage::ToggleSubscript);
    act->setIcon(FbIcon("format-text-subscript"));
    act->setText(QObject::tr("Su&bscript"));
    tool->addAction(act);
}

//---------------------------------------------------------------------------
//  FbTextEdit
//---------------------------------------------------------------------------

FbTextEdit::FbTextEdit(QWidget *parent, QObject *owner)
    : FbTextBase(parent)
    , m_owner(qobject_cast<QMainWindow*>(owner))
    , m_noteView(0)
    , m_thread(0)
    , dockTree(0)
    , dockNote(0)
    , dockImgs(0)
    , dockInsp(0)
{
    FbTextPage * p = new FbTextPage(this);
    setContextMenuPolicy(Qt::CustomContextMenu);
    connect(this, SIGNAL(customContextMenuRequested(QPoint)), SLOT(contextMenu(QPoint)));
    connect(p, SIGNAL(linkHovered(QString,QString,QString)), SLOT(linkHovered(QString,QString,QString)));
    connect(p->undoStack(), SIGNAL(cleanChanged(bool)), SLOT(cleanChanged(bool)));
    setPage(p);
}

FbTextEdit::~FbTextEdit()
{
    if (m_noteView) delete m_noteView;
}

FbTextPage *FbTextEdit::page()
{
    return qobject_cast<FbTextPage*>(FbTextBase::page());
}

FbStore *FbTextEdit::store()
{
    return page()->manager()->store();
}

QAction * FbTextEdit::act(Fb::Actions index) const
{
    return m_actions[index];
}

QAction * FbTextEdit::pAct(QWebPage::WebAction index) const
{
    return pageAction(index);
}

void FbTextEdit::setAction(Fb::Actions index, QAction *action)
{
    m_actions.insert(index, action);
}

void FbTextEdit::connectActions(QToolBar *tool)
{
    m_actions.connect();

    for (QAction *action: m_actions) {
        if (FbTextAction *a = qobject_cast<FbTextAction*>(action)) {
            a->connectAction();
        }
    }

    connect(act(Fb::EditFind), SIGNAL(triggered()), SLOT(find()));

    connect(act(Fb::InsertImage), SIGNAL(triggered()), SLOT(insertImage()));
    connect(act(Fb::InsertLink), SIGNAL(triggered()), SLOT(insertLink()));
    connect(act(Fb::InsertNote), SIGNAL(triggered()), SLOT(insertNote()));

    connect(act(Fb::ViewContents), SIGNAL(triggered(bool)), SLOT(viewContents(bool)));
    connect(act(Fb::ViewPictures), SIGNAL(triggered(bool)), SLOT(viewPictures(bool)));
    connect(act(Fb::ViewFootnotes), SIGNAL(triggered(bool)), SLOT(viewFootnotes(bool)));
    connect(act(Fb::ViewInspector), SIGNAL(triggered(bool)), SLOT(viewInspector(bool)));

    connect(act(Fb::ZoomIn), SIGNAL(triggered()), SLOT(zoomIn()));
    connect(act(Fb::ZoomOut), SIGNAL(triggered()), SLOT(zoomOut()));
    connect(act(Fb::ZoomReset), SIGNAL(triggered()), SLOT(zoomReset()));

    /*
    connect(actionTitle, SIGNAL(triggered()), textPage, SLOT(insertTitle()));
    connect(actionAnnot, SIGNAL(triggered()), textPage, SLOT(insertAnnot()));
    connect(actionAuthor, SIGNAL(triggered()), textPage, SLOT(insertAuthor()));
    connect(actionEpigraph, SIGNAL(triggered()), textPage, SLOT(insertEpigraph()));
    connect(actionSubtitle, SIGNAL(triggered()), textPage, SLOT(insertSubtitle()));
    connect(actionSection, SIGNAL(triggered()), textPage, SLOT(insertSection()));
    connect(actionStanza, SIGNAL(triggered()), textPage, SLOT(insertStanza()));
    connect(actionPoem, SIGNAL(triggered()), textPage, SLOT(insertPoem()));
    connect(actionDate, SIGNAL(triggered()), textPage, SLOT(insertDate()));
    connect(actionBody, SIGNAL(triggered()), textPage, SLOT(insertBody()));

    connect(actionSectionAdd, SIGNAL(triggered()), textPage, SLOT(createSection()));
    connect(actionSectionDel, SIGNAL(triggered()), textPage, SLOT(deleteSection()));
    connect(actionTextTitle, SIGNAL(triggered()), textPage, SLOT(createTitle()));
    */

    tool->clear();

    tool->addSeparator();
    tool->addAction(act(Fb::EditUndo));
    tool->addAction(act(Fb::EditRedo));

    tool->addSeparator();
    tool->addAction(act(Fb::EditCut));
    tool->addAction(act(Fb::EditCopy));
    tool->addAction(act(Fb::EditPaste));

    tool->addSeparator();
    tool->addAction(act(Fb::TextBold));
    tool->addAction(act(Fb::TextItalic));
    tool->addAction(act(Fb::TextStrike));
    tool->addAction(act(Fb::TextSup));
    tool->addAction(act(Fb::TextSub));

    tool->addSeparator();
    tool->addAction(act(Fb::SectionAdd));
    tool->addAction(act(Fb::SectionDel));
    tool->addAction(act(Fb::TextTitle));

    tool->addSeparator();
    tool->addAction(act(Fb::InsertImage));
    tool->addAction(act(Fb::InsertNote));
    tool->addAction(act(Fb::InsertLink));
    tool->addAction(act(Fb::InsertSection));
}

void FbTextEdit::disconnectActions()
{
    m_actions.disconnect();
    for (QAction *action: m_actions) {
        if (FbTextAction *a = qobject_cast<FbTextAction*>(action)) {
            a->disconnectAction();
        }
    }
    viewContents(false);
    viewPictures(false);
    viewInspector(false);
}

#ifdef QT_DEBUG
void FbTextEdit::exportHtml()
{
    FbSaveDialog dlg(this, tr("Save As..."));
    dlg.selectFile("filename.htm");
    if (!dlg.exec()) return;
    QString fileName = dlg.fileName();
    if (fileName.isEmpty()) return;

    QFile file(fileName);
    if (!file.open(QFile::WriteOnly)) return;
    QTextStream out(&file);
    out.setCodec("UTF-8");
    out << toHtml();

    QFileInfo fileInfo(fileName);
    QString dirName = fileInfo.path() + "/" + fileInfo.completeBaseName() + "/";
    QDir().mkpath(dirName);

    FbNetworkAccessManager *m = qobject_cast<FbNetworkAccessManager*>(page()->networkAccessManager());
    if (!m) return;

    int count = m->count();
    for (int i = 0; i < count; ++i) {
        QFile file(dirName + m->info(i, 0).toString());
        if (file.open(QFile::WriteOnly)) {
            file.write(m->data(i));
        }
    }
}
#endif // QT_DEBUG

void FbTextEdit::viewContents(bool show)
{
    if (show) {
        if (dockTree) { dockTree->show(); return; }
        dockTree = new FbDockWidget(tr("Contents"), this);
        dockTree->setWidget(new FbTreeWidget(this, m_owner));
        connect(dockTree, SIGNAL(visibilityChanged(bool)), act(Fb::ViewContents), SLOT(setChecked(bool)));
        connect(dockTree, SIGNAL(destroyed()), SLOT(treeDestroyed()));
        m_owner->addDockWidget(Qt::LeftDockWidgetArea, dockTree);
    } else if (dockTree) {
        dockTree->deleteLater();
        dockTree = 0;
    }
}

void FbTextEdit::viewPictures(bool show)
{
    if (show) {
        if (dockImgs) { dockImgs->show(); return; }
        dockImgs = new FbDockWidget(tr("Pictures"), this);
        dockImgs->setWidget(new FbImgsWidget(this, m_owner));
        connect(dockImgs, SIGNAL(visibilityChanged(bool)), act(Fb::ViewPictures), SLOT(setChecked(bool)));
        connect(dockImgs, SIGNAL(destroyed()), SLOT(imgsDestroyed()));
        m_owner->addDockWidget(Qt::RightDockWidgetArea, dockImgs);
    } else if (dockImgs) {
        dockImgs->deleteLater();
        dockImgs = 0;
    }
}

void FbTextEdit::viewFootnotes(bool show)
{
    if (show) {
        if (dockNote) { dockNote->show(); return; }
        dockNote = new FbDockWidget(tr("Footnotes"), this);
        dockNote->setWidget(new FbNotesWidget(this, m_owner));
        connect(dockNote, SIGNAL(visibilityChanged(bool)), act(Fb::ViewFootnotes), SLOT(setChecked(bool)));
        connect(dockNote, SIGNAL(destroyed()), SLOT(noteDestroyed()));
        m_owner->addDockWidget(Qt::RightDockWidgetArea, dockNote);
    } else if (dockNote) {
        dockNote->deleteLater();
        dockNote = 0;
    }
}

void FbTextEdit::viewInspector(bool show)
{
    if (show) {
        if (dockInsp) { dockInsp->show(); return; }
        QWebInspector *inspector = new QWebInspector(this);
        inspector->setPage(page());
        dockInsp = new QDockWidget(tr("Web inspector"), this);
        dockInsp->setFeatures(QDockWidget::DockWidgetClosable|QDockWidget::DockWidgetMovable|QDockWidget::DockWidgetFloatable);
        dockInsp->setWidget(inspector);
        connect(dockInsp, SIGNAL(visibilityChanged(bool)), act(Fb::ViewInspector), SLOT(setChecked(bool)));
        m_owner->addDockWidget(Qt::BottomDockWidgetArea, dockInsp);
    } else if (dockInsp) {
        dockInsp->hide();
    }
}

void FbTextEdit::hideDocks()
{
    return;
    if (dockTree) {
        dockTree->deleteLater();
        dockTree = 0;
    }
    if (dockImgs) {
        dockImgs->deleteLater();
        dockImgs = 0;
    }
    if (dockInsp) {
        dockInsp->hide();
    }
}

void FbTextEdit::treeDestroyed()
{
    m_actions[Fb::ViewContents]->setChecked(false);
    dockTree = 0;
}

void FbTextEdit::imgsDestroyed()
{
    m_actions[Fb::ViewPictures]->setChecked(false);
    dockImgs = 0;
}

void FbTextEdit::noteDestroyed()
{
    m_actions[Fb::ViewFootnotes]->setChecked(false);
    dockNote = 0;
}

FbNoteView & FbTextEdit::noteView()
{
    if (m_noteView) return *m_noteView;
    m_noteView = new FbNoteView(qobject_cast<QWidget*>(parent()), url());
    m_noteView->setPage(new FbTextPage(this));
    m_noteView->page()->setNetworkAccessManager(page()->networkAccessManager());
    m_noteView->page()->setContentEditable(false);
    m_noteView->setGeometry(QRect(100, 100, 400, 200));
    return *m_noteView;
}

QWebElement FbTextEdit::body()
{
    return doc().findFirst("body");
}

QWebElement FbTextEdit::doc()
{
    return page()->mainFrame()->documentElement();
}

void FbTextEdit::mouseMoveEvent(QMouseEvent *event)
{
    m_point = event->pos();
    QWebView::mouseMoveEvent(event);
}

void FbTextEdit::cleanChanged(bool clean)
{
    emit modificationChanged(!clean);
}

void FbTextEdit::contextMenu(const QPoint &pos)
{
    QMenu menu, *submenu;

    submenu = menu.addMenu(tr("Fo&rmat"));
    submenu->addAction(pageAction(QWebPage::RemoveFormat));
    submenu->addSeparator();
    submenu->addAction(pageAction(QWebPage::ToggleBold));
    submenu->addAction(pageAction(QWebPage::ToggleItalic));
    submenu->addAction(pageAction(QWebPage::ToggleStrikethrough));
    submenu->addAction(pageAction(QWebPage::ToggleSuperscript));
    submenu->addAction(pageAction(QWebPage::ToggleSubscript));

    menu.addSeparator();

    menu.addAction(pageAction(QWebPage::Cut));
    menu.addAction(pageAction(QWebPage::Copy));
    menu.addAction(pageAction(QWebPage::Paste));
    menu.addAction(pageAction(QWebPage::PasteAndMatchStyle));

    menu.exec(mapToGlobal(pos));
}

void FbTextEdit::linkHovered(const QString &link, const QString &title, const QString &textContent)
{
    Q_UNUSED(title);
    Q_UNUSED(textContent);

    const QString href = QUrl(link).fragment();
    if (href.isEmpty()) {
        if (m_noteView) m_noteView->hide();
        return;
    }

    const QString query = QString("fb\\:section#%1").arg(href);
    const QWebElement element = doc().findFirst(query);
    if (element.isNull()) {
        if (m_noteView) m_noteView->hide();
        return;
    }

    QRect rect = geometry();
    QSize size = element.geometry().size() + QSize(2, 4);
    int center = rect.size().height() / 2;
    int h = size.height();
    if (h > center) size.setHeight(center - 10);
    int x = (rect.size().width() - size.width()) / 2;
    int y = m_point.y();
    if ( y > h ) y = y - h - 10; else y = y + 10;
    QPoint point = QPoint(x, y) + rect.topLeft();
    noteView().hint(element, QRect(point, size));
}

bool FbTextEdit::save(QIODevice *device, const QString &codec)
{
    FbSaveWriter writer(*this, device);
    if (!codec.isEmpty()) writer.setCodec(codec.toLatin1());
    bool ok = FbSaveHandler(writer).save();
    if (ok) page()->undoStack()->setClean();
    return ok;
}

bool FbTextEdit::save(QByteArray *array)
{
    FbSaveWriter writer(*this, array);
    return FbSaveHandler(writer).save();
}

bool FbTextEdit::save(QString *string, int &anchor, int &focus)
{
    FbSaveWriter writer(*this, string);
    bool ok = FbSaveHandler(writer).save();
    anchor = writer.anchor();
    focus = writer.focus();
    return ok;
}

QString FbTextEdit::toHtml()
{
    return page()->mainFrame()->toHtml();
}

void FbTextEdit::zoomIn()
{
    qreal zoom = zoomFactor();
    setZoomFactor(zoom * 1.1);
}

void FbTextEdit::zoomOut()
{
    qreal zoom = zoomFactor();
    setZoomFactor(zoom * 0.9);
}

void FbTextEdit::zoomReset()
{
    setZoomFactor(1);
}

bool FbTextEdit::actionEnabled(QWebPage::WebAction action)
{
    QAction *act = pageAction(action);
    return act ? act->isEnabled() : false;
}

bool FbTextEdit::actionChecked(QWebPage::WebAction action)
{
    QAction *act = pageAction(action);
    return act ? act->isChecked() : false;
}

bool FbTextEdit::BoldChecked()
{
    return pageAction(QWebPage::ToggleBold)->isChecked();
}

bool FbTextEdit::ItalicChecked()
{
    return pageAction(QWebPage::ToggleItalic)->isChecked();
}

bool FbTextEdit::StrikeChecked()
{
    return pageAction(QWebPage::ToggleStrikethrough)->isChecked();
}

bool FbTextEdit::SubChecked()
{
    return pageAction(QWebPage::ToggleSubscript)->isChecked();
}

bool FbTextEdit::SupChecked()
{
    return pageAction(QWebPage::ToggleSuperscript)->isChecked();
}

void FbTextEdit::find()
{
    FbTextFindDlg dlg(*this);
    dlg.exec();
}

void FbTextEdit::insertImage()
{
    FbImageDlg dlg(this);
    if (dlg.exec()) {
        QString name = dlg.result();
        if (name.isEmpty()) return;
        QUrl url; url.setFragment(name);
        execCommand("insertImage", url.toString());
    }
}

void FbTextEdit::insertNote()
{
    FbNoteDlg dlg(this);
    dlg.exec();
}

void FbTextEdit::insertLink()
{
    bool ok;
    QString text = QInputDialog::getText(this, tr("Insert hyperlink"), tr("URL:"), QLineEdit::Normal, QString(), &ok);
    if (ok && !text.isEmpty()) execCommand("CreateLink", text);
}

void FbTextEdit::execCommand(const QString &cmd, const QString &arg)
{
    QString javascript = QString("document.execCommand(\"%1\",false,\"%2\")").arg(cmd).arg(arg);
    page()->mainFrame()->evaluateJavaScript(javascript);
}

//---------------------------------------------------------------------------
//  FbTextFrame
//---------------------------------------------------------------------------

FbTextFrame::FbTextFrame(QWidget *parent)
    : QFrame(parent)
{
    setFrameShape(QFrame::StyledPanel);
    setFrameShadow(QFrame::Sunken);

    QLayout * layout = new QVBoxLayout(this);
    layout->setSpacing(0);
    layout->setMargin(0);
    setLayout(layout);
}