CartoDB-SQL-API/app/sql/psql_wrapper.js

'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;