knex/src/dialects/mssql/schema/compiler.js

// MySQL Schema Compiler
// -------
var inherits       = require('inherits');
var SchemaCompiler = require('../../../schema/compiler');
var assign         = require('lodash/object/assign');

function SchemaCompiler_MSSQL(client, builder) {
  SchemaCompiler.call(this, client, builder)
}
inherits(SchemaCompiler_MSSQL, SchemaCompiler)

assign(SchemaCompiler_MSSQL.prototype, {

  dropTableIfExists: function(tableName) {
    var name = this.formatter.wrap(prefixedTableName(this.schema, tableName));
    this.pushQuery('if object_id(\'' + name +'\', \'U\') is not null drop table ' + name);
  },

  // Rename a table on the schema.
  renameTable: function(tableName, to) {
    this.pushQuery('rename table ' + this.formatter.wrap(tableName) + ' to ' + this.formatter.wrap(to));
  },

  // Check whether a table exists on the query.
  hasTable: function(tableName) {
    this.pushQuery({
      sql: 'SELECT object_id FROM sys.tables WHERE object_id = Object_ID(N\'' + this.formatter.parameter(tableName) + '\')',
      output: function(resp) {
        return resp.length > 0;
      }
    });
  },

  // Check whether a column exists on the schema.
  hasColumn: function(tableName, column) {
    this.pushQuery({
      sql: 'SELECT object_id FROM sys.columns WHERE Name Like ' + this.formatter.parameter(column) +
      ' AND object_id = Object_ID(N\'' + this.formatter.wrap(tableName) + '\')',
      output: function(resp) {
        return resp.length > 0;
      }
    });
  }

})

function prefixedTableName(prefix, table) {
  return prefix ? `${prefix}.${table}` : table;
}

module.exports = SchemaCompiler_MSSQL;