2014-04-15 13:10:32 -04:00
|
|
|
// SQLite3 Query Builder & Compiler
|
|
|
|
// -------
|
2014-04-08 16:25:57 -04:00
|
|
|
module.exports = function(client) {
|
|
|
|
|
|
|
|
var _ = require('lodash');
|
|
|
|
var inherits = require('inherits');
|
|
|
|
|
|
|
|
var QueryBuilder = require('../../query/builder');
|
|
|
|
var QueryCompiler = require('../../query/compiler');
|
|
|
|
|
2014-04-15 13:10:32 -04:00
|
|
|
// Query Builder
|
|
|
|
// -------
|
2014-04-08 16:25:57 -04:00
|
|
|
function QueryBuilder_SQLite3() {
|
2014-04-15 11:43:47 -04:00
|
|
|
this.client = client;
|
2014-04-08 16:25:57 -04:00
|
|
|
QueryBuilder.apply(this, arguments);
|
|
|
|
}
|
|
|
|
inherits(QueryBuilder_SQLite3, QueryBuilder);
|
|
|
|
|
2014-04-15 13:10:32 -04:00
|
|
|
// Query Compiler
|
|
|
|
// -------
|
2014-04-08 16:25:57 -04:00
|
|
|
function QueryCompiler_SQLite3() {
|
|
|
|
this.formatter = new client.Formatter();
|
|
|
|
QueryCompiler.apply(this, arguments);
|
|
|
|
}
|
|
|
|
inherits(QueryCompiler_SQLite3, QueryCompiler);
|
|
|
|
|
2014-04-15 11:43:47 -04:00
|
|
|
// The locks are not applicable in SQLite3
|
2014-04-08 16:25:57 -04:00
|
|
|
QueryCompiler_SQLite3.prototype.forShare =
|
|
|
|
QueryCompiler_SQLite3.prototype.forUpdate = function() {
|
|
|
|
return '';
|
|
|
|
};
|
|
|
|
|
2014-04-15 11:43:47 -04:00
|
|
|
// SQLite requires us to build the multi-row insert as a listing of select with
|
|
|
|
// unions joining them together. So we'll build out this list of columns and
|
|
|
|
// then join them all together with select unions to complete the queries.
|
2014-04-08 16:25:57 -04:00
|
|
|
QueryCompiler_SQLite3.prototype.insert = function() {
|
|
|
|
var insert = this.single.insert;
|
|
|
|
var sql = 'insert into ' + this.tableName + ' ';
|
2014-04-15 11:43:47 -04:00
|
|
|
if (_.isEmpty(this.single.insert)) return sql + 'default values';
|
|
|
|
var insertData = this._prepInsert(insert);
|
|
|
|
sql += '(' + this.formatter.columnize(insertData.columns) + ')';
|
|
|
|
if (insertData.values.length === 1) {
|
2014-04-16 01:23:50 -04:00
|
|
|
return sql + ' values (' + this.formatter.parameterize(insertData.values[0]) + ')';
|
2014-04-08 16:25:57 -04:00
|
|
|
}
|
|
|
|
var blocks = [];
|
2014-04-15 11:43:47 -04:00
|
|
|
for (var i = 0, l = insertData.values.length; i < l; i++) {
|
2014-04-08 16:25:57 -04:00
|
|
|
var block = blocks[i] = [];
|
2014-04-15 11:43:47 -04:00
|
|
|
var current = insertData.values[i];
|
|
|
|
for (var i2 = 0, l2 = insertData.columns.length; i2 < l2; i2++) {
|
|
|
|
block.push(this.formatter.parameter(current[i2]) + ' as ' + this.formatter.wrap(insertData.columns[i2]));
|
2014-04-08 16:25:57 -04:00
|
|
|
}
|
|
|
|
blocks[i] = block.join(', ');
|
|
|
|
}
|
|
|
|
return sql + ' select ' + blocks.join(' union all select ');
|
|
|
|
};
|
|
|
|
|
|
|
|
// Adds a `order by` clause to the query, using "collate nocase" for the sort.
|
2014-04-15 13:10:32 -04:00
|
|
|
QueryCompiler_SQLite3.prototype.order = function() {
|
|
|
|
var orders = this.grouped.order;
|
2014-04-16 01:23:50 -04:00
|
|
|
if (!orders) return '';
|
2014-04-15 13:10:32 -04:00
|
|
|
return _.map(orders, function(order) {
|
|
|
|
var cols = _.isArray(order.value) ? order.value : [order.value];
|
|
|
|
return 'order by ' + this.formatter.columnize(cols) + ' collate nocase ' + this.formatter.direction(order.direction);
|
|
|
|
}, this);
|
2014-04-08 16:25:57 -04:00
|
|
|
};
|
|
|
|
|
|
|
|
// Compiles an `update` query.
|
|
|
|
QueryCompiler_SQLite3.prototype.update = function() {
|
2014-04-16 01:23:50 -04:00
|
|
|
var updateData = this._prepUpdate(this.single.update);
|
|
|
|
var joins = this.join();
|
|
|
|
var wheres = this.where();
|
|
|
|
return 'update ' + this.tableName +
|
|
|
|
(joins ? ' ' + joins : '') +
|
|
|
|
' set ' + updateData.join(', ') +
|
|
|
|
(wheres ? ' ' + wheres : '');
|
2014-04-08 16:25:57 -04:00
|
|
|
};
|
|
|
|
|
|
|
|
// Compile a truncate table statement into SQL.
|
|
|
|
QueryCompiler_SQLite3.prototype.truncate = function() {
|
|
|
|
var table = this.tableName;
|
|
|
|
return {
|
|
|
|
sql: 'delete from sqlite_sequence where name = ' + this.tableName,
|
|
|
|
output: function() {
|
|
|
|
return this.query({sql: 'delete from ' + table});
|
|
|
|
}
|
|
|
|
};
|
|
|
|
};
|
|
|
|
|
2014-04-21 23:08:59 -04:00
|
|
|
// Compiles a `columnInfo` query
|
|
|
|
QueryCompiler_SQLite3.prototype.columnInfo = function() {
|
|
|
|
return {
|
|
|
|
sql: 'PRAGMA table_info(' + this.single.table +')',
|
|
|
|
output: function(resp) {
|
|
|
|
var maxLengthRegex = /.*\((\d+)\)/;
|
|
|
|
return _.reduce(resp, function (columns, val) {
|
|
|
|
var type = val.type;
|
|
|
|
var maxLength = (maxLength = type.match(maxLengthRegex)) && maxLength[1];
|
|
|
|
type = maxLength ? type.split('(')[0] : type;
|
|
|
|
columns[val.name] = {
|
|
|
|
type: type.toLowerCase(),
|
2014-05-04 23:53:44 -04:00
|
|
|
maxLength: maxLength,
|
|
|
|
nullable: !val.notnull,
|
|
|
|
defaultValue: val.dflt_value
|
2014-04-21 23:08:59 -04:00
|
|
|
};
|
|
|
|
return columns;
|
|
|
|
}, {});
|
|
|
|
}
|
|
|
|
};
|
|
|
|
};
|
|
|
|
|
2014-04-08 16:25:57 -04:00
|
|
|
client.QueryBuilder = QueryBuilder_SQLite3;
|
|
|
|
client.QueryCompiler = QueryCompiler_SQLite3;
|
|
|
|
|
|
|
|
};
|