@contember/schema-migrations 1.2.0-alpha.1 → 1.2.0-alpha.12

package/src/modifications/indexes/CreateIndexModification.ts

@@ -0,0 +1,83 @@
+import { MigrationBuilder } from '@contember/database-migrations'
+import { Model, Schema } from '@contember/schema'
+import { SchemaUpdater, updateEntity, updateModel } from '../utils/schemaUpdateUtils'
+import { createModificationType, Differ, ModificationHandler } from '../ModificationHandler'
+import { acceptFieldVisitor } from '@contember/schema-utils'
+
+export class CreateIndexModificationHandler implements ModificationHandler<CreateIndexModificationData> {
+	constructor(private readonly data: CreateIndexModificationData, private readonly schema: Schema) {}
+
+	public createSql(builder: MigrationBuilder): void {
+		const entity = this.schema.model.entities[this.data.entityName]
+		if (entity.view) {
+			return
+		}
+		const fields = this.data.index.fields
+		const columns = fields.map(fieldName => {
+			return acceptFieldVisitor(this.schema.model, entity, fieldName, {
+				visitColumn: ({ column }) => {
+					return column.columnName
+				},
+				visitManyHasOne: ({ relation }) => {
+					return relation.joiningColumn.columnName
+				},
+				visitOneHasOneOwning: ({ relation }) => {
+					return relation.joiningColumn.columnName
+				},
+				visitOneHasMany: () => {
+					throw new Error(`Cannot create index on 1:m relation in ${entity.name}.${fieldName}`)
+				},
+				visitOneHasOneInverse: () => {
+					throw new Error(`Cannot create index on 1:1 inverse relation in ${entity.name}.${fieldName}`)
+				},
+				visitManyHasManyOwning: () => {
+					throw new Error(`Cannot create index on m:m relation in ${entity.name}.${fieldName}`)
+				},
+				visitManyHasManyInverse: () => {
+					throw new Error(`Cannot create index on m:m inverse relation in ${entity.name}.${fieldName}`)
+				},
+			})
+		})
+		builder.addIndex(entity.tableName, [...columns], {
+			name: this.data.index.name,
+		})
+	}
+
+	public getSchemaUpdater(): SchemaUpdater {
+		return updateModel(
+			updateEntity(this.data.entityName, ({ entity }) => ({
+				...entity,
+				indexes: {
+					...entity.indexes,
+					[this.data.index.name]: this.data.index,
+				},
+			})),
+		)
+	}
+
+	describe() {
+		return {
+			message: `Create index(${this.data.index.fields.join(', ')}) on entity ${this.data.entityName}`,
+		}
+	}
+}
+
+export const createIndexModification = createModificationType({
+	id: 'createIndex',
+	handler: CreateIndexModificationHandler,
+})
+
+export class CreateIndexDiffer implements Differ {
+	createDiff(originalSchema: Schema, updatedSchema: Schema) {
+		return Object.values(updatedSchema.model.entities).flatMap(entity =>
+			Object.values(entity.indexes)
+				.filter(it => !originalSchema.model.entities[entity.name].indexes[it.name])
+				.map(index => createIndexModification.createModification({ entityName: entity.name, index })),
+		)
+	}
+}
+
+export interface CreateIndexModificationData {
+	entityName: string
+	index: Model.Index
+}

package/src/modifications/indexes/RemoveIndexModification.ts

@@ -0,0 +1,68 @@
+import { MigrationBuilder } from '@contember/database-migrations'
+import { Schema } from '@contember/schema'
+import { SchemaUpdater, updateEntity, updateModel } from '../utils/schemaUpdateUtils'
+import { createModificationType, Differ, ModificationHandler } from '../ModificationHandler'
+import deepEqual from 'fast-deep-equal'
+
+export class RemoveIndexModificationHandler implements ModificationHandler<RemoveIndexModificationData>  {
+	constructor(private readonly data: RemoveIndexModificationData, private readonly schema: Schema) {}
+
+	public createSql(builder: MigrationBuilder): void {
+		const entity = this.schema.model.entities[this.data.entityName]
+		if (entity.view) {
+			return
+		}
+		builder.dropIndex(entity.tableName, [], {
+			name: this.data.indexName,
+		})
+	}
+
+	public getSchemaUpdater(): SchemaUpdater {
+		return updateModel(
+			updateEntity(this.data.entityName, ({ entity }) => {
+				const { [this.data.indexName]: removed, ...indexes } = entity.indexes
+				return {
+					...entity,
+					indexes,
+				}
+			}),
+		)
+	}
+
+	describe() {
+		const fields = this.schema.model.entities[this.data.entityName].indexes[this.data.indexName].fields
+		return { message: `Remove index (${fields.join(', ')}) on entity ${this.data.entityName}` }
+	}
+}
+
+export const removeIndexModification = createModificationType({
+	id: 'removeIndex',
+	handler: RemoveIndexModificationHandler,
+})
+
+
+export class RemoveIndexDiffer implements Differ {
+
+	createDiff(originalSchema: Schema, updatedSchema: Schema) {
+		return Object.values(originalSchema.model.entities).flatMap(entity =>
+			Object.values(entity.indexes)
+				.filter(
+					it =>
+						updatedSchema.model.entities[entity.name] &&
+						(!updatedSchema.model.entities[entity.name].indexes[it.name] ||
+							!deepEqual(it.fields, updatedSchema.model.entities[entity.name].indexes[it.name].fields)),
+				)
+				.map(index =>
+					removeIndexModification.createModification({
+						entityName: entity.name,
+						indexName: index.name,
+					}),
+				),
+		)
+	}
+}
+
+export interface RemoveIndexModificationData {
+	entityName: string
+	indexName: string
+}

package/src/modifications/indexes/index.ts

@@ -0,0 +1,2 @@
+export * from './CreateIndexModification'
+export * from './RemoveIndexModification'

package/src/modifications/relations/ConvertOneHasManyToManyHasManyRelationModification.ts

@@ -1,16 +1,20 @@
 import { MigrationBuilder } from '@contember/database-migrations'
 import { Model, Schema } from '@contember/schema'
 import { addField, SchemaUpdater, updateEntity, updateModel, updateSchema } from '../utils/schemaUpdateUtils'
-import { ModificationHandler, ModificationHandlerOptions, ModificationHandlerStatic } from '../ModificationHandler'
+import {
+	createModificationType,
+	Differ,
+	ModificationHandler,
+	ModificationHandlerOptions,
+} from '../ModificationHandler'
 import { isInverseRelation, isOwningRelation } from '@contember/schema-utils'
 import { updateRelations } from '../utils/diffUtils'
-import { UpdateFieldNameModification } from '../fields'
 import { createJunctionTableSql } from '../utils/createJunctionTable'
 import { wrapIdentifier } from '@contember/database'
 import { normalizeManyHasManyRelation, PartialManyHasManyRelation } from './normalization'
+import { updateFieldNameModification } from '../fields'
 
-export const ConvertOneHasManyToManyHasManyRelationModification: ModificationHandlerStatic<ConvertOneHasManyToManyHasManyRelationModificationData> = class {
-	static id = 'convertOneHasManyToManyHasManyRelation'
+export class ConvertOneHasManyToManyHasManyRelationModificationHandler implements ModificationHandler<ConvertOneHasManyToManyHasManyRelationModificationData> {
 	private subModification: ModificationHandler<any>
 
 	constructor(
@@ -18,7 +22,7 @@ export const ConvertOneHasManyToManyHasManyRelationModification: ModificationHan
 		private readonly schema: Schema,
 		private readonly options: ModificationHandlerOptions,
 	) {
-		this.subModification = new UpdateFieldNameModification(
+		this.subModification = updateFieldNameModification.createHandler(
 			{
 				entityName: this.data.entityName,
 				fieldName: this.data.fieldName,
@@ -63,11 +67,33 @@ export const ConvertOneHasManyToManyHasManyRelationModification: ModificationHan
 		}
 	}
 
-	static createModification(data: ConvertOneHasManyToManyHasManyRelationModificationData) {
-		return { modification: this.id, ...data }
+
+
+
+	private getRelation(): { entity: Model.Entity; relation: Model.ManyHasOneRelation } {
+		const entity = this.schema.model.entities[this.data.entityName]
+		const relation = entity.fields[this.data.fieldName]
+		if (relation.type !== Model.RelationType.ManyHasOne) {
+			throw new Error()
+		}
+		return { entity, relation }
 	}
+}
+
+export const convertOneHasManyToManyHasManyRelationModification = createModificationType({
+	id: 'convertOneHasManyToManyHasManyRelation',
+	handler: ConvertOneHasManyToManyHasManyRelationModificationHandler,
+})
 
-	static createDiff(originalSchema: Schema, updatedSchema: Schema) {
+export interface ConvertOneHasManyToManyHasManyRelationModificationData {
+	entityName: string
+	fieldName: string
+	owningSide: PartialManyHasManyRelation
+	inverseSide?: Model.ManyHasManyInverseRelation
+}
+
+export class ConvertOneHasManyToManyHasManyRelationDiffer implements Differ {
+	createDiff(originalSchema: Schema, updatedSchema: Schema) {
 		return updateRelations(originalSchema, updatedSchema, ({ originalRelation, updatedRelation, updatedEntity }) => {
 			if (
 				isInverseRelation(originalRelation) &&
@@ -81,7 +107,7 @@ export const ConvertOneHasManyToManyHasManyRelationModification: ModificationHan
 				if (owningSide.type !== Model.RelationType.ManyHasMany || !isOwningRelation(owningSide)) {
 					throw new Error()
 				}
-				return ConvertOneHasManyToManyHasManyRelationModification.createModification({
+				return convertOneHasManyToManyHasManyRelationModification.createModification({
 					entityName: updatedRelation.target,
 					fieldName: owningOldName,
 					owningSide,
@@ -91,20 +117,4 @@ export const ConvertOneHasManyToManyHasManyRelationModification: ModificationHan
 			return undefined
 		})
 	}
-
-	private getRelation(): { entity: Model.Entity; relation: Model.ManyHasOneRelation } {
-		const entity = this.schema.model.entities[this.data.entityName]
-		const relation = entity.fields[this.data.fieldName]
-		if (relation.type !== Model.RelationType.ManyHasOne) {
-			throw new Error()
-		}
-		return { entity, relation }
-	}
-}
-
-export interface ConvertOneHasManyToManyHasManyRelationModificationData {
-	entityName: string
-	fieldName: string
-	owningSide: PartialManyHasManyRelation
-	inverseSide?: Model.ManyHasManyInverseRelation
 }

package/src/modifications/relations/ConvertOneToManyRelationModification.ts

@@ -1,14 +1,18 @@
 import { MigrationBuilder } from '@contember/database-migrations'
 import { Model, Schema } from '@contember/schema'
 import { SchemaUpdater, updateEntity, updateField, updateModel, updateSchema } from '../utils/schemaUpdateUtils'
-import { ModificationHandler, ModificationHandlerOptions, ModificationHandlerStatic } from '../ModificationHandler'
+import {
+	createModificationType,
+	Differ,
+	ModificationHandler,
+	ModificationHandlerOptions,
+} from '../ModificationHandler'
 import { isOwningRelation, NamingHelper } from '@contember/schema-utils'
 import { updateRelations } from '../utils/diffUtils'
-import { UpdateFieldNameModification } from '../fields'
 import { NoopModification } from '../NoopModification'
+import { updateFieldNameModification } from '../fields'
 
-export const ConvertOneToManyRelationModification: ModificationHandlerStatic<ConvertOneToManyRelationModificationData> = class {
-	static id = 'convertOneToManyRelation'
+export class ConvertOneToManyRelationModificationHandler implements ModificationHandler<ConvertOneToManyRelationModificationData> {
 	private subModification: ModificationHandler<any>
 
 	constructor(
@@ -18,7 +22,7 @@ export const ConvertOneToManyRelationModification: ModificationHandlerStatic<Con
 	) {
 		const { relation } = this.getRelation()
 		this.subModification = data.newInverseSideFieldName && relation.inversedBy ?
-			new UpdateFieldNameModification(
+			updateFieldNameModification.createHandler(
 				{
 					entityName: relation.target,
 					fieldName: relation.inversedBy,
@@ -77,11 +81,29 @@ export const ConvertOneToManyRelationModification: ModificationHandlerStatic<Con
 		}
 	}
 
-	static createModification(data: ConvertOneToManyRelationModificationData) {
-		return { modification: this.id, ...data }
+	private getRelation() {
+		const entity = this.schema.model.entities[this.data.entityName]
+		const relation = entity.fields[this.data.fieldName]
+		if (relation.type !== Model.RelationType.OneHasOne || !isOwningRelation(relation)) {
+			throw new Error()
+		}
+		return { entity, relation }
 	}
+}
+
+export interface ConvertOneToManyRelationModificationData {
+	entityName: string
+	fieldName: string
+	newInverseSideFieldName?: string
+}
+
+export const convertOneToManyRelationModification = createModificationType({
+	id: 'convertOneToManyRelation',
+	handler: ConvertOneToManyRelationModificationHandler,
+})
 
-	static createDiff(originalSchema: Schema, updatedSchema: Schema) {
+export class ConvertOneToManyRelationDiffer implements Differ {
+	createDiff(originalSchema: Schema, updatedSchema: Schema) {
 		return updateRelations(originalSchema, updatedSchema, ({ originalRelation, updatedRelation, updatedEntity }) => {
 			if (
 				isOwningRelation(originalRelation) &&
@@ -90,7 +112,7 @@ export const ConvertOneToManyRelationModification: ModificationHandlerStatic<Con
 				updatedRelation.type === Model.RelationType.ManyHasOne
 			) {
 				const isInverseSideRenamed = originalRelation.inversedBy && updatedRelation.inversedBy && originalRelation.inversedBy !== updatedRelation.inversedBy
-				return ConvertOneToManyRelationModification.createModification({
+				return convertOneToManyRelationModification.createModification({
 					entityName: updatedEntity.name,
 					fieldName: updatedRelation.name,
 					...(isInverseSideRenamed ? {
@@ -101,19 +123,4 @@ export const ConvertOneToManyRelationModification: ModificationHandlerStatic<Con
 			return undefined
 		})
 	}
-
-	private getRelation() {
-		const entity = this.schema.model.entities[this.data.entityName]
-		const relation = entity.fields[this.data.fieldName]
-		if (relation.type !== Model.RelationType.OneHasOne || !isOwningRelation(relation)) {
-			throw new Error()
-		}
-		return { entity, relation }
-	}
-}
-
-export interface ConvertOneToManyRelationModificationData {
-	entityName: string
-	fieldName: string
-	newInverseSideFieldName?: string
 }

package/src/modifications/relations/CreateRelationInverseSideModification.ts

@@ -1,12 +1,11 @@
 import { MigrationBuilder } from '@contember/database-migrations'
 import { Model, Schema } from '@contember/schema'
 import { addField, SchemaUpdater, updateEntity, updateField, updateModel } from '../utils/schemaUpdateUtils'
-import { ModificationHandlerStatic } from '../ModificationHandler'
+import { createModificationType, Differ, ModificationHandler } from '../ModificationHandler'
 import { createFields } from '../utils/diffUtils'
 import { isInverseRelation, isRelation } from '@contember/schema-utils'
 
-export const CreateRelationInverseSideModification: ModificationHandlerStatic<CreateRelationInverseSideModificationData> = class {
-	static id = 'createRelationInverseSide'
+export class CreateRelationInverseSideModificationHandler implements ModificationHandler<CreateRelationInverseSideModificationData> {
 	constructor(private readonly data: CreateRelationInverseSideModificationData, private readonly schema: Schema) {}
 
 	public createSql(builder: MigrationBuilder): void {}
@@ -27,25 +26,28 @@ export const CreateRelationInverseSideModification: ModificationHandlerStatic<Cr
 	describe() {
 		return { message: `Add relation ${this.data.entityName}.${this.data.relation.name}` }
 	}
+}
 
-	static createModification(data: CreateRelationInverseSideModificationData) {
-		return { modification: this.id, ...data }
-	}
+export interface CreateRelationInverseSideModificationData {
+	entityName: string
+	relation: Model.AnyRelation & Model.InverseRelation
+}
 
-	static createDiff(originalSchema: Schema, updatedSchema: Schema) {
+export const createRelationInverseSideModification = createModificationType({
+	id: 'createRelationInverseSide',
+	handler: CreateRelationInverseSideModificationHandler,
+})
+
+export class CreateRelationInverseSideDiffer implements Differ {
+	createDiff(originalSchema: Schema, updatedSchema: Schema) {
 		return createFields(originalSchema, updatedSchema, ({ newField, updatedEntity }) => {
 			if (!isRelation(newField) || !isInverseRelation(newField)) {
 				return undefined
 			}
-			return CreateRelationInverseSideModification.createModification({
+			return createRelationInverseSideModification.createModification({
 				entityName: updatedEntity.name,
 				relation: newField,
 			})
 		})
 	}
 }
-
-export interface CreateRelationInverseSideModificationData {
-	entityName: string
-	relation: Model.AnyRelation & Model.InverseRelation
-}

package/src/modifications/relations/CreateRelationModification.ts

@@ -8,7 +8,12 @@ import {
 } from '@contember/schema-utils'
 import { MigrationBuilder } from '@contember/database-migrations'
 import { addField, SchemaUpdater, updateEntity, updateModel } from '../utils/schemaUpdateUtils'
-import { ModificationHandlerOptions, ModificationHandlerStatic } from '../ModificationHandler'
+import {
+	createModificationType,
+	Differ,
+	ModificationHandler,
+	ModificationHandlerOptions,
+} from '../ModificationHandler'
 import { isIt } from '../../utils/isIt'
 import { createFields } from '../utils/diffUtils'
 import { getPrimaryColumnType } from '../utils/getPrimaryColumnType'
@@ -16,8 +21,7 @@ import { createJunctionTableSql } from '../utils/createJunctionTable'
 import { normalizeManyHasManyRelation, PartialManyHasManyRelation } from './normalization'
 
 
-export const CreateRelationModification: ModificationHandlerStatic<CreateRelationModificationData> = class {
-	static id = 'createRelation'
+export class CreateRelationModificationHandler implements ModificationHandler<CreateRelationModificationData> {
 
 	constructor(
 		private readonly data: CreateRelationModificationData,
@@ -55,18 +59,18 @@ export const CreateRelationModification: ModificationHandlerStatic<CreateRelatio
 			})
 		}
 		acceptRelationTypeVisitor(this.schema.model, entity, this.getNormalizedOwningSide(), {
-			visitManyHasOne: ({}, relation, {}, _) => {
+			visitManyHasOne: ({ relation }) => {
 				createOwningSide(relation)
 				builder.addIndex(entity.tableName, relation.joiningColumn.columnName)
 			},
 			visitOneHasMany: () => {},
-			visitOneHasOneOwning: ({}, relation, {}, _) => {
+			visitOneHasOneOwning: ({ relation }) => {
 				createOwningSide(relation)
 				const uniqueConstraintName = NamingHelper.createUniqueConstraintName(entity.name, [relation.name])
 				builder.addConstraint(entity.tableName, uniqueConstraintName, { unique: [relation.joiningColumn.columnName] })
 			},
 			visitOneHasOneInverse: () => {},
-			visitManyHasManyOwning: ({}, relation, {}, _) => {
+			visitManyHasManyOwning: ({ relation }) => {
 				createJunctionTableSql(builder, this.options.systemSchema,  entity, targetEntity, relation)
 			},
 			visitManyHasManyInverse: () => {},
@@ -91,11 +95,32 @@ export const CreateRelationModification: ModificationHandlerStatic<CreateRelatio
 		return { message: `Add relation ${this.data.entityName}.${this.data.owningSide.name}`, failureWarning }
 	}
 
-	static createModification(data: CreateRelationModificationData) {
-		return { modification: this.id, ...data }
+
+	private getNormalizedOwningSide(): Model.AnyOwningRelation {
+		if (this.data.owningSide.type === Model.RelationType.ManyHasMany) {
+			return normalizeManyHasManyRelation(this.data.owningSide)
+		}
+		return this.data.owningSide
 	}
+}
+
+
+export type CreateRelationModificationData = {
+	entityName: string
+	owningSide:
+		| Model.ManyHasOneRelation
+		| Model.OneHasOneOwningRelation
+		| PartialManyHasManyRelation
+	inverseSide?: Model.AnyInverseRelation
+}
 
-	static createDiff(originalSchema: Schema, updatedSchema: Schema) {
+export const createRelationModification = createModificationType({
+	id: 'createRelation',
+	handler: CreateRelationModificationHandler,
+})
+
+export class CreateRelationDiffer implements Differ {
+	createDiff(originalSchema: Schema, updatedSchema: Schema) {
 		return createFields(originalSchema, updatedSchema, ({ newField, updatedEntity }) => {
 			if (!isRelation(newField) || !isOwningRelation(newField)) {
 				return undefined
@@ -106,28 +131,11 @@ export const CreateRelationModification: ModificationHandlerStatic<CreateRelatio
 			if (inverseSide && !(isRelation(inverseSide) && isInverseRelation(inverseSide))) {
 				throw new Error()
 			}
-			return CreateRelationModification.createModification({
+			return createRelationModification.createModification({
 				entityName: updatedEntity.name,
 				owningSide: newField,
 				...(inverseSide ? { inverseSide } : {}),
 			})
 		})
 	}
-
-	private getNormalizedOwningSide(): Model.AnyOwningRelation {
-		if (this.data.owningSide.type === Model.RelationType.ManyHasMany) {
-			return normalizeManyHasManyRelation(this.data.owningSide)
-		}
-		return this.data.owningSide
-	}
-}
-
-
-export type CreateRelationModificationData = {
-	entityName: string
-	owningSide:
-		| Model.ManyHasOneRelation
-		| Model.OneHasOneOwningRelation
-		| PartialManyHasManyRelation
-	inverseSide?: Model.AnyInverseRelation
 }

package/src/modifications/relations/DisableOrphanRemovalModification.ts

@@ -1,12 +1,11 @@
 import { MigrationBuilder } from '@contember/database-migrations'
 import { Model, Schema } from '@contember/schema'
 import { SchemaUpdater, updateEntity, updateField, updateModel } from '../utils/schemaUpdateUtils'
-import { ModificationHandlerStatic } from '../ModificationHandler'
+import { createModificationType, Differ, ModificationHandler } from '../ModificationHandler'
 import { isOwningRelation } from '@contember/schema-utils'
 import { updateRelations } from '../utils/diffUtils'
 
-export const DisableOrphanRemovalModification: ModificationHandlerStatic<DisableOrphanRemovalModificationData> = class {
-	static id = 'disableOrphanRemoval'
+export class DisableOrphanRemovalModificationHandler implements ModificationHandler<DisableOrphanRemovalModificationData> {
 
 	constructor(private readonly data: DisableOrphanRemovalModificationData, private readonly schema: Schema) {}
 
@@ -27,12 +26,20 @@ export const DisableOrphanRemovalModification: ModificationHandlerStatic<Disable
 			message: `Disable orphan removal on ${this.data.entityName}.${this.data.fieldName}`,
 		}
 	}
+}
 
-	static createModification(data: DisableOrphanRemovalModificationData) {
-		return { modification: this.id, ...data }
-	}
+export interface DisableOrphanRemovalModificationData {
+	entityName: string
+	fieldName: string
+}
 
-	static createDiff(originalSchema: Schema, updatedSchema: Schema) {
+export const disableOrphanRemovalModification = createModificationType({
+	id: 'disableOrphanRemoval',
+	handler: DisableOrphanRemovalModificationHandler,
+})
+
+export class DisableOrphanRemovalDiffer implements Differ {
+	createDiff(originalSchema: Schema, updatedSchema: Schema) {
 		return updateRelations(originalSchema, updatedSchema, ({ originalRelation, updatedRelation, updatedEntity }) => {
 			if (
 				isOwningRelation(originalRelation) &&
@@ -42,7 +49,7 @@ export const DisableOrphanRemovalModification: ModificationHandlerStatic<Disable
 				originalRelation.orphanRemoval &&
 				!updatedRelation.orphanRemoval
 			) {
-				return DisableOrphanRemovalModification.createModification({
+				return disableOrphanRemovalModification.createModification({
 					entityName: updatedEntity.name,
 					fieldName: updatedRelation.name,
 				})
@@ -51,7 +58,3 @@ export const DisableOrphanRemovalModification: ModificationHandlerStatic<Disable
 		})
 	}
 }
-export interface DisableOrphanRemovalModificationData {
-	entityName: string
-	fieldName: string
-}

package/src/modifications/relations/EnableOrphanRemovalModification.ts

@@ -1,12 +1,11 @@
 import { MigrationBuilder } from '@contember/database-migrations'
 import { Model, Schema } from '@contember/schema'
 import { SchemaUpdater, updateEntity, updateField, updateModel } from '../utils/schemaUpdateUtils'
-import { ModificationHandlerStatic } from '../ModificationHandler'
+import { createModificationType, Differ, ModificationHandler } from '../ModificationHandler'
 import { isOwningRelation } from '@contember/schema-utils'
 import { updateRelations } from '../utils/diffUtils'
 
-export const EnableOrphanRemovalModification: ModificationHandlerStatic<EnableOrphanRemovalModificationData> = class {
-	static id = 'enableOrphanRemoval'
+export class EnableOrphanRemovalModificationHandler implements ModificationHandler<EnableOrphanRemovalModificationData> {
 	constructor(private readonly data: EnableOrphanRemovalModificationData, private readonly schema: Schema) {}
 
 	public createSql(builder: MigrationBuilder): void {}
@@ -30,11 +29,20 @@ export const EnableOrphanRemovalModification: ModificationHandlerStatic<EnableOr
 		}
 	}
 
-	static createModification(data: EnableOrphanRemovalModificationData) {
-		return { modification: this.id, ...data }
-	}
+}
+
+export interface EnableOrphanRemovalModificationData {
+	entityName: string
+	fieldName: string
+}
 
-	static createDiff(originalSchema: Schema, updatedSchema: Schema) {
+export const enableOrphanRemovalModification = createModificationType({
+	id: 'enableOrphanRemoval',
+	handler: EnableOrphanRemovalModificationHandler,
+})
+
+export class EnableOrphanRemovalDiffer implements Differ {
+	 createDiff(originalSchema: Schema, updatedSchema: Schema) {
 		return updateRelations(originalSchema, updatedSchema, ({ originalRelation, updatedRelation, updatedEntity }) => {
 			if (
 				isOwningRelation(originalRelation) &&
@@ -44,7 +52,7 @@ export const EnableOrphanRemovalModification: ModificationHandlerStatic<EnableOr
 				!originalRelation.orphanRemoval &&
 				updatedRelation.orphanRemoval
 			) {
-				return EnableOrphanRemovalModification.createModification({
+				return enableOrphanRemovalModification.createModification({
 					entityName: updatedEntity.name,
 					fieldName: updatedRelation.name,
 				})
@@ -53,8 +61,3 @@ export const EnableOrphanRemovalModification: ModificationHandlerStatic<EnableOr
 		})
 	}
 }
-
-export interface EnableOrphanRemovalModificationData {
-	entityName: string
-	fieldName: string
-}