From e39fe772bbc4ac63b7376ade10e19c4104badc9d Mon Sep 17 00:00:00 2001
From: Peng Xiao <pengxiao@outlook.com>
Date: Mon, 20 Jun 2022 03:13:23 +0800
Subject: [PATCH] fix: paste shapes with bindings

---
 .../apps/tldraw-logseq/src/hooks/usePaste.ts  | 38 ++++++++++++++++---
 tldraw/packages/core/src/lib/TLApp/TLApp.ts   |  8 +++-
 2 files changed, 40 insertions(+), 6 deletions(-)

diff --git a/tldraw/apps/tldraw-logseq/src/hooks/usePaste.ts b/tldraw/apps/tldraw-logseq/src/hooks/usePaste.ts
index ecb953d26..1db968f14 100644
--- a/tldraw/apps/tldraw-logseq/src/hooks/usePaste.ts
+++ b/tldraw/apps/tldraw-logseq/src/hooks/usePaste.ts
@@ -3,6 +3,7 @@ import {
   fileToBase64,
   getSizeFromSrc,
   TLAsset,
+  TLBinding,
   TLShapeModel,
   uniqueId,
 } from '@tldraw/core'
@@ -19,6 +20,7 @@ export function usePaste() {
 
     const assetsToCreate: ImageAsset[] = []
     const shapesToCreate: TLShapeModel[] = []
+    const bindingsToCreate: TLBinding[] = []
 
     async function handleImage(item: ClipboardItem) {
       const firstImageType = item.types.find(type => type.startsWith('image'))
@@ -62,10 +64,9 @@ export function usePaste() {
               maxY: (shape.point?.[1] ?? point[1]) + (shape.size?.[1] ?? 4),
             }))
           )
-          const clonedShape = data.shapes.map((shape: TLShapeModel) => {
+          const clonedShapes = shapes.map((shape: TLShapeModel) => {
             return {
               ...shape,
-              handles: {}, // TODO: may add this later?
               id: uniqueId(),
               parentId: app.currentPageId,
               point: [
@@ -74,7 +75,34 @@ export function usePaste() {
               ],
             }
           })
-          shapesToCreate.push(...clonedShape)
+          shapesToCreate.push(...clonedShapes)
+
+          // Try to rebinding the shapes to the new assets
+          shapesToCreate.forEach((s, idx) => {
+            if (s.handles) {
+              Object.values(s.handles).forEach(h => {
+                if (h.bindingId) {
+                  // try to bind the new shape
+                  const binding = app.currentPage.bindings[h.bindingId]
+                  // if the copied binding from/to is in the source
+                  const oldFromIdx = shapes.findIndex(s => s.id === binding.fromId)
+                  const oldToIdx = shapes.findIndex(s => s.id === binding.toId)
+                  if (binding && oldFromIdx !== -1 && oldToIdx !== -1) {
+                    const newBinding: TLBinding = {
+                      ...binding,
+                      id: uniqueId(),
+                      fromId: shapesToCreate[oldFromIdx].id,
+                      toId: shapesToCreate[oldToIdx].id,
+                    }
+                    bindingsToCreate.push(newBinding)
+                    h.bindingId = newBinding.id
+                  } else {
+                    h.bindingId = undefined
+                  }
+                }
+              })
+            }
+          })
         }
       }
     }
@@ -91,7 +119,7 @@ export function usePaste() {
       }
     }
 
-    const allShapesToAdd = [
+    const allShapesToAdd: TLShapeModel[] = [
       ...assetsToCreate.map((asset, i) => ({
         id: uniqueId(),
         type: 'image',
@@ -103,9 +131,9 @@ export function usePaste() {
       })),
       ...shapesToCreate,
     ]
-
     app.createAssets(assetsToCreate)
     app.createShapes(allShapesToAdd)
+    app.currentPage.updateBindings(Object.fromEntries(bindingsToCreate.map(b => [b.id, b])))
 
     app.setSelectedShapes(allShapesToAdd.map(s => s.id))
   }, [])
diff --git a/tldraw/packages/core/src/lib/TLApp/TLApp.ts b/tldraw/packages/core/src/lib/TLApp/TLApp.ts
index cdafc74c8..9e189f308 100644
--- a/tldraw/packages/core/src/lib/TLApp/TLApp.ts
+++ b/tldraw/packages/core/src/lib/TLApp/TLApp.ts
@@ -2,7 +2,7 @@
 /* eslint-disable @typescript-eslint/no-non-null-assertion */
 /* eslint-disable @typescript-eslint/no-explicit-any */
 import { Vec } from '@tldraw/vec'
-import { action, computed, makeObservable, observable } from 'mobx'
+import { action, computed, makeObservable, observable, toJS } from 'mobx'
 import { BoundsUtils, KeyUtils } from '~utils'
 import {
   TLSelectTool,
@@ -470,6 +470,12 @@ export class TLApp<
     } else {
       this.selectionRotation = 0
     }
+    if (process.env.NODE_ENV === 'development') {
+      console.log(
+        'setSelectedShapes',
+        newSelectedShapes.map(s => toJS(s.serialized))
+      )
+    }
     return this
   }