trilium/src/routes/api/sql.ts

"use strict";

import sql = require('../../services/sql');
import becca = require('../../becca/becca');
import { Request } from 'express';
import ValidationError = require('../../errors/validation_error');

function getSchema() {
    const tableNames = sql.getColumn(`SELECT name FROM sqlite_master WHERE type='table' AND name NOT LIKE 'sqlite_%' ORDER BY name`);
    const tables = [];

    for (const tableName of tableNames) {
        tables.push({
            name: tableName,
            columns: sql.getRows(`PRAGMA table_info(${tableName})`)
        });
    }

    return tables;
}

function execute(req: Request) {
    const note = becca.getNoteOrThrow(req.params.noteId);

    const content = note.getContent();
    if (typeof content !== "string") {
        throw new ValidationError("Invalid note type.");
    }

    const queries = content.split("\n---");

    try {
        const results = [];

        for (let query of queries) {
            query = query.trim();

            while (query.startsWith('-- ')) {
                // Query starts with one or more SQL comments, discard these before we execute.
                const pivot = query.indexOf('\n');
                query = pivot > 0 ? query.substr(pivot + 1).trim() : "";
            }
            
            if (!query) {
                continue;
            }

            if (query.toLowerCase().startsWith('select') || query.toLowerCase().startsWith('with')) {
                results.push(sql.getRows(query));
            }
            else {
                results.push(sql.execute(query));
            }
        }

        return {
            success: true,
            results
        };
    }
    catch (e: any) {
        return {
            success: false,
            error: e.message
        };
    }
}

export = {
    getSchema,
    execute
};