From f987f351cd59fbbf51c99588cd8244ab31bb6e8d Mon Sep 17 00:00:00 2001
From: maxime <maximebatista18@gmail.com>
Date: Mon, 25 Mar 2024 16:50:26 +0100
Subject: [PATCH] fix steps in visualizer

---
 src/components/editor/StepsTree.tsx | 56 ++++++++++++++---------------
 src/domains/TacticContentDomains.ts | 20 ++++++++++-
 src/style/player.css                |  2 +-
 3 files changed, 48 insertions(+), 30 deletions(-)

diff --git a/src/components/editor/StepsTree.tsx b/src/components/editor/StepsTree.tsx
index 0015a18..ddb19ee 100644
--- a/src/components/editor/StepsTree.tsx
+++ b/src/components/editor/StepsTree.tsx
@@ -15,12 +15,12 @@ export interface StepsTreeProps {
 }
 
 export default function StepsTree({
-    root,
-    selectedStepId,
-    onAddChildren,
-    onRemoveNode,
-    onStepSelected,
-}: StepsTreeProps) {
+                                      root,
+                                      selectedStepId,
+                                      onAddChildren,
+                                      onRemoveNode,
+                                      onStepSelected,
+                                  }: StepsTreeProps) {
     return (
         <div className="steps-tree">
             <StepsTreeNode
@@ -46,13 +46,13 @@ interface StepsTreeContentProps {
 }
 
 function StepsTreeNode({
-    node,
-    rootNode,
-    selectedStepId,
-    onAddChildren,
-    onRemoveNode,
-    onStepSelected,
-}: StepsTreeContentProps) {
+                           node,
+                           rootNode,
+                           selectedStepId,
+                           onAddChildren,
+                           onRemoveNode,
+                           onStepSelected,
+                       }: StepsTreeContentProps) {
     const ref = useRef<HTMLDivElement>(null)
 
     return (
@@ -67,7 +67,8 @@ function StepsTreeNode({
                             next: "step-piece-" + child.id,
                         },
                     ]}
-                    onSegmentsChanges={() => {}}
+                    onSegmentsChanges={() => {
+                    }}
                     forceStraight={true}
                     wavy={false}
                     readOnly={true}
@@ -79,15 +80,14 @@ function StepsTreeNode({
             <StepPiece
                 id={node.id}
                 isSelected={selectedStepId === node.id}
-                onAddButtonClicked={() => {
-                    if (onAddChildren) onAddChildren(node)
-                }}
+                onAddButtonClicked={onAddChildren
+                    ? () => onAddChildren(node)
+                    : undefined
+                }
                 onRemoveButtonClicked={
-                    rootNode.id === node.id
+                    rootNode.id === node.id || !onRemoveNode
                         ? undefined
-                        : () => {
-                              if (onRemoveNode) onRemoveNode(node)
-                          }
+                        : () => onRemoveNode(node)
                 }
                 onSelected={() => {
                     if (onStepSelected) onStepSelected(node)
@@ -126,13 +126,13 @@ interface StepPieceProps {
 }
 
 function StepPiece({
-    id,
-    isSelected,
-    onAddButtonClicked,
-    onRemoveButtonClicked,
-    onSelected,
-    children,
-}: StepPieceProps) {
+                       id,
+                       isSelected,
+                       onAddButtonClicked,
+                       onRemoveButtonClicked,
+                       onSelected,
+                       children,
+                   }: StepPieceProps) {
     return (
         <div
             id={"step-piece-" + id}
diff --git a/src/domains/TacticContentDomains.ts b/src/domains/TacticContentDomains.ts
index 07fba44..7774f36 100644
--- a/src/domains/TacticContentDomains.ts
+++ b/src/domains/TacticContentDomains.ts
@@ -408,6 +408,7 @@ export function drainTerminalStateOnChildContent(
             childContent.components,
         )
 
+
         if (!childComponent) {
             //if the child does not contain the parent's component, add it to the children's content.
             childContent = {
@@ -418,6 +419,22 @@ export function drainTerminalStateOnChildContent(
             continue
         }
 
+        if (childComponent.type !== parentComponent.type)
+            throw Error("child and parent components are not of the same type.")
+
+        if (childComponent.type === "ball" && parentComponent.type === "ball") {
+            gotUpdated = true
+            childContent = updateComponent(
+                {
+                    ...childComponent,
+                    frozen: true,
+                    pos: parentComponent.pos,
+                },
+                childContent,
+            )
+
+        }
+
         // ensure that the component is a player
         if (
             parentComponent.type !== "player" ||
@@ -439,7 +456,7 @@ export function drainTerminalStateOnChildContent(
                 newContentResult?.components,
             )
         }
-        // update the position of the player if it has been moved
+        // update the position of the component if it has been moved
         // also force update if the child component is not frozen (the component was introduced previously by the child step but the parent added it afterward)
         if (
             !childComponent.frozen ||
@@ -459,6 +476,7 @@ export function drainTerminalStateOnChildContent(
 
     const initialChildCompsCount = childContent.components.length
 
+    //remove players if they are not present on the parent's anymore
     for (const component of childContent.components) {
         if (
             component.type !== "phantom" &&
diff --git a/src/style/player.css b/src/style/player.css
index 8817474..13b0e11 100644
--- a/src/style/player.css
+++ b/src/style/player.css
@@ -8,7 +8,7 @@
 
 .from-parent .player-piece {
     color: white;
-    background-color: var(--player-from-parent-color);
+    background-color: var(--player-from-parent-color) !important;
 }
 
 .player-content {