lib/schema/columncompiler

Column Compiler Used for designating column definitions

during the table "create" / "alter" statements.

var _ = require('lodash'); var Raw = require('../raw'); function ColumnCompiler(tableCompiler, columnBuilder) { this.tableCompiler = tableCompiler; this.args = columnBuilder._args; this.type = columnBuilder._type; this.grouped = _.groupBy(columnBuilder._statements, 'grouping'); this.modified = columnBuilder._modifiers; this.isIncrements = (this.type.toLowerCase().indexOf('increments') !== -1); this.initCompiler(); }

To convert to sql, we first go through and build the column as it would be in the insert statement

ColumnCompiler.prototype.toSQL = function() { this.pushQuery(this.compileColumn()); return this.sequence; };

Compiles a column.

ColumnCompiler.prototype.compileColumn = function() { return this.formatter.wrap(this.getColumnName()) + ' ' + this.getColumnType() + this.getModifiers(); };

Assumes the autoincrementing key is named id if not otherwise specified.

ColumnCompiler.prototype.getColumnName = function() { var value = _.first(this.args); if (value) return value; if (this.isIncrements) { return 'id'; } else { throw new Error('You did not specify a column name for the ' + this.type + 'column.'); } }; ColumnCompiler.prototype.getColumnType = function() { var type = this[this.type.toLowerCase()]; return _.isFunction(type) ? type.apply(this, _.rest(this.args)) : type; }; ColumnCompiler.prototype.getModifiers = function() { var modifiers = []; if (this.type.indexOf('increments') === -1) { for (var i = 0, l = this.modifiers.length; i < l; i++) { var modifier = this.modifiers[i]; if (_.has(this.modified, modifier)) { var val = this[modifier].apply(this, this.modified[modifier]); if (val) modifiers.push(val); } } } return modifiers.length > 0 ? ' ' + modifiers.join(' ') : ''; };

Types

ColumnCompiler.prototype.increments = 'integer not null primary key autoincrement'; ColumnCompiler.prototype.bigincrements = 'integer not null primary key autoincrement'; ColumnCompiler.prototype.integer = ColumnCompiler.prototype.smallint = ColumnCompiler.prototype.mediumint = 'integer'; ColumnCompiler.prototype.biginteger = 'bigint'; ColumnCompiler.prototype.varchar = function(length) { return 'varchar(' + this._num(length, 255) + ')'; }; ColumnCompiler.prototype.text = 'text'; ColumnCompiler.prototype.tinyint = 'tinyint'; ColumnCompiler.prototype.floating = function(precision, scale) { return 'float(' + this._num(precision, 8) + ', ' + this._num(scale, 2) + ')'; }; ColumnCompiler.prototype.decimal = function(precision, scale) { return 'decimal(' + this._num(precision, 8) + ', ' + this._num(scale, 2) + ')'; }; ColumnCompiler.prototype.binary = 'blob'; ColumnCompiler.prototype.bool = 'boolean'; ColumnCompiler.prototype.date = 'date'; ColumnCompiler.prototype.datetime = 'datetime'; ColumnCompiler.prototype.time = 'time'; ColumnCompiler.prototype.timestamp = 'timestamp'; ColumnCompiler.prototype.enu = 'varchar'; ColumnCompiler.prototype.bit = ColumnCompiler.prototype.json = 'text'; ColumnCompiler.prototype.uuid = 'char(36)'; ColumnCompiler.prototype.specificType = function(type) { return type; };

Modifiers

ColumnCompiler.prototype.nullable = function(nullable) { return nullable === false ? 'not null' : 'null'; }; ColumnCompiler.prototype.notNullable = function() { return this.nullable(false); }; ColumnCompiler.prototype.defaultTo = function(value) { if (value === void 0) { return ''; } else if (value instanceof Raw) { value = value.toQuery(); } else if (this.method === 'bool') { if (value === 'false') value = 0; value = (value ? 1 : 0); } else if (value === true || value === false) { value = parseInt(value, 10); } else { value = "'" + value + "'"; } return 'default ' + value; }; ColumnCompiler.prototype._num = function(val, fallback) { if (val == null) return fallback; var number = parseInt(val, 10); return isNaN(number) ? fallback : number; }; module.exports = ColumnCompiler;