FE: added support of <applyDropStatements> option

Author: Vitalii4as

forward_engineering/api.js

@@ -1,39 +1,61 @@
 const reApi = require('../reverse_engineering/api');
 const { createLogger } = require('../reverse_engineering/helpers/loggerHelper');
 const applyToInstanceHelper = require('./applyToInstanceHelper');
+const { commentDropStatements } = require('./helpers/commentDropStatements');
+const { DROP_STATEMENTS } = require('./helpers/constants');
 
 module.exports = {
 	generateScript(data, logger, callback, app) {
-		const {
-			getAlterContainersScripts,
-			getAlterCollectionsScripts,
-			getAlterViewScripts,
-			getAlterModelDefinitionsScripts,
-		} = require('./helpers/alterScriptFromDeltaHelper');
-
-		const collection = JSON.parse(data.jsonSchema);
-		if (!collection) {
-			throw new Error(
-				'"comparisonModelCollection" is not found. Alter script can be generated only from Delta model',
+		try {
+			const {
+				getAlterContainersScripts,
+				getAlterCollectionsScripts,
+				getAlterViewScripts,
+				getAlterModelDefinitionsScripts,
+			} = require('./helpers/alterScriptFromDeltaHelper');
+
+			const collection = JSON.parse(data.jsonSchema);
+			if (!collection) {
+				throw new Error(
+					'"comparisonModelCollection" is not found. Alter script can be generated only from Delta model',
+				);
+			}
+
+			const dbVersion = data.modelData[0]?.dbVersion;
+			const containersScripts = getAlterContainersScripts(collection);
+			const collectionsScripts = getAlterCollectionsScripts(collection, app, dbVersion);
+			const viewScripts = getAlterViewScripts(collection, app);
+			const modelDefinitionsScripts = getAlterModelDefinitionsScripts(collection, app);
+			const script = [
+				...containersScripts,
+				...collectionsScripts,
+				...viewScripts,
+				...modelDefinitionsScripts,
+			].join('\n\n');
+
+			const applyDropStatements = data.options?.additionalOptions?.some(
+				option => option.id === 'applyDropStatements' && option.value,
 			);
-		}
 
-		const dbVersion = data.modelData[0]?.dbVersion;
-		const containersScripts = getAlterContainersScripts(collection);
-		const collectionsScripts = getAlterCollectionsScripts(collection, app, dbVersion);
-		const viewScripts = getAlterViewScripts(collection, app);
-		const modelDefinitionsScripts = getAlterModelDefinitionsScripts(collection, app);
+			callback(null, applyDropStatements ? script : commentDropStatements(script));
+		} catch (error) {
+			logger.log('error', { message: error.message, stack: error.stack }, 'PostgreSQL Forward-Engineering Error');
 
-		callback(
-			null,
-			[...containersScripts, ...collectionsScripts, ...viewScripts, ...modelDefinitionsScripts].join('\n\n'),
-		);
+			callback({ message: error.message, stack: error.stack });
+		}
 	},
 	generateViewScript(data, logger, callback, app) {
 		callback(new Error('Forward-Engineering of delta model on view level is not supported'));
 	},
 	generateContainerScript(data, logger, callback, app) {
-		callback(new Error('Forward-Engineering of delta model on container level is not supported'));
+		try {
+			data.jsonSchema = data.collections[0];
+			this.generateScript(data, logger, callback, app);
+		} catch (error) {
+			logger.log('error', { message: error.message, stack: error.stack }, 'PostgreSQL Forward-Engineering Error');
+
+			callback({ message: error.message, stack: error.stack });
+		}
 	},
 	getDatabases(connectionInfo, logger, callback, app) {
 		logger.progress({ message: 'Find all databases' });
@@ -60,4 +82,21 @@ module.exports = {
 	testConnection(connectionInfo, logger, callback, app) {
 		reApi.testConnection(connectionInfo, logger, callback, app).then(callback, callback);
 	},
+	isDropInStatements(data, logger, callback, app) {
+		try {
+			const cb = (error, script = '') =>
+				callback(
+					error,
+					DROP_STATEMENTS.some(statement => script.includes(statement)),
+				);
+
+			if (data.level === 'container') {
+				this.generateContainerScript(data, logger, cb, app);
+			} else {
+				this.generateScript(data, logger, cb, app);
+			}
+		} catch (e) {
+			callback({ message: e.message, stack: e.stack });
+		}
+	},
 };

forward_engineering/config.json

@@ -26,6 +26,15 @@
 	},
 	"compMode": {
 		"entity": true,
-		"container": false
-	}
+		"container": true
+	},
+	"additionalOptions": [
+		{
+			"id": "applyDropStatements",
+			"value": false,
+			"forUpdate": true,
+			"name": "Apply Drop Statements",
+			"isDropInStatements": true
+		}
+	]
 }

forward_engineering/helpers/commentDropStatements.js

@@ -0,0 +1,17 @@
+const { DROP_STATEMENTS } = require('./constants');
+
+const commentDropStatements = (script = '') =>
+	script
+		.split('\n')
+		.map(line => {
+			if (DROP_STATEMENTS.some(statement => line.includes(statement))) {
+				return `-- ${line}`;
+			} else {
+				return line;
+			}
+		})
+		.join('\n');
+
+module.exports = {
+	commentDropStatements,
+};

forward_engineering/helpers/constants.js

@@ -0,0 +1,13 @@
+const DROP_STATEMENTS = [
+	'DROP SCHEMA',
+	'DROP TABLE',
+	'DROP COLUMN',
+	'DROP DOMAIN',
+	'DROP TYPE',
+	'DROP ATTRIBUTE',
+	'DROP VIEW',
+];
+
+module.exports = {
+	DROP_STATEMENTS,
+};