From 666dd7dd439df4d84a1f4ad9ba555f3486cf3eb3 Mon Sep 17 00:00:00 2001
From: F-star <hhbuhuiai@163.com>
Date: Sat, 12 Oct 2024 14:35:24 +0800
Subject: [PATCH] feat: support paste image

---
 packages/core/src/clipboard.ts | 50 +++++++++++++++++++++++++++++++++-
 1 file changed, 49 insertions(+), 1 deletion(-)

diff --git a/packages/core/src/clipboard.ts b/packages/core/src/clipboard.ts
index 6f0628c0..0f6af865 100644
--- a/packages/core/src/clipboard.ts
+++ b/packages/core/src/clipboard.ts
@@ -8,8 +8,14 @@ import {
 import { generateNKeysBetween } from 'fractional-indexing';
 
 import { type SuikaEditor } from './editor';
-import { type GraphicsAttrs, isFrameGraphics, SuikaGraphics } from './graphics';
+import {
+  type GraphicsAttrs,
+  isFrameGraphics,
+  SuikaGraphics,
+  SuikaRect,
+} from './graphics';
 import { isCanvasGraphics } from './graphics/canvas';
+import { PaintType } from './paint';
 import { toSVG } from './to_svg';
 import { Transaction } from './transaction';
 import { type IEditorPaperData } from './type';
@@ -46,6 +52,21 @@ export class ClipboardManager {
       ) {
         return;
       }
+
+      if (clipboardData.files.length > 0) {
+        for (const file of clipboardData.files) {
+          if (file.type.includes('image')) {
+            const reader = new FileReader();
+            reader.onload = (e) => {
+              const base64 = e.target?.result as string;
+              this.createGraphicsWithImg(base64);
+            };
+            reader.readAsDataURL(file);
+          }
+        }
+        return;
+      }
+
       const pastedData = clipboardData.getData('Text');
       this.addGraphsFromClipboard(pastedData);
     };
@@ -64,6 +85,33 @@ export class ClipboardManager {
     };
   }
 
+  private async createGraphicsWithImg(imgUrl: string) {
+    const editor = this.editor;
+    await editor.imgManager.addImg(imgUrl);
+    const img = editor.imgManager.getImg(imgUrl);
+    const center = editor.viewportManager.getCenter();
+    if (img) {
+      const rectGraphics = new SuikaRect(
+        {
+          objectName: '',
+          width: img.width,
+          height: img.height,
+          fill: [{ type: PaintType.Image, attrs: { src: imgUrl } }],
+        },
+        {
+          advancedAttrs: {
+            x: center.x - img.width / 2,
+            y: center.y - img.height / 2,
+          },
+          doc: editor.doc,
+        },
+      );
+      editor.sceneGraph.addItems([rectGraphics]);
+      editor.doc.getCanvas().insertChild(rectGraphics);
+      editor.render();
+    }
+  }
+
   copy() {
     const snapshot = this.getSelectedItemsSnapshot();
     if (!snapshot) {