150 lines
4.0 KiB
JavaScript
150 lines
4.0 KiB
JavaScript
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;
|
|
}
|
|
};
|