loopback-datasource-juggler/lib/sql.js

module.exports = BaseSQL;

/**
 * Base SQL class
 */
function BaseSQL() {
}

BaseSQL.prototype.relational = true;

BaseSQL.prototype.query = function () {
    throw new Error('query method should be declared in connector');
};

BaseSQL.prototype.command = function (sql, callback) {
    return this.query(sql, callback);
};

BaseSQL.prototype.queryOne = function (sql, callback) {
    return this.query(sql, function (err, data) {
        if (err) return callback(err);
        callback(err, data[0]);
    });
};

/**
 * Look up the data source by model name
 * @param model The model name
 * @returns {DataSource} The data source
 */
BaseSQL.prototype.getDataSource = function(model) {
    var m = this._models[model];
    if(!m) {
        console.trace('Model not found: ' + model);
    }
    return m.model.dataSource;
}

/**
 * Get the table name for a given model
 * @param model The model name
 * @returns {String} The table name
 */
BaseSQL.prototype.table = function (model) {
    var name = this.getDataSource(model).tableName(model);
    var dbName = this.dbName;
    if(typeof dbName === 'function') {
        name = dbName(name);
    }
    return name;
};

/**
 * Get the column name for given model property
 * @param model The model name
 * @param property The property name
 * @returns {String} The column name
 */
BaseSQL.prototype.column = function (model, property) {
    var name = this.getDataSource(model).columnName(model, property);
    var dbName = this.dbName;
    if(typeof dbName === 'function') {
        name = dbName(name);
    }
    return name;
};

/**
 * Get the column name for given model property
 * @param model The model name
 * @param property The property name
 * @returns {Object} The column metadata
 */
BaseSQL.prototype.columnMetadata = function (model, property) {
    return this.getDataSource(model).columnMetadata(model, property);
};

/**
 * Get the corresponding property name for a given column name
 * @param model The model name
 * @param column The column name
 * @returns {*}
 */
BaseSQL.prototype.propertyName = function (model, column) {
    var props = this._models[model].properties;
    for(var p in props) {
        if(this.column(model, p) === column) {
            return p;
        }
    }
    return null;
};

/**
 * Get the id property name
 * @param model The model name
 * @returns {String} The id property name
 */
BaseSQL.prototype.idName = function (model) {
    return this.getDataSource(model).idName(model);
};

/**
 * Get the id property names
 * @param model The model name
 * @returns {[String]} The id property names
 */
BaseSQL.prototype.idNames = function (model) {
    return this.getDataSource(model).idNames(model);
};

/**
 * Get the id column name
 * @param model The model name
 * @returns {String} The column name
 */
BaseSQL.prototype.idColumn = function (model) {
    var name = this.getDataSource(model).idColumnName(model);;
    var dbName = this.dbName;
    if(typeof dbName === 'function') {
        name = dbName(name);
    }
    return name;
};

/**
 * Get the escaped id column name
 * @param model The model name
 * @returns {String} the escaped id column name
 */
BaseSQL.prototype.idColumnEscaped = function (model) {
    return this.escapeName(this.getDataSource(model).idColumnName(model));
};

/**
 * Get the id index (sequence number, starting from 1)
 * @param model The model name
 * @param prop The property name
 * @returns {Number} The id index, undefined if the property is not part of the primary key
 */
BaseSQL.prototype.id = function (model, prop) {
    var p = this._models[model].properties[prop];
    if(!p) {
        console.trace('Property not found: ' + model +'.' + prop);
    }
    return p.id;
};

/**
 * Escape the name for the underlying database
 * @param name The name
 */
BaseSQL.prototype.escapeName = function (name) {
    throw new Error('escapeName method should be declared in connector');
};

/**
 * Get the escaped table name
 * @param model The model name
 * @returns {String} the escaped table name
 */
BaseSQL.prototype.tableEscaped = function (model) {
    return this.escapeName(this.table(model));
};

/**
 * Get the escaped column name for a given model property
 * @param model The model name
 * @param property The property name
 * @returns {String} The escaped column name
 */
BaseSQL.prototype.columnEscaped = function (model, property) {
    return this.escapeName(this.column(model, property));
};

BaseSQL.prototype.define = function (descr) {
    if (!descr.settings) descr.settings = {};
    this._models[descr.model.modelName] = descr;
};

BaseSQL.prototype.defineProperty = function (model, prop, params) {
    this._models[model].properties[prop] = params;
};

BaseSQL.prototype.save = function (model, data, callback) {
    var sql = 'UPDATE ' + this.tableEscaped(model) + ' SET '
        + this.toFields(model, data) + ' WHERE ' + this.idColumnEscaped(model) + ' = ' + Number(data.id);

    this.query(sql, function (err) {
        callback(err);
    });
};


BaseSQL.prototype.exists = function (model, id, callback) {
    var sql = 'SELECT 1 FROM ' +
        this.tableEscaped(model) + ' WHERE ' + this.idColumnEscaped(model) + ' = ' + Number(id) + ' LIMIT 1';

    this.query(sql, function (err, data) {
        if (err) return callback(err);
        callback(null, data.length === 1);
    });
};

BaseSQL.prototype.find = function find(model, id, callback) {
    var sql = 'SELECT * FROM ' +
        this.tableEscaped(model) + ' WHERE ' + this.idColumnEscaped(model) + ' = ' + id + ' LIMIT 1';

    this.query(sql, function (err, data) {
        if (data && data.length === 1) {
            data[0].id = id;
        } else {
            data = [null];
        }
        callback(err, this.fromDatabase(model, data[0]));
    }.bind(this));
};

BaseSQL.prototype.delete = BaseSQL.prototype.destroy = function destroy(model, id, callback) {
    var sql = 'DELETE FROM ' +
        this.tableEscaped(model) + ' WHERE ' + this.idColumnEscaped(model) + ' = ' + id;

    this.command(sql, function (err) {
        callback(err);
    });
};

BaseSQL.prototype.deleteAll = BaseSQL.prototype.destroyAll = function destroyAll(model, callback) {
    this.command('DELETE FROM ' + this.tableEscaped(model), function (err) {
        if (err) {
            return callback(err, []);
        }
        callback(err);
    }.bind(this));
};

BaseSQL.prototype.count = function count(model, callback, where) {
    var self = this;
    var props = this._models[model].properties;

    this.queryOne('SELECT count(*) as cnt FROM ' +
        this.tableEscaped(model) + ' ' + buildWhere(where), function (err, res) {
        if (err) return callback(err);
        callback(err, res && res.cnt);
    });

    function buildWhere(conds) {
        var cs = [];
        Object.keys(conds || {}).forEach(function (key) {
            var keyEscaped = self.columnEscaped(model, key);
            if (conds[key] === null) {
                cs.push(keyEscaped + ' IS NULL');
            } else {
                cs.push(keyEscaped + ' = ' + self.toDatabase(props[key], conds[key]));
            }
        });
        return cs.length ? ' WHERE ' + cs.join(' AND ') : '';
    }
};

BaseSQL.prototype.updateAttributes = function updateAttrs(model, id, data, cb) {
    data.id = id;
    this.save(model, data, cb);
};

BaseSQL.prototype.disconnect = function disconnect() {
    this.client.end();
};

/**
 * Recreate the tables for the given models
 * @param models A model name or an array of model names, if not present, apply to all models defined in the connector
 * @param cb Callback function
 */
BaseSQL.prototype.automigrate = function (models, cb) {
    var self = this;
    var wait = 0;
    if ((!cb) && ('function' === typeof models)) {
        cb = models;
        models = undefined;
    }
    // First argument is a model name
    if ('string' === typeof models) {
        models = [models];
    }

    models = models || Object.keys(this._models);
    models.forEach(function (model) {
        if (model in self._models) {
            wait++;
            self.dropTable(model, function () {
                // console.log('drop', model);
                self.createTable(model, function (err) {
                    // console.log('create', model);
                    if (err) console.log(err);
                    done();
                });
            });
        }
    });
    if (wait === 0) cb();

    function done() {
        if (--wait === 0 && cb) {
            cb();
        }
    }
};

BaseSQL.prototype.dropTable = function (model, cb) {
    this.command('DROP TABLE IF EXISTS ' + this.tableEscaped(model), cb);
};

BaseSQL.prototype.createTable = function (model, cb) {
    this.command('CREATE TABLE ' + this.tableEscaped(model) +
        ' (\n  ' + this.propertiesSQL(model) + '\n)', cb);
};