@mikro-orm/knex/dialects/mssql/MsSqlTableCompiler.js

TypeScript ORM for Node.js based on Data Mapper, Unit of Work and Identity Map patterns. Supports MongoDB, MySQL, PostgreSQL and SQLite databases as well as usage with vanilla JavaScript.

"use strict";
Object.defineProperty(exports, "__esModule", { value: true });
exports.MsSqlTableCompiler = void 0;
const MonkeyPatchable_1 = require("../../MonkeyPatchable");
class MsSqlTableCompiler extends MonkeyPatchable_1.MonkeyPatchable.MsSqlTableCompiler {
    lowerCase = true;
    addColumnsPrefix = 'add ';
    dropColumnPrefix = 'drop column ';
    alterColumnPrefix = 'alter column ';
    alterColumns(columns, colBuilder) {
        for (let i = 0, l = colBuilder.length; i < l; i++) {
            const builder = colBuilder[i];
            if (builder.modified.defaultTo) {
                const schema = this.schemaNameRaw || 'dbo';
                const baseQuery = `declare @constraint${i} varchar(100) = (select default_constraints.name from sys.all_columns`
                    + ' join sys.tables on all_columns.object_id = tables.object_id'
                    + ' join sys.schemas on tables.schema_id = schemas.schema_id'
                    + ' join sys.default_constraints on all_columns.default_object_id = default_constraints.object_id'
                    + ` where schemas.name = '${schema}' and tables.name = '${this.tableNameRaw}' and all_columns.name = '${builder.getColumnName()}')`
                    + ` if @constraint${i} is not null exec('alter table ${this.tableNameRaw} drop constraint ' + @constraint${i})`;
                this.pushQuery(baseQuery);
            }
        }
        // in SQL server only one column can be altered at a time
        columns.sql.forEach((sql) => {
            this.pushQuery({
                sql: `alter table ${this.tableName()} ${this.alterColumnPrefix.toLowerCase()}${sql}`,
                bindings: columns.bindings,
            });
        });
    }
    dropForeign(columns, constraintName) {
        /* istanbul ignore next */
        constraintName = constraintName
            ? this.formatter.wrap(constraintName)
            : this._indexCommand('foreign', this.tableNameRaw, columns);
        this.pushQuery(`alter table ${this.tableName()} drop constraint ${constraintName}`);
    }
}
exports.MsSqlTableCompiler = MsSqlTableCompiler;