HCK-7253: fix getting target script for documentation

hackolade · Jul 16, 2024 · 0863ade · 0863ade
1 parent d625a20
commit 0863ade
Show file tree

Hide file tree

Showing 3 changed files with 16 additions and 6 deletions.
diff --git a/forward_engineering/api.js b/forward_engineering/api.js
@@ -18,10 +18,7 @@ module.exports = {
 					(data.entityData[entityId] || [])[0] || {},
 				),
 			);
-			if (
-				data.options.targetScriptOptions &&
-				data.options.targetScriptOptions.keyword === 'containerSettingsJson'
-			) {
+			if (data.options?.targetScriptOptions?.keyword === 'containerSettingsJson') {
 				const uniqueKeys = _.get(data.containerData, '[0].uniqueKey', []);
 				const scriptData = {
 					partitionKey: getPartitionKey(_)(data.containerData),
@@ -43,7 +40,6 @@ module.exports = {
 
 			const script = buildAzureCLIScript(_)({
 				...data,
-				shellName: data.options.targetScriptOptions.keyword.split('azureCli')[1].toLowerCase(),
 			});
 
 			if (withSamples || !insertSamplesOption.value) {

diff --git a/forward_engineering/helpers/azureCLIScriptHelpers/buildAzureCLIScript.js b/forward_engineering/helpers/azureCLIScriptHelpers/buildAzureCLIScript.js
@@ -4,6 +4,7 @@ const { getUniqueKeyPolicyScript } = require('../getUniqueKeyPolicyScript');
 const { getCliParamsDelimiter } = require('./getCliParamsDelimiter');
 const getIndexPolicyScript = require('../getIndexPolicyScript');
 const getPartitionKey = require('../getPartitionKey');
+const { getCliShellName } = require('./getCliShellName');
 const {
 	CLI,
 	DATABASE,
@@ -16,7 +17,8 @@ const {
 
 const buildAzureCLIScript =
 	_ =>
-	({ modelData, containerData, shellName }) => {
+	({ modelData, containerData, options }) => {
+		const shellName = getCliShellName(options?.targetScriptOptions);
 		const cliParamsDelimiter = getCliParamsDelimiter(shellName);
 		const escapeAndWrapInQuotes = string => wrapInSingleQuotes(escapeShellCommand(shellName, string));
 

diff --git a/forward_engineering/helpers/azureCLIScriptHelpers/getCliShellName.js b/forward_engineering/helpers/azureCLIScriptHelpers/getCliShellName.js
@@ -0,0 +1,12 @@
+/**
+ * @param {object} targetScriptOptions
+ * @returns {string}
+ */
+const getCliShellName = (targetScriptOptions = {}) => {
+	const keyword = targetScriptOptions.keyword ?? '';
+	const [, shellName = ''] = keyword.split('azureCli');
+
+	return shellName.toLowerCase();
+};
+
+module.exports = { getCliShellName };