#include "qownnotesmarkdowntextedit.h"

#include <utils/gui.h>
#include <utils/listutils.h>
#include <utils/misc.h>
#include <utils/schema.h>

#include <QApplication>
#include <QClipboard>
#include <QDebug>
#include <QDesktopServices>
#include <QDir>
#include <QDragEnterEvent>
#include <QDragMoveEvent>
#include <QDropEvent>
#include <QEvent>
#include <QEventLoop>
#include <QFileInfo>
#include <QFont>
#include <QFontDatabase>
#include <QHash>
#include <QHelpEvent>
#include <QImageReader>
#include <QJSEngine>
#include <QKeyEvent>
#include <QLineEdit>
#include <QMenu>
#include <QMimeData>
#include <QNetworkAccessManager>
#include <QNetworkReply>
#include <QNetworkRequest>
#include <QPainter>
#include <QPixmap>
#include <QRegularExpression>
#include <QSet>
#include <QTextBlock>
#include <QTextCursor>
#include <QTextDocument>
#include <QTextDocumentFragment>
#include <QTextEdit>
#include <QTextFormat>
#include <QTextLayout>
#include <QTimer>
#include <QToolTip>
#include <QUrl>
#include <QVariant>
#include <QVector>
#include <QWidgetAction>
#include <QtGlobal>
#include <algorithm>

#include "entities/notefolder.h"
#include "helpers/qownnotesmarkdownhighlighter.h"
#include "helpers/qownspellchecker.h"
#include "libraries/qmarkdowntextedit/linenumberarea.h"
#include "mainwindow.h"
#ifdef LANGUAGETOOL_ENABLED
#include "services/languagetoolchecker.h"
#endif
#ifdef HARPER_ENABLED
#include "services/harperchecker.h"
#endif
#include "dialogs/markdowntabledialog.h"
#include "services/cloudservice.h"
#include "services/markdownlspclient.h"
#include "services/markdownlspdocumenttracker.h"
#include "services/markdownlspignoredrules.h"
#include "services/nextclouddeckservice.h"
#include "services/openaiservice.h"
#include "services/scriptingservice.h"
#include "services/settingsservice.h"
#include "utils/urlhandler.h"
#include "version.h"

namespace {
constexpr int kFoldIndicatorPadding = 4;
QHash<QString, QSet<QString>> s_foldedHeadingStateByNoteReference;

static QChar accentForDeadKey(int key) {
    switch (key) {
        case Qt::Key_Dead_Acute:
            return QChar(0x0301);
        case Qt::Key_Dead_Grave:
            return QChar(0x0300);
        case Qt::Key_Dead_Circumflex:
            return QChar(0x0302);
        case Qt::Key_Dead_Diaeresis:
            return QChar(0x0308);
        case Qt::Key_Dead_Tilde:
            return QChar(0x0303);
        default:
            return QChar();
    }
}

static QChar spacingAccentForCombiningMark(QChar accent) {
    switch (accent.unicode()) {
        case 0x0300:
            return QLatin1Char('`');
        case 0x0301:
            return QChar(0x00B4);
        case 0x0302:
            return QLatin1Char('^');
        case 0x0303:
            return QLatin1Char('~');
        case 0x0308:
            return QChar(0x00A8);
        default:
            return QChar();
    }
}

static QString composeDeadKey(QChar accent, QChar character) {
    return QString(character).append(accent).normalized(QString::NormalizationForm_C);
}

struct SetextHeadingUnderline {
    int level = 0;
    int leadingSpaces = 0;
    int markerCount = 0;

    bool isValid() const { return level > 0; }
};

static int markdownHeadingIndent(const QString &line) {
    int i = 0;
    while (i < line.size() && i < 3 && line.at(i) == QLatin1Char(' ')) {
        ++i;
    }

    return i;
}

static int atxHeadingLevel(const QString &line, int *headingMarkerStart = nullptr,
                           int *headingMarkerEnd = nullptr) {
    int i = markdownHeadingIndent(line);
    const int markerStart = i;
    while (i < line.size() && line.at(i) == QLatin1Char('#')) {
        ++i;
    }

    const int headingLevel = i - markerStart;
    if (headingLevel <= 0 || headingLevel > 6) {
        return 0;
    }

    if (i < line.size() && line.at(i) != QLatin1Char(' ') && line.at(i) != QLatin1Char('\t')) {
        return 0;
    }

    if (headingMarkerStart) {
        *headingMarkerStart = markerStart;
    }

    if (headingMarkerEnd) {
        *headingMarkerEnd = i;
    }

    return headingLevel;
}

static SetextHeadingUnderline parseSetextHeadingUnderline(const QString &line) {
    SetextHeadingUnderline underline;

    int i = markdownHeadingIndent(line);
    underline.leadingSpaces = i;

    if (i >= line.size()) {
        return underline;
    }

    const QChar marker = line.at(i);
    if (marker != QLatin1Char('=') && marker != QLatin1Char('-')) {
        return underline;
    }

    const int markerStart = i;
    while (i < line.size() && line.at(i) == marker) {
        ++i;
    }

    underline.markerCount = i - markerStart;

    while (i < line.size() && (line.at(i) == QLatin1Char(' ') || line.at(i) == QLatin1Char('\t'))) {
        ++i;
    }

    if (i != line.size()) {
        return {};
    }

    underline.level = (marker == QLatin1Char('=')) ? 1 : 2;
    return underline;
}

static int boundedHeadingLevel(const int headingLevel, const int levelDelta) {
#if __cplusplus >= 201703L
    return std::clamp(headingLevel + levelDelta, 1, 6);
#else
    return qBound(1, headingLevel + levelDelta, 6);
#endif
}

QString changeHeadingDepth(const QString &text, const int levelDelta) {
    QString normalizedText = text;
    normalizedText.replace(QChar(0x2029), QLatin1Char('\n'));

    QStringList lines = normalizedText.split(QLatin1Char('\n'),
#if QT_VERSION < QT_VERSION_CHECK(5, 15, 0)
                                             QString::KeepEmptyParts
#else
                                             Qt::KeepEmptyParts
#endif
    );
    QStringList updatedLines;
    updatedLines.reserve(lines.size());

    for (int lineIndex = 0; lineIndex < lines.size(); ++lineIndex) {
        const QString &line = lines.at(lineIndex);

        int headingMarkerStart = 0;
        int headingMarkerEnd = 0;
        const int atxLevel = atxHeadingLevel(line, &headingMarkerStart, &headingMarkerEnd);
        if (atxLevel > 0) {
            const int newHeadingLevel = boundedHeadingLevel(atxLevel, levelDelta);
            if (newHeadingLevel == atxLevel) {
                updatedLines.append(line);
                continue;
            }

            updatedLines.append(line.left(headingMarkerStart) +
                                QString(newHeadingLevel, QLatin1Char('#')) +
                                line.mid(headingMarkerEnd));
            continue;
        }

        if (lineIndex + 1 < lines.size()) {
            const SetextHeadingUnderline underline =
                parseSetextHeadingUnderline(lines.at(lineIndex + 1));
            const int titleIndent = markdownHeadingIndent(line);
            const QString titleText = line.mid(titleIndent).trimmed();

            if (underline.isValid() && !titleText.isEmpty()) {
                const int newHeadingLevel = boundedHeadingLevel(underline.level, levelDelta);
                if (newHeadingLevel <= 2) {
                    updatedLines.append(line);

                    if (newHeadingLevel == underline.level) {
                        updatedLines.append(lines.at(lineIndex + 1));
                    } else {
                        const int underlineWidth = std::max(
                            underline.markerCount, std::max(static_cast<int>(titleText.size()), 3));
                        const QChar marker =
                            (newHeadingLevel == 1) ? QLatin1Char('=') : QLatin1Char('-');
                        updatedLines.append(QString(underline.leadingSpaces, QLatin1Char(' ')) +
                                            QString(underlineWidth, marker));
                    }
                } else {
                    updatedLines.append(line.left(titleIndent) +
                                        QString(newHeadingLevel, QLatin1Char('#')) +
                                        QStringLiteral(" ") + titleText);
                }

                ++lineIndex;
                continue;
            }
        }

        updatedLines.append(line);
    }

    return updatedLines.join(QLatin1Char('\n'));
}

struct InnerSelectionCandidate {
    int innerStart = -1;
    int innerEnd = -1;

    bool isValid() const { return innerStart >= 0 && innerEnd >= innerStart; }
    int length() const { return innerEnd - innerStart; }
};

static void addSelectionCandidate(InnerSelectionCandidate &best, const int innerStart,
                                  const int innerEnd, const int targetStart, const int targetEnd) {
    if (innerStart < 0 || innerEnd < innerStart || innerStart > targetStart ||
        innerEnd < targetEnd) {
        return;
    }

    if (!best.isValid() || (innerEnd - innerStart) < best.length() ||
        ((innerEnd - innerStart) == best.length() && innerStart > best.innerStart)) {
        best.innerStart = innerStart;
        best.innerEnd = innerEnd;
    }
}

static bool isEscapedToken(const QString &text, const int pos) {
    int backslashCount = 0;

    for (int i = pos - 1; i >= 0 && text.at(i) == QLatin1Char('\\'); --i) {
        ++backslashCount;
    }

    return (backslashCount % 2) == 1;
}

static void addStackedTokenCandidates(const QString &text, const QString &openToken,
                                      const QString &closeToken, const int targetStart,
                                      const int targetEnd, InnerSelectionCandidate &best) {
    QVector<int> openPositions;
    const int maxStart = text.size() - std::min(openToken.size(), closeToken.size());

    for (int i = 0; i <= maxStart;) {
        if (text.mid(i, closeToken.size()) == closeToken && !openPositions.isEmpty()) {
            const int openPos = openPositions.takeLast();
            addSelectionCandidate(best, openPos + openToken.size(), i, targetStart, targetEnd);
            i += closeToken.size();
            continue;
        }

        if (text.mid(i, openToken.size()) == openToken) {
            openPositions.append(i);
            i += openToken.size();
            continue;
        }

        ++i;
    }
}

static void addRepeatedTokenCandidates(const QString &text, const QString &token,
                                       const int targetStart, const int targetEnd,
                                       InnerSelectionCandidate &best, const bool escapable) {
    int openPos = -1;

    for (int i = 0; i <= text.size() - token.size();) {
        if (text.mid(i, token.size()) != token || (escapable && isEscapedToken(text, i))) {
            ++i;
            continue;
        }

        if (openPos < 0) {
            openPos = i;
        } else {
            addSelectionCandidate(best, openPos + token.size(), i, targetStart, targetEnd);
            openPos = -1;
        }

        i += token.size();
    }
}

static void addMarkdownRangeCandidate(const QString &blockText,
                                      MarkdownHighlighter *markdownHighlighter,
                                      const MarkdownHighlighter::RangeType rangeType,
                                      const int blockNumber, const int probePosition,
                                      const int targetStart, const int targetEnd,
                                      InnerSelectionCandidate &best) {
    if (!markdownHighlighter || probePosition < 0 || probePosition >= blockText.size()) {
        return;
    }

    const auto range = markdownHighlighter->getSpanRange(rangeType, blockNumber, probePosition);
    if (range.first < 0 || range.second < 0 || range.first >= blockText.size()) {
        return;
    }

    if (rangeType == MarkdownHighlighter::RangeType::CodeSpan) {
        int delimiterLength = 0;

        while ((range.first + delimiterLength) < blockText.size() &&
               blockText.at(range.first + delimiterLength) == QLatin1Char('`')) {
            ++delimiterLength;
        }

        if (delimiterLength > 0) {
            addSelectionCandidate(best, range.first + delimiterLength, range.second, targetStart,
                                  targetEnd);
        }

        return;
    }

    const QChar marker = blockText.at(range.first);
    if (rangeType != MarkdownHighlighter::RangeType::Emphasis ||
        (marker != QLatin1Char('*') && marker != QLatin1Char('_'))) {
        return;
    }

    int leftMarkerStart = range.first;
    while (leftMarkerStart > 0 && blockText.at(leftMarkerStart - 1) == marker) {
        --leftMarkerStart;
    }

    const int delimiterLength = range.first - leftMarkerStart + 1;
    int rightMarkerIndex = range.second;

    if (rightMarkerIndex >= blockText.size() || blockText.at(rightMarkerIndex) != marker) {
        --rightMarkerIndex;
    }

    const int rightMarkerStart = rightMarkerIndex - delimiterLength + 1;
    if (delimiterLength <= 0 || rightMarkerStart < 0 || rightMarkerIndex < 0 ||
        blockText.at(rightMarkerIndex) != marker) {
        return;
    }

    for (int i = rightMarkerStart; i <= rightMarkerIndex; ++i) {
        if (blockText.at(i) != marker) {
            return;
        }
    }

    addSelectionCandidate(best, leftMarkerStart + delimiterLength, rightMarkerStart, targetStart,
                          targetEnd);
}

struct WikiLinkCompletionContext {
    QString filterText;
    int startPosition = -1;
    int cursorPosition = -1;
};

static bool currentWikiLinkCompletionContext(const QPlainTextEdit *edit,
                                             WikiLinkCompletionContext &context) {
    if (!edit) {
        return false;
    }

    QTextCursor cursor = edit->textCursor();
    if (cursor.hasSelection()) {
        return false;
    }

    const QTextBlock block = cursor.block();
    const QString blockText = block.text();
    const int positionInBlock = cursor.positionInBlock();
    const QString leftText = blockText.left(positionInBlock);
    const int openPos = leftText.lastIndexOf(QStringLiteral("[["));
    if (openPos < 0) {
        return false;
    }

    const QString candidate = leftText.mid(openPos + 2);
    if (candidate.contains(QChar('|')) || candidate.contains(QStringLiteral("]]")) ||
        candidate.contains(QChar('[')) || candidate.contains(QChar(']')) ||
        candidate.contains(QChar('\n'))) {
        return false;
    }

    context.filterText = candidate.trimmed();
    context.startPosition = block.position() + openPos + 2;
    context.cursorPosition = cursor.position();
    return true;
}
}    // namespace

// Initialize static member
QOwnNotesMarkdownTextEdit *QOwnNotesMarkdownTextEdit::_activeAutocompleteEditor = nullptr;

QOwnNotesMarkdownTextEdit::QOwnNotesMarkdownTextEdit(QWidget *parent)
    : QMarkdownTextEdit(parent, false) {
    // We need to set the internal variable to true, because we start with a highlighter
    _highlightingEnabled = true;
    _highlighter = nullptr;
    if (!parent || parent->objectName() != QStringLiteral("LogWidget")) {
        _highlighter = new QOwnNotesMarkdownHighlighter(document());

        connect(_highlighter, &QOwnNotesMarkdownHighlighter::highlightingFinished, this,
                &QOwnNotesMarkdownTextEdit::refreshFoldingSidebar);
        connect(_highlighter, &QOwnNotesMarkdownHighlighter::highlightingFinished, this,
                &QOwnNotesMarkdownTextEdit::scheduleRestoreCurrentFoldedHeadingState);

        _markdownLspTracker = new MarkdownLspDocumentTracker(this);

        setStyles();
        updateSettings();
    }

    // Initialize AI autocomplete timer (only for note editors, not log widget)
    _aiAutocompleteTimer = new QTimer(this);
    _aiAutocompleteTimer->setSingleShot(true);
    _aiAutocompleteTimer->setInterval(500);    // Wait 500ms after typing stops
    connect(_aiAutocompleteTimer, &QTimer::timeout, this,
            &QOwnNotesMarkdownTextEdit::requestAiAutocomplete);

    _markdownLspChangeTimer = new QTimer(this);
    _markdownLspChangeTimer->setSingleShot(true);
    _markdownLspChangeTimer->setInterval(200);
    connect(_markdownLspChangeTimer, &QTimer::timeout, this,
            &QOwnNotesMarkdownTextEdit::sendMarkdownLspChange);

    // NOTE: Callback registration is now done globally in OpenAiService constructor
    // We just need to register this editor as active if it's a note editor
    if (!parent || parent->objectName() != QStringLiteral("LogWidget")) {
        qDebug() << __func__ << " - Registering as active editor";
        registerAsActiveEditor();
    } else {
        qDebug() << __func__ << " - Skipping registration for non-editor widget:" << objectName();
    }

    // Use DirectConnection to ensure signal is delivered immediately before widget can be destroyed
    // This prevents the "0 receivers" problem when widgets are recreated
    bool conn1 = connect(
        OpenAiService::instance(), &OpenAiService::autocompleteCompleted, this,
        [this](const QString &result) {
            qDebug() << "*** LAMBDA RECEIVED autocompleteCompleted signal for widget:" << this
                     << objectName();
            this->onAiAutocompleteCompleted(result);
        },
        Qt::DirectConnection);
    bool conn2 = connect(OpenAiService::instance(), &OpenAiService::autocompleteErrorOccurred, this,
                         &QOwnNotesMarkdownTextEdit::onAiAutocompleteTimeout, Qt::DirectConnection);

    qDebug() << __func__
             << " - AI autocomplete signals connected (DirectConnection), conn1:" << conn1
             << "conn2:" << conn2;
    qDebug() << __func__ << " - Connected to OpenAiService instance:" << OpenAiService::instance();

    // Test: Call the slot directly to verify it exists
    qDebug() << __func__ << " - Testing slot by calling it directly...";
    onAiAutocompleteCompleted("TEST DIRECT CALL");
    qDebug() << __func__ << " - Direct call completed";

    // Connect to text changes to trigger autocomplete
    connect(this, &QOwnNotesMarkdownTextEdit::textChanged, this, [this]() {
        // Only trigger if autocomplete is enabled
        if (OpenAiService::getAutocompleteEnabled() && OpenAiService::getEnabled()) {
            // Skip for log widgets
            if (objectName() == QStringLiteral("logTextEdit")) {
                return;
            }
            // Don't clear or restart timer if we're currently inserting a suggestion
            if (_isInsertingAiSuggestion) {
                return;
            }
            clearAiAutocompleteSuggestion();
            _aiAutocompleteTimer->start();
        }
    });

    SettingsService settings;
    MarkdownHighlighter::HighlightingOptions options;

    if (settings.value(QStringLiteral("fullyHighlightedBlockquotes")).toBool()) {
        options |= MarkdownHighlighter::HighlightingOption ::FullyHighlightedBlockQuote;
    }
    if (settings.value(QStringLiteral("MainWindow/noteTextView.underline")).toBool()) {
        options |= MarkdownHighlighter::HighlightingOption ::Underline;
    }

    // set the highlighting options
    if (_highlighter) {
        _highlighter->setHighlightingOptions(options);

        // re-initialize the highlighting rules if we are using some options
        if (options != MarkdownHighlighter::HighlightingOption::None) {
            _highlighter->initHighlightingRules();
        }
    }

    // ignores note clicks in QMarkdownTextEdit in the note text edit
    setIgnoredClickUrlSchemata(QStringList({"note", "task", "deck", "wikilink"}));

    connect(this, &QOwnNotesMarkdownTextEdit::zoomIn, this, [this]() { onZoom(/*in=*/true); });
    connect(this, &QOwnNotesMarkdownTextEdit::zoomOut, this, [this]() { onZoom(/*in=*/false); });

    connect(this, &QOwnNotesMarkdownTextEdit::urlClicked, this, [this](const QString &url) {
        if (!MainWindow::instance()) {
            qWarning() << "No MainWindow! shouldn't happen!";
            return;
        }
        // Use the openUrl method which properly handles the openInNewTab flag
        openUrl(url, _openLinkInNewTab);
    });

    connect(MainWindow::instance(), &MainWindow::settingsChanged, this,
            &QOwnNotesMarkdownTextEdit::updateSettings);

    connect(MarkdownLspIgnoredRules::instance(), &MarkdownLspIgnoredRules::ignoredRulesChanged,
            this, &QOwnNotesMarkdownTextEdit::refreshMarkdownLspDiagnostics);

#ifdef LANGUAGETOOL_ENABLED
    connect(LanguageToolChecker::instance(), &LanguageToolChecker::blockMatchesUpdated, this,
            [this](const QVector<int> &blockNumbers) {
                if (!document() || !highlighter()) {
                    return;
                }

                if (blockNumbers.isEmpty()) {
                    highlighter()->rehighlight();
                    return;
                }

                for (const int blockNumber : blockNumbers) {
                    const QTextBlock block = document()->findBlockByNumber(blockNumber);
                    if (block.isValid()) {
                        highlighter()->rehighlightBlock(block);
                    }
                }
            });
#endif

#ifdef HARPER_ENABLED
    connect(HarperChecker::instance(), &HarperChecker::blockMatchesUpdated, this,
            [this](const QVector<int> &blockNumbers) {
                if (!document() || !highlighter()) {
                    return;
                }

                if (blockNumbers.isEmpty()) {
                    highlighter()->rehighlight();
                    return;
                }

                for (const int blockNumber : blockNumbers) {
                    const QTextBlock block = document()->findBlockByNumber(blockNumber);
                    if (block.isValid()) {
                        highlighter()->rehighlightBlock(block);
                    }
                }
            });
#endif

    setContextMenuPolicy(Qt::CustomContextMenu);
    connect(this, &QOwnNotesMarkdownTextEdit::customContextMenuRequested, this,
            &QOwnNotesMarkdownTextEdit::onContextMenu);

    refreshFoldingSidebar();
}

/*
 * Prevent infinite loops of QResizeEvents in conjunction with setPaperMargins()
 * when the widget is resized (and leave some space for line numbers too)
 * See: https://github.com/pbek/QOwnNotes/issues/2679
 */
QSize QOwnNotesMarkdownTextEdit::minimumSizeHint() const {
    int lineWidthLeftMargin = _lineNumArea->lineNumAreaWidth();

    // Let the min size be the defaultMinSize + lineNumAreaWidth + paper margin
    auto sizeHint = QMarkdownTextEdit::minimumSizeHint();
    sizeHint.rwidth() += lineWidthLeftMargin + 10;

    return sizeHint;
}

void QOwnNotesMarkdownTextEdit::onZoom(bool in) {
    FontModificationMode mode = in ? Increase : Decrease;
    const int fontSize = modifyFontSize(mode);

    auto mainWindow = MainWindow::instance();
    if (mainWindow && MainWindow::isInDistractionFreeMode()) {
        setPaperMargins();
        if (in) {
            mainWindow->showStatusBarMessage(tr("Increased font size to %1 pt").arg(fontSize),
                                             QStringLiteral("🔤"), 3000);
        } else {
            mainWindow->showStatusBarMessage(tr("Decreased font size to %1 pt").arg(fontSize),
                                             QStringLiteral("🔤"), 3000);
        }
    }

    setPaperMargins();
}

/**
 * Sets the format style
 *
 * @param index
 * @param styles
 */
void QOwnNotesMarkdownTextEdit::setFormatStyle(MarkdownHighlighter::HighlighterState index) {
    QTextCharFormat format;
    Utils::Schema::schemaSettings->setFormatStyle(index, format);

    if (index == MarkdownHighlighter::HighlighterState::WikiLink) {
        format.setFontUnderline(true);
        format.setUnderlineStyle(QTextCharFormat::DotLine);
    } else if (index == MarkdownHighlighter::HighlighterState::WikiLinkBroken) {
        format.setFontUnderline(true);
        format.setUnderlineStyle(QTextCharFormat::DashUnderline);
    }

    if (_highlighter) {
        _highlighter->setTextFormat(index, format);
    }
}

/**
 * Overrides the font size style if overrideInterfaceFontSize was set to prevent
 * Utils::Gui::updateInterfaceFontSize from overriding the default text size on
 * Windows 10
 *
 * @param fontSize
 */
void QOwnNotesMarkdownTextEdit::overrideFontSizeStyle(int fontSize) {
    bool overrideInterfaceFontSize =
        SettingsService().value(QStringLiteral("overrideInterfaceFontSize"), false).toBool();

    // remove old style
    QString stylesheet = styleSheet().remove(QRegularExpression(
        QRegularExpression::escape(
            QOWNNOTESMARKDOWNTEXTEDIT_OVERRIDE_FONT_SIZE_STYLESHEET_PRE_STRING) +
        ".*" +
        QRegularExpression::escape(
            QOWNNOTESMARKDOWNTEXTEDIT_OVERRIDE_FONT_SIZE_STYLESHEET_POST_STRING)));

    if (overrideInterfaceFontSize) {
        // using pt is important here, px didn't work properly
        stylesheet +=
            QStringLiteral(QOWNNOTESMARKDOWNTEXTEDIT_OVERRIDE_FONT_SIZE_STYLESHEET_PRE_STRING) +
            "QOwnNotesMarkdownTextEdit {font-size: " + QString::number(fontSize) + "pt;}" +
            QStringLiteral(QOWNNOTESMARKDOWNTEXTEDIT_OVERRIDE_FONT_SIZE_STYLESHEET_POST_STRING);
    }

    setStyleSheet(stylesheet);
}

/**
 * Sets the highlighting styles for the text edit
 */
void QOwnNotesMarkdownTextEdit::setStyles() {
    QFont font = Utils::Schema::schemaSettings->getEditorTextFont();
    setFont(font);

    // workaround for Windows 10 if overrideInterfaceFontSize was set
    overrideFontSizeStyle(font.pointSize());

    // set the tab stop to the width of 4 spaces in the editor
    const int tabStop = 4;
    QFontMetrics metrics(font);

#if QT_VERSION < QT_VERSION_CHECK(5, 11, 0)
    setTabStopWidth(tabStop * metrics.width(' '));
#else
    setTabStopDistance(tabStop * metrics.horizontalAdvance(' '));
#endif

    setFormatStyle(MarkdownHighlighter::HighlighterState::H1);
    setFormatStyle(MarkdownHighlighter::HighlighterState::H2);
    setFormatStyle(MarkdownHighlighter::HighlighterState::H3);
    setFormatStyle(MarkdownHighlighter::HighlighterState::H4);
    setFormatStyle(MarkdownHighlighter::HighlighterState::H5);
    setFormatStyle(MarkdownHighlighter::HighlighterState::H6);
    setFormatStyle(MarkdownHighlighter::HighlighterState::HorizontalRule);
    setFormatStyle(MarkdownHighlighter::HighlighterState::List);
    setFormatStyle(MarkdownHighlighter::HighlighterState::CheckBoxChecked);
    setFormatStyle(MarkdownHighlighter::HighlighterState::CheckBoxUnChecked);
    setFormatStyle(MarkdownHighlighter::HighlighterState::Bold);
    setFormatStyle(MarkdownHighlighter::HighlighterState::Italic);
    setFormatStyle(MarkdownHighlighter::HighlighterState::StUnderline);
    setFormatStyle(MarkdownHighlighter::HighlighterState::BlockQuote);
    setFormatStyle(MarkdownHighlighter::HighlighterState::CodeBlock);
    setFormatStyle(MarkdownHighlighter::HighlighterState::Comment);
    setFormatStyle(MarkdownHighlighter::HighlighterState::MaskedSyntax);
    setFormatStyle(MarkdownHighlighter::HighlighterState::Image);
    setFormatStyle(MarkdownHighlighter::HighlighterState::InlineCodeBlock);
    setFormatStyle(MarkdownHighlighter::HighlighterState::Link);
    setFormatStyle(MarkdownHighlighter::HighlighterState::LinkInternal);
    setFormatStyle(MarkdownHighlighter::HighlighterState::WikiLink);
    setFormatStyle(MarkdownHighlighter::HighlighterState::WikiLinkBroken);
    setFormatStyle(MarkdownHighlighter::HighlighterState::Table);
    setFormatStyle(MarkdownHighlighter::HighlighterState::BrokenLink);
    setFormatStyle(MarkdownHighlighter::HighlighterState::TrailingSpace);

    setFormatStyle(MarkdownHighlighter::HighlighterState::CodeType);
    setFormatStyle(MarkdownHighlighter::HighlighterState::CodeKeyWord);
    setFormatStyle(MarkdownHighlighter::HighlighterState::CodeComment);
    setFormatStyle(MarkdownHighlighter::HighlighterState::CodeString);
    setFormatStyle(MarkdownHighlighter::HighlighterState::CodeNumLiteral);
    setFormatStyle(MarkdownHighlighter::HighlighterState::CodeBuiltIn);
    setFormatStyle(MarkdownHighlighter::HighlighterState::CodeOther);

#ifdef Q_OS_WIN32
    // set the selection background color to a light blue if not in dark mode
    if (!SettingsService().value(QStringLiteral("darkMode")).toBool()) {
        // light green (#9be29b) could be another choice, but be aware that
        // this color will be used for mouse and keyboard selections too
        setStyleSheet(styleSheet() +
                      "QWidget {selection-color: #ffffff;"
                      "selection-background-color: #3399ff}");
    }
#endif
}

/**
 * Modifies the font size of the text edit
 */
int QOwnNotesMarkdownTextEdit::modifyFontSize(FontModificationMode mode) {
    SettingsService settings;
    QFont font = this->font();
    int fontSize = font.pointSize();
    bool doSetStyles = false;

    // modify the text edit default font
    QString fontString = settings.value(QStringLiteral("MainWindow/noteTextEdit.font")).toString();
    if (!fontString.isEmpty()) {
        font.fromString(fontString);

        fontSize = font.pointSize();

        switch (mode) {
            case FontModificationMode::Increase:
                fontSize++;
                doSetStyles = true;
                break;
            case FontModificationMode::Decrease:
                fontSize--;

                if (fontSize < 5) {
                    fontSize = 5;
                } else {
                    doSetStyles = true;
                }
                break;
            default:
                QPlainTextEdit textEdit;
                int newFontSize = textEdit.font().pointSize();
                if (fontSize != newFontSize) {
                    fontSize = newFontSize;
                    doSetStyles = true;
                }
        }

        if (fontSize > 0) {
            font.setPointSize(fontSize);
        }

        // store the font settings
        settings.setValue(QStringLiteral("MainWindow/noteTextEdit.font"), font.toString());
    }

    // modify the text edit code font
    fontString = settings.value(QStringLiteral("MainWindow/noteTextEdit.code.font")).toString();
    if (!fontString.isEmpty()) {
        font.fromString(fontString);

        int codeFontSize = font.pointSize();

        switch (mode) {
            case FontModificationMode::Increase:
                codeFontSize++;
                doSetStyles = true;
                break;
            case FontModificationMode::Decrease:
                codeFontSize--;

                if (codeFontSize < 5) {
                    codeFontSize = 5;
                } else {
                    doSetStyles = true;
                }
                break;
            default:
                QPlainTextEdit textEdit;
                int newCodeFontSize = textEdit.font().pointSize();
                if (codeFontSize != newCodeFontSize) {
                    codeFontSize = newCodeFontSize;
                    doSetStyles = true;
                }
        }

        if (codeFontSize > 0) {
            font.setPointSize(codeFontSize);
        }

        // store the font settings
        settings.setValue(QStringLiteral("MainWindow/noteTextEdit.code.font"), font.toString());
    }

    if (doSetStyles) {
        this->setStyles();
        if (_highlighter) {
            _highlighter->rehighlight();
        }
    }

    return fontSize;
}

/**
 * Handles clicked urls (including relative urls)
 *
 * examples:
 * - <https://www.qownnotes.org> opens the webpage
 * - <file:///path/to/my/file/QOwnNotes.pdf> opens the file
 * "/path/to/my/file/QOwnNotes.pdf" if the operating system
 * supports that handler
 */
void QOwnNotesMarkdownTextEdit::openUrl(const QString &urlString, bool openInNewTab) {
    qDebug() << "QOwnNotesMarkdownTextEdit " << __func__ << " - 'urlString': " << urlString
             << " - 'openInNewTab': " << openInNewTab;

    QString notesPath = NoteFolder::currentLocalPath();
    QString windowsSlash = QString();

#ifdef Q_OS_WIN32
    // we need another slash for Windows
    windowsSlash = QStringLiteral("/");
#endif

    auto urlCopy = urlString;

    // parse for relative file urls and make them absolute
    urlCopy.replace(QRegularExpression(QStringLiteral("^file:[\\/]{2}([^\\/].+)$")),
                    QStringLiteral("file://") + windowsSlash + notesPath + QStringLiteral("/\\1"));

    // Check if this is a note URL that should be handled specially
    QUrl url(urlCopy);
    const QString scheme = url.scheme();

    // If it's a note URL, noteid URL, file URL in note folder, has no scheme (relative link),
    // or is a Nextcloud Deck card URL, use UrlHandler which knows how to handle these properly
    if (scheme == QStringLiteral("note") || scheme == QStringLiteral("noteid") ||
        scheme == QStringLiteral("wikilink") || scheme == QStringLiteral("file") ||
        scheme.isEmpty() || Note::fileUrlIsNoteInCurrentNoteFolder(url) ||
        NextcloudDeckService::isCardUrl(urlCopy)) {
        // Use UrlHandler for note and Deck URLs with the openInNewTab flag
        UrlHandler().openUrl(urlCopy, openInNewTab);
    } else {
        // For other URLs (http, https, etc.), use the base class implementation
        QMarkdownTextEdit::openUrl(urlCopy, openInNewTab);
    }
}

// void QOwnNotesMarkdownTextEdit::setViewportMargins(
//        int left, int top, int right, int bottom) {
//    QMarkdownTextEdit::setViewportMargins(left, top, right, bottom);
//}

/**
 * Sets the viewport margins for the distraction free mode
 */
void QOwnNotesMarkdownTextEdit::setPaperMargins(int width) {
    SettingsService settings;
    bool isInDistractionFreeMode =
        settings.value(QStringLiteral("DistractionFreeMode/isEnabled")).toBool();
    bool editorWidthInDFMOnly =
        settings.value(QStringLiteral("Editor/editorWidthInDFMOnly"), true).toBool();

    if (isInDistractionFreeMode || !editorWidthInDFMOnly) {
        int margin = 0;

        if (width == -1) {
            width = this->width();
        }

        int editorWidthMode =
            settings.value(QStringLiteral("DistractionFreeMode/editorWidthMode")).toInt();

        if (editorWidthMode != Full) {
            QFontMetrics metrics(font());

            int characterAmount = 0;
            switch (editorWidthMode) {
                case Medium:
                    characterAmount = 80;
                    break;
                case Wide:
                    characterAmount = 100;
                    break;
                case Custom:
                    characterAmount =
                        settings.value(QStringLiteral("DistractionFreeMode/editorWidthCustom"), 80)
                            .toInt();
                    break;
                default:
                case Narrow:
                    characterAmount = 60;
                    break;
            }

                // set the size of characterAmount times the size of "O"
                // characters
#if QT_VERSION < QT_VERSION_CHECK(5, 11, 0)
            int proposedEditorWidth = metrics.width(QStringLiteral("O").repeated(characterAmount));
#else
            int proposedEditorWidth =
                metrics.horizontalAdvance(QStringLiteral("O").repeated(characterAmount));
#endif

            // Apply a factor to correct the faulty calculated margin
            // Use a different factor for monospaced fonts
            // TODO(pbek): I don't know better way to get around this yet
            proposedEditorWidth /= usesMonospacedFont() ? 0.95 : 1.332;

            // calculate the margin to be applied
            margin = (width - proposedEditorWidth) / 2;

            if (margin < 0) {
                margin = 0;
            }
        }

        setViewportMargins(margin, 20, margin, 0);
    } else {
        int lineWidthLeftMargin = lineNumberArea()->lineNumAreaWidth();

        setLineNumberLeftMarginOffset(10);
        setViewportMargins(10 + lineWidthLeftMargin, 10, 10, 0);
    }
}

/**
 * Try to determine if the used font is monospaced
 *
 * @return
 */
bool QOwnNotesMarkdownTextEdit::usesMonospacedFont() {
    QFontMetrics metrics(font());

#if QT_VERSION < QT_VERSION_CHECK(5, 11, 0)
    int widthNarrow = metrics.width(QStringLiteral("iiiii"));
    int widthWide = metrics.width(QStringLiteral("WWWWW"));
#else
    int widthNarrow = metrics.horizontalAdvance(QStringLiteral("iiiii"));
    int widthWide = metrics.horizontalAdvance(QStringLiteral("WWWWW"));
#endif

    return widthNarrow == widthWide;
}

void QOwnNotesMarkdownTextEdit::toggleCase() {
    QTextCursor c = textCursor();
    // Save positions to restore everything at the end
    const int selectionStart = c.selectionStart();
    const int selectionEnd = c.selectionEnd();
    const int cPos = c.position();

    QString selectedText = c.selectedText();
    const bool textWasSelected = !selectedText.isEmpty();

    // if no text is selected: automatically select the Word under the Cursor
    if (selectedText.isEmpty()) {
        c.select(QTextCursor::WordUnderCursor);
        selectedText = c.selectedText();
    }

    // cycle text through lowercase, uppercase, start case, and sentence case
    c.insertText(Utils::Misc::cycleTextCase(selectedText));

    if (textWasSelected) {
        // select the text again to maybe do another operation on it
        // keep the original cursor position
        if (cPos == selectionStart) {
            c.setPosition(selectionEnd, QTextCursor::MoveAnchor);
            c.setPosition(selectionStart, QTextCursor::KeepAnchor);
        } else {
            c.setPosition(selectionStart, QTextCursor::MoveAnchor);
            c.setPosition(selectionEnd, QTextCursor::KeepAnchor);
        }
    } else {
        // Just restore the Cursor Position if no text was selected
        c.setPosition(cPos, QTextCursor::MoveAnchor);
    }
    // Restore the visible cursor
    setTextCursor(c);
}

void QOwnNotesMarkdownTextEdit::selectEnclosedText() {
    QTextCursor cursor = textCursor();
    const QTextBlock block = cursor.block();
    if (!block.isValid()) {
        return;
    }

    const QString blockText = block.text();
    if (blockText.isEmpty()) {
        return;
    }

    const int blockStart = block.position();
    int targetStart = cursor.positionInBlock();
    int targetEnd = targetStart;

    if (cursor.hasSelection()) {
        QTextCursor selectionStartCursor(document());
        selectionStartCursor.setPosition(cursor.selectionStart());

        const int selectionEndPos = std::max(cursor.selectionStart(), cursor.selectionEnd() - 1);
        QTextCursor selectionEndCursor(document());
        selectionEndCursor.setPosition(selectionEndPos);

        if (selectionStartCursor.block() != block || selectionEndCursor.block() != block) {
            return;
        }

        targetStart = cursor.selectionStart() - blockStart;
        targetEnd = cursor.selectionEnd() - blockStart;
    }

    InnerSelectionCandidate bestCandidate;
    const int lastBlockPosition = blockText.size() - 1;
    const int probePosition =
        qBound(0, std::min(targetStart, lastBlockPosition), lastBlockPosition);

    addMarkdownRangeCandidate(blockText, highlighter(), MarkdownHighlighter::RangeType::CodeSpan,
                              block.blockNumber(), probePosition, targetStart, targetEnd,
                              bestCandidate);
    addMarkdownRangeCandidate(blockText, highlighter(), MarkdownHighlighter::RangeType::Emphasis,
                              block.blockNumber(), probePosition, targetStart, targetEnd,
                              bestCandidate);

    addStackedTokenCandidates(blockText, QStringLiteral("[["), QStringLiteral("]]"), targetStart,
                              targetEnd, bestCandidate);
    addStackedTokenCandidates(blockText, QStringLiteral("("), QStringLiteral(")"), targetStart,
                              targetEnd, bestCandidate);
    addStackedTokenCandidates(blockText, QStringLiteral("["), QStringLiteral("]"), targetStart,
                              targetEnd, bestCandidate);
    addStackedTokenCandidates(blockText, QStringLiteral("{"), QStringLiteral("}"), targetStart,
                              targetEnd, bestCandidate);

    addRepeatedTokenCandidates(blockText, QStringLiteral("~~"), targetStart, targetEnd,
                               bestCandidate, false);
    addRepeatedTokenCandidates(blockText, QStringLiteral("\""), targetStart, targetEnd,
                               bestCandidate, true);
    addRepeatedTokenCandidates(blockText, QStringLiteral("'"), targetStart, targetEnd,
                               bestCandidate, true);

    if (!bestCandidate.isValid()) {
        return;
    }

    cursor.setPosition(blockStart + bestCandidate.innerStart);
    cursor.setPosition(blockStart + bestCandidate.innerEnd, QTextCursor::KeepAnchor);
    setTextCursor(cursor);
}

void QOwnNotesMarkdownTextEdit::insertCodeBlock() {
    QTextCursor c = this->textCursor();
    QString selectedText = c.selection().toPlainText();

    if (selectedText.isEmpty()) {
        // insert multi-line code block if cursor is in an empty line
        if (c.atBlockStart() && c.atBlockEnd()) {
            c.insertText(QStringLiteral("```\n\n```"));
            c.movePosition(QTextCursor::Left, QTextCursor::MoveAnchor, 3);
        } else {
            c.insertText(QStringLiteral("``"));
        }

        c.movePosition(QTextCursor::Left, QTextCursor::MoveAnchor);
        setTextCursor(c);
    } else {
        bool addNewline = false;

        // if the selected text has multiple lines add a multi-line code block
        if (selectedText.contains(QStringLiteral("\n"))) {
            // add another newline if there is no newline at the end of the
            // selected text
            const QString endNewline =
                selectedText.endsWith(QLatin1String("\n")) ? QString() : QStringLiteral("\n");

            selectedText =
                QStringLiteral("``\n") + selectedText + endNewline + QStringLiteral("``");
            addNewline = true;
        }

        c.insertText(QStringLiteral("`") + selectedText + QStringLiteral("`"));

        if (addNewline) {
            c.insertText(QStringLiteral("\n"));
        }
    }
}

void QOwnNotesMarkdownTextEdit::insertWikiLink() {
    QTextCursor cursor = textCursor();
    const QString selectedText = cursor.selectedText();

    if (!selectedText.isEmpty()) {
        cursor.insertText(QStringLiteral("[[") + selectedText + QStringLiteral("]]"));
        setTextCursor(cursor);
        return;
    }

    cursor.insertText(QStringLiteral("[[]]"));
    cursor.movePosition(QTextCursor::Left, QTextCursor::MoveAnchor, 2);
    setTextCursor(cursor);

    if (Note::isWikiLinkSupportEnabled()) {
        QTimer::singleShot(0, this, &QOwnNotesMarkdownTextEdit::onAutoCompleteRequested);
    }
}

void QOwnNotesMarkdownTextEdit::onAutoCompleteRequested() {
    if (isReadOnly() || !Utils::Misc::isNoteEditingAllowed()) {
        if (openLinkAtCursorPosition()) {
            MainWindow::instance()->showStatusBarMessage(
                tr("An url was opened at the current cursor position"), QStringLiteral("📃"), 5000);
        }

        return;
    }

    // attempt to toggle a checkbox at the cursor position
    if (Utils::Gui::toggleCheckBoxAtCursor(this)) {
        return;
    }

    QString wikiFilterText;
    int wikiReplaceLength = 0;
    QStringList wikiResultList;
    const bool wikiContextActive =
        Note::isWikiLinkSupportEnabled() &&
        wikiLinkAutoComplete(wikiResultList, wikiFilterText, wikiReplaceLength);

    // Don't treat typing inside a wiki-link target as an activation request.
    // This prevents completing the leading "[[" from opening or creating the note.
    if (!wikiContextActive && openLinkAtCursorPosition()) {
        MainWindow::instance()->showStatusBarMessage(
            tr("An url was opened at the current cursor position"), QStringLiteral("📃"), 5000);
        return;
    }

    if (_markdownLspEnabled && _markdownLspClient && !_markdownLspUri.isEmpty()) {
        const QTextCursor cursor = textCursor();
        const int line = cursor.blockNumber();
        const int character = cursor.positionInBlock();
        _markdownLspCompletionRequestId =
            _markdownLspClient->requestCompletion(_markdownLspUri, line, character);
        if (_markdownLspCompletionRequestId >= 0) {
            return;
        }
    }

    // attempt a Markdown table auto-format
    if (Utils::Gui::autoFormatTableAtCursor(this)) {
        return;
    }

    QMenu menu;

    if (wikiContextActive) {
        for (const QString &text : Utils::asConst(wikiResultList)) {
            auto *action = menu.addAction(text);
            action->setData(text);
            action->setWhatsThis(QStringLiteral("wikilink-autocomplete"));
        }
    }

    double resultValue;
    if (solveEquation(resultValue)) {
        const QString text = QString::number(resultValue);
        auto *action = menu.addAction(QStringLiteral("= ") + text);
        action->setData(text);
        action->setWhatsThis(QStringLiteral("equation"));
    }

    QStringList resultList;
    if (!wikiContextActive && autoComplete(resultList)) {
        for (const QString &text : Utils::asConst(resultList)) {
            auto *action = menu.addAction(text);
            action->setData(text);
            action->setWhatsThis(QStringLiteral("autocomplete"));
        }
    }

    // load texts from scripts to show in the autocompletion list
    const QStringList autocompletionList = ScriptingService::instance()->callAutocompletionHook();
    if (!autocompletionList.isEmpty()) {
        auto *action = menu.addAction(QString());
        action->setSeparator(true);

        for (const QString &text : autocompletionList) {
            auto *newAction = menu.addAction(text);
            newAction->setData(text);
            newAction->setWhatsThis(QStringLiteral("autocomplete"));
        }
    }

    QPoint globalPos = mapToGlobal(cursorRect().bottomRight());

    // compensate viewport margins
    globalPos.setY(globalPos.y() + viewportMargins().top());
    globalPos.setX(globalPos.x() + viewportMargins().left());

    if (menu.actions().count() > 0) {
        QAction *selectedItem = menu.exec(globalPos);
        if (selectedItem) {
            const QString text = selectedItem->data().toString();
            const QString type = selectedItem->whatsThis();

            if (text.isEmpty()) {
                return;
            }

            if (type == QStringLiteral("wikilink-autocomplete")) {
                WikiLinkCompletionContext context;
                if (!currentWikiLinkCompletionContext(this, context)) {
                    return;
                }

                QTextCursor c = textCursor();
                c.setPosition(context.startPosition, QTextCursor::MoveAnchor);
                c.setPosition(context.cursorPosition, QTextCursor::KeepAnchor);
                c.insertText(text);

                const QString rightText = toPlainText().mid(c.position(), 2);
                if (rightText != QStringLiteral("]]")) {
                    c.insertText(QStringLiteral("]]"));
                    c.movePosition(QTextCursor::Left, QTextCursor::MoveAnchor, 2);
                }

                setTextCursor(c);
            } else if (type == QStringLiteral("autocomplete")) {
                // overwrite the currently written word
                QTextCursor c = textCursor();
                c.movePosition(QTextCursor::StartOfWord, QTextCursor::KeepAnchor);
                c.insertText(text + QStringLiteral(" "));
            } else {
                insertPlainText(text);
            }
        }
    }
}

/**
 * Returns the text from the current cursor to the start of the word in the
 * note text edit
 *
 * @param withPreviousCharacters also get more characters at the beginning
 *                               to get characters like "@" that are not
 *                               word-characters
 * @return
 */
QString QOwnNotesMarkdownTextEdit::currentWord(bool withPreviousCharacters) const {
    QTextCursor c = textCursor();

    // get the text from the current word
    c.movePosition(QTextCursor::EndOfWord);
    c.movePosition(QTextCursor::StartOfWord, QTextCursor::KeepAnchor);

    QString text = c.selectedText();

    if (withPreviousCharacters) {
        static const QRegularExpression re(QStringLiteral("^[\\s\\n][^\\s]*"));
        do {
            c.movePosition(QTextCursor::Left, QTextCursor::KeepAnchor);
            text = c.selectedText();
        } while (!(re.match(text).hasMatch() || c.atBlockStart()));
    }

    return text.trimmed();
}

QString QOwnNotesMarkdownTextEdit::currentBlock() const {
    QTextCursor cursor = textCursor();
    QTextBlock currentBlock = cursor.block();
    return currentBlock.text();
}

/**
 * Tries to find words that start with the current word in the note text edit
 *
 * @param resultList
 * @return
 */
bool QOwnNotesMarkdownTextEdit::autoComplete(QStringList &resultList) const {
    // get the text from the current cursor to the start of the word
    const QString text = currentWord();
    qDebug() << __func__ << " - 'text': " << text;

    if (text.isEmpty()) {
        return false;
    }

    const QString noteText = toPlainText();

    // find all items that match our current word
    resultList =
        noteText
            .split(QRegularExpression(QStringLiteral("[^\\w\\d]"),
                                      QRegularExpression::UseUnicodePropertiesOption),
#if QT_VERSION < QT_VERSION_CHECK(5, 15, 0)
                   QString::SkipEmptyParts)
#else
                   Qt::SkipEmptyParts)
#endif
            .filter(QRegularExpression(QStringLiteral("^") + QRegularExpression::escape(text),
                                       QRegularExpression::CaseInsensitiveOption));

    // we only want each word once
    resultList.removeDuplicates();

    // remove the text we already entered
    resultList.removeOne(text);

    if (resultList.count() == 0) {
        return false;
    }

    qDebug() << __func__ << " - 'resultList': " << resultList;

    return true;
}

bool QOwnNotesMarkdownTextEdit::wikiLinkAutoComplete(QStringList &resultList, QString &filterText,
                                                     int &replaceLength) const {
    WikiLinkCompletionContext context;
    if (!currentWikiLinkCompletionContext(this, context)) {
        return false;
    }

    filterText = context.filterText;
    replaceLength = context.cursorPosition - context.startPosition;

    QSet<QString> results;
    const QVector<Note> notes = Note::fetchAll();
    for (const Note &note : notes) {
        results.insert(note.getName());

        const QString subfolderPath = note.relativeNoteSubFolderPath();
        if (!subfolderPath.isEmpty()) {
            results.insert(subfolderPath + QStringLiteral("/") + note.getName());
        }
    }

    resultList = results.values();
    std::sort(resultList.begin(), resultList.end(),
              [](const QString &a, const QString &b) { return a.toLower() < b.toLower(); });

    if (!filterText.isEmpty()) {
        resultList = resultList.filter(QRegularExpression(
            QRegularExpression::escape(filterText), QRegularExpression::CaseInsensitiveOption));
    }

    resultList.removeDuplicates();
    resultList.removeAll(filterText);
    return !resultList.isEmpty();
}

/**
 * Tries to find an equation in the current line and solves it
 *
 * @param returnValue
 * @return
 */
bool QOwnNotesMarkdownTextEdit::solveEquation(double &returnValue) {
    QTextCursor c = textCursor();

    // get the text from the current cursor to the start of the line
    c.movePosition(QTextCursor::StartOfBlock, QTextCursor::KeepAnchor);
    QString text = c.selectedText();
    qDebug() << __func__ << " - 'text': " << text;

    QString equation = text;

    // replace "," with "." to allow "," as coma
    equation.replace(QLatin1Char(','), QLatin1Char('.'));

    // remove leading list characters
    equation.remove(QRegularExpression(QStringLiteral(R"(^\s*[\-*+] )")));

    // match all numbers and basic operations like +, -, * and /
    QRegularExpressionMatch match =
        QRegularExpression(QStringLiteral(R"(([\d\.,+\-*\/\(\)\s]+)\s*=)")).match(equation);

    if (!match.hasMatch()) {
        if (equation.trimmed().endsWith(QChar('='))) {
            MainWindow::instance()->showStatusBarMessage(
                tr("No equation was found in front of the cursor"), QStringLiteral("🧮"), 5000);
        }

        return false;
    }

    equation = match.captured(1);
    qDebug() << __func__ << " - 'equation': " << equation;

    QJSEngine engine;
    // evaluate our equation
    QJSValue result = engine.evaluate(equation);
    double resultValue = result.toNumber();
    qDebug() << __func__ << " - 'resultValue': " << resultValue;

    // compensate for subtraction errors with 0
    if ((resultValue < 0.0001) && (resultValue > 0)) {
        resultValue = 0;
    }

    MainWindow::instance()->showStatusBarMessage(
        tr("Result for equation: %1 = %2").arg(equation, QString::number(resultValue)),
        QStringLiteral("🧮"), 10000);

    // check if cursor is after the "="
    match = QRegularExpression(QStringLiteral("=\\s*$")).match(text);
    if (!match.hasMatch()) {
        return false;
    }

    returnValue = resultValue;
    return true;
}

void QOwnNotesMarkdownTextEdit::insertBlockQuote() {
    QTextCursor c = textCursor();
    QString selectedText = c.selectedText();

    if (selectedText.isEmpty()) {
        c.insertText(QStringLiteral("> "));
        setTextCursor(c);
    } else {
        // this only applies to the start of the selected block
        selectedText.replace(QRegularExpression(QStringLiteral("^")), QStringLiteral("> "));

        // transform Unicode line endings
        // this newline character seems to be used in multi-line selections
        const QString newLine = QString::fromUtf8(QByteArray::fromHex("e280a9"));
        selectedText.replace(newLine, QStringLiteral("\n> "));

        // remove the block quote if it was placed at the end of the text
        selectedText.remove(QRegularExpression(QStringLiteral("> $")));

        c.insertText(selectedText);
    }
}

QTextCursor QOwnNotesMarkdownTextEdit::fullLineSelectionCursor() const {
    QTextCursor cursor = textCursor();
    if (!cursor.hasSelection()) {
        cursor.movePosition(QTextCursor::StartOfBlock);
        cursor.movePosition(QTextCursor::EndOfBlock, QTextCursor::KeepAnchor);
        return cursor;
    }

    const int selectionStart = cursor.selectionStart();
    int selectionEnd = cursor.selectionEnd();

    QTextCursor lineCursor(document());
    lineCursor.setPosition(selectionStart);
    lineCursor.movePosition(QTextCursor::StartOfBlock);

    QTextCursor endCursor(document());
    endCursor.setPosition(selectionEnd);
    if (selectionEnd > selectionStart && endCursor.atBlockStart()) {
        endCursor.movePosition(QTextCursor::PreviousCharacter);
    }
    endCursor.movePosition(QTextCursor::EndOfBlock);

    lineCursor.setPosition(endCursor.position(), QTextCursor::KeepAnchor);
    return lineCursor;
}

bool QOwnNotesMarkdownTextEdit::replaceFullLineSelection(const QString &text) {
    QTextCursor cursor = fullLineSelectionCursor();
    if (!cursor.hasSelection()) {
        return false;
    }

    const int start = cursor.selectionStart();
    cursor.beginEditBlock();
    cursor.insertText(text);
    cursor.setPosition(start);
    cursor.setPosition(start + text.size(), QTextCursor::KeepAnchor);
    cursor.endEditBlock();
    setTextCursor(cursor);
    return true;
}

bool QOwnNotesMarkdownTextEdit::changeHeadingDepthOfSelection(const int levelDelta) {
    QTextCursor cursor = fullLineSelectionCursor();
    if (!cursor.hasSelection()) {
        return false;
    }

    return replaceFullLineSelection(changeHeadingDepth(cursor.selectedText(), levelDelta));
}

QMargins QOwnNotesMarkdownTextEdit::viewportMargins() {
    return QMarkdownTextEdit::viewportMargins();
}

void QOwnNotesMarkdownTextEdit::setText(const QString &text) {
    // Set a search delay of 300ms for text with more than 20k characters
    setSearchWidgetDebounceDelay(text.size() > 20000 ? 300 : 0);
    _foldingStateRestorePending = !_currentNoteReference.isEmpty();
    _foldingStateRestoreAttempts = 0;

    QMarkdownTextEdit::setText(text);
}

void QOwnNotesMarkdownTextEdit::setCurrentNoteReference(const QString &noteReference) {
    const QString effectiveNoteReference = _headingFoldingEnabled ? noteReference : QString();

    if (_currentNoteReference == effectiveNoteReference) {
        _foldingStateRestorePending = !effectiveNoteReference.isEmpty();
        _foldingStateRestoreAttempts = 0;
        return;
    }

    storeCurrentFoldedHeadingState();
    _currentNoteReference = effectiveNoteReference;
    _foldingStateRestorePending = !_currentNoteReference.isEmpty();
    _foldingStateRestoreAttempts = 0;
}

/**
 * Since spell checking can only be enabled and disabled globally this allows to
 * disable it in one QOwnNotesMarkdownTextEdit
 */
void QOwnNotesMarkdownTextEdit::disableSpellChecking() { _isSpellCheckingDisabled = true; }

void QOwnNotesMarkdownTextEdit::setSpellCheckingEnabled(bool enabled) {
    QOwnSpellChecker::instance()->setActive(enabled);
}

bool QOwnNotesMarkdownTextEdit::isSpellCheckingEnabled() {
    return QOwnSpellChecker::instance()->isActive();
}

void QOwnNotesMarkdownTextEdit::resizeEvent(QResizeEvent *event) {
    setPaperMargins();
    QMarkdownTextEdit::resizeEvent(event);
}

void QOwnNotesMarkdownTextEdit::paintEvent(QPaintEvent *event) {
    if (_showMarkdownImagePreviews) {
        QMarkdownTextEdit::paintEvent(event);
        paintMarkdownImagePreviews();
    } else {
        QMarkdownTextEdit::paintEvent(event);
    }
}

int QOwnNotesMarkdownTextEdit::sidebarAdditionalWidth() const {
    if (objectName() == QStringLiteral("logTextEdit") || !_headingFoldingEnabled ||
        !_hasFoldableHeadings) {
        return 0;
    }

    return fontMetrics().height() + (kFoldIndicatorPadding * 2);
}

bool QOwnNotesMarkdownTextEdit::isHeadingBlock(const QTextBlock &block, int *level) {
    if (!block.isValid()) {
        return false;
    }

    const int state = block.userState();
    if (state < MarkdownHighlighter::H1 || state > MarkdownHighlighter::H6) {
        return false;
    }

    if (level != nullptr) {
        *level = state - MarkdownHighlighter::H1 + 1;
    }

    return true;
}

bool QOwnNotesMarkdownTextEdit::foldRegionForHeaderBlock(const QTextBlock &headerBlock,
                                                         FoldRegion &region) const {
    int level = 0;
    if (!isHeadingBlock(headerBlock, &level)) {
        return false;
    }

    QTextBlock block = headerBlock.next();
    QTextBlock lastContentBlock;

    while (block.isValid()) {
        int nextLevel = 0;
        if (isHeadingBlock(block, &nextLevel) && nextLevel <= level) {
            break;
        }

        lastContentBlock = block;
        block = block.next();
    }

    if (!lastContentBlock.isValid()) {
        return false;
    }

    region.headerBlock = headerBlock;
    region.firstContentBlock = headerBlock.next();
    region.lastContentBlock = lastContentBlock;
    return true;
}

bool QOwnNotesMarkdownTextEdit::isHeadingFolded(const QTextBlock &headerBlock) const {
    FoldRegion region;
    return foldRegionForHeaderBlock(headerBlock, region) && !region.firstContentBlock.isVisible();
}

QString QOwnNotesMarkdownTextEdit::headingStateKey(const QTextBlock &headerBlock,
                                                   QHash<QString, int> &headingOccurrences) {
    int level = 0;
    if (!isHeadingBlock(headerBlock, &level)) {
        return QString();
    }

    const QString headingText = headerBlock.text();
    const int occurrence = ++headingOccurrences[headingText];
    return QStringLiteral("%1:%2:%3").arg(level).arg(occurrence).arg(headingText);
}

bool QOwnNotesMarkdownTextEdit::setFoldRegionFolded(const FoldRegion &region, bool folded) {
    if (!region.firstContentBlock.isValid() || !region.lastContentBlock.isValid()) {
        return false;
    }

    const bool currentlyFolded = !region.firstContentBlock.isVisible();
    if (currentlyFolded == folded) {
        return false;
    }

    QTextCursor cursor = textCursor();
    const int firstPosition = region.firstContentBlock.position();
    const int lastPosition = region.lastContentBlock.position() + region.lastContentBlock.length();

    if (folded && cursor.position() >= firstPosition && cursor.position() < lastPosition) {
        cursor.setPosition(region.headerBlock.position());
        setTextCursor(cursor);
    }

    QTextBlock block = region.firstContentBlock;
    while (block.isValid()) {
        block.setVisible(!folded);
        if (folded) {
            block.setLineCount(0);
        } else {
            QTextLayout *layout = block.layout();
            block.setLineCount(layout ? qMax(1, layout->lineCount()) : 1);
        }

        if (block == region.lastContentBlock) {
            break;
        }

        block = block.next();
    }

    document()->markContentsDirty(firstPosition, lastPosition - firstPosition);
    viewport()->update();
    lineNumberArea()->update();
    ensureCursorVisible();
    storeCurrentFoldedHeadingState();
    return true;
}

bool QOwnNotesMarkdownTextEdit::setHeadingFolded(const QTextBlock &headerBlock, bool folded) {
    FoldRegion region;
    return foldRegionForHeaderBlock(headerBlock, region) && setFoldRegionFolded(region, folded);
}

bool QOwnNotesMarkdownTextEdit::hasFoldableHeadings() const {
    QTextBlock block = document()->firstBlock();
    while (block.isValid()) {
        FoldRegion region;
        if (foldRegionForHeaderBlock(block, region)) {
            return true;
        }

        block = block.next();
    }

    return false;
}

void QOwnNotesMarkdownTextEdit::refreshFoldingSidebar() {
    if (!_headingFoldingEnabled) {
        _hasFoldableHeadings = false;
        updateLineNumberAreaWidth(0);
        lineNumberArea()->update();
        return;
    }

    const bool hadFoldableHeadings = _hasFoldableHeadings;
    _hasFoldableHeadings = hasFoldableHeadings();

    if (hadFoldableHeadings != _hasFoldableHeadings) {
        updateLineNumberAreaWidth(0);
    }

    lineNumberArea()->update();
}

void QOwnNotesMarkdownTextEdit::paintSidebar(QPainter *painter, const QRect &eventRect) {
    const int additionalWidth = sidebarAdditionalWidth();
    if (additionalWidth <= 0) {
        return;
    }

    QTextBlock block = firstVisibleBlock();
    int top = qRound(blockBoundingGeometry(block).translated(contentOffset()).top());
    top += viewportMargins().top();
    int bottom = top;

    const QColor iconColor = palette().color(QPalette::Active, QPalette::Text);
    const QColor borderColor = iconColor.lighter(130);

    painter->save();
    painter->setRenderHint(QPainter::Antialiasing, true);

    while (block.isValid() && top <= eventRect.bottom()) {
        top = bottom;
        bottom = top + qRound(blockBoundingRect(block).height());

        if (block.isVisible() && bottom >= eventRect.top()) {
            FoldRegion region;
            if (foldRegionForHeaderBlock(block, region)) {
                const int indicatorSize = qMax(7, fontMetrics().height() - 8);
                const int x = kFoldIndicatorPadding;
                const int y =
                    top + qMax(0, (qRound(blockBoundingRect(block).height()) - indicatorSize) / 2);
                const QRect indicatorRect(x, y, indicatorSize, indicatorSize);
                const bool folded = isHeadingFolded(block);

                painter->setPen(borderColor);
                painter->setBrush(Qt::NoBrush);
                painter->drawRect(indicatorRect.adjusted(0, 0, -1, -1));

                painter->setPen(QPen(iconColor, 1.5));
                painter->drawLine(indicatorRect.left() + 2, indicatorRect.center().y(),
                                  indicatorRect.right() - 2, indicatorRect.center().y());

                if (folded) {
                    painter->drawLine(indicatorRect.center().x(), indicatorRect.top() + 2,
                                      indicatorRect.center().x(), indicatorRect.bottom() - 2);
                }
            }
        }

        block = block.next();
    }

    painter->restore();
}

bool QOwnNotesMarkdownTextEdit::headerBlockAtSidebarPosition(const QPoint &pos,
                                                             QTextBlock &headerBlock) const {
    if (pos.x() > sidebarAdditionalWidth()) {
        return false;
    }

    QTextBlock block = firstVisibleBlock();
    int top = qRound(blockBoundingGeometry(block).translated(contentOffset()).top());
    top += QPlainTextEdit::viewportMargins().top();
    int bottom = top;

    while (block.isValid()) {
        top = bottom;
        bottom = top + qRound(blockBoundingRect(block).height());

        if (block.isVisible() && pos.y() >= top && pos.y() <= bottom) {
            FoldRegion region;
            if (foldRegionForHeaderBlock(block, region)) {
                headerBlock = block;
                return true;
            }

            return false;
        }

        if (top > pos.y()) {
            return false;
        }

        block = block.next();
    }

    return false;
}

void QOwnNotesMarkdownTextEdit::storeCurrentFoldedHeadingState() {
    if (!_headingFoldingEnabled || _currentNoteReference.isEmpty() ||
        _isApplyingStoredFoldingState) {
        return;
    }

    QSet<QString> foldedHeadingState;
    QHash<QString, int> headingOccurrences;
    QTextBlock block = document()->firstBlock();

    while (block.isValid()) {
        if (isHeadingFolded(block)) {
            const QString headingKey = headingStateKey(block, headingOccurrences);
            if (!headingKey.isEmpty()) {
                foldedHeadingState.insert(headingKey);
            }
        } else if (isHeadingBlock(block)) {
            headingStateKey(block, headingOccurrences);
        }

        block = block.next();
    }

    if (foldedHeadingState.isEmpty()) {
        s_foldedHeadingStateByNoteReference.remove(_currentNoteReference);
    } else {
        s_foldedHeadingStateByNoteReference[_currentNoteReference] = foldedHeadingState;
    }
}

void QOwnNotesMarkdownTextEdit::scheduleRestoreCurrentFoldedHeadingState() {
    if (!_foldingStateRestorePending || _currentNoteReference.isEmpty()) {
        return;
    }

    QTimer::singleShot(0, this, &QOwnNotesMarkdownTextEdit::restoreCurrentFoldedHeadingState);
}

void QOwnNotesMarkdownTextEdit::restoreCurrentFoldedHeadingState() {
    if (!_foldingStateRestorePending || _currentNoteReference.isEmpty() ||
        !_headingFoldingEnabled) {
        return;
    }

    const QSet<QString> foldedHeadingState =
        s_foldedHeadingStateByNoteReference.value(_currentNoteReference);
    _foldingStateRestorePending = false;

    if (foldedHeadingState.isEmpty()) {
        return;
    }

    if (!_hasFoldableHeadings && _foldingStateRestoreAttempts < 5) {
        ++_foldingStateRestoreAttempts;
        QTimer::singleShot(0, this, &QOwnNotesMarkdownTextEdit::restoreCurrentFoldedHeadingState);
        return;
    }

    _isApplyingStoredFoldingState = true;
    _foldingStateRestoreAttempts = 0;
    QHash<QString, int> headingOccurrences;
    QTextBlock block = document()->firstBlock();

    while (block.isValid()) {
        const QString headingKey = headingStateKey(block, headingOccurrences);
        if (!headingKey.isEmpty() && foldedHeadingState.contains(headingKey)) {
            setHeadingFolded(block, true);
        }

        block = block.next();
    }

    _isApplyingStoredFoldingState = false;
    refreshFoldingSidebar();
}

bool QOwnNotesMarkdownTextEdit::sidebarMousePressEvent(QMouseEvent *event) {
    if (event == nullptr || event->button() != Qt::LeftButton || sidebarAdditionalWidth() <= 0) {
        return false;
    }

    QTextBlock headerBlock;
    if (!headerBlockAtSidebarPosition(event->pos(), headerBlock)) {
        return false;
    }

    const bool handled = setHeadingFolded(headerBlock, !isHeadingFolded(headerBlock));
    if (handled) {
        event->accept();
    }

    return handled;
}

void QOwnNotesMarkdownTextEdit::foldAllHeadings() {
    if (!_headingFoldingEnabled) {
        return;
    }

    QTextBlock block = document()->firstBlock();
    while (block.isValid()) {
        setHeadingFolded(block, true);
        block = block.next();
    }

    refreshFoldingSidebar();
}

void QOwnNotesMarkdownTextEdit::unfoldAllHeadings() {
    QTextBlock block = document()->firstBlock();
    while (block.isValid()) {
        setHeadingFolded(block, false);
        block = block.next();
    }

    refreshFoldingSidebar();
}

/**
 * Resolves a raw Markdown image source string to a canonical URL string
 * suitable for loading (file://, data:image/, http(s)://, or Nextcloud
 * /core/preview path).
 */
static QString resolveMarkdownImageSource(const QString &rawSource,
                                          const QString &noteDirectoryPath) {
    QString source = rawSource.trimmed();
    if (source.isEmpty()) {
        return QString();
    }

    // Handle angle-bracket wrapped paths: <path with spaces.png>
    if (source.startsWith(QLatin1Char('<')) && source.endsWith(QLatin1Char('>')) &&
        source.size() > 2) {
        source = source.mid(1, source.size() - 2).trimmed();
    }

    // Data URIs are returned as-is
    if (source.startsWith(QLatin1String("data:image/"), Qt::CaseInsensitive)) {
        return source;
    }

    // Strip optional Markdown image title: path "title" or path 'title'
    if (!source.startsWith(QLatin1Char('<'))) {
        int splitPos = -1;
        bool inSingleQuote = false;
        bool inDoubleQuote = false;
        for (int i = 0; i < source.size(); ++i) {
            const QChar ch = source.at(i);
            if (ch == QLatin1Char('"') && !inSingleQuote) {
                inDoubleQuote = !inDoubleQuote;
                continue;
            }
            if (ch == QLatin1Char('\'') && !inDoubleQuote) {
                inSingleQuote = !inSingleQuote;
                continue;
            }
            if (!inSingleQuote && !inDoubleQuote && ch.isSpace()) {
                splitPos = i;
                break;
            }
        }
        if (splitPos > 0) {
            source = source.left(splitPos).trimmed();
        }
    }

    if (source.isEmpty()) {
        return QString();
    }

    const QUrl sourceUrl(source);
    if (sourceUrl.isLocalFile()) {
        return QLatin1String("file://") + sourceUrl.toLocalFile();
    }

    if (!sourceUrl.scheme().isEmpty()) {
        return source;
    }

    // Nextcloud preview paths are resolved later with authentication
    if (source.startsWith(QLatin1String("/core/preview"))) {
        return source;
    }

    if (QFileInfo(source).isAbsolute()) {
        return QLatin1String("file://") + source;
    }

    return QLatin1String("file://") + QDir(noteDirectoryPath).absoluteFilePath(source);
}

/**
 * Loads a pixmap from a resolved image source string and caches the result.
 * Failed loads are also cached to avoid repeated expensive lookups.
 */
static QPixmap loadMarkdownImagePixmap(const QString &resolvedSource) {
    static QHash<QString, QPixmap> imageCache;
    static QSet<QString> failedImageCache;
    static const QRegularExpression srcRegex(QStringLiteral("src=\"([^\"]+)\""));

    if (resolvedSource.isEmpty()) {
        return QPixmap();
    }

    if (imageCache.contains(resolvedSource)) {
        return imageCache.value(resolvedSource);
    }

    if (failedImageCache.contains(resolvedSource)) {
        return QPixmap();
    }

    QImage image;

    if (resolvedSource.startsWith(QLatin1String("file://"))) {
        const QString localPath = resolvedSource.mid(7);
        QFileInfo info(localPath);
        if (!info.exists() || !info.isFile()) {
            failedImageCache.insert(resolvedSource);
            return QPixmap();
        }

        QImageReader reader(localPath);
        reader.setAutoTransform(true);
        image = reader.read();
    } else if (resolvedSource.startsWith(QLatin1String("data:image/"), Qt::CaseInsensitive)) {
        const int markerPos = resolvedSource.indexOf(QStringLiteral(";base64,"));
        if (markerPos < 0) {
            failedImageCache.insert(resolvedSource);
            return QPixmap();
        }

        const QString base64 = resolvedSource.mid(markerPos + 8);
        image = QImage::fromData(QByteArray::fromBase64(base64.toLatin1()));
    } else if (resolvedSource.startsWith(QLatin1String("/core/preview"))) {
        if (!CloudService::isCloudSupportEnabled()) {
            failedImageCache.insert(resolvedSource);
            return QPixmap();
        }

        int width = 0;
        const QString imgTag =
            QStringLiteral("<img src=\"") + resolvedSource + QStringLiteral("\" alt=\"img\"/>");
        const QString inlineTag =
            CloudService::instance()->nextcloudPreviewImageTagToInlineImageTag(imgTag, width);
        const QRegularExpressionMatch srcMatch = srcRegex.match(inlineTag);
        if (!srcMatch.hasMatch()) {
            failedImageCache.insert(resolvedSource);
            return QPixmap();
        }

        const QString dataUrl = srcMatch.captured(1);
        const int markerPos = dataUrl.indexOf(QStringLiteral(";base64,"));
        if (markerPos < 0) {
            failedImageCache.insert(resolvedSource);
            return QPixmap();
        }

        const QString base64 = dataUrl.mid(markerPos + 8);
        image = QImage::fromData(QByteArray::fromBase64(base64.toLatin1()));
    } else {
        // Remote URL: do not block the paint event with a synchronous download.
        // Start an async fetch; the result will be stored in the cache and the
        // viewport will be refreshed once the download completes.
        static QSet<QString> pendingDownloadCache;
        if (!pendingDownloadCache.contains(resolvedSource)) {
            pendingDownloadCache.insert(resolvedSource);

            // Use a shared network manager to avoid spawning one per request.
            static QNetworkAccessManager *netManager = nullptr;
            if (!netManager) {
                netManager = new QNetworkAccessManager();
            }

            QNetworkRequest request((QUrl(resolvedSource)));
#if QT_VERSION >= QT_VERSION_CHECK(5, 9, 0)
            // RedirectPolicyAttribute was introduced in Qt 5.9
            request.setAttribute(QNetworkRequest::RedirectPolicyAttribute,
                                 QNetworkRequest::NoLessSafeRedirectPolicy);
#endif
            request.setHeader(QNetworkRequest::UserAgentHeader,
                              Utils::Misc::friendlyUserAgentString());

            QNetworkReply *reply = netManager->get(request);
            QObject::connect(reply, &QNetworkReply::finished, reply, [reply, resolvedSource]() {
                reply->deleteLater();
                pendingDownloadCache.remove(resolvedSource);

                if (reply->error() != QNetworkReply::NoError) {
                    failedImageCache.insert(resolvedSource);
                    return;
                }

                const QByteArray data = reply->readAll();
                const QImage img = QImage::fromData(data);
                if (img.isNull()) {
                    failedImageCache.insert(resolvedSource);
                    return;
                }

                const QPixmap pix = QPixmap::fromImage(img);
                if (pix.isNull()) {
                    failedImageCache.insert(resolvedSource);
                    return;
                }

                imageCache.insert(resolvedSource, pix);

                // Trigger a repaint on all markdown text edit widgets so the
                // newly downloaded image is shown without the user having to
                // interact with the editor.
                const auto widgets = QApplication::allWidgets();
                for (QWidget *w : widgets) {
                    if (auto *edit = qobject_cast<QOwnNotesMarkdownTextEdit *>(w)) {
                        if (edit->viewport()) {
                            edit->viewport()->update();
                        }
                    }
                }
            });
        }
        // Return null for now; the next paint will pick up the cached pixmap.
        return QPixmap();
    }

    if (image.isNull()) {
        failedImageCache.insert(resolvedSource);
        return QPixmap();
    }

    const QPixmap pixmap = QPixmap::fromImage(image);
    if (pixmap.isNull()) {
        failedImageCache.insert(resolvedSource);
        return QPixmap();
    }

    imageCache.insert(resolvedSource, pixmap);
    return pixmap;
}

void QOwnNotesMarkdownTextEdit::paintMarkdownImagePreviews() {
    if (objectName() == QStringLiteral("logTextEdit")) {
        return;
    }

    auto *mainWindow = MainWindow::instance();
    QString noteDirectoryPath = NoteFolder::currentLocalPath();
    if (mainWindow) {
        const QString currentNotePath = mainWindow->getCurrentNote().fullNoteFilePath();
        if (!currentNotePath.isEmpty()) {
            noteDirectoryPath = QFileInfo(currentNotePath).absolutePath();
        }
    }

    static const QRegularExpression imageRegex(
        QStringLiteral(R"(!\[[^\]]*\]\(([^\n\)]*)\)(?:\s*\{[^}]*\})?)"));

    struct PreviewDrawItem {
        QRect targetRect;
        QPixmap pixmap;
    };
    QVector<PreviewDrawItem> drawItems;

    QTextBlock block = firstVisibleBlock();
    const qreal contentX = contentOffset().x();
    const QRect viewportRect = viewport()->rect();

    // Thumbnail size: as tall as one text line, square
    const int thumbSize = fontMetrics().height();
    const int thumbSpacing = 4;

    int top = qRound(blockBoundingGeometry(block).translated(contentOffset()).top());
    int bottom = top + qRound(blockBoundingRect(block).height());

    while (block.isValid()) {
        if (!block.isVisible()) {
            block = block.next();
            continue;
        }

        if (bottom < 0) {
            block = block.next();
            top = bottom;
            bottom = top + qRound(blockBoundingRect(block).height());
            continue;
        }

        if (top > viewportRect.bottom()) {
            break;
        }

        const QString text = block.text();
        QTextLayout *layout = block.layout();
        if (!layout) {
            block = block.next();
            top = bottom;
            bottom = top + qRound(blockBoundingRect(block).height());
            continue;
        }

        QRegularExpressionMatchIterator iterator = imageRegex.globalMatch(text);

        while (iterator.hasNext()) {
            const QRegularExpressionMatch match = iterator.next();
            const int tagEnd = match.capturedEnd(0);
            const QTextLine line = layout->lineForTextPosition(tagEnd - 1);
            if (!line.isValid()) {
                continue;
            }

            // Only show inline image previews when no visible text follows the
            // image tag on the same layout line to avoid cursor/caret mismatch.
            const int lineTextEnd = line.textStart() + line.textLength();
            if (tagEnd < lineTextEnd) {
                const QString trailingText = text.mid(tagEnd, lineTextEnd - tagEnd).trimmed();
                if (!trailingText.isEmpty()) {
                    continue;
                }
            }

            const QString resolvedSource =
                resolveMarkdownImageSource(match.captured(1), noteDirectoryPath);
            const QPixmap pixmap = loadMarkdownImagePixmap(resolvedSource);
            if (pixmap.isNull()) {
                continue;
            }

            // Find the line that contains the end of the image tag and compute
            // the pixel X position right after the closing parenthesis
            // cursorToX gives us the X coordinate of the character at tagEnd
            // within the line's local coordinate system
            const qreal cursorX = line.cursorToX(tagEnd);
            QRectF lineRect = line.rect();
            lineRect.translate(contentX, top);

            const int x = qRound(lineRect.left() + cursorX) + thumbSpacing;
            const int y = qRound(lineRect.top() + (lineRect.height() - thumbSize) / 2.0);
            const QRect targetRect(x, y, thumbSize, thumbSize);

            if (!viewportRect.intersects(targetRect)) {
                continue;
            }

            drawItems.append({targetRect, pixmap});
        }

        block = block.next();
        top = bottom;
        bottom = top + qRound(blockBoundingRect(block).height());
    }

    QPainter painter(viewport());
    painter.setRenderHint(QPainter::Antialiasing, true);
    painter.setRenderHint(QPainter::SmoothPixmapTransform, true);
    for (const auto &item : drawItems) {
        // Draw a subtle rounded border around the thumbnail
        painter.setPen(QPen(QColor(120, 120, 120, 160)));
        painter.setBrush(QColor(255, 255, 255, 140));
        painter.drawRoundedRect(item.targetRect.adjusted(-1, -1, 1, 1), 2, 2);

        // Scale pixmap to fit the square thumbnail, keeping aspect ratio
        const QPixmap scaledPixmap =
            item.pixmap.scaled(thumbSize, thumbSize, Qt::KeepAspectRatio, Qt::SmoothTransformation);
        QRect pixmapRect(QPoint(0, 0), scaledPixmap.size());
        pixmapRect.moveCenter(item.targetRect.center());
        painter.drawPixmap(pixmapRect.topLeft(), scaledPixmap);
    }
}

bool QOwnNotesMarkdownTextEdit::canInsertFromMimeData(const QMimeData *source) const {
    return (!source->hasUrls());
}

void QOwnNotesMarkdownTextEdit::dragEnterEvent(QDragEnterEvent *event) {
    if (event->mimeData()->hasUrls()) {
        event->acceptProposedAction();
        return;
    }

    QMarkdownTextEdit::dragEnterEvent(event);
}

void QOwnNotesMarkdownTextEdit::dragMoveEvent(QDragMoveEvent *event) {
    if (event->mimeData()->hasUrls()) {
        event->acceptProposedAction();
        return;
    }

    QMarkdownTextEdit::dragMoveEvent(event);
}

void QOwnNotesMarkdownTextEdit::dropEvent(QDropEvent *event) {
    if (event->mimeData()->hasUrls()) {
        if (auto mainWindow = MainWindow::instance()) {
            event->acceptProposedAction();
            mainWindow->handleInsertingFromMimeData(event->mimeData());
            return;
        }
    }

    QMarkdownTextEdit::dropEvent(event);
}

/**
 * Handles pasting from clipboard
 */
void QOwnNotesMarkdownTextEdit::insertFromMimeData(const QMimeData *source) {
    // if there is text in the clipboard do the normal pasting process
    if (source->hasText()) {
        QMarkdownTextEdit::insertFromMimeData(source);
    } else if (auto mainWindow = MainWindow::instance()) {
        // to more complex pasting if there was no text (and a main window
        // was set)
        mainWindow->handleInsertingFromMimeData(source);
    }
}

/**
 * Handles the settings of the Markdown textedit
 */
void QOwnNotesMarkdownTextEdit::updateSettings() {
    // we need a blocker, otherwise the "change" events will fire
    const QSignalBlocker blocker(this);
    Q_UNUSED(blocker)

    SettingsService settings;
    QMarkdownTextEdit::AutoTextOptions options;

    if (settings.value(QStringLiteral("Editor/autoBracketClosing"), true).toBool()) {
        options |= QMarkdownTextEdit::AutoTextOption::BracketClosing;
    }

    if (settings.value(QStringLiteral("Editor/autoBracketRemoval"), true).toBool()) {
        options |= QMarkdownTextEdit::AutoTextOption::BracketRemoval;
    }

    setAutoTextOptions(options);

    auto spellchecker = QOwnSpellChecker::instance();
    if (spellchecker) {
        // spell check active/inactive
        bool spellcheckerActive = settings.value(QStringLiteral("checkSpelling"), true).toBool();
        spellchecker->setActive(spellcheckerActive);

        QString lang =
            settings.value(QStringLiteral("spellCheckLanguage"), QStringLiteral("auto")).toString();
        if (lang == QStringLiteral("auto")) {
            spellchecker->setAutoDetect(true);
        } else {
            spellchecker->setAutoDetect(false);
            spellchecker->setCurrentLanguage(lang);
        }
    }

#ifdef LANGUAGETOOL_ENABLED
    auto *languageToolChecker = LanguageToolChecker::instance();
    if (languageToolChecker) {
        if ((objectName() == QStringLiteral("noteTextEdit")) ||
            (objectName() == QStringLiteral("encryptedNoteTextEdit")) || objectName().isEmpty()) {
            languageToolChecker->setTextEdit(this);
            languageToolChecker->scheduleCheck(true);
        } else {
            languageToolChecker->clearForTextEdit(this);
        }
    }
#endif

#ifdef HARPER_ENABLED
    auto *harperChecker = HarperChecker::instance();
    if (harperChecker) {
        if ((objectName() == QStringLiteral("noteTextEdit")) ||
            (objectName() == QStringLiteral("encryptedNoteTextEdit")) || objectName().isEmpty()) {
            harperChecker->setTextEdit(this);
            harperChecker->scheduleCheck(true);
        } else {
            harperChecker->clearForTextEdit(this);
        }
    }
#endif

    // highlighting is always disabled for logTextEdit
    if (objectName() != QStringLiteral("logTextEdit")) {
        // enable or disable Markdown highlighting
        bool highlightingEnabled =
            settings.value(QStringLiteral("markdownHighlightingEnabled"), true).toBool();

        setHighlightingEnabled(highlightingEnabled);

        if (highlightingEnabled) {
            // set the new highlighting styles
            setStyles();
            if (_highlighter) {
                _highlighter->rehighlight();
            }
        }
    }

    const bool hlCurrLine =
        settings.value(QStringLiteral("Editor/highlightCurrentLine"), true).toBool();
    setHighlightCurrentLine(hlCurrLine);

    // Hide formatting syntax on non-cursor blocks (Typora-like)
    const bool hideFormattingSyntax =
        settings.value(QStringLiteral("Editor/hideFormattingSyntax"), false).toBool();
    if (_highlighter) {
        _highlighter->setHideFormattingSyntax(hideFormattingSyntax);
        _highlighter->setCurrentCursorBlockNumber(textCursor().blockNumber());
    }

    const bool hangingIndentEnabled =
        settings.value(QStringLiteral("Editor/hangingIndent"), false).toBool();
    setHangingIndentEnabled(hangingIndentEnabled);
    _headingFoldingEnabled =
        settings.value(QStringLiteral("Editor/headingFolding"), false).toBool();
    if (!_headingFoldingEnabled) {
        setCurrentNoteReference(QString());
        unfoldAllHeadings();
    }
    _showMarkdownImagePreviews =
        settings.value(QStringLiteral("Editor/showMarkdownImagePreviews"), true).toBool();
    viewport()->update();
    refreshFoldingSidebar();
    const auto color = Utils::Schema::schemaSettings->getBackgroundColor(
        MarkdownHighlighter::HighlighterState::CurrentLineBackgroundColor);
    setCurrentLineHighlightColor(color);

    _centerCursor = settings.value(QStringLiteral("Editor/centerCursor")).toBool();
    QMarkdownTextEdit::updateSettings();

    if (_markdownLspInitialized) {
        applyMarkdownLspSettings();
    }
}

void QOwnNotesMarkdownTextEdit::onContextMenu(QPoint pos) {
    auto *spellCheckMenu = spellCheckContextMenu(pos);
    const QTextCursor cursorAtMouse = cursorForPosition(pos);
    auto *lspMenu = markdownLspContextMenu(cursorAtMouse);

    const QPoint globalPos = this->viewport()->mapToGlobal(pos);
    QMenu *menu = this->createStandardContextMenu();
    if (lspMenu) {
        menu->insertMenu(menu->actions().constFirst(), lspMenu);
    }
    if (spellCheckMenu) {
        // insert spell check at the top if available
        menu->insertMenu(menu->actions().constFirst(), spellCheckMenu);
    }

    const bool isAllowNoteEditing = Utils::Misc::isNoteEditingAllowed();
    const bool isTextSelected = textCursor().hasSelection();

    const QString linkTextActionName =
        isTextSelected ? tr("&Link selected text") : tr("Insert &link");
    QAction *linkTextAction = menu->addAction(linkTextActionName, this, []() {
        MainWindow::instance()->insertTextLinkAction()->trigger();
    });
    linkTextAction->setEnabled(isAllowNoteEditing);

    QString blockQuoteTextActionName = isTextSelected
                                           ? tr("Block &quote selected text",
                                                "Action to apply a block quote formatting to the "
                                                "selected text")
                                           : tr("Insert block &quote");
    QAction *blockQuoteTextAction = menu->addAction(blockQuoteTextActionName);
    connect(blockQuoteTextAction, &QAction::triggered, this,
            &QOwnNotesMarkdownTextEdit::insertBlockQuote);
    blockQuoteTextAction->setEnabled(isAllowNoteEditing);

    if (isTextSelected) {
        QMenu *listOperationsMenu = menu->addMenu(tr("List operations"));
        listOperationsMenu->setEnabled(isAllowNoteEditing);

        listOperationsMenu->addAction(MainWindow::instance()->toggleCheckboxesAction());
        listOperationsMenu->addAction(MainWindow::instance()->createOrderedListAction());
        listOperationsMenu->addAction(MainWindow::instance()->createAlphabeticalListAction());
        listOperationsMenu->addAction(MainWindow::instance()->createUnorderedListAction());
        listOperationsMenu->addAction(MainWindow::instance()->createCheckboxListAction());
        listOperationsMenu->addAction(MainWindow::instance()->clearListFormattingAction());
        listOperationsMenu->addAction(MainWindow::instance()->orderCheckboxesAction());

        QMenu *markdownOperationsMenu = menu->addMenu(tr("Markdown operations"));
        markdownOperationsMenu->setEnabled(isAllowNoteEditing);

        markdownOperationsMenu->addAction(MainWindow::instance()->increaseHeadingDepthAction());
        markdownOperationsMenu->addAction(MainWindow::instance()->decreaseHeadingDepthAction());

        menu->addAction(MainWindow::instance()->searchTextOnWebAction());
        menu->addAction(MainWindow::instance()->findNoteAction());
    }

    //     searchAction->setEnabled(isTextSelected);
    //     QAction *searchAction =
    //         menu->addAction(ui->actionSearch_text_on_the_web->text());
    //     searchAction->setShortcut(ui->actionSearch_text_on_the_web->shortcut());

    QAction *copyCodeBlockAction = menu->addAction(tr("Copy code block"));
    copyCodeBlockAction->setIcon(
        QIcon::fromTheme(QStringLiteral("edit-copy"),
                         QIcon(QStringLiteral(":icons/breeze-qownnotes/16x16/edit-copy.svg"))));
    const QTextBlock currentTextBlock = cursorForPosition(pos).block();
    const int userState = currentTextBlock.userState();
    const bool isCodeSpan = highlighter()->isPosInACodeSpan(
        currentTextBlock.blockNumber(), cursorForPosition(pos).positionInBlock());
    copyCodeBlockAction->setEnabled(MarkdownHighlighter::isCodeBlock(userState) || isCodeSpan);
    connect(copyCodeBlockAction, &QAction::triggered, this,
            [this, isCodeSpan, currentTextBlock, pos]() {
                // copy the text from a copy block around currentTextBlock to the
                // clipboard
                if (isCodeSpan) {
                    const auto codeSpanRange = highlighter()->getSpanRange(
                        MarkdownHighlighter::RangeType::CodeSpan, currentTextBlock.blockNumber(),
                        cursorForPosition(pos).positionInBlock());
                    QApplication::clipboard()->setText(currentTextBlock.text().mid(
                        codeSpanRange.first + 1, codeSpanRange.second - codeSpanRange.first - 1));
                } else {
                    Utils::Gui::copyCodeBlockText(currentTextBlock);
                }
            });

    menu->addSeparator();

    if (_markdownLspEnabled && _markdownLspClient && !_markdownLspUri.isEmpty()) {
        QMenu *lspMenu = menu->addMenu(tr("Markdown LSP"));
        QAction *formatDocumentAction = lspMenu->addAction(tr("Format document"));
        formatDocumentAction->setEnabled(isAllowNoteEditing);
        connect(formatDocumentAction, &QAction::triggered, this,
                [this]() { requestMarkdownLspFormatting(false); });

        QAction *formatSelectionAction = lspMenu->addAction(tr("Format selection"));
        formatSelectionAction->setEnabled(isAllowNoteEditing && isTextSelected);
        connect(formatSelectionAction, &QAction::triggered, this,
                [this]() { requestMarkdownLspFormatting(true); });

        menu->addSeparator();
    }

    // add table column insertion actions if cursor is in a table
    if (Utils::Gui::isTableAtCursor(this)) {
        QAction *addColumnLeftAction = menu->addAction(tr("Add table column left"));
        addColumnLeftAction->setEnabled(isAllowNoteEditing);
        connect(addColumnLeftAction, &QAction::triggered, this,
                [this]() { Utils::Gui::insertTableColumnLeft(this); });

        QAction *addColumnRightAction = menu->addAction(tr("Add table column right"));
        addColumnRightAction->setEnabled(isAllowNoteEditing);
        connect(addColumnRightAction, &QAction::triggered, this,
                [this]() { Utils::Gui::insertTableColumnRight(this); });

        QAction *addRowAboveAction = menu->addAction(tr("Add table row above"));
        addRowAboveAction->setEnabled(isAllowNoteEditing);
        connect(addRowAboveAction, &QAction::triggered, this,
                [this]() { Utils::Gui::insertTableRowAbove(this); });

        QAction *addRowBelowAction = menu->addAction(tr("Add table row below"));
        addRowBelowAction->setEnabled(isAllowNoteEditing);
        connect(addRowBelowAction, &QAction::triggered, this,
                [this]() { Utils::Gui::insertTableRowBelow(this); });

        QAction *editTableAction = menu->addAction(tr("Edit table"));
        editTableAction->setEnabled(isAllowNoteEditing);
        connect(editTableAction, &QAction::triggered, this, [this]() {
            auto *dialog = new MarkdownTableDialog(this, this);
            dialog->exec();
            delete dialog;
        });

        menu->addSeparator();
    }

    // add the print menu
    QMenu *printMenu = menu->addMenu(tr("Print"));
    QIcon printIcon =
        QIcon::fromTheme(QStringLiteral("document-print"),
                         QIcon(QStringLiteral(":icons/breeze-qownnotes/16x16/document-print.svg")));
    printMenu->setIcon(printIcon);

    // add the print selected text action
    QAction *printTextAction = printMenu->addAction(tr("Print selected text"));
    printTextAction->setEnabled(isTextSelected);
    printTextAction->setIcon(printIcon);
    connect(printTextAction, &QAction::triggered, this, [this]() {
        // print the selected text
        auto mainWindow = MainWindow::instance();
        auto *textEdit = new QOwnNotesMarkdownTextEdit(this);
        textEdit->setPlainText(mainWindow->selectedNoteTextEditText());
        mainWindow->printTextDocument(textEdit->document(), true);
    });

    // add the print selected text (preview) action
    QAction *printHTMLAction = printMenu->addAction(tr("Print selected text (preview)"));
    printHTMLAction->setEnabled(isTextSelected);
    printHTMLAction->setIcon(printIcon);
    connect(printHTMLAction, &QAction::triggered, this, [this]() {
        // print the selected text (preview)
        auto mainWindow = MainWindow::instance();
        auto note = mainWindow->getCurrentNote();
        QString html = note.textToMarkdownHtml(
            mainWindow->selectedNoteTextEditText(), NoteFolder::currentLocalPath(),
            mainWindow->getMaxImageWidth(), Utils::Misc::useInternalExportStylingForPreview());
        auto *textEdit = new QTextEdit(this);
        textEdit->setHtml(html);
        mainWindow->printTextDocument(textEdit->document());
    });

    // add the export menu
    QMenu *exportMenu = menu->addMenu(tr("Export"));
    exportMenu->setIcon(QIcon::fromTheme(
        QStringLiteral("document-export"),
        QIcon(QStringLiteral(":icons/breeze-qownnotes/16x16/document-export.svg"))));

    QIcon pdfIcon = QIcon::fromTheme(
        QStringLiteral("application-pdf"),
        QIcon(QStringLiteral(":icons/breeze-qownnotes/16x16/application-pdf.svg")));

    // add the export selected text action
    QAction *exportTextAction = exportMenu->addAction(tr("Export selected text as PDF"));
    exportTextAction->setEnabled(isTextSelected);
    exportTextAction->setIcon(pdfIcon);
    connect(exportTextAction, &QAction::triggered, this, [this]() {
        // export the selected text as PDF
        auto mainWindow = MainWindow::instance();
        auto *textEdit = new QOwnNotesMarkdownTextEdit(this);
        textEdit->setPlainText(mainWindow->selectedNoteTextEditText());
        mainWindow->exportNoteAsPDF(textEdit->document(), true);
    });

    // add the export selected text (preview) action
    QAction *exportHTMLAction = exportMenu->addAction(tr("Export selected text as PDF (preview)"));
    exportHTMLAction->setEnabled(isTextSelected);
    exportHTMLAction->setIcon(pdfIcon);
    connect(exportHTMLAction, &QAction::triggered, this, []() {
        // export the selected text (preview) as PDF
        auto mainWindow = MainWindow::instance();
        auto note = mainWindow->getCurrentNote();
        QString html = note.textToMarkdownHtml(
            mainWindow->selectedNoteTextEditText(), NoteFolder::currentLocalPath(),
            mainWindow->getMaxImageWidth(), Utils::Misc::useInternalExportStylingForPreview());
        html = Utils::Misc::parseTaskList(html, false);
        QTextDocument doc;
        doc.setHtml(html);
        mainWindow->exportNoteAsPDF(&doc);
    });

    menu->addSeparator();

    // add some other existing menu entries
    auto mainWindow = MainWindow::instance();
    QMenu *selectMenu = menu->addMenu(tr("Select"));
    selectMenu->addAction(mainWindow->selectEnclosedTextAction());
    menu->addAction(mainWindow->pasteImageAction());
    menu->addAction(mainWindow->autocompleteAction());
    menu->addAction(mainWindow->splitNoteAtPosAction());

    // add the custom actions to the context menu
    const auto customActions = mainWindow->customTextEditActions();
    if (!customActions.isEmpty()) {
        // add the scripts menu
        QIcon scriptIcon = QIcon::fromTheme(
            QStringLiteral("story-editor"),
            QIcon(QStringLiteral(":icons/breeze-qownnotes/16x16/story-editor.svg")));
        menu->addSeparator();
        QMenu *scriptMenu = menu->addMenu(tr("Custom actions"));
        scriptMenu->setIcon(scriptIcon);
        scriptMenu->addActions(customActions);
    }

    menu->exec(globalPos);
}

QMenu *QOwnNotesMarkdownTextEdit::spellCheckContextMenu(QPoint pos) {
    auto spellchecker = QOwnSpellChecker::instance();

    // obtain the cursor at current mouse position
    QTextCursor cursorAtMouse = cursorForPosition(pos);
    const int mousePos = cursorAtMouse.position();

    QTextCursor cursor = textCursor();
    if (MarkdownHighlighter::isCodeBlock(cursor.block().userState())) {
        return nullptr;
    }

    auto *menu = new QMenu(this);
    bool hasEntries = false;

#ifdef LANGUAGETOOL_ENABLED
    addLanguageToolMenuSection(menu, cursorAtMouse, cursor, hasEntries);
#endif

#ifdef HARPER_ENABLED
    addHarperMenuSection(menu, cursorAtMouse, cursor, hasEntries);
#endif

    if (!spellchecker || !spellchecker->isActive() || _isSpellCheckingDisabled) {
        if (!hasEntries) {
            delete menu;
            return nullptr;
        }

        menu->setTitle(tr("Spelling"));
        return menu;
    }

    // Check if the user clicked a selected word
    const bool selectedWordClicked = cursor.hasSelection() && mousePos >= cursor.selectionStart() &&
                                     mousePos <= cursor.selectionEnd();

    // Get the word under the (mouse-)cursor and see if it is misspelled.
    // Don't include apostrophes at the start/end of the word in the selection.
    QTextCursor wordSelectCursor(cursorAtMouse);
    wordSelectCursor.clearSelection();
    wordSelectCursor.select(QTextCursor::WordUnderCursor);
    QString selectedWord = wordSelectCursor.selectedText();
    const int selectedWordBlock = wordSelectCursor.blockNumber();

    bool isMouseCursorInsideWord = true;
    if ((mousePos < wordSelectCursor.selectionStart() ||
         mousePos >= wordSelectCursor.selectionEnd()) &&
        (selectedWord.length() > 1)) {
        isMouseCursorInsideWord = false;
    }

    // Clear the selection again, we re-select it below (without the
    // apostrophes).
    wordSelectCursor.setPosition(wordSelectCursor.position() - selectedWord.size());
    if (selectedWord.startsWith(QLatin1Char('\'')) || selectedWord.startsWith(QLatin1Char('\"'))) {
        selectedWord = selectedWord.right(selectedWord.size() - 1);
        wordSelectCursor.movePosition(QTextCursor::NextCharacter, QTextCursor::MoveAnchor);
    }
    if (selectedWord.endsWith(QLatin1Char('\'')) || selectedWord.endsWith(QLatin1Char('\"'))) {
        selectedWord.chop(1);
    }

    wordSelectCursor.movePosition(QTextCursor::NextCharacter, QTextCursor::KeepAnchor,
                                  selectedWord.size());

    const bool wordIsMisspelled = isMouseCursorInsideWord && spellchecker->isActive() &&
                                  !selectedWord.isEmpty() &&
                                  spellchecker->isWordMisspelled(selectedWord);

    if (!wordIsMisspelled) {
        if (!hasEntries) {
            delete menu;
            return nullptr;
        }

        menu->setTitle(tr("Spelling"));
        return menu;
    }

    if (!selectedWordClicked) {
        // If the user clicked on a misspelled word, select that word.
        setTextCursor(wordSelectCursor);
        cursor = textCursor();
    }

    if (hasEntries) {
        menu->addSeparator();
    }

    // Add the suggestions to the menu
    const QStringList reps = spellchecker->suggestionsForWord(selectedWord, cursor, 8);
    if (reps.isEmpty()) {
        QAction *suggestionsAction = menu->addAction(tr("No suggestions for %1").arg(selectedWord));
        suggestionsAction->setEnabled(false);
    } else {
        for (const QString &rep : reps) {
            menu->addAction(rep, this, [rep, this, cursor]() mutable {
                if (!cursor.isNull()) {
                    // Turn off read-only mode first so the note will be stored
                    if (auto *mw = MainWindow::instance()) {
                        mw->allowNoteEditing();
                    }
                    cursor.insertText(rep);
                    setTextCursor(cursor);
                }
            });
        }
    }

    menu->addSeparator();

    menu->addAction(tr("Ignore"), this,
                    [selectedWord]() { QOwnSpellChecker::instance()->ignoreWord(selectedWord); });
    menu->addAction(tr("Add to Dictionary"), this, [this, selectedWord, selectedWordBlock]() {
        QOwnSpellChecker::instance()->addWordToDictionary(selectedWord);
        if (highlighter() && document()) {
            const QTextBlock block = document()->findBlockByNumber(selectedWordBlock);
            if (block.isValid()) {
                highlighter()->rehighlightBlock(block);
            }
        }
    });

    menu->setTitle(tr("Spelling"));

    return menu;
}

QMenu *QOwnNotesMarkdownTextEdit::markdownLspContextMenu(const QTextCursor &cursorAtMouse) {
    if (!_markdownLspEnabled || !_markdownLspClient || _markdownLspDiagnostics.isEmpty()) {
        return nullptr;
    }

    // Find a diagnostic whose range covers the cursor position
    const int cursorLine = cursorAtMouse.blockNumber();
    const int cursorCol = cursorAtMouse.positionInBlock();
    const MarkdownLspClient::Diagnostic *match = nullptr;
    const auto &constMarkdownLspDiagnostics = _markdownLspDiagnostics;
    for (const auto &diag : constMarkdownLspDiagnostics) {
        if (cursorLine >= diag.range.startLine && cursorLine <= diag.range.endLine) {
            // For single-line diagnostics also check column range
            if (diag.range.startLine == diag.range.endLine) {
                if (cursorCol >= diag.range.startCharacter &&
                    cursorCol <= diag.range.endCharacter) {
                    match = &diag;
                    break;
                }
            } else {
                match = &diag;
                break;
            }
        }
    }

    if (!match) {
        return nullptr;
    }

    auto *lspMenu = new QMenu(match->message, this);
    auto *ignoredRules = MarkdownLspIgnoredRules::instance();

    // Request code actions for this diagnostic and wait briefly for the response
    // using a local event loop (same pattern as QDialog::exec)
    _markdownLspCodeActionRequestId =
        _markdownLspClient->requestCodeActions(_markdownLspUri, match->range, {*match});

    if (_markdownLspCodeActionRequestId >= 0) {
        QVector<MarkdownLspClient::CodeAction> receivedActions;
        int receivedId = -1;

        // Wait up to 2 seconds for the response
        QEventLoop loop;
        QTimer timeout;
        timeout.setSingleShot(true);
        timeout.setInterval(2000);
        connect(&timeout, &QTimer::timeout, &loop, &QEventLoop::quit);
        connect(_markdownLspClient, &MarkdownLspClient::codeActionsReceived, &loop,
                [&](int id, const QVector<MarkdownLspClient::CodeAction> &actions) {
                    receivedId = id;
                    receivedActions = actions;
                    loop.quit();
                });
        timeout.start();
        loop.exec();

        if (receivedId == _markdownLspCodeActionRequestId) {
            _markdownLspCodeActionRequestId = -1;
            const auto &constReceivedActions = receivedActions;
            for (const auto &action : constReceivedActions) {
                lspMenu->addAction(action.title, this, [this, action]() {
                    if (action.hasEdits()) {
                        applyMarkdownLspTextEdits(action.edits);
                    } else if (action.hasCommand()) {
                        _markdownLspClient->executeCommand(action.command);
                    }
                });
            }
        }
    }

    const QString ruleId = match->ruleId();
    if (!ruleId.isEmpty() && ignoredRules && !ignoredRules->isRuleIgnored(ruleId)) {
        const QString displayRuleName = match->displayRuleName();
        const QString actionText = displayRuleName.isEmpty()
                                       ? tr("Ignore this rule globally")
                                       : tr("Ignore rule %1 globally").arg(displayRuleName);
        lspMenu->addAction(actionText, this, [this, ignoredRules, ruleId]() {
            ignoredRules->ignoreRule(ruleId);
            refreshMarkdownLspDiagnostics();
        });
    }

    return lspMenu;
}

#ifdef LANGUAGETOOL_ENABLED
void QOwnNotesMarkdownTextEdit::addLanguageToolMenuSection(QMenu *menu,
                                                           const QTextCursor &cursorAtMouse,
                                                           const QTextCursor &selectedCursor,
                                                           bool &hasEntries) {
    auto *checker = LanguageToolChecker::instance();
    if ((checker == nullptr) || !checker->isEnabled()) {
        return;
    }

    QTextCursor matchCursor(cursorAtMouse);
    matchCursor.clearSelection();
    const auto blockMatch = checker->matchAtPosition(matchCursor, matchCursor.positionInBlock());
    if ((blockMatch.blockNumber < 0) || checker->isRuleIgnored(blockMatch.match.ruleId)) {
        return;
    }

    QTextCursor replacementCursor(selectedCursor);
    const int blockPosition = matchCursor.block().position();
    replacementCursor.setPosition(blockPosition + blockMatch.match.offset);
    replacementCursor.setPosition(blockPosition + blockMatch.match.offset + blockMatch.match.length,
                                  QTextCursor::KeepAnchor);

    const QString category = blockMatch.match.ruleCategory.isEmpty()
                                 ? tr("LanguageTool")
                                 : blockMatch.match.ruleCategory;
    QString headerText = category;
    if (!blockMatch.match.shortMessage.isEmpty()) {
        headerText += QStringLiteral(": ") + blockMatch.match.shortMessage;
    } else if (!blockMatch.match.message.isEmpty()) {
        headerText += QStringLiteral(": ") + blockMatch.match.message;
    }

    QAction *headerAction = menu->addAction(headerText);
    headerAction->setEnabled(false);

    if (blockMatch.match.replacements.isEmpty()) {
        QAction *noSuggestionsAction = menu->addAction(tr("No suggestions"));
        noSuggestionsAction->setEnabled(false);
    } else {
        const QStringList replacements = blockMatch.match.replacements.mid(0, 8);
        for (const QString &replacement : replacements) {
            menu->addAction(replacement, this, [this, replacementCursor, replacement]() mutable {
                applyLanguageToolReplacement(replacementCursor, replacement);
            });
        }
    }

    if (!blockMatch.match.ruleUrl.isEmpty()) {
        const auto url = blockMatch.match.ruleUrl;
        menu->addAction(tr("More info..."), this,
                        [url]() { QDesktopServices::openUrl(QUrl(url)); });
    }

    const auto ruleId = blockMatch.match.ruleId;
    menu->addAction(tr("Ignore this rule"), this, [this, checker, ruleId]() {
        checker->ignoreRule(ruleId);
        if (highlighter()) {
            highlighter()->rehighlight();
        }
    });

    // Add "Ignore this word" option for typo/spelling matches
    const QString matchedWord = replacementCursor.selectedText();
    if (!matchedWord.isEmpty()) {
        menu->addAction(tr("Ignore word \"%1\"").arg(matchedWord), this,
                        [this, checker, matchedWord]() {
                            checker->ignoreWord(matchedWord);
                            if (highlighter()) {
                                highlighter()->rehighlight();
                            }
                        });
    }

    hasEntries = true;
}

void QOwnNotesMarkdownTextEdit::applyLanguageToolReplacement(const QTextCursor &cursor,
                                                             const QString &replacement) {
    QTextCursor mutableCursor(cursor);
    if (mutableCursor.isNull()) {
        return;
    }

    // Turn off read-only mode first so the note will be stored
    if (auto *mw = MainWindow::instance()) {
        mw->allowNoteEditing();
    }

    mutableCursor.insertText(replacement);
    setTextCursor(mutableCursor);
}
#endif

#ifdef HARPER_ENABLED
void QOwnNotesMarkdownTextEdit::addHarperMenuSection(QMenu *menu, const QTextCursor &cursorAtMouse,
                                                     const QTextCursor &selectedCursor,
                                                     bool &hasEntries) {
    auto *checker = HarperChecker::instance();
    if ((checker == nullptr) || !checker->isEnabled()) {
        return;
    }

    QTextCursor matchCursor(cursorAtMouse);
    matchCursor.clearSelection();
    const auto blockMatch = checker->matchAtPosition(matchCursor, matchCursor.positionInBlock());
    if ((blockMatch.blockNumber < 0) || checker->isRuleIgnored(blockMatch.match.ruleId)) {
        return;
    }

    if (hasEntries) {
        menu->addSeparator();
    }

    QTextCursor replacementCursor(selectedCursor);
    const int blockPosition = matchCursor.block().position();
    replacementCursor.setPosition(blockPosition + blockMatch.match.offset);
    replacementCursor.setPosition(blockPosition + blockMatch.match.offset + blockMatch.match.length,
                                  QTextCursor::KeepAnchor);

    const QString category =
        blockMatch.match.ruleCategory.isEmpty() ? tr("Harper") : blockMatch.match.ruleCategory;
    QString headerText = category;
    if (!blockMatch.match.shortMessage.isEmpty()) {
        headerText += QStringLiteral(": ") + blockMatch.match.shortMessage;
    } else if (!blockMatch.match.message.isEmpty()) {
        headerText += QStringLiteral(": ") + blockMatch.match.message;
    }

    QAction *headerAction = menu->addAction(headerText);
    headerAction->setEnabled(false);

    if (blockMatch.match.replacements.isEmpty()) {
        QAction *noSuggestionsAction = menu->addAction(tr("No suggestions"));
        noSuggestionsAction->setEnabled(false);
    } else {
        const QStringList replacements = blockMatch.match.replacements.mid(0, 8);
        for (const QString &replacement : replacements) {
            menu->addAction(replacement, this, [this, replacementCursor, replacement]() mutable {
                applyHarperReplacement(replacementCursor, replacement);
            });
        }
    }

    const auto ruleId = blockMatch.match.ruleId;
    menu->addAction(tr("Ignore this rule"), this, [this, checker, ruleId]() {
        checker->ignoreRule(ruleId);
        if (highlighter()) {
            highlighter()->rehighlight();
        }
    });

    const QString matchedWord = replacementCursor.selectedText();
    if (!matchedWord.isEmpty()) {
        menu->addAction(tr("Ignore word \"%1\"").arg(matchedWord), this,
                        [this, checker, matchedWord]() {
                            checker->ignoreWord(matchedWord);
                            if (highlighter()) {
                                highlighter()->rehighlight();
                            }
                        });
    }

    hasEntries = true;
}

void QOwnNotesMarkdownTextEdit::applyHarperReplacement(const QTextCursor &cursor,
                                                       const QString &replacement) {
    QTextCursor mutableCursor(cursor);
    if (mutableCursor.isNull()) {
        return;
    }

    if (auto *mw = MainWindow::instance()) {
        mw->allowNoteEditing();
    }

    mutableCursor.insertText(replacement);
    setTextCursor(mutableCursor);
}
#endif

bool QOwnNotesMarkdownTextEdit::eventFilter(QObject *obj, QEvent *event) {
    if (event->type() == QEvent::KeyPress) {
        auto *keyEvent = static_cast<QKeyEvent *>(event);

        if (objectName() == QStringLiteral("encryptedNoteTextEdit") ||
            objectName() == QStringLiteral("noteTextEdit")) {
            // deactivating the search widget has priority
            if ((keyEvent->key() == Qt::Key_Escape) && _searchWidget->isVisible()) {
                _searchWidget->deactivate();
                return true;
            } else if ((keyEvent->key() == Qt::Key_R) &&
                       keyEvent->modifiers().testFlag(Qt::ControlModifier) &&
                       !Utils::Misc::isNoteEditingAllowed()) {
                MainWindow::instance()->allowNoteEditing();
            } else if (!Utils::Misc::isNoteEditingAllowed()) {
                const auto noModifierKeys = QList<int>()
                                            << Qt::Key_Return << Qt::Key_Enter << Qt::Key_Space
                                            << Qt::Key_Backspace << Qt::Key_Delete << Qt::Key_Tab
                                            << Qt::Key_Backtab << Qt::Key_Minus << Qt::Key_ParenLeft
                                            << Qt::Key_BraceLeft << Qt::Key_BracketLeft
                                            << Qt::Key_Plus << Qt::Key_Comma << Qt::Key_Period;

                if ((keyEvent->key() == Qt::Key_Space) &&
                    keyEvent->modifiers().testFlag(Qt::ControlModifier)) {
                    if (openLinkAtCursorPosition()) {
                        MainWindow::instance()->showStatusBarMessage(
                            tr("An url was opened at the current cursor position"),
                            QStringLiteral("📃"), 5000);
                    }

                    return true;
                }

                const auto controlModifierKeys = QList<int>() << Qt::Key_V;

                // show notification if user tries to edit a note while
                // note editing is turned off
                if (((keyEvent->key() < 128 || noModifierKeys.contains(keyEvent->key())) &&
                     keyEvent->modifiers().testFlag(Qt::NoModifier)) ||
                    (controlModifierKeys.contains(keyEvent->key()) &&
                     keyEvent->modifiers().testFlag(Qt::ControlModifier) && isReadOnly())) {
                    if (MainWindow::instance() && MainWindow::instance()->doNoteEditingCheck()) {
                        // If the answer is overridden to Yes ("Don't ask again" with "Yes"),
                        // what you type then only enables note editing, but is not typed in
                        // the editor. We need to re-send the event after enabling editing.
                        // BUT, we should do that only if the msgbox is overridden to Yes,
                        // not if manually answered.
                        // You may see: https://github.com/pbek/QOwnNotes/issues/2421
                        // This check is partially copied from utils/gui.cpp showMessage()
                        SettingsService settings;
                        const QString settingsKey =
                            QStringLiteral("MessageBoxOverride/readonly-mode-allow");
                        auto overrideButton = static_cast<QMessageBox::StandardButton>(
                            settings.value(settingsKey, QMessageBox::NoButton).toInt());
                        if (overrideButton == QMessageBox::Yes) {
                            // overridden to answer yes: re-send the event
                            return QMarkdownTextEdit::eventFilter(obj, event);
                        }
                    }

                    return true;
                }
            } else {
                // Disable note editing if Escape key was pressed and
                // read-only mode feature is enabled
                if (keyEvent->key() == Qt::Key_Escape && Utils::Misc::isReadOnlyModeEnabled()) {
                    MainWindow::instance()->disallowNoteEditing();

                    return true;
                } else if ((keyEvent->key() == Qt::Key_Tab) ||
                           (keyEvent->key() == Qt::Key_Backtab)) {
                    // handle entered tab and reverse tab keys
                    return handleTabEntered(keyEvent->key() == Qt::Key_Backtab,
                                            Utils::Misc::indentCharacters());
                }
            }
        }
    }

    return QMarkdownTextEdit::eventFilter(obj, event);
}

void QOwnNotesMarkdownTextEdit::updateIgnoredClickUrlRegexps() {
    // Deck card URLs are now fully handled by openUrl() via UrlHandler, so we
    // must not add them to the ignored-click list — doing so would cause
    // openLinkAtCursorPosition() to return early and swallow the Ctrl+Click,
    // preventing the Nextcloud Deck dialog from opening in the note editor.
    // Clear any previously set regexps to avoid stale state.
    setIgnoredClickUrlRegexps({});
}

/**
 * Requests AI autocomplete for the current text
 */
void QOwnNotesMarkdownTextEdit::requestAiAutocomplete() {
    qDebug() << __func__ << " - called";

    if (!OpenAiService::getAutocompleteEnabled() || !OpenAiService::getEnabled()) {
        qDebug() << __func__ << " - autocomplete not enabled, returning";
        return;
    }

    qDebug() << __func__ << " - autocomplete is enabled, proceeding";

    // Don't autocomplete if there's a selection
    QTextCursor cursor = textCursor();
    if (cursor.hasSelection()) {
        qDebug() << __func__ << " - cursor has selection, returning";
        return;
    }

    // Get the current text context (e.g., last 200 characters)
    cursor.movePosition(QTextCursor::Start, QTextCursor::KeepAnchor);
    QString context = cursor.selectedText();

    // Limit context to last 500 characters to avoid too large prompts
    if (context.length() > 500) {
        context = context.right(500);
    }

    qDebug() << __func__ << " - context:" << context;

    // Don't request if context is too short
    if (context.trimmed().length() < 10) {
        qDebug() << __func__ << " - context too short, returning";
        return;
    }

    // Store the current cursor position
    _aiAutocompletePosition = textCursor().position();

    qDebug() << __func__ << " - calling completeAsync, position:" << _aiAutocompletePosition;

    // Request completion from OpenAI service
    OpenAiService::instance()->completeAsync(context);
}

/**
 * Shows the AI autocomplete suggestion
 */
void QOwnNotesMarkdownTextEdit::showAiAutocompleteSuggestion(const QString &suggestion) {
    qDebug() << "=== showAiAutocompleteSuggestion CALLED ===" << this;
    qDebug() << __func__ << " - Widget:" << objectName();

    SettingsService settings;
    bool enabled = settings.value(QStringLiteral("ai/autocompleteEnabled")).toBool();
    qDebug() << __func__ << " - ai/autocompleteEnabled setting:" << enabled;

    if (!enabled) {
        qDebug() << __func__ << " - autocomplete not enabled in settings, returning";
        return;
    }

    qDebug() << __func__ << " - 'suggestion': " << suggestion;

    if (suggestion.isEmpty()) {
        qDebug() << __func__ << " - suggestion is empty, returning";
        return;
    }

    // Check if cursor hasn't moved too far from the autocomplete position
    int currentPos = textCursor().position();
    qDebug() << __func__ << " - 'currentPos': " << currentPos
             << " '_aiAutocompletePosition': " << _aiAutocompletePosition;

    if (currentPos < _aiAutocompletePosition || currentPos > _aiAutocompletePosition + 50) {
        // Cursor has moved too much, don't show suggestion
        qDebug() << __func__ << " - cursor moved too much, returning";
        return;
    }

    _aiAutocompleteSuggestion = suggestion;
    _isInsertingAiSuggestion = true;

    qDebug() << __func__ << " - inserting suggestion text...";

    // Insert the suggestion with a gray color format
    QTextCursor cursor = textCursor();
    cursor.beginEditBlock();

    // Store the original format to restore it later
    QTextCharFormat originalFormat = cursor.charFormat();

    // Create the suggestion format
    QTextCharFormat format;
    format.setForeground(QColor(128, 128, 128));    // Gray color
    format.setFontItalic(true);

    cursor.insertText(_aiAutocompleteSuggestion, format);

    // Select the suggestion so it can be easily replaced
    cursor.setPosition(currentPos);
    cursor.movePosition(QTextCursor::Right, QTextCursor::KeepAnchor,
                        _aiAutocompleteSuggestion.length());

    // Reset the character format to the original to prevent the italic format
    // from affecting subsequent text
    cursor.setCharFormat(originalFormat);

    cursor.endEditBlock();
    setTextCursor(cursor);

    qDebug() << __func__ << " - suggestion inserted and selected successfully!";

    _isInsertingAiSuggestion = false;
}

/**
 * Clears the AI autocomplete suggestion
 */
void QOwnNotesMarkdownTextEdit::clearAiAutocompleteSuggestion() {
    if (_aiAutocompleteSuggestion.isEmpty()) {
        return;
    }

    _isInsertingAiSuggestion = true;

    QTextCursor cursor = textCursor();

    // If we have selected text that matches our suggestion, delete it
    if (cursor.hasSelection()) {
        QString selectedText = cursor.selectedText();
        if (selectedText == _aiAutocompleteSuggestion) {
            cursor.removeSelectedText();
            setTextCursor(cursor);
        }
    }

    _aiAutocompleteSuggestion.clear();
    _aiAutocompletePosition = -1;
    _isInsertingAiSuggestion = false;
}

/**
 * Accepts the current AI autocomplete suggestion
 */
void QOwnNotesMarkdownTextEdit::acceptAiAutocompleteSuggestion() {
    if (_aiAutocompleteSuggestion.isEmpty()) {
        return;
    }

    _isInsertingAiSuggestion = true;

    QTextCursor cursor = textCursor();

    // If we have the suggestion selected, replace it with normal formatted text
    if (cursor.hasSelection()) {
        QString selectedText = cursor.selectedText();
        if (selectedText == _aiAutocompleteSuggestion) {
            // Remove the formatted suggestion
            cursor.removeSelectedText();
            // Insert as normal text
            cursor.insertText(_aiAutocompleteSuggestion);
            setTextCursor(cursor);
        }
    }

    _aiAutocompleteSuggestion.clear();
    _aiAutocompletePosition = -1;
    _isInsertingAiSuggestion = false;
}

/**
 * Called when AI autocomplete is completed
 */
void QOwnNotesMarkdownTextEdit::onAiAutocompleteCompleted(const QString &result) {
    qDebug() << "=== onAiAutocompleteCompleted CALLED ===" << this;
    qDebug() << __func__ << " - Widget:" << objectName()
             << "Parent:" << (parent() ? parent()->objectName() : "null");
    qDebug() << __func__ << " - 'result': " << result;

    if (result.isEmpty()) {
        qDebug() << __func__ << " - result is empty, returning";
        return;
    }

    qDebug() << __func__ << " - Processing result, length:" << result.length();

    // Extract the first line or first sentence as suggestion
    QString suggestion = result.trimmed();

    // Take only the first line or up to 100 characters
    int newlinePos = suggestion.indexOf('\n');
    if (newlinePos > 0 && newlinePos < 100) {
        suggestion = suggestion.left(newlinePos);
    } else if (suggestion.length() > 100) {
        suggestion = suggestion.left(100);
        // Try to break at a word boundary
        int lastSpace = suggestion.lastIndexOf(' ');
        if (lastSpace > 50) {
            suggestion = suggestion.left(lastSpace);
        }
    }

    qDebug() << __func__ << " - 'suggestion': " << suggestion;
    qDebug() << __func__ << " - 'cursor position': " << textCursor().position();
    qDebug() << __func__ << " - '_aiAutocompletePosition': " << _aiAutocompletePosition;

    qDebug() << __func__ << " - Calling showAiAutocompleteSuggestion...";
    showAiAutocompleteSuggestion(suggestion);
    qDebug() << __func__ << " - showAiAutocompleteSuggestion returned";
}

/**
 * Called when AI autocomplete request times out or errors
 */
void QOwnNotesMarkdownTextEdit::onAiAutocompleteTimeout(const QString &errorString) {
    qDebug() << __func__ << " - error:" << errorString;
    // Just clear the state, don't show error to user
    _aiAutocompleteSuggestion.clear();
    _aiAutocompletePosition = -1;
}

/**
 * Override keyPressEvent to handle Tab and Escape for autocomplete
 */
void QOwnNotesMarkdownTextEdit::keyPressEvent(QKeyEvent *e) {
    if (!isReadOnly()) {
        const QChar deadKeyAccent = accentForDeadKey(e->key());
        if (!deadKeyAccent.isNull() && e->text().isEmpty()) {
            _pendingDeadKey = deadKeyAccent;
            e->accept();
            return;
        }

        if (!_pendingDeadKey.isNull()) {
            if (e->key() == Qt::Key_Escape) {
                _pendingDeadKey = QChar();
            } else if (!(e->modifiers() &
                         (Qt::ControlModifier | Qt::AltModifier | Qt::MetaModifier)) &&
                       e->text().size() == 1) {
                const QChar accent = _pendingDeadKey;
                _pendingDeadKey = QChar();

                const QChar character = e->text().at(0);
                const QString composed = composeDeadKey(accent, character);
                QTextCursor cursor = textCursor();
                cursor.insertText(composed.size() == 1
                                      ? composed
                                      : QString(spacingAccentForCombiningMark(accent)));
                if (composed.size() != 1 && character != QLatin1Char(' ')) {
                    cursor.insertText(e->text());
                }
                setTextCursor(cursor);
                e->accept();
                return;
            }
        }
    } else {
        _pendingDeadKey = QChar();
    }

    // Handle Tab key to accept AI autocomplete suggestion
    if (e->key() == Qt::Key_Tab && !_aiAutocompleteSuggestion.isEmpty()) {
        acceptAiAutocompleteSuggestion();
        e->accept();
        return;
    }

    // Handle Escape key to dismiss AI autocomplete suggestion
    if (e->key() == Qt::Key_Escape && !_aiAutocompleteSuggestion.isEmpty()) {
        clearAiAutocompleteSuggestion();
        e->accept();
        return;
    }

    // Clear suggestion on any other key press
    if (!_aiAutocompleteSuggestion.isEmpty()) {
        clearAiAutocompleteSuggestion();
    }

    // Call parent implementation
    QMarkdownTextEdit::keyPressEvent(e);

    if (!e->isAccepted() || !Note::isWikiLinkSupportEnabled()) {
        return;
    }

    if (e->text() == QStringLiteral("[") &&
        (e->modifiers() == Qt::NoModifier || e->modifiers() == Qt::ShiftModifier)) {
        // Only trigger automatic note filename selection if the setting is enabled
        const bool autoSelect =
            SettingsService()
                .value(QStringLiteral("Editor/wikiLinkFileNameAutoSelect"), false)
                .toBool();
        if (!autoSelect) {
            return;
        }

        WikiLinkCompletionContext context;
        if (currentWikiLinkCompletionContext(this, context)) {
            QTimer::singleShot(0, this, &QOwnNotesMarkdownTextEdit::onAutoCompleteRequested);
        }
    }
}

/**
 * Override focusInEvent to register this editor as active when it receives focus
 */
void QOwnNotesMarkdownTextEdit::focusInEvent(QFocusEvent *e) {
    // Register as the active editor for autocomplete when receiving focus
    // Skip for log widgets
    if (objectName() != QStringLiteral("logTextEdit")) {
        qDebug() << __func__ << " - Registering as active editor:" << this << objectName();
        registerAsActiveEditor();
#ifdef LANGUAGETOOL_ENABLED
        LanguageToolChecker::instance()->setTextEdit(this);
#endif
#ifdef HARPER_ENABLED
        HarperChecker::instance()->setTextEdit(this);
#endif
    }

    // Call parent implementation
    QMarkdownTextEdit::focusInEvent(e);
}

/**
 * Override inputMethodQuery to fix IME candidate window position on Windows.
 * When viewport margins are set (e.g. for paper margins), the cursor rectangle
 * returned to the IME must be offset by the top/left margin so that the
 * candidate window appears adjacent to the cursor rather than overlapping it.
 */
QVariant QOwnNotesMarkdownTextEdit::inputMethodQuery(Qt::InputMethodQuery property) const {
    QVariant result = QPlainTextEdit::inputMethodQuery(property);

    const bool isCursorQuery =
        (property == Qt::ImCursorRectangle || property == Qt::ImAnchorRectangle);
    if (isCursorQuery) {
        // viewportMargins() is non-const in this class, so cast away const to call it
        const QMargins vm = const_cast<QOwnNotesMarkdownTextEdit *>(this)->viewportMargins();
        if (vm.top() != 0 || vm.left() != 0) {
            if (result.userType() == QMetaType::QRectF) {
                result = result.toRectF().translated(qreal(vm.left()), qreal(vm.top()));
            }
        }
    }

    return result;
}

/**
 * Register this editor as the active one for AI autocomplete
 */
void QOwnNotesMarkdownTextEdit::registerAsActiveEditor() {
    qDebug() << __func__ << " - Registering editor:" << this << objectName();
    _activeAutocompleteEditor = this;
    // Also store in QApplication property for access from OpenAiService callback
    qApp->setProperty("activeAutocompleteEditor", QVariant::fromValue<QObject *>(this));
}

/**
 * Unregister this editor from receiving AI autocomplete
 */
void QOwnNotesMarkdownTextEdit::unregisterAsActiveEditor() {
    qDebug() << __func__ << " - Unregistering editor:" << this << objectName();
    if (_activeAutocompleteEditor == this) {
        _activeAutocompleteEditor = nullptr;
        // Also clear QApplication property
        qApp->setProperty("activeAutocompleteEditor", QVariant::fromValue<QObject *>(nullptr));
    }
}

/**
 * Get the currently active editor for AI autocomplete
 */
QOwnNotesMarkdownTextEdit *QOwnNotesMarkdownTextEdit::getActiveEditorForAutocomplete() {
    return _activeAutocompleteEditor;
}

QOwnNotesMarkdownTextEdit::~QOwnNotesMarkdownTextEdit() {
    qDebug() << "*** QOwnNotesMarkdownTextEdit DESTROYED ***" << this << objectName();
    closeMarkdownLspDocument();
#ifdef LANGUAGETOOL_ENABLED
    LanguageToolChecker::instance()->clearForTextEdit(this);
#endif
#ifdef HARPER_ENABLED
    HarperChecker::instance()->clearForTextEdit(this);
#endif
    // Unregister if this was the active editor
    unregisterAsActiveEditor();
}

void QOwnNotesMarkdownTextEdit::setMarkdownLspDocumentPath(const QString &filePath,
                                                           const QString &text) {
    if (!_markdownLspEnabled || !_markdownLspTracker) {
        return;
    }

    const QString uri = QUrl::fromLocalFile(filePath).toString();
    if (uri.isEmpty()) {
        return;
    }

    if (_markdownLspUri == uri) {
        return;
    }

    // Close previous document via the tracker
    if (_markdownLspTracker->isOpen()) {
        _markdownLspTracker->close();
    }

    _markdownLspUri = uri;
    _markdownLspVersion = 1;

    if (_markdownLspClient) {
        _markdownLspTracker->open(_markdownLspUri, text);
    }
}

void QOwnNotesMarkdownTextEdit::closeMarkdownLspDocument() {
    if (_markdownLspTracker && _markdownLspTracker->isOpen()) {
        _markdownLspTracker->close();
    }

    _markdownLspUri.clear();
    _markdownLspVersion = 0;
    _markdownLspAllDiagnostics.clear();
    _markdownLspDiagnostics.clear();

    if (auto *h = dynamic_cast<QOwnNotesMarkdownHighlighter *>(highlighter())) {
        h->clearMarkdownLspDiagnostics();
    }
}

void QOwnNotesMarkdownTextEdit::initializeMarkdownLsp() {
    _markdownLspInitialized = true;
    applyMarkdownLspSettings();
}

void QOwnNotesMarkdownTextEdit::applyMarkdownLspSettings() {
    SettingsService settings;
    const bool enabled =
        settings.value(QStringLiteral("Editor/markdownLspEnabled"), false).toBool();
    const QString command =
        settings.value(QStringLiteral("Editor/markdownLspCommand"), QStringLiteral("marksman"))
            .toString();
    const QStringList arguments =
        settings.value(QStringLiteral("Editor/markdownLspArguments")).toStringList();
    const bool verboseLogging =
        settings.value(QStringLiteral("Editor/markdownLspVerboseLogging"), false).toBool();
    const bool serverConfigurationChanged =
        (_markdownLspCommand != command) || (_markdownLspArguments != arguments);

    if (!enabled) {
        _markdownLspEnabled = false;
        _markdownLspCommand = command;
        _markdownLspArguments = arguments;
        if (_markdownLspClient) {
            closeMarkdownLspDocument();
            _markdownLspClient->shutdown();
        }
        return;
    }

    if (!_markdownLspClient) {
        _markdownLspClient = new MarkdownLspClient(this);
        connect(_markdownLspClient, &MarkdownLspClient::completionReceived, this,
                &QOwnNotesMarkdownTextEdit::showMarkdownLspCompletions);
        connect(_markdownLspClient, &MarkdownLspClient::formattingReceived, this,
                &QOwnNotesMarkdownTextEdit::applyMarkdownLspFormatting);
        connect(_markdownLspClient, &MarkdownLspClient::diagnosticsReceived, this,
                &QOwnNotesMarkdownTextEdit::showMarkdownLspDiagnostics);
        connect(_markdownLspClient, &MarkdownLspClient::errorMessage, this,
                [this](const QString &message) {
                    if (!message.trimmed().isEmpty()) {
                        qWarning() << "Markdown LSP:" << message.trimmed();
                    }
                });
        connect(_markdownLspClient, &MarkdownLspClient::serverInitialized, this, [this]() {
            if (auto *mw = MainWindow::instance()) {
                mw->showStatusBarMessage(tr("Markdown LSP server connected"), 3000);
            }

            // Configure the tracker's sync kind based on the server's capability
            if (_markdownLspTracker && _markdownLspClient) {
                const auto kind = _markdownLspClient->serverSyncKind();
                if (kind == MarkdownLspClient::SyncIncremental) {
                    _markdownLspTracker->setSyncKind(MarkdownLspDocumentTracker::SyncIncremental);
                } else {
                    _markdownLspTracker->setSyncKind(MarkdownLspDocumentTracker::SyncFull);
                }
            }
        });

        // Wire the tracker to the client and document
        if (_markdownLspTracker) {
            _markdownLspTracker->setClient(_markdownLspClient);
            _markdownLspTracker->setDocument(document());
        }
    }

    QString reopenUri;
    QString reopenText;
    const bool restartClient = serverConfigurationChanged && _markdownLspClient->isRunning();
    if (restartClient) {
        reopenUri = _markdownLspUri;
        reopenText = toPlainText();
        closeMarkdownLspDocument();
        _markdownLspClient->shutdown();
    }

    _markdownLspCommand = command;
    _markdownLspArguments = arguments;
    _markdownLspClient->setServerCommand(command, arguments);
    _markdownLspClient->setVerboseLogging(verboseLogging);
    const bool needsStart = !_markdownLspClient->isRunning();
    if (needsStart) {
        if (!_markdownLspClient->start()) {
            return;
        }

        const QString rootPath = NoteFolder::currentLocalPath();
        _markdownLspClient->initialize(rootPath, QStringLiteral("QOwnNotes"),
                                       QStringLiteral(VERSION));
    }

    _markdownLspEnabled = true;

    if (!reopenUri.isEmpty()) {
        _markdownLspUri = reopenUri;
        _markdownLspVersion = 1;
        _markdownLspTracker->open(_markdownLspUri, reopenText);
        return;
    }

    if (needsStart && !_markdownLspUri.isEmpty()) {
        _markdownLspVersion = 1;
        _markdownLspTracker->open(_markdownLspUri, toPlainText());
    }
}

void QOwnNotesMarkdownTextEdit::scheduleMarkdownLspChange() {
    if (!_markdownLspEnabled || !_markdownLspClient || _markdownLspUri.isEmpty()) {
        return;
    }

    _markdownLspPendingText = toPlainText();
    _markdownLspChangeTimer->start();
}

void QOwnNotesMarkdownTextEdit::sendMarkdownLspChange() {
    if (!_markdownLspEnabled || !_markdownLspClient || _markdownLspUri.isEmpty()) {
        return;
    }

    _markdownLspVersion++;
    _markdownLspClient->didChange(_markdownLspUri, _markdownLspPendingText, _markdownLspVersion);
}

void QOwnNotesMarkdownTextEdit::showMarkdownLspCompletions(int requestId,
                                                           const QStringList &items) {
    if (requestId != _markdownLspCompletionRequestId || _markdownLspCompletionRequestId == -1) {
        return;
    }

    if (items.isEmpty()) {
        _markdownLspCompletionRequestId = -1;
        return;
    }

    const QString initialFilter = currentWord();

    QMenu menu;
    auto *filterEdit = new QLineEdit(&menu);
    filterEdit->setPlaceholderText(tr("Filter completions"));
    filterEdit->setClearButtonEnabled(true);
    filterEdit->setText(initialFilter);

    auto *filterAction = new QWidgetAction(&menu);
    filterAction->setDefaultWidget(filterEdit);
    menu.addAction(filterAction);
    menu.addSeparator();

    QList<QAction *> completionActions;
    for (const QString &text : items) {
        auto *action = menu.addAction(text);
        action->setData(text);
        action->setWhatsThis(QStringLiteral("autocomplete"));
        completionActions.append(action);
    }

    _markdownLspCompletionRequestId = -1;

    if (menu.actions().isEmpty()) {
        return;
    }

    auto updateFilter = [filterEdit, completionActions]() {
        const QString query = filterEdit->text().trimmed();
        const bool hasQuery = !query.isEmpty();
        for (QAction *action : completionActions) {
            if (!action) {
                continue;
            }
            const bool matches = !hasQuery || action->text().contains(query, Qt::CaseInsensitive);
            action->setVisible(matches);
        }
    };

    connect(filterEdit, &QLineEdit::textChanged, this, [updateFilter]() { updateFilter(); });
    connect(filterEdit, &QLineEdit::returnPressed, this, [this, completionActions]() {
        for (QAction *action : completionActions) {
            if (action && action->isVisible()) {
                action->trigger();
                return;
            }
        }
    });
    updateFilter();

    QPoint globalPos = mapToGlobal(cursorRect().bottomRight());
    globalPos.setY(globalPos.y() + viewportMargins().top());
    globalPos.setX(globalPos.x() + viewportMargins().left());

    QTimer::singleShot(0, filterEdit, [filterEdit]() {
        filterEdit->setFocus(Qt::PopupFocusReason);
        if (filterEdit->text().isEmpty()) {
            filterEdit->selectAll();
        } else {
            filterEdit->setCursorPosition(filterEdit->text().size());
        }
    });

    QAction *selectedItem = menu.exec(globalPos);
    if (!selectedItem) {
        return;
    }

    const QString text = selectedItem->data().toString();
    if (text.isEmpty()) {
        return;
    }

    QTextCursor c = textCursor();
    c.movePosition(QTextCursor::StartOfWord, QTextCursor::KeepAnchor);
    c.insertText(text + QStringLiteral(" "));
}

void QOwnNotesMarkdownTextEdit::showMarkdownLspDiagnostics(
    const QString &uri, const QVector<MarkdownLspClient::Diagnostic> &diagnostics) {
    // Normalize both URIs by percent-decoding before comparing, since some LSP
    // servers (e.g. rumdl) encode spaces as %20 while we store the raw path
    const QString normalizedUri = QUrl::fromPercentEncoding(uri.toUtf8());
    const QString normalizedCurrentUri = QUrl::fromPercentEncoding(_markdownLspUri.toUtf8());
    if (normalizedUri != normalizedCurrentUri) {
        return;
    }

    _markdownLspAllDiagnostics = diagnostics;
    applyMarkdownLspDiagnostics(filteredMarkdownLspDiagnostics(diagnostics));
}

QVector<MarkdownLspClient::Diagnostic> QOwnNotesMarkdownTextEdit::filteredMarkdownLspDiagnostics(
    const QVector<MarkdownLspClient::Diagnostic> &diagnostics) const {
    auto *ignoredRules = MarkdownLspIgnoredRules::instance();
    if (!ignoredRules) {
        return diagnostics;
    }

    QVector<MarkdownLspClient::Diagnostic> filteredDiagnostics;
    filteredDiagnostics.reserve(diagnostics.size());
    for (const MarkdownLspClient::Diagnostic &diagnostic : diagnostics) {
        if (ignoredRules->isRuleIgnored(diagnostic.ruleId())) {
            continue;
        }

        filteredDiagnostics.append(diagnostic);
    }

    return filteredDiagnostics;
}

void QOwnNotesMarkdownTextEdit::applyMarkdownLspDiagnostics(
    const QVector<MarkdownLspClient::Diagnostic> &diagnostics) {
    auto *h = dynamic_cast<QOwnNotesMarkdownHighlighter *>(highlighter());
    if (!h || !document()) {
        _markdownLspDiagnostics = diagnostics;
        return;
    }

    // Collect blocks that had old diagnostics so we can clear their underlines
    QSet<int> dirtyLines;
    for (const MarkdownLspClient::Diagnostic &diag : _markdownLspDiagnostics) {
        for (int line = diag.range.startLine; line <= diag.range.endLine; ++line) {
            dirtyLines.insert(line);
        }
    }

    _markdownLspDiagnostics = diagnostics;
    h->setMarkdownLspDiagnostics(diagnostics);

    // Also collect blocks that have new diagnostics
    for (const MarkdownLspClient::Diagnostic &diag : diagnostics) {
        for (int line = diag.range.startLine; line <= diag.range.endLine; ++line) {
            dirtyLines.insert(line);
        }
    }

    if (dirtyLines.isEmpty()) {
        return;
    }

    const QSet<int> &constDirtyLines = dirtyLines;
    for (const int line : constDirtyLines) {
        const QTextBlock block = document()->findBlockByNumber(line);
        if (block.isValid()) {
            h->rehighlightBlock(block);
        }
    }
}

void QOwnNotesMarkdownTextEdit::refreshMarkdownLspDiagnostics() {
    applyMarkdownLspDiagnostics(filteredMarkdownLspDiagnostics(_markdownLspAllDiagnostics));
}

void QOwnNotesMarkdownTextEdit::applyMarkdownLspTextEdits(
    const QVector<MarkdownLspClient::TextEdit> &edits) {
    if (edits.isEmpty()) {
        return;
    }

    _markdownLspApplyingEdits = true;

    // Suppress the tracker so it doesn't record the programmatic edits
    if (_markdownLspTracker) {
        _markdownLspTracker->setSuppressed(true);
    }

    struct ResolvedEdit {
        int start = 0;
        int end = 0;
        QString text;
    };

    QVector<ResolvedEdit> resolved;
    resolved.reserve(edits.size());

    for (const MarkdownLspClient::TextEdit &edit : edits) {
        const QTextBlock startBlock = document()->findBlockByNumber(edit.range.startLine);
        const QTextBlock endBlock = document()->findBlockByNumber(edit.range.endLine);
        if (!startBlock.isValid() || !endBlock.isValid()) {
            continue;
        }

        const int startPosition = startBlock.position() + edit.range.startCharacter;
        const int endPosition = endBlock.position() + edit.range.endCharacter;
        if (endPosition < startPosition) {
            continue;
        }

        ResolvedEdit resolvedEdit;
        resolvedEdit.start = startPosition;
        resolvedEdit.end = endPosition;
        resolvedEdit.text = edit.newText;
        resolved.append(resolvedEdit);
    }

    std::sort(resolved.begin(), resolved.end(),
              [](const ResolvedEdit &left, const ResolvedEdit &right) {
                  if (left.start == right.start) {
                      return left.end > right.end;
                  }
                  return left.start > right.start;
              });

    // Clear the diagnostic cache before applying edits so that the rehighlight
    // triggered by endEditBlock() will not re-apply the old LSP underlines.
    _markdownLspDiagnostics.clear();

    if (auto *h = dynamic_cast<QOwnNotesMarkdownHighlighter *>(highlighter())) {
        h->clearMarkdownLspDiagnostics();
    }

    QTextCursor cursor(document());
    cursor.beginEditBlock();
    for (const ResolvedEdit &edit : resolved) {
        cursor.setPosition(edit.start);
        cursor.setPosition(edit.end, QTextCursor::KeepAnchor);
        cursor.insertText(edit.text);
    }
    cursor.endEditBlock();

    _markdownLspApplyingEdits = false;

    if (_markdownLspTracker) {
        _markdownLspTracker->setSuppressed(false);
    }

    if (_markdownLspEnabled && _markdownLspTracker) {
        // Quick-fixes invalidate the current diagnostics immediately, so refresh
        // the LSP state right away instead of waiting for the debounce timer.
        _markdownLspTracker->flushFullSync();
    }
}

void QOwnNotesMarkdownTextEdit::applyMarkdownLspFormatting(
    int requestId, const QVector<MarkdownLspClient::TextEdit> &edits) {
    if (requestId != _markdownLspFormattingRequestId &&
        requestId != _markdownLspRangeFormattingRequestId) {
        return;
    }

    if (requestId == _markdownLspFormattingRequestId) {
        _markdownLspFormattingRequestId = -1;
    }
    if (requestId == _markdownLspRangeFormattingRequestId) {
        _markdownLspRangeFormattingRequestId = -1;
    }

    applyMarkdownLspTextEdits(edits);
}

void QOwnNotesMarkdownTextEdit::requestMarkdownLspFormatting(bool useSelection) {
    if (!_markdownLspEnabled || !_markdownLspClient || _markdownLspUri.isEmpty()) {
        return;
    }

    const int tabSize = 4;
    const bool insertSpaces = true;

    if (useSelection) {
        QTextCursor cursor = textCursor();
        if (!cursor.hasSelection()) {
            return;
        }

        const int startPos = cursor.selectionStart();
        const int endPos = cursor.selectionEnd();

        QTextBlock startBlock = document()->findBlock(startPos);
        QTextBlock endBlock = document()->findBlock(endPos);
        if (!startBlock.isValid() || !endBlock.isValid()) {
            return;
        }

        MarkdownLspClient::DiagnosticRange range;
        range.startLine = startBlock.blockNumber();
        range.startCharacter = startPos - startBlock.position();
        range.endLine = endBlock.blockNumber();
        range.endCharacter = endPos - endBlock.position();

        _markdownLspRangeFormattingRequestId = _markdownLspClient->requestRangeFormatting(
            _markdownLspUri, range, tabSize, insertSpaces);
        return;
    }

    _markdownLspFormattingRequestId =
        _markdownLspClient->requestDocumentFormatting(_markdownLspUri, tabSize, insertSpaces);
}
