fix: added missing functions

hackolade · Nov 14, 2023 · 3cc5777 · 3cc5777
1 parent 8372d5b
commit 3cc5777
Showing 1 changed file with 9 additions and 0 deletions.
diff --git a/forward_engineering/helpers/commentDeactivatedHelper.js b/forward_engineering/helpers/commentDeactivatedHelper.js
@@ -1,3 +1,6 @@
+const MULTILINE_COMMENT = /(\n\/\*\n[\s\S]*?\n\s\*\/\n)|(\n\/\*\n[\s\S]*?\n\s\*\/)$/gi;
+const STARTS_QUERY = ['//', '--'];
+
 module.exports = (_) => {
 	const commentIfDeactivated = (statement, data, isPartOfLine) => {
 		if (_.has(data, 'isActivated') && !data.isActivated) {
@@ -12,7 +15,13 @@ module.exports = (_) => {
 		return statement;
 	};
 
+	const filterDeactivatedQuery = query => query.replace(MULTILINE_COMMENT, '');
+
+	const queryIsDeactivated = (query = '') => STARTS_QUERY.some(startQuery => query.startsWith(startQuery));
+
 	return {
 		commentIfDeactivated,
+		filterDeactivatedQuery,
+		queryIsDeactivated
 	};
 };