ztextedit.cpp

#include "ztextedit.h"
#include "daemon.h"
#include "mdialog.h"
#include <DGuiApplicationHelper>
#include <DInputDialog>
#include <QDebug>

DGUI_USE_NAMESPACE
ZTextEdit::ZTextEdit(QWidget* parent, bool DStyle)
    : DTextEdit(parent)
    , ctrlPressed(false)
    , hackDStyle(!DStyle)
    , marginBottom(0)
    , marginTop(0)
{
    const QSignalBlocker blocker(this);

    auto p = palette();
    qDebug() << p.currentColorGroup();
    p.setBrush(QPalette::Dark, QBrush(QColor(130, 130, 130)));
    setPalette(p);
    setTextInteractionFlags(textInteractionFlags() | Qt::LinksAccessibleByMouse);
    connect(this, &DTextEdit::currentCharFormatChanged, this, &ZTextEdit::updateCharFormat);
    sample = textCursor().charFormat();
    connect(this, &DTextEdit::textChanged, [this]() {
        //        qDebug() << "textChanged emitted on" << idx;
        Daemon::instance()->updateNoteContent(idx, QTextDocumentFragment(this->document()));
        commitModifications();
        //此处应采用事件队列，降低修改操作频率
    });
}

void ZTextEdit::keyPressEvent(QKeyEvent* e)
{

    if (e->key() == Qt::Key_Control)
        ctrlPressed = true;
    DTextEdit::keyPressEvent(e);
}
void ZTextEdit::keyReleaseEvent(QKeyEvent* e)
{
    if (e->key() == Qt::Key_Control)
        ctrlPressed = false;
    DTextEdit::keyPressEvent(e);
}
void ZTextEdit::mouseReleaseEvent(QMouseEvent* e)
{
    if (ctrlPressed) {
        auto cursor = textCursor();
        if (cursor.charFormat().isAnchor())
            QDesktopServices::openUrl(QUrl(cursor.charFormat().anchorHref()));
    }
    QWidget::mousePressEvent(e);
}
bool ZTextEdit::event(QEvent* e)
{
    if (hackDStyle)
        return QTextEdit::event(e);
    else
        return DTextEdit::event(e);
}

void ZTextEdit::insertFromMimeData(const QMimeData* src)
{
    if (!src->hasHtml() && src->hasImage()) {
        pInsImage(src->imageData());
    } else if (src->hasText() and !src->hasHtml()) {
        textCursor().insertText(src->text());
    } else {
        //        qDebug() << "insert html detected";
        auto cursor = textCursor();
        cursor.beginEditBlock();
        if (cursor.hasSelection())
            cursor.removeSelectedText();

        auto doc = src->html();
        //remove title marks
        doc.replace(QRegExp("<h\\d[^>]*>"), "<b>");
        doc.replace(QRegExp("</h\\d>"), "</b>");
        //remove table marks
        doc.remove(QRegExp("</?table[^>]*>"));
        doc.remove(QRegExp("</?tr[^>]*>"));
        doc.remove(QRegExp("</?th[^>]*>"));
        doc.replace(QRegExp("</?td[^>]*>"), "&nbsp;");
        doc.remove(QRegExp("</?tbody[^>]*>"));
        doc.remove(QRegExp("</?tfoot[^>]*>"));
        doc.remove(QRegExp("</?thead[^>]*>"));

        auto currentCharFormat = this->currentCharFormat();
        auto fmt = QTextCharFormat();
        auto blkfmt = QTextBlockFormat();

        auto reservedProperties = {
            QTextFormat::FontWeight,
            QTextFormat::TextUnderlineStyle,
            QTextFormat::FontItalic,
            QTextFormat::FontStrikeOut,
            QTextFormat::TextUnderlineStyle,
            QTextFormat::AnchorHref,
            QTextFormat::AnchorName,
            QTextFormat::IsAnchor
        };
        std::list<std::pair<QTextFormat::Property, QVariant>> filteredProperties = {
            { QTextFormat::BackgroundBrush, QVariant::fromValue(Qt::black) },
            { QTextFormat::ForegroundBrush, QVariant::fromValue(Qt::black) },
            { QTextFormat::FontPixelSize, QVariant::fromValue(int(ZTextEdit::DefaultFontPointSize)) },
            { QTextFormat::BlockAlignment, QVariant::fromValue(int(Qt::AlignLeft)) },
            { QTextFormat::LineHeight, QVariant::fromValue(0) },
            { QTextFormat::BlockTopMargin, QVariant::fromValue(qreal(0)) },
            { QTextFormat::BlockBottomMargin, QVariant::fromValue(qreal(0)) },
        };
        for (auto i : reservedProperties)
            fmt.clearProperty(i);
        for (auto i : filteredProperties)
            fmt.setProperty(i.first, i.second);
        for (auto i : filteredProperties)
            blkfmt.setProperty(i.first, i.second);

        int uninsertedPos = cursor.position();

        cursor.insertHtml(doc);
        int insertedPos = cursor.position();
        cursor.setPosition(uninsertedPos, QTextCursor::MoveAnchor);
        cursor.setPosition(insertedPos, QTextCursor::KeepAnchor);

        //remove invalid formats
        cursor.mergeCharFormat(fmt);
        cursor.mergeBlockFormat(blkfmt);

        cursor.setPosition(insertedPos, QTextCursor::MoveAnchor); //reset cursor
        //        setTextBackgroundColor(currentCharFormat.background().color());
        //        if (!uninsertedPos) //在空白textEdit中粘贴文本会，然后全选删除，会出现原因不明的光标下格式错误。?
        setCurrentCharFormat(currentCharFormat.toCharFormat());
        cursor.endEditBlock();

        updateResourcesList(toHtml());
        updateResources();
    }
}

inline void ZTextEdit::fEdit(QTextCharFormat& fmt)
{
    auto selection = textCursor();
    selection.beginEditBlock();
    if (!selection.hasSelection())
        mergeCurrentCharFormat(fmt);
    else
        selection.mergeCharFormat(fmt);
    selection.endEditBlock();
}
void ZTextEdit::fBold(bool b)
{
    auto fmt = QTextCharFormat();
    fmt.setFontWeight(b ? 68 : QFont::Normal);
    fEdit(fmt);
}
void ZTextEdit::fItalic(bool b)
{
    auto fmt = QTextCharFormat();
    fmt.setFontItalic(b);
    fEdit(fmt);
}
void ZTextEdit::fUnderline(bool b)
{
    auto fmt = QTextCharFormat();
    fmt.setFontUnderline(b);
    fEdit(fmt);
}
void ZTextEdit::fLinethrough(bool b)
{
    auto fmt = QTextCharFormat();
    fmt.setFontStrikeOut(b);
    fEdit(fmt);
}
void ZTextEdit::fUrl(bool b)
{
    if (b)
        return;
    auto fmt = QTextCharFormat();
    fmt.setAnchor(false);
    fmt.setFontUnderline(false);
    fmt.setForeground(sample.foreground());
    fmt.setAnchorHref("");
    fmt.setAnchorName("");
    fEdit(fmt);
}

void ZTextEdit::pSplitline()
{
    auto cursor = textCursor();
    cursor.beginEditBlock();
    cursor.insertHtml("<hr /><br />");
    cursor.deletePreviousChar();
    cursor.endEditBlock();
}
void ZTextEdit::pInsImage(const QVariant& data)
{
    //	qDebug() << size() << sizeHint();
    auto cursor = textCursor();
    QString imgHash;
    if (data.type() != QVariant::Image) {
        auto imageName = QFileDialog::getOpenFileName(this, "", "", "Images(*.jpg *.png *.svg *.tif *.bmp)");
        if (imageName.isEmpty())
            return;
        imgHash = Daemon::instance()->calcImageHash(QUrl::fromLocalFile(imageName));
    } else {
        QByteArray dataRaw;
        QBuffer buffer(&dataRaw);
        buffer.open(QIODevice::WriteOnly);
        data.value<QImage>().save(&buffer, "PNG");
        imgHash = Daemon::instance()->calcImageHash(dataRaw);
    }
    if (!resources.contains(imgHash))
        resources.append(imgHash);
    document()->addResource(QTextDocument::ImageResource, QUrl("" + imgHash),
        QVariant(processImage(Daemon::instance()->fetchImageData(imgHash))));
    QTextBlockFormat fmt;
    fmt.setAlignment(Qt::AlignCenter);
    cursor.insertText("\n");
    cursor.movePosition(QTextCursor::Up);
    cursor.insertBlock(fmt);
    cursor.insertHtml("<img src=\"" + imgHash + "\"/>");
    cursor.movePosition(QTextCursor::NextBlock);
}
void ZTextEdit::pInsUrl(bool b)
{
    if (!b)
        return;
    auto cursor = textCursor();
    cursor.beginEditBlock();
    auto bak = cursor.charFormat();
    cursor.insertHtml("<span>" + getUrlString() + "</span>");
    setCurrentCharFormat(bak);
    cursor.endEditBlock();
    updateCharFormat(currentCharFormat());
}
void ZTextEdit::pInsList()
{
    auto cursor = textCursor();
    cursor.beginEditBlock();
    QTextListFormat fmt = QTextListFormat();
    fmt.setStyle(QTextListFormat::ListCircle);
    cursor.insertList(fmt);
    cursor.endEditBlock();
}
void ZTextEdit::pInsOrderedList()
{
    auto cursor = textCursor();
    cursor.beginEditBlock();
    QTextListFormat fmt = QTextListFormat();
    fmt.setStyle(QTextListFormat::ListDecimal);
    cursor.insertList(fmt);
    cursor.endEditBlock();
}
void ZTextEdit::pInsCode()
{
    auto cursor = textCursor();
    cursor.beginEditBlock();
    QTextFrameFormat framefmt;
    framefmt.setBorder(2);
    auto frame = cursor.insertFrame(framefmt);
    cursor.insertText(DInputDialog::getMultiLineText(this, "Enter Your Code", "code"));
    cursor.movePosition(QTextCursor::PreviousBlock);
    //	cursor.setPosition(frame->lastPosition() + 1);
    cursor.endEditBlock();
    cursor.setPosition(document()->rootFrame()->firstPosition());
}
QString ZTextEdit::getUrlString()
{
    auto res = mDialog::getMultiText("Enter the Url", QList<QString>({ "name", "url" }));
    if (!res[1].contains("http"))
        res[1] = "http://" + res[1];
    return "<a href=\"" + res[1] + "\">" + res[0] + "</a>";
}
void ZTextEdit::updateCharFormat(const QTextCharFormat& nwfmt)
{
    if (textCursor().hasSelection())
        return;
    emit fBoldState(nwfmt.fontWeight() >= QFont::DemiBold);
    emit fItalicState(nwfmt.fontItalic());
    emit fUnderlineState(nwfmt.fontUnderline());
    emit fLinethroughState(nwfmt.fontStrikeOut());
    emit pIsUrl(nwfmt.isAnchor());
}
void ZTextEdit::setHtml(const QString& html)
{
    //    qDebug() << "call ZTextEdit::setHtml";
    qDebug() << html;
    updateResourcesList(html);
    updateResources();
    DTextEdit::setHtml(html);
    qDebug() << toHtml();
    updateDocumentFormat();
    qDebug() << toHtml();
}
void ZTextEdit::setBottomMargin(int d)
{
    marginBottom = d;
    { // init document format
        const QSignalBlocker blocker(this);
        QTextFrameFormat fmt;
        fmt.setBottomMargin(marginBottom);
        document()->rootFrame()->setFrameFormat(fmt);
    }
}
void ZTextEdit::setTopMargin(int d)
{
    marginTop = d;
}
void ZTextEdit::updateDocumentFormat()
{
    const QSignalBlocker blocker(this);
    QTextFrameFormat fmt;
    fmt.setBottomMargin(marginBottom);
    fmt.setTopMargin(marginTop);
    document()->rootFrame()->setFrameFormat(fmt);
}
QImage ZTextEdit::processImage(const QByteArray& data)
{
    QImage img = QImage::fromData(data);
    double ori = double(img.width()) / img.height(), view = double(size().width()) / size().height();
    if (ori > view) {
        if (size().width() < img.width())
            return img.scaledToWidth(int(size().width() * ImagePercentage));
        else
            return img;
    } else {
        if (size().height() < img.height())
            return img.scaledToHeight(int(size().height() * ImagePercentage));
        else
            return img;
    }
}
void ZTextEdit::resizeEvent(QResizeEvent* e)
{
    updateResources();
    DTextEdit::resizeEvent(e);
}
void ZTextEdit::updateResources()
{
    for (auto i : resources) {
        document()->addResource(QTextDocument::ImageResource, "" + i, (processImage(Daemon::instance()->fetchImageData(i))));
    }
}
void ZTextEdit::updateResourcesList(QString html)
{
    QRegExp reg("<img src=\"([^\"]*)\"");
    resources.clear();
    int from = 0;
    while ((from = reg.indexIn(html, from)) != -1) {
        qDebug() << "captured url" << reg.cap(1);
        if (reg.cap(1).right(2) != "==") {
            auto raw = Daemon::instance()->fetchRemoteResource(reg.cap(1));
            auto hash = Daemon::instance()->calcImageHash(raw);
            //            qDebug() << "hash=" << hash;
            html.replace('\"' + reg.cap(1) + '\"', '\"' + hash + '\"');
            if (!resources.contains(hash))
                resources.append(hash);
        } else
            resources.append(reg.cap(1));
        from += reg.matchedLength();
    }
    auto lst = std::unique(resources.begin(), resources.end());
    resources.erase(lst, resources.end());
    qDebug() << lst - resources.begin() << "resources in total captured";
}

void ZTextEdit::commitModifications()
{
    auto ori = idx;
    idx = Daemon::instance()->commitChange(idx);
    emit indexChanged(ori, idx);
}
void ZTextEdit::setNote(const ZNote& d)
{
    idx = InnerIndex(d.lastModified());
    const QSignalBlocker blocker(this);
    setHtml(d.getHtml());
    moveCursor(QTextCursor::End);
}