riot-katex/riot-katex.js

(function() {
    /**
     * Check and set a global guard variable.
     * If this content script is injected into the same page again,
     * it will do nothing next time.
     */
    if (window.hasRun) {
        return;
    }
    window.hasRun = true;

    /**
     * Set up math delimiters.
     */
    const math_config = [
        {left: "$$", right: "$$", display: true},
        {left: "\\(", right: "\\)", display: false},
        {left: "\\[", right: "\\]", display: true}
    ]

    /**
     * Render math in a node if it hasn't already be done
     * and register an event on the edit button.
     */
    function renderMath(node) {
        let li = node.closest('li');

        // we have already rendered this element
        if (li.getAttribute('originalContent')) {
            return;
        }

        let og_content = (' ' + node.textContent).slice(1);
        li.setAttribute('originalContent', og_content);
        li.querySelector('.mx_MessageActionBar_maskButton[title="Edit"]')
            .addEventListener("click", event => {
                event.target.closest('li').setAttribute('originalContent', '');
                node.textContent = og_content;
            });
        renderMathInElement(node);
    }

    /**
     * Initialize Extension as soon as matrix has been loaded.
     */
    function init() {
        // render with KaTeX as soon as the message appears
        function listen_on_chat_element() {
            const chat_elem = document.querySelector('.mx_RoomView_MessageList');
            for(let node of chat_elem.querySelectorAll('.mx_MTextBody')) {
                renderMath(node);
            }

            const callback = function(mutationsList, observer) {
                for(let mutation of mutationsList) {
                    if(mutation.type === 'attributes' && mutation.attributeName === 'class') {
                        let node = mutation.target.querySelector('.mx_MTextBody');
                        renderMath(node);
                    }

                    if (mutation.type === 'childList' && mutation.addedNodes.length > 0) {
                        for(let node of mutation.addedNodes) {
                            node = node.querySelector('.mx_MTextBody');
                            renderMath(node);
                        }
                    }
                }
            };

            const observer = new MutationObserver(callback);
            observer.observe(chat_elem, { attributes: true, childList: true, subtree: true });
            return observer;
        }

        let last_title = ''
        const header = document.querySelector('title');
        // when changing the chat room, we have to create a new listener
        function change_chat_element(mutationsList, observer) {
            if(last_title == header.textContent)
                return;

            last_title = header.textContent;
            if (chat_observer) {
                chat_observer.disconnect();
            }

            chat_observer = listen_on_chat_element();
        }

        // start listening
        let chat_observer = listen_on_chat_element();

        // start listening on the header as well
        const header_obs = new MutationObserver(change_chat_element)
        header_obs.observe(header, { attributes: true, childList: true, subtree: true });
    }

    // a clever hack to check if riot has been loaded yet
    function wait_for_matrix() {
        if(!('matrixChat' in window.wrappedJSObject
             && window.wrappedJSObject.matrixChat.firstSyncComplete))
            return setTimeout(wait_for_matrix, 500);

        init();
    }

    wait_for_matrix();
})();