Merge pull request #62 from ArzamastsevVladyslav/feature/HCK-3559-pk-fk-constraiints

Feature/hck 3559 pk fk constraiints

Author: ArzamastsevVladyslav

forward_engineering/alterScript/alterScriptBuilder.js

@@ -0,0 +1,103 @@
+const {getAlterScriptDtos} = require("./alterScriptFromDeltaHelper");
+
+const {AlterScriptDto} = require('./types/AlterScriptDto');
+
+/**
+ * @return {(dtos: AlterScriptDto[], shouldApplyDropStatements: boolean) => string}
+ * */
+const joinAlterScriptDtosIntoScript = (_) => (dtos, shouldApplyDropStatements) => {
+    const {commentIfDeactivated} = require('../utils/general')(_);
+    return dtos.map((dto) => {
+        if (dto.isActivated === false) {
+            return dto.scripts
+                .map((scriptDto) => commentIfDeactivated(scriptDto.script, {
+                    isActivated: false,
+                    isPartOfLine: false,
+                }));
+        }
+        if (!shouldApplyDropStatements) {
+            return dto.scripts
+                .map((scriptDto) => commentIfDeactivated(scriptDto.script, {
+                    isActivated: !scriptDto.isDropScript,
+                    isPartOfLine: false,
+                }));
+        }
+        return dto.scripts.map((scriptDto) => scriptDto.script);
+    })
+        .flat()
+        .filter(Boolean)
+        .map((scriptLine) => scriptLine.trim())
+        .filter(Boolean)
+        .join('\n\n');
+}
+
+/**
+ * @param data {CoreData}
+ * @param app {App}
+ * @return {string}
+ * */
+const buildEntityLevelAlterScript = (data, app) => {
+    const _ = app.require('lodash');
+    const alterScriptDtos = getAlterScriptDtos(data, app);
+    const shouldApplyDropStatements = data.options?.additionalOptions?.some(
+        option => option.id === 'applyDropStatements' && option.value,
+    );
+
+    return joinAlterScriptDtosIntoScript(_)(alterScriptDtos, shouldApplyDropStatements);
+}
+
+/**
+ * @param data {CoreData}
+ * @param app {App}
+ * @return { boolean}
+ * */
+const doesEntityLevelAlterScriptContainDropStatements = (data, app) => {
+    const alterScriptDtos = getAlterScriptDtos(data, app);
+    return alterScriptDtos
+        .some(alterScriptDto => alterScriptDto.isActivated && alterScriptDto
+            .scripts.some(scriptModificationDto => scriptModificationDto.isDropScript));
+}
+
+const mapCoreDataForContainerLevelScripts = (data) => {
+    return {
+        ...data,
+        jsonSchema: data.collections[0],
+        internalDefinitions: Object.values(data.internalDefinitions)[0],
+    }
+}
+
+/**
+ * @param data {CoreData}
+ * @param app {App}
+ * @return {string}
+ * */
+const buildContainerLevelAlterScript = (data, app) => {
+    const preparedData = mapCoreDataForContainerLevelScripts(data);
+    const _ = app.require('lodash');
+    const alterScriptDtos = getAlterScriptDtos(preparedData, app);
+    const shouldApplyDropStatements = preparedData.options?.additionalOptions?.some(
+        option => option.id === 'applyDropStatements' && option.value,
+    );
+
+    return joinAlterScriptDtosIntoScript(_)(alterScriptDtos, shouldApplyDropStatements);
+}
+
+/**
+ * @param data {CoreData}
+ * @param app {App}
+ * @return { boolean}
+ * */
+const doesContainerLevelAlterScriptContainDropStatements = (data, app) => {
+    const preparedData = mapCoreDataForContainerLevelScripts(data);
+    const alterScriptDtos = getAlterScriptDtos(preparedData, app);
+    return alterScriptDtos
+        .some(alterScriptDto => alterScriptDto.isActivated && alterScriptDto
+            .scripts.some(scriptModificationDto => scriptModificationDto.isDropScript));
+}
+
+module.exports = {
+    buildEntityLevelAlterScript,
+    doesEntityLevelAlterScriptContainDropStatements,
+    buildContainerLevelAlterScript,
+    doesContainerLevelAlterScriptContainDropStatements
+}