redraw arrows when editing existing document

NDLANO · Mar 8, 2022 · 7804f69 · 7804f69
1 parent 1dfd205
commit 7804f69
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 1 deletion.
diff --git a/src/components/Grid/Grid.tsx b/src/components/Grid/Grid.tsx
@@ -104,7 +104,7 @@ export const Grid: FC<GridProps> = ({
   const [arrowItems, setArrowItems] = useState(initialArrowItems ?? []);
   const [classicArrowItems, setClassicArrowItems] = useState<
     Array<ClassicArrowItemType>
-  >([]);
+  >(initialArrowItems as ClassicArrowItemType[] ?? []);
   const [occupiedCells, setOccupiedCells] = useState<Array<OccupiedCell>>([]);
   const [boxStartIndex, setBoxStartIndex] = useState<number | null>(null);
   const [isDragging, setIsDragging] = useState(false);
@@ -136,6 +136,7 @@ export const Grid: FC<GridProps> = ({
   useEffectOnce(() => {
     // eslint-disable-next-line no-param-reassign
     updateGrid.current = updateLocalGrid;
+    updateGridDimensions({ numberOfColumns, numberOfRows });
   });
 
   const elementRef = useRef<HTMLDivElement>(null);

diff --git a/src/types/ArrowItemType.ts b/src/types/ArrowItemType.ts
@@ -17,4 +17,7 @@ export type ArrowItemType = CommonItemType & {
 
   startPosition: Position;
   endPosition: Position;
+
+  startGridPosition: Position;
+  endGridPosition: Position;
 };