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 * * @return {Object} - Result promise */ rawSqlFromDef(queryName, params) { const query = fs.readFileSync(`${queryName}.sql`, 'utf8'); return this.rawSql(query, params); }, /** * 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; } };