knex/lib/schema/compiler.js

var _ = require('lodash');

// The "SchemaCompiler" takes all of the query statements which have been
// gathered in the "SchemaBuilder" and turns them into an array of
// properly formatted / bound query strings.
function SchemaCompiler(builder) {
  this.builder   = builder;
  this.formatter = new this.Formatter();
  this.sequence  = [];
}

SchemaCompiler.prototype.toSQL = function() {
  var sequence = this.builder._sequence;
  for (var i = 0, l = sequence.length; i < l; i++) {
    var query = sequence[i];
    this[query.method].apply(this, query.args);
  }
  return this.sequence;
};

// Push a new query onto the compiled "sequence" stack,
// creating a new formatter, returning the compiler.
SchemaCompiler.prototype.pushQuery = function(query) {
  if (_.isString(query)) {
    query = {sql: query};
  } else {
    query = query;
  }
  query.bindings = this.formatter.bindings;
  this.sequence.push(query);
  this.formatter = new this.Formatter();
};

SchemaCompiler.prototype.createTable = function(tableName, fn) {
  new this.client.TableCompiler(this, new this.client.TableBuilder(tableName)._create(fn)).toSQL('create');
};
SchemaCompiler.prototype.alterTable = function(tableName, fn) {
  new this.client.TableCompiler(this, new this.client.TableBuilder(tableName)._alter(fn)).toSQL('alter');
};
SchemaCompiler.prototype.dropTable = function(tableName) {
  this.pushQuery('drop table ' + this.formatter.wrap(tableName));
};
SchemaCompiler.prototype.dropTableIfExists = function(tableName) {
  this.pushQuery('drop table if exists ' + this.formatter.wrap(tableName));
};

module.exports = SchemaCompiler;