HCK-13407: add primary, unique constraints handling for the script options

forward_engineering/helpers/alterScriptFromDeltaHelper.js

@@ -29,6 +29,7 @@ const getAlterCollectionsScripts = (collection, app, options) => {
 		getDeleteColumnScript,
 		getModifyColumnScript,
 		getModifyCollectionScript,
+		getModifyCollectionKeysScript,
 	} = require('./alterScriptHelpers/alterEntityHelper')(app, options);
 
 	const createCollectionsScripts = [collection.properties?.entities?.properties?.added?.items]
@@ -72,13 +73,20 @@ const getAlterCollectionsScripts = (collection, app, options) => {
 		.map(item => Object.values(item.properties)[0])
 		.flatMap(getModifyColumnScript);
 
+	const modifyCollectionKeysScripts = [collection.properties?.entities?.properties?.modified?.items]
+		.flat()
+		.filter(Boolean)
+		.map(item => Object.values(item.properties)[0])
+		.flatMap(getModifyCollectionKeysScript);
+
 	return [
 		...createCollectionsScripts,
 		...deleteCollectionScripts,
 		...addColumnScripts,
 		...modifyCollectionScripts,
 		...deleteColumnScripts,
 		...modifyColumnScript,
+		...modifyCollectionKeysScripts,
 	]
 		.filter(Boolean)
 		.map(script => script.trim());

forward_engineering/helpers/alterScriptHelpers/alterEntityHelper.js

@@ -3,6 +3,8 @@ const { getTableName } = require('../general');
 const { getEntityName } = require('../../utils/general');
 const { createColumnDefinitionBySchema } = require('./createColumnDefinition');
 const { checkFieldPropertiesChanged, modifyGroupItems, setIndexKeys } = require('./common');
+const { getModifyPkScripts } = require('./entityHelper/primaryKeyHelper');
+const { getModifyUkScripts } = require('./entityHelper/uniqueKeyHelper');
 
 const alterEntityHelper = (app, options) => {
 	const ddlProvider = require('../../ddlProvider')(null, options, app);
@@ -150,6 +152,13 @@ const alterEntityHelper = (app, options) => {
 		return [...renameColumnScripts, ...changeTypeScripts];
 	};
 
+	const getModifyCollectionKeysScript = collection => {
+		const modifyPkScripts = getModifyPkScripts(collection, options);
+		const modifyUkScripts = getModifyUkScripts(collection, options);
+
+		return [...modifyPkScripts, ...modifyUkScripts].filter(Boolean);
+	};
+
 	const hydrateIndex =
 		({ idToNameHashTable, idToActivatedHashTable, ddlProvider, tableData, schemaData }) =>
 		index => {
@@ -165,6 +174,7 @@ const alterEntityHelper = (app, options) => {
 		getAddColumnScript,
 		getDeleteColumnScript,
 		getModifyColumnScript,
+		getModifyCollectionKeysScript,
 	};
 };
 

forward_engineering/helpers/alterScriptHelpers/entityHelper/primaryKeyHelper.js

@@ -0,0 +1,26 @@
+const { getModifyKeyScripts } = require('./sharedKeyConstraintHelper');
+
+/**
+ * Primary Key constraint configuration
+ */
+const PRIMARY_KEY_CONFIG = {
+	constraintType: 'PRIMARY KEY',
+	compModKeyName: 'primaryKey',
+	columnKeyProperty: 'primaryKey',
+	compositeKeyProperty: 'compositePrimaryKey',
+};
+
+/**
+ * Get all modify PK scripts (both composite and regular)
+ *
+ * @param {Object} collection
+ * @param {Object} options
+ * @return {string[]}
+ */
+const getModifyPkScripts = (collection, options) => {
+	return getModifyKeyScripts(collection, PRIMARY_KEY_CONFIG, options);
+};
+
+module.exports = {
+	getModifyPkScripts,
+};