neochat/src/libneochat/chatmarkdownhelper.cpp

// SPDX-FileCopyrightText: 2025 James Graham <james.h.graham@protonmail.com>
// SPDX-License-Identifier: GPL-2.0-only OR GPL-3.0-only OR LicenseRef-KDE-Accepted-GPL

#include "chatmarkdownhelper.h"

#include <QTextCursor>
#include <QTextDocument>
#include <qtextcursor.h>

#include "chattextitemhelper.h"
#include "richformat.h"

namespace
{
struct MarkdownSyntax {
    QLatin1String sequence;
    bool closable = false;
    bool lineStart = false;
    RichFormat::Format format;
};

const QList<MarkdownSyntax> syntax = {
    MarkdownSyntax{.sequence = "*"_L1, .closable = true, .format = RichFormat::Italic},
    MarkdownSyntax{.sequence = "**"_L1, .closable = true, .format = RichFormat::Bold},
    MarkdownSyntax{.sequence = "# "_L1, .lineStart = true, .format = RichFormat::Heading1},
    MarkdownSyntax{.sequence = "## "_L1, .lineStart = true, .format = RichFormat::Heading2},
    MarkdownSyntax{.sequence = "### "_L1, .lineStart = true, .format = RichFormat::Heading3},
    MarkdownSyntax{.sequence = "#### "_L1, .lineStart = true, .format = RichFormat::Heading4},
    MarkdownSyntax{.sequence = "##### "_L1, .lineStart = true, .format = RichFormat::Heading5},
    MarkdownSyntax{.sequence = "###### "_L1, .lineStart = true, .format = RichFormat::Heading6},
    MarkdownSyntax{.sequence = "> "_L1, .lineStart = true, .format = RichFormat::Quote},
    MarkdownSyntax{.sequence = "* "_L1, .lineStart = true, .format = RichFormat::UnorderedList},
    MarkdownSyntax{.sequence = "- "_L1, .lineStart = true, .format = RichFormat::UnorderedList},
    MarkdownSyntax{.sequence = "1. "_L1, .lineStart = true, .format = RichFormat::OrderedList},
    MarkdownSyntax{.sequence = "1) "_L1, .lineStart = true, .format = RichFormat::OrderedList},
    MarkdownSyntax{.sequence = "`"_L1, .closable = true, .format = RichFormat::InlineCode},
    MarkdownSyntax{.sequence = "```"_L1, .lineStart = true, .format = RichFormat::Code},
    MarkdownSyntax{.sequence = "~~"_L1, .closable = true, .format = RichFormat::Strikethrough},
    MarkdownSyntax{.sequence = "__"_L1, .closable = true, .format = RichFormat::Underline},
};

std::optional<bool> checkSequence(const QString &currentString, const QString &nextChar, bool lineStart = false)
{
    QList<MarkdownSyntax> partialMatches;
    std::optional<MarkdownSyntax> fullMatch = std::nullopt;
    auto it = syntax.cbegin();
    while ((it = std::find_if(it,
                              syntax.cend(),
                              [currentString, nextChar](const MarkdownSyntax &syntax) {
                                  return syntax.sequence == currentString || syntax.sequence.startsWith(QString(currentString + nextChar));
                              }))
           != syntax.cend()) {
        if (it->lineStart ? lineStart : true) {
            if (it->sequence == currentString) {
                fullMatch = *it;
            } else {
                partialMatches += *it;
            }
        }
        ++it;
    }

    if (partialMatches.length() > 0) {
        return false;
    }
    if (fullMatch) {
        return true;
    }
    return std::nullopt;
}

std::optional<MarkdownSyntax> syntaxForSequence(const QString &sequence)
{
    const auto it = std::find_if(syntax.cbegin(), syntax.cend(), [sequence](const MarkdownSyntax &syntax) {
        return syntax.sequence == sequence;
    });
    if (it == syntax.cend()) {
        return std::nullopt;
    }
    return *it;
}
}

ChatMarkdownHelper::ChatMarkdownHelper(QObject *parent)
    : QObject(parent)
{
}

ChatTextItemHelper *ChatMarkdownHelper::textItem() const
{
    return m_textItem;
}

void ChatMarkdownHelper::setTextItem(ChatTextItemHelper *textItem)
{
    if (textItem == m_textItem) {
        return;
    }

    if (m_textItem) {
        m_textItem->disconnect(this);
    }

    m_textItem = textItem;

    if (m_textItem) {
        connect(m_textItem, &ChatTextItemHelper::textItemChanged, this, &ChatMarkdownHelper::textItemChanged);
        connect(m_textItem, &ChatTextItemHelper::textItemChanged, this, [this]() {
            m_startPos = m_textItem->cursorPosition();
            m_endPos = m_startPos;
            if (m_startPos == 0) {
                m_currentState = Pre;
            }
        });
        connect(m_textItem, &ChatTextItemHelper::contentsChange, this, &ChatMarkdownHelper::checkMarkdown);
    }

    Q_EMIT textItemChanged();
}

void ChatMarkdownHelper::checkMarkdown(int position, int charsRemoved, int charsAdded)
{
    auto cursor = m_textItem->textCursor();
    if (cursor.isNull()) {
        return;
    }

    if (charsRemoved - charsAdded > 0) {
        if (position < m_startPos) {
            m_startPos = position;
        }
        m_endPos -= charsRemoved;
        cursor.setPosition(m_endPos);
        cursor.setPosition(m_endPos + (cursor.atBlockEnd() ? 0 : 1), QTextCursor::KeepAnchor);
        const auto nextChar = cursor.selectedText();
        m_currentState = m_startPos == 0 || nextChar == u' ' ? Pre : None;
        return;
    }

    for (auto i = 1; i <= charsAdded - charsRemoved; ++i) {
        cursor.setPosition(m_startPos);
        cursor.setPosition(m_endPos, QTextCursor::KeepAnchor);
        const auto currentMarkdown = cursor.selectedText();
        cursor.setPosition(m_endPos);
        cursor.setPosition(m_endPos + 1, QTextCursor::KeepAnchor);
        const auto nextChar = cursor.selectedText();
        cursor.setPosition(m_startPos);

        const auto result = checkSequence(currentMarkdown, nextChar, cursor.atBlockStart());

        switch (m_currentState) {
        case None:
            if (nextChar == u' ' || cursor.atBlockEnd()) {
                m_currentState = Pre;
            }
            ++m_startPos;
            m_endPos = m_startPos;
            break;
        case Pre:
            if (!result && RichFormat::formatsAtCursor(cursor).length() == 0) {
                m_currentState = None;
            } else if (result && !*result) {
                m_currentState = Started;
                ++m_endPos;
                break;
            }
            ++m_startPos;
            m_endPos = m_startPos;
            break;
        case Started:
            if (!result) {
                m_currentState = Pre;
                ++m_startPos;
                m_endPos = m_startPos;
                break;
            } else if (!*result) {
                ++m_endPos;
                break;
            }
            complete();
            break;
        }
    }
}

void ChatMarkdownHelper::complete()
{
    auto cursor = m_textItem->textCursor();
    if (cursor.isNull()) {
        return;
    }
    cursor.beginEditBlock();
    cursor.setPosition(m_startPos);
    cursor.setPosition(m_endPos, QTextCursor::KeepAnchor);
    const auto syntax = syntaxForSequence(cursor.selectedText());
    cursor.removeSelectedText();

    if (m_currentFormats.contains(syntax->format)) {
        m_currentFormats.remove(syntax->format);
    } else if (syntax->closable) {
        m_currentFormats.insert(syntax->format, m_startPos);
    }

    cursor.setPosition(m_startPos);
    cursor.setPosition(m_startPos + 1, QTextCursor::KeepAnchor);
    const auto nextChar = cursor.selectedText();
    const auto result = checkSequence({}, nextChar, cursor.atBlockStart());
    m_currentState = result ? Started : Pre;

    // cursor.setPosition(m_startPos + 1);
    cursor.movePosition(QTextCursor::NextCharacter, QTextCursor::KeepAnchor);
    if (syntax) {
        const auto formatType = RichFormat::typeForFormat(syntax->format);
        if (formatType == RichFormat::Block) {
            Q_EMIT unhandledBlockFormat(syntax->format);
        } else {
            m_textItem->mergeFormatOnCursor(syntax->format, cursor);
        }
    }

    m_startPos = result ? m_startPos : m_startPos + 1;
    m_endPos = result ? m_startPos + 1 : m_startPos;

    cursor.endEditBlock();
}

void ChatMarkdownHelper::handleExternalFormatChange()
{
    auto cursor = m_textItem->textCursor();
    if (cursor.isNull()) {
        return;
    }
    cursor.setPosition(m_startPos);
    m_currentState = RichFormat::formatsAtCursor(cursor).length() > 0 ? Pre : None;
}

#include "moc_chatmarkdownhelper.cpp"