fluidb/cozo-lib-nodejs/index.js

/*
 * Copyright 2022, The Cozo Project Authors.
 *
 * This Source Code Form is subject to the terms of the Mozilla Public License, v. 2.0.
 * If a copy of the MPL was not distributed with this file,
 * You can obtain one at https://mozilla.org/MPL/2.0/.
 */

const binary = require('@mapbox/node-pre-gyp');
const path = require('path');
const binding_path = binary.find(path.resolve(path.join(__dirname, './package.json')));
const native = require(binding_path);

class CozoDb {
    constructor(engine, path, options) {
        this.db_id = native.open_db(engine || 'mem', path || 'data.db', JSON.stringify(options || {}))
    }

    close() {
        native.close_db(this.db_id)
    }

    run(script, params) {
        return new Promise((resolve, reject) => {
            params = params || {};
            native.query_db(this.db_id, script, params, (err, result) => {
                if (err) {
                    reject(JSON.parse(err))
                } else {
                    resolve(result)
                }
            })
        })
    }

    exportRelations(relations, as_objects) {
        return new Promise((resolve, reject) => {
            const rels_str = JSON.stringify({relations, as_objects: as_objects || false});
            native.export_relations(this.db_id, rels_str, (result_str) => {
                const result = JSON.parse(result_str);
                if (result.ok) {
                    resolve(result)
                } else {
                    reject(result)
                }
            })
        })
    }

    importRelations(data) {
        return new Promise((resolve, reject) => {
            const rels_str = JSON.stringify(data);
            native.import_relations(this.db_id, rels_str, (result_str) => {
                const result = JSON.parse(result_str);
                if (result.ok) {
                    resolve(result)
                } else {
                    reject(result)
                }
            })
        })
    }

    importRelationsFromBackup(path, rels) {
        return new Promise((resolve, reject) => {
            native.import_relations(this.db_id, path, JSON.stringify(rels), (result_str) => {
                const result = JSON.parse(result_str);
                if (result.ok) {
                    resolve(result)
                } else {
                    reject(result)
                }
            })
        })
    }

    backup(path) {
        return new Promise((resolve, reject) => {
            native.backup_db(this.db_id, path, (result_str) => {
                const result = JSON.parse(result_str);
                if (result.ok) {
                    resolve(result)
                } else {
                    reject(result)
                }
            })
        })
    }

    restore(path) {
        return new Promise((resolve, reject) => {
            native.restore_db(this.db_id, path, (result_str) => {
                const result = JSON.parse(result_str);
                if (result.ok) {
                    resolve(result)
                } else {
                    reject(result)
                }
            })
        })
    }

    register_callback(relation, cb, capacity = -1) {
        return native.register_callback(this.db_id, relation, cb, capacity)
    }

    unregister_callback(cb_id) {
        return native.unregister_callback(this.db_id, cb_id)
    }

    register_named_rule(name, arity, cb) {
        return native.register_named_rule(this.db_id, name, arity, async (ret_id, inputs, options) => {
            let ret = undefined;
            try {
                ret = await cb(inputs, options);
            } catch (e) {
                console.error(e);
                native.respond_to_named_rule_invocation(ret_id, '' + e);
                return;
            }
            try {
                native.respond_to_named_rule_invocation(ret_id, ret);
            } catch (e) {
                console.error(e);
            }
        })
    }

    unregister_named_rule(name) {
        return native.unregister_named_rule(this.db_id, name)
    }
}

module.exports = {CozoDb: CozoDb}