Update token duplicate and removing to include attached tokens

src/components/konva/Token.tsx

@@ -145,7 +145,11 @@ function Token({
       }
     }
     setDragging(true);
-    onTokenDragStart(event, tokenState.id);
+    onTokenDragStart(
+      event,
+      tokenState.id,
+      attachedTokensRef.current.map((token) => token.id())
+    );
   }
 
   function handleDragMove(event: Konva.KonvaEventObject<DragEvent>) {
@@ -211,11 +215,16 @@ function Token({
     });
 
     setDragging(false);
+
+    onTokenDragEnd(
+      event,
+      tokenState.id,
+      attachedTokensRef.current.map((token) => token.id())
+    );
+
     attachmentCharactersRef.current = [];
     attachedTokensRef.current = [];
     setAttachmentOverCharacter(false);
-
-    onTokenDragEnd(event, tokenState.id);
   }
 
   function handleClick() {

src/components/token/TokenDragOverlay.tsx

@@ -1,23 +1,29 @@
 import DragOverlay from "../map/DragOverlay";
 import { TokenStateRemoveHandler } from "../../types/Events";
-import { TokenState } from "../../types/TokenState";
+import { TokenDraggingOptions } from "../../types/Token";
 
 type TokenDragOverlayProps = {
   onTokenStateRemove: TokenStateRemoveHandler;
-  tokenState: TokenState;
-  dragging: boolean;
+  draggingOptions: TokenDraggingOptions;
 };
 
 function TokenDragOverlay({
   onTokenStateRemove,
-  tokenState,
-  dragging,
+  draggingOptions,
 }: TokenDragOverlayProps) {
   function handleTokenRemove() {
-    onTokenStateRemove([tokenState.id]);
+    onTokenStateRemove([
+      draggingOptions.tokenStateId,
+      ...draggingOptions.attachedTokenStateIds,
+    ]);
   }
 
-  return <DragOverlay dragging={dragging} onRemove={handleTokenRemove} />;
+  return (
+    <DragOverlay
+      dragging={draggingOptions.dragging}
+      onRemove={handleTokenRemove}
+    />
+  );
 }
 
 export default TokenDragOverlay;

src/hooks/useMapTokens.tsx

@@ -60,17 +60,23 @@ function useMapTokens(
 
   function handleTokenDragStart(
     _: KonvaEventObject<DragEvent>,
-    tokenStateId: string
+    tokenStateId: string,
+    attachedTokenStateIds: string[]
   ) {
     if (duplicateToken) {
-      const state = mapState?.tokens[tokenStateId];
-      if (state) {
-        onTokensStateCreate([{ ...state, id: uuid() }]);
+      let newStates: TokenState[] = [];
+      for (let id of [tokenStateId, ...attachedTokenStateIds]) {
+        const state = mapState?.tokens[id];
+        if (state) {
+          newStates.push({ ...state, id: uuid() });
+        }
       }
+      onTokensStateCreate(newStates);
     }
     setTokenDraggingOptions({
       dragging: true,
       tokenStateId,
+      attachedTokenStateIds,
     });
   }
 
@@ -173,14 +179,10 @@ function useMapTokens(
     />
   );
 
-  const tokenDraggingState =
-    tokenDraggingOptions && mapState?.tokens[tokenDraggingOptions.tokenStateId];
-
-  const tokenDragOverlay = tokenDraggingOptions && tokenDraggingState && (
+  const tokenDragOverlay = tokenDraggingOptions && (
     <TokenDragOverlay
       onTokenStateRemove={handleTokenStateRemove}
-      tokenState={tokenDraggingState}
-      dragging={!!(tokenDraggingOptions && tokenDraggingOptions.dragging)}
+      draggingOptions={tokenDraggingOptions}
     />
   );
 

src/types/Events.ts

@@ -34,7 +34,8 @@ export type TokenMenuCloseChangeEventHandler = () => void;
 export type TokenSettingsChangeEventHandler = (change: Partial<Token>) => void;
 export type TokenDragEventHandler = (
   event: Konva.KonvaEventObject<DragEvent>,
-  tokenStateId: string
+  tokenStateId: string,
+  attachedTokenStateIds: string[]
 ) => void;
 
 export type NoteCreateEventHander = (notes: Note[]) => void;

src/types/Token.ts

@@ -39,4 +39,5 @@ export type TokenMenuOptions = {
 export type TokenDraggingOptions = {
   dragging: boolean;
   tokenStateId: string;
+  attachedTokenStateIds: string[];
 };