trilium/src/public/javascripts/dialogs/sql_console.js

"use strict";

const sqlConsole = (function() {
    const $dialog = $("#sql-console-dialog");
    const $query = $('#sql-console-query');
    const $executeButton = $('#sql-console-execute');
    const $resultHead = $('#sql-console-results thead');
    const $resultBody = $('#sql-console-results tbody');

    let codeEditor;

    function showDialog() {
        glob.activeDialog = $dialog;

        $dialog.dialog({
            modal: true,
            width: $(window).width(),
            height: $(window).height(),
            open: function() {
                initEditor();
            }
        });
    }

    async function initEditor() {
        if (!codeEditor) {
            await requireLibrary(CODE_MIRROR);

            CodeMirror.keyMap.default["Shift-Tab"] = "indentLess";
            CodeMirror.keyMap.default["Tab"] = "indentMore";

            // removing Escape binding so that Escape will propagate to the dialog (which will close on escape)
            delete CodeMirror.keyMap.basic["Esc"];

            CodeMirror.modeURL = 'libraries/codemirror/mode/%N/%N.js';

            codeEditor = CodeMirror($query[0], {
                value: "",
                viewportMargin: Infinity,
                indentUnit: 4,
                highlightSelectionMatches: {showToken: /\w/, annotateScrollbar: false}
            });

            codeEditor.setOption("mode", "text/x-sqlite");
            CodeMirror.autoLoadMode(codeEditor, "sql");
        }

        codeEditor.focus();
    }

    async function execute(e) {
        // stop from propagating upwards (dangerous especially with ctrl+enter executable javascript notes)
        e.preventDefault();
        e.stopPropagation();

        const sqlQuery = codeEditor.getValue();

        const result = await server.post("sql/execute", {
            query: sqlQuery
        });

        if (!result.success) {
            showError(result.error);
            return;
        }
        else {
            showMessage("Query was executed successfully.");
        }

        const rows = result.rows;

        $resultHead.empty();
        $resultBody.empty();

        if (rows.length > 0) {
            const result = rows[0];
            const rowEl = $("<tr>");

            for (const key in result) {
                rowEl.append($("<th>").html(key));
            }

            $resultHead.append(rowEl);
        }

        for (const result of rows) {
            const rowEl = $("<tr>");

            for (const key in result) {
                rowEl.append($("<td>").html(result[key]));
            }

            $resultBody.append(rowEl);
        }
    }

    $(document).bind('keydown', 'alt+o', showDialog);

    $query.bind('keydown', 'ctrl+return', execute);

    $executeButton.click(execute);

    return {
        showDialog
    };
})();