Conditional Norms

src/pages/VisProgPage/visualProgrammingUI/nodes/NormNode.tsx

@@ -8,6 +8,7 @@ import { Toolbar } from '../components/NodeComponents';
 import styles from '../../VisProg.module.css';
 import { TextField } from '../../../../components/TextField';
 import useFlowStore from '../VisProgStores';
+import { BasicBeliefReduce } from './BasicBeliefNode';
 
 /**
  * The default data dot a phase node
@@ -19,6 +20,7 @@ import useFlowStore from '../VisProgStores';
 export type NormNodeData = {
   label: string;
   droppable: boolean;
+  conditions: string[]; // List of (basic) belief nodes' ids.
   norm: string;
   hasReduce: boolean;
   critical: boolean;
@@ -67,7 +69,14 @@ export default function NormNode(props: NodeProps<NormNode>) {
           onChange={(e) => setCritical(e.target.checked)}
         />
       </div>
+
+      {data.conditions.length > 0 && (<div className={"flex-row gap-md align-center"} data-testid="norm-condition-information">
+        <label htmlFor={checkbox_id}>{data.conditions.length} condition{data.conditions.length > 1 ? "s" : ""}/ belief{data.conditions.length > 1 ? "s" : ""} attached.</label>
+      </div>)}
+
+  
       <Handle type="source" position={Position.Right} id="norms"/>
+      <Handle type="target" position={Position.Bottom} id="norms"/>
     </div>
   </>;
 };
@@ -78,14 +87,29 @@ export default function NormNode(props: NodeProps<NormNode>) {
  * @param node The Node Properties of this node.
  * @param _nodes all the nodes in the graph
  */
-export function NormReduce(node: Node, _nodes: Node[]) {
+export function NormReduce(node: Node, nodes: Node[]) {
   const data = node.data as NormNodeData;
-    return {
-      id: node.id,
-      label: data.label,
-      norm: data.norm,
-      critical: data.critical,
-    }
+
+  // conditions nodes - make sure to check for empty arrays
+  let conditionNodes: Node[] = [];
+  if (data.conditions) 
+    conditionNodes = nodes.filter((node) => data.conditions.includes(node.id));
+
+   // Build the result object
+  const result: Record<string, unknown> = {
+    id: node.id,
+    label: data.label,
+    norm: data.norm,
+    critical: data.critical,
+  };
+
+  // Go over our conditionNodes. They should either be Basic (OR TODO: Inferred)
+  const reducer = BasicBeliefReduce;
+  result["basic_beliefs"] = conditionNodes.map((condition) => reducer(condition, nodes))
+
+  // When the Inferred is being implemented, you should follow the same kind of structure that PhaseNode has,
+  // dividing the conditions into basic and inferred, then calling the correct reducer on them.
+  return result
 }
 
 /**
@@ -94,7 +118,11 @@ export function NormReduce(node: Node, _nodes: Node[]) {
  * @param _sourceNodeId the source of the received connection
  */
 export function NormConnectionTarget(_thisNode: Node, _sourceNodeId: string) {
-  // no additional connection logic exists yet
+  const data = _thisNode.data as NormNodeData;
+  // If we got a belief connected, this is a condition for the norm.
+  if ((useFlowStore.getState().nodes.find((node) => node.id === _sourceNodeId && node.type === 'basic_belief' /* TODO: Add the option for an inferred belief */))) {
+      data.conditions.push(_sourceNodeId);
+    }
 }
 
 /**
@@ -112,7 +140,11 @@ export function NormConnectionSource(_thisNode: Node, _targetNodeId: string) {
  * @param _sourceNodeId the source of the disconnected connection
  */
 export function NormDisconnectionTarget(_thisNode: Node, _sourceNodeId: string) {
-  // no additional connection logic exists yet
+  const data = _thisNode.data as NormNodeData;
+  // If we got a belief connected, this is a condition for the norm.
+  if ((useFlowStore.getState().nodes.find((node) => node.id === _sourceNodeId && node.type === 'basic_belief' /* TODO: Add the option for an inferred belief */))) {
+    data.conditions = data.conditions.filter(id => id != _sourceNodeId);
+    }
 }
 
 /**