From 1a90a5374b60eed3a75014731b2cedd8bf416636 Mon Sep 17 00:00:00 2001
From: mohamedsalem401 <salemmohamed780@gmail.com>
Date: Thu, 13 Mar 2025 07:34:32 +0200
Subject: [PATCH] Refactor DataCondition

---
 .../conditional_variables/DataCondition.ts    | 24 +++++++++----------
 1 file changed, 12 insertions(+), 12 deletions(-)

diff --git a/packages/core/src/data_sources/model/conditional_variables/DataCondition.ts b/packages/core/src/data_sources/model/conditional_variables/DataCondition.ts
index fc72ef2f0..483b399b9 100644
--- a/packages/core/src/data_sources/model/conditional_variables/DataCondition.ts
+++ b/packages/core/src/data_sources/model/conditional_variables/DataCondition.ts
@@ -26,8 +26,8 @@ export interface LogicGroupProps {
 export interface DataConditionProps {
   type: typeof DataConditionType;
   condition: ConditionProps;
-  ifTrue: any;
-  ifFalse: any;
+  ifTrue?: any;
+  ifFalse?: any;
 }
 
 interface DataConditionPropsDefined extends Omit<DataConditionProps, 'condition'> {
@@ -42,8 +42,8 @@ export class DataCondition extends Model<DataConditionPropsDefined> {
   constructor(
     props: {
       condition: ConditionProps;
-      ifTrue: any;
-      ifFalse: any;
+      ifTrue?: any;
+      ifFalse?: any;
     },
     opts: { em: EditorModel; onValueChange?: () => void },
   ) {
@@ -62,7 +62,7 @@ export class DataCondition extends Model<DataConditionPropsDefined> {
     this.listenToDataVariables();
     this._onValueChange = opts.onValueChange;
 
-    this.on('change:condition change:ifTrue change:ifFalse', () => {
+    this.listenTo(this, 'change:condition change:ifTrue change:ifFalse', () => {
       this.listenToDataVariables();
       this._onValueChange?.();
     });
@@ -89,8 +89,8 @@ export class DataCondition extends Model<DataConditionPropsDefined> {
   }
 
   getDataValue(skipDynamicValueResolution: boolean = false): any {
-    const ifTrue = this.get('ifTrue');
-    const ifFalse = this.get('ifFalse');
+    const ifTrue = this.getIfTrue();
+    const ifFalse = this.getIfFalse();
 
     const isConditionTrue = this.isTrue();
     if (skipDynamicValueResolution) {
@@ -141,8 +141,8 @@ export class DataCondition extends Model<DataConditionPropsDefined> {
 
   getDependentDataVariables() {
     const dataVariables: DataVariableProps[] = this.conditionEvaluator.getDependentDataVariables();
-    const ifTrue = this.get('ifTrue');
-    const ifFalse = this.get('ifFalse');
+    const ifTrue = this.getIfTrue();
+    const ifFalse = this.getIfFalse();
     if (isDataVariable(ifTrue)) dataVariables.push(ifTrue);
     if (isDataVariable(ifFalse)) dataVariables.push(ifFalse);
 
@@ -155,12 +155,12 @@ export class DataCondition extends Model<DataConditionPropsDefined> {
   }
 
   toJSON() {
-    const ifTrue = this.get('ifTrue');
-    const ifFalse = this.get('ifFalse');
+    const ifTrue = this.getIfTrue();
+    const ifFalse = this.getIfFalse();
 
     return {
       type: DataConditionType,
-      condition: this.conditionEvaluator,
+      condition: this.conditionEvaluator.toJSON(),
       ifTrue,
       ifFalse,
     };