salix/print/core/database.js

const mysql = require('mysql2/promise');
const config = require('./config.js');
const fs = require('fs-extra');
const PromisePoolConnection = mysql.PromisePoolConnection;

module.exports = {
    init() {
        if (!this.pool) {
            this.pool = mysql.createPool(config.mysql);
            this.pool.on('connection', connection => {
                connection.config.namedPlaceholders = true;
            });
        }
    },

    /**
     * Makes a query from a raw sql
     * @param {String} query - The raw SQL query
     * @param {Object} params - Parameterized values
     * @param {Object} connection - Optional pool connection
     *
     * @return {Object} - Result promise
     */
    rawSql(query, params, connection) {
        let pool = this.pool;
        if (params instanceof PromisePoolConnection)
            connection = params;
        if (connection) pool = connection;

        return pool.query(query, params).then(([rows]) => {
            return rows;
        });
    },

    /**
     * Makes a query from a SQL file
     * @param {String} queryName - The SQL file name
     * @param {Object} params - Parameterized values
     * @param {Object} connection - Optional pool connection
     *
     * @return {Object} - Result promise
     */
    rawSqlFromDef(queryName, params, connection) {
        const query = fs.readFileSync(`${queryName}.sql`, 'utf8');

        return this.rawSql(query, params, connection);
    },

    /**
     * Returns the first row from a given raw sql
     * @param {String} query - The raw SQL query
     * @param {Object} params - Parameterized values
     *
     * @return {Object} - Result promise
     */
    findOne(query, params) {
        return this.rawSql(query, params).then(([row]) => row);
    },

    /**
     * Returns the first row from a given SQL file
     * @param {String} queryName - The SQL file name
     * @param {Object} params - Parameterized values
     *
     * @return {Object} - Result promise
     */
    findOneFromDef(queryName, params) {
        return this.rawSqlFromDef(queryName, params)
            .then(([row]) => row);
    },

    /**
     * Returns the first property from a given raw sql
     * @param {String} query - The raw SQL query
     * @param {Object} params - Parameterized values
     *
     * @return {Object} - Result promise
     */
    findValue(query, params) {
        return this.findOne(query, params).then(row => {
            return Object.values(row)[0];
        });
    },

    /**
     * Returns the first property from a given SQL file
     * @param {String} queryName - The SQL file name
     * @param {Object} params - Parameterized values
     *
     * @return {Object} - Result promise
     */
    findValueFromDef(queryName, params) {
        return this.findOneFromDef(queryName, params).then(row => {
            return Object.values(row)[0];
        });
    },

    /**
     * Returns the content of a SQL file
     * @param {String} queryName - The SQL file name
     *
     * @return {Object} - SQL
     */
    getSqlFromDef(queryName) {
        return fs.readFileSync(`${queryName}.sql`, 'utf8');
    },

    /**
     * Merges two SQL strings
     *
     * @param {String} query - Input SQL
     * @param {String} sql - String to merge
     *
     * @return {String} - Merged SQL string
     */
    merge(query, sql) {
        return query += `\r\n ${sql}`;
    },

    /**
     * Builds where string
     *
     * @param {Object} args - Query params
     * @param {Function} callback - Callback
     *
     * @return {String} - Where Sql string
     */
    buildWhere(args, callback) {
        let sql = '';

        for (let prop in args) {
            const value = args[prop];
            if (!value) continue;

            const sanitizedValue = mysql.escape(value);
            const conditional = callback(prop, sanitizedValue);

            if (!conditional) continue;

            if (sql) sql += ' AND ';
            sql += conditional;
        }
        let where = '';
        if (sql)
            where = `WHERE ${sql}`;

        return where;
    }
};