pepperplus-ui/src/pages/VisProgPage/visualProgrammingUI/nodes/GoalNode.tsx

import {
  type NodeProps,
  Position,
  type Node,
} from '@xyflow/react';
import { Toolbar } from '../components/NodeComponents';
import styles from '../../VisProg.module.css';
import { TextField } from '../../../../components/TextField';
import {MultiConnectionHandle} from "../components/RuleBasedHandle.tsx";
import {allowOnlyConnectionsFromHandle, allowOnlyConnectionsFromType} from "../HandleRules.ts";
import useFlowStore from '../VisProgStores';
import {DoesPlanIterate, PlanReduce, type Plan } from '../components/Plan';
import PlanEditorDialog from '../components/PlanEditor';
import { MultilineTextField } from '../../../../components/MultilineTextField';
import { defaultPlan } from '../components/Plan.default.ts';
import { deleteGoalInPlanByID, insertGoalInPlan } from '../components/PlanEditingFunctions.tsx';

/**
 * The default data dot a phase node
 * @param label: the label of this phase
 * @param droppable: whether this node is droppable from the drop bar (initialized as true)
 * @param desciption: description of the goal - this will be checked for completion
 * @param hasReduce: whether this node has reducing functionality (true by default)
 * @param can_fail: whether this plan should be checked- this plan could possible fail
 * @param plan: The (possible) attached plan to this goal
 */
export type GoalNodeData = {
  label: string;
  name: string;
  description: string;
  droppable: boolean;
  achieved: boolean;
  hasReduce: boolean;
  can_fail: boolean;
  plan?: Plan;
};

export type GoalNode = Node<GoalNodeData>


/**
 * Defines how a Goal node should be rendered
 * @param props NodeProps, like id, label, children
 * @returns React.JSX.Element
 */
export default function GoalNode({id, data}: NodeProps<GoalNode>) {
  const {updateNodeData} = useFlowStore();

  const text_input_id = `goal_${id}_text_input`;
  const checkbox_id = `goal_${id}_checkbox`;
  const planIterate = DoesPlanIterate(data.plan);

  const setDescription = (value: string) => {
    updateNodeData(id, {...data, description: value});
  }

  const setName= (value: string) => {
    updateNodeData(id, {...data, name: value})
  }

  const setFailable = (value: boolean) => {
    updateNodeData(id, {...data, can_fail: value});
  }

  return <>
    <Toolbar nodeId={id} allowDelete={true}/>
    <div className={`${styles.defaultNode} ${styles.nodeGoal} flex-col gap-sm`}>
      <div className={"flex-row gap-md"}>
        <label htmlFor={text_input_id}>Goal:</label>
        <TextField
          id={text_input_id}
          value={data.name}
          setValue={(val) => setName(val)}
          placeholder={"To ..."}
        />
      </div>
      
      {data.can_fail && (<div>
        <label htmlFor={text_input_id}>Description/ Condition of goal:</label>
        <div className={"flex-wrap"}>
          <MultilineTextField
            id={text_input_id}
            value={data.description}
            setValue={setDescription}
            placeholder={"Describe the condition of this goal..."}
          />
        </div>
      </div>)}
      <div>
        <label> {!data.plan ? "No plan set to execute while goal is not reached. 🔴" : "Will follow plan '" + data.plan.name + "' until all steps complete. 🟢"} </label>
      </div>
      {data.plan && (<div className={"flex-row gap-md align-center " + (planIterate ? "" : styles.planNoIterate)}>
        {planIterate ? "" : <s></s>}
        <label htmlFor={checkbox_id}>{!planIterate ? "This plan always succeeds!" : "Check if this plan fails"}:</label>
        <input
          id={checkbox_id}
          type={"checkbox"}
          disabled={!planIterate}
          checked={!planIterate || data.can_fail}
          onChange={(e) => planIterate ? setFailable(e.target.checked) : setFailable(false)}
        />
      </div>
)}

      <div>
        <PlanEditorDialog
                  plan={data.plan}
                  onSave={(plan) => {
                    updateNodeData(id, {
                      ...data,
                      plan,
                    });
                  }} 
                  description={data.name}
        />
      </div>
      <MultiConnectionHandle type="source" position={Position.Right} id="GoalSource" rules={[
        allowOnlyConnectionsFromHandle([{nodeType:"phase",handleId:"data"}]),
      ]}/>

      <MultiConnectionHandle type="target" position={Position.Bottom} id="GoalTarget" rules={[allowOnlyConnectionsFromType(["goal"])]}/>


    </div>
  </>;
}


/**
 * Reduces each Goal, including its children down into its relevant data.
 * @param node The Node Properties of this node.
 * @param _nodes all the nodes in the graph
 */
export function GoalReduce(node: Node, _nodes: Node[]) {
  const data = node.data as GoalNodeData;
    return {  
      id: node.id,
      name: data.name,
      description: data.description,
      can_fail: data.can_fail,  
      plan: data.plan ? PlanReduce(_nodes, data.plan) : "",
    }
}

/**
 * This function is called whenever a connection is made with this node type as the target
 * @param _thisNode the node of this node type which function is called
 * @param _sourceNodeId the source of the received connection
 */
export function GoalConnectionTarget(_thisNode: Node, _sourceNodeId: string) {
  // Goals should only be targeted by other goals, for them to be part of our plan.
  const nodes = useFlowStore.getState().nodes;
  const otherNode = nodes.find((x) => x.id === _sourceNodeId)
  if (!otherNode || otherNode.type !== "goal") return;

  const data = _thisNode.data as GoalNodeData 

  // First, let's see if we have a plan currently. If not, let's create a default plan with this goal inside.:)
  if (!data.plan) {
    data.plan = insertGoalInPlan({...structuredClone(defaultPlan), id: crypto.randomUUID()} as Plan, otherNode as GoalNode)
  }
  
  // Else, lets just insert this goal into our current plan.
  else {
    data.plan = insertGoalInPlan(structuredClone(data.plan), otherNode as GoalNode)
  }
}

/**
 * This function is called whenever a connection is made with this node type as the source
 * @param _thisNode the node of this node type which function is called
 * @param _targetNodeId the target of the created connection
 */
export function GoalConnectionSource(_thisNode: Node, _targetNodeId: string) {
  // no additional connection logic exists yet
}

/**
 * This function is called whenever a connection is disconnected with this node type as the target
 * @param _thisNode the node of this node type which function is called
 * @param _sourceNodeId the source of the disconnected connection
 */
export function GoalDisconnectionTarget(_thisNode: Node, _sourceNodeId: string) {
  // We should probably check if our disconnection was by a goal, since it would mean we have to remove it from our plan list.
  const data = _thisNode.data as GoalNodeData 
  data.plan = deleteGoalInPlanByID(structuredClone(data.plan) as Plan, _sourceNodeId)
}

/**
 * This function is called whenever a connection is disconnected with this node type as the source
 * @param _thisNode the node of this node type which function is called
 * @param _targetNodeId the target of the diconnected connection
 */
export function GoalDisconnectionSource(_thisNode: Node, _targetNodeId: string) {
  // no additional connection logic exists yet
}