node-query/lib/Driver.js

'use strict';

let helpers = require('./helpers');

/**
 * Base Database Driver
 *
 * @module driver
 */
let d = {
	identifierStartChar: '"',
	identifierEndChar: '"',
	tablePrefix: null,
	hasTruncate: true,

	/**
	 * Low level function for naive quoting of strings

	 * @param {String} str - The sql fragment to quote
	 * @return {String} - The quoted sql fragment
	 * @private
	 */
	_quote(str) {
		return (helpers.isString(str) && ! (str.startsWith(d.identifierStartChar) || str.endsWith(d.identifierEndChar)))
			? `${d.identifierStartChar}${str}${d.identifierEndChar}`
			: str;
	},

	/**
	 * Set the limit clause

	 * @param {String} sql - SQL statement to modify
	 * @param {Number} limit - Maximum number of rows to fetch
	 * @param {Number|null} offset - Number of rows to skip
	 * @return {String} - Modified SQL statement
	 */
	limit(sql, limit, offset) {
		sql +=  ` LIMIT ${limit}`;

		if (helpers.isNumber(offset))
		{
			sql += ` OFFSET ${offset}`;
		}

		return sql;
	},

	/**
	 * Quote database table name, and set prefix
	 *
	 * @param {String} table - Table name to quote
	 * @return {String} - Quoted table name
	 */
	quoteTable(table) {
		// Quote after prefix
		return d.quoteIdentifiers(table);
	},

	/**
	 * Use the driver's escape character to quote identifiers
	 *
	 * @param {String|String[]} - String or array of strings to quote identifiers
	 * @return {String|String[]} - Quoted identifier(s)
	 */
	quoteIdentifiers(str) {
		let hiers, raw;
		let pattern = new RegExp(
			`${d.identifierStartChar}(`
				+ '([a-zA-Z0-9_]+)' + '(\((.*?)\))'
				+ `)${d.identifierEndChar}`, 'ig');

		// Recurse for arrays of identifiiers
		if (Array.isArray(str))
		{
			return str.map(d.quoteIdentifiers);
		}

		// Handle commas
		if (str.includes(','))
		{
			let parts = str.split(',').map(helpers.stringTrim);
			str = parts.map(d.quoteIdentifiers).join(',');
		}

		// Split identifiers by period
		hiers = str.split('.').map(d._quote);
		raw = hiers.join('.');

		// Fix functions
		if (raw.includes('(') && raw.includes(')'))
		{
			let funcs = pattern.exec(raw);

			// Unquote the function
			raw = raw.replace(funcs[0], funcs[1]);

			// Quote the identifiers inside of the parens
			let inParens = funcs[3].substring(1, funcs[3].length - 1);
			raw = raw.replace(inParens, d.quoteIdentifiers(inParens));
		}

		return raw;
	},

	/**
	 * SQL to truncate the passed table
	 *
	 * @param {String} table - Table to truncate
	 * @return {String} - Truncation SQL
	 */
	truncate(table) {
		let sql = (d.hasTruncate)
			? 'TRUNCATE '
			: 'DELETE FROM ';

		sql += d.quoteTable(table);

		return sql;
	},

	/**
	 * SQL to insert a group of rows
	 *
	 * @param {String} table - The table to insert to
	 * @param {Array} [data] - The array of object containing data to insert
	 * @return {String} - Query and data to insert
	 */
	insertBatch(table, data) {
		let vals = [],
			fields = Object.keys(data[0]),
			sql = '',
			params = [],
			paramString = '',
			paramList = [];

		// Get the data values to insert, so they can
		// be parameterized
		data.forEach(obj => {
			Object.keys(obj).forEach(key => {
				vals.push(obj[key]);
			});
		});

		// Get the field names from the keys of the first
		// object inserted
		table = d.quoteTable(table);

		sql += `INSERT INTO ${table} (${d.quoteIdentifiers(fields).join(',')}) VALUES `;

		// Create placeholder groups
		params = Array(fields.length).fill('?');
		paramString = `(${params.join(',')})`;
		paramList = Array(data.length).fill(paramString);

		sql += paramList.join(',');

		return {
			sql: sql,
			values: vals,
		};
	},
};

module.exports = d;