133 lines
4.0 KiB
JavaScript
133 lines
4.0 KiB
JavaScript
'use strict';
|
|
|
|
var _ = require('underscore'),
|
|
util = require('util'),
|
|
SORT_ORDER_OPTIONS = {ASC: 1, DESC: 1},
|
|
REGEX_SELECT = /^\s*SELECT\s/i,
|
|
REGEX_INTO = /\sINTO\s+([^\s]+|"([^"]|"")*")\s*$/i;
|
|
|
|
function PSQLWrapper(sql) {
|
|
this.sqlQuery = sql.replace(/;\s*$/, '');
|
|
this.sqlClauses = {
|
|
orderBy: '',
|
|
limit: ''
|
|
};
|
|
}
|
|
|
|
/**
|
|
* Only window select functions (NOTE: "values" will be broken, "with" will be broken)
|
|
*
|
|
* @param {number} limit
|
|
* @param {number} offset
|
|
* @returns {PSQLWrapper}
|
|
*/
|
|
PSQLWrapper.prototype.window = function (limit, offset) {
|
|
if (!_.isNumber(limit) || !_.isNumber(offset)) {
|
|
return this;
|
|
}
|
|
this.sqlClauses.limit = util.format(' LIMIT %d OFFSET %d', limit, offset);
|
|
return this;
|
|
};
|
|
|
|
/**
|
|
*
|
|
* @param {string} column The name of the column to sort by
|
|
* @param {string} sortOrder Whether it's ASC or DESC ordering
|
|
* @returns {PSQLWrapper}
|
|
*/
|
|
PSQLWrapper.prototype.orderBy = function (column, sortOrder) {
|
|
if (!_.isString(column) || _.isEmpty(column)) {
|
|
return this;
|
|
}
|
|
this.sqlClauses.orderBy = util.format(' ORDER BY "%s"', column);
|
|
|
|
if (!_.isUndefined(sortOrder)) {
|
|
sortOrder = sortOrder.toUpperCase();
|
|
if (SORT_ORDER_OPTIONS[sortOrder]) {
|
|
this.sqlClauses.orderBy += util.format(' %s', sortOrder);
|
|
}
|
|
}
|
|
return this;
|
|
};
|
|
|
|
/**
|
|
* Builds an SQL query with extra clauses based on the builder calls.
|
|
*
|
|
* @returns {string} The SQL query with the extra clauses
|
|
*/
|
|
PSQLWrapper.prototype.query = function () {
|
|
if (_.isEmpty(this.sqlClauses.orderBy) && _.isEmpty(this.sqlClauses.limit)) {
|
|
return this.sqlQuery;
|
|
}
|
|
// Strip comments
|
|
this.sqlQuery = this.sqlQuery.replace(/(^|\n)\s*--.*\n/g, '');
|
|
|
|
var cte = '';
|
|
|
|
if (this.sqlQuery.match(/^\s*WITH\s/i)) {
|
|
|
|
var rem = this.sqlQuery, // analyzed portion of sql
|
|
q, // quote char
|
|
n = 0, // nested parens level
|
|
s = 0, // 0:outQuote, 1:inQuote
|
|
l;
|
|
while (1) {
|
|
l = rem.search(/[("')]/);
|
|
// console.log("REM Is " + rem);
|
|
if (l < 0) {
|
|
// console.log("Malformed SQL");
|
|
return this.sqlQuery;
|
|
}
|
|
var f = rem.charAt(l);
|
|
// console.log("n:" + n + " s:" + s + " l:" + l + " charAt(l):" + f + " charAt(l+1):" + rem.charAt(l+1));
|
|
if (s == 0) {
|
|
if (f == '(') {
|
|
++n;
|
|
} else if (f == ')') {
|
|
if (!--n) { // end of CTE
|
|
cte += rem.substr(0, l + 1);
|
|
rem = rem.substr(l + 1);
|
|
//console.log("Out of cte, rem is " + rem);
|
|
if (rem.search(/^s*,/) < 0) {
|
|
break;
|
|
} else {
|
|
continue; // cte and rem already updated
|
|
}
|
|
}
|
|
} else { // enter quoting
|
|
s = 1;
|
|
q = f;
|
|
}
|
|
} else if (f == q) {
|
|
if (rem.charAt(l + 1) == f) {
|
|
++l; // escaped
|
|
} else {
|
|
s = 0; // exit quoting
|
|
}
|
|
}
|
|
cte += rem.substr(0, l + 1);
|
|
rem = rem.substr(l + 1);
|
|
}
|
|
/*
|
|
console.log("cte: " + cte);
|
|
console.log("rem: " + rem);
|
|
*/
|
|
this.sqlQuery = rem; //sql.substr(l+1);
|
|
}
|
|
|
|
//console.log("SQL " + sql);
|
|
//console.log(" does " + ( sql.match(REGEX_SELECT) ? '' : 'not ' ) + "match REGEX_SELECT " + REGEX_SELECT);
|
|
//console.log(" does " + ( sql.match(REGEX_INTO) ? '' : 'not ' ) + "match REGEX_INTO " + REGEX_INTO);
|
|
|
|
if (this.sqlQuery.match(REGEX_SELECT) && !this.sqlQuery.match(REGEX_INTO)) {
|
|
return util.format(
|
|
'%sSELECT * FROM (%s) AS cdbq_1%s%s',
|
|
cte, this.sqlQuery, this.sqlClauses.orderBy, this.sqlClauses.limit
|
|
);
|
|
}
|
|
|
|
return cte + this.sqlQuery;
|
|
};
|
|
|
|
module.exports = PSQLWrapper;
|