RE: implement reverse-engineering of unique keys with single key as unique on field level

Author: Vitalii4as

reverse_engineering/helpers/postgresHelpers/reorganizeConstraints.js

@@ -0,0 +1,101 @@
+const reorganizeConstraints = (attributes, entityLevel) => {
+	return [reorganizePrimaryKeys, reorganizeUniqueKeys].reduce(
+		({ attributes, entityLevel }, reorganize) => reorganize(attributes, entityLevel),
+		{ attributes, entityLevel },
+	);
+};
+
+const reorganizePrimaryKeys = (attributes, entityLevel) => {
+	const isSinglePrimaryKey = entityLevel.primaryKey?.[0]?.compositePrimaryKey?.length !== 1;
+
+	if (isSinglePrimaryKey) {
+		return { attributes, entityLevel };
+	}
+
+	const primaryKey = entityLevel.primaryKey[0];
+	const attributeName = primaryKey.compositePrimaryKey[0];
+
+	return {
+		attributes: setAttributesPrimaryKeyData(attributes, attributeName, primaryKey),
+		entityLevel: clearPrimaryKeys(entityLevel),
+	};
+};
+
+const reorganizeUniqueKeys = (attributes, entityLevel) => {
+	return (entityLevel.uniqueKey || []).reduce(
+		({ attributes, entityLevel }, uniqueKey) => {
+			const hasSingleUniqueKey = uniqueKey.compositeUniqueKey.length === 1;
+			if (!hasSingleUniqueKey) {
+				return { attributes, entityLevel };
+			}
+
+			const attributeName = uniqueKey.compositeUniqueKey[0];
+
+			return {
+				attributes: setAttributesUniqueKeyData(attributes, attributeName, uniqueKey),
+				entityLevel: filterUniqueKeys(entityLevel, uniqueKey),
+			};
+		},
+		{ attributes, entityLevel },
+	);
+};
+
+const setAttributesPrimaryKeyData = (attributes, attributeName, primaryKey) => {
+	return attributes.map(attribute => {
+		if (attribute.name !== attributeName) {
+			return attribute;
+		}
+
+		return setPrimaryKeyData(attribute, primaryKey);
+	});
+};
+
+const setPrimaryKeyData = (attribute, primaryKey) => {
+	return {
+		...attribute,
+		primaryKey: true,
+		primaryKeyOptions: getOptions(primaryKey),
+	};
+};
+
+const setAttributesUniqueKeyData = (attributes, attributeName, uniqueKey) => {
+	return attributes.map(attribute => {
+		if (attribute.name !== attributeName) {
+			return attribute;
+		}
+
+		return setUniqueKeyData(attribute, uniqueKey);
+	});
+};
+
+const setUniqueKeyData = (attribute, uniqueKey) => {
+	return {
+		...attribute,
+		unique: true,
+		uniqueKeyOptions: getOptions(uniqueKey),
+	};
+};
+
+const getOptions = key => {
+	return [
+		{
+			constraintName: key.constraintName,
+			indexInclude: key.indexInclude,
+			indexStorageParameters: key.indexStorageParameters,
+			indexTablespace: key.indexTablespace,
+			indexComment: key.indexComment,
+		},
+	];
+};
+
+const clearPrimaryKeys = entityLevel => {
+	return { ...entityLevel, primaryKey: [] };
+};
+
+const filterUniqueKeys = (entityLevel, uniqueKey) => {
+	const filteredKeys = entityLevel.uniqueKey.filter(key => key.constraintName !== uniqueKey.constraintName);
+
+	return { ...entityLevel, uniqueKey: filteredKeys };
+};
+
+module.exports = { reorganizeConstraints };

reverse_engineering/helpers/postgresHelpers/reorganizePrimaryKeys.js

@@ -45,7 +45,7 @@ const {
 } = require('./postgresHelpers/viewHelper');
 const { setDependencies: setDependenciesInTriggerHelper, getTriggers } = require('./postgresHelpers/triggerHelper');
 const queryConstants = require('./queryConstants');
-const { reorganizePrimaryKeys } = require('./postgresHelpers/reorganizePrimaryKeys');
+const { reorganizeConstraints } = require('./postgresHelpers/reorganizeConstraints');
 
 let currentSshTunnel = null;
 let _ = null;
@@ -305,7 +305,7 @@ module.exports = {
 			targetAttributes = setSubtypeFromSampledJsonValues(targetAttributes, documents);
 		}
 
-		const { attributes, entityLevel: updatedEntityLevel } = reorganizePrimaryKeys(targetAttributes, entityLevel);
+		const { attributes, entityLevel: updatedEntityLevel } = reorganizeConstraints(targetAttributes, entityLevel);
 
 		return {
 			name: tableName,