feat: add copy image in share menu

invoke-ai · Feb 4, 2023 · 81e174b · 81e174b
1 parent 702a929
commit 81e174b
Show file tree

Hide file tree

Showing 2 changed files with 25 additions and 0 deletions.
diff --git a/invokeai/frontend/public/locales/parameters/en.json b/invokeai/frontend/public/locales/parameters/en.json
@@ -46,6 +46,7 @@
   "sendTo": "Send to",
   "sendToImg2Img": "Send to Image to Image",
   "sendToUnifiedCanvas": "Send To Unified Canvas",
+  "copyImage": "Copy Image",
   "copyImageToLink": "Copy Image To Link",
   "downloadImage": "Download Image",
   "openInViewer": "Open In Viewer",

diff --git a/invokeai/frontend/src/features/gallery/components/CurrentImageButtons.tsx b/invokeai/frontend/src/features/gallery/components/CurrentImageButtons.tsx
@@ -131,6 +131,22 @@ const CurrentImageButtons = () => {
     dispatch(setActiveTab('img2img'));
   };
 
+  const handleCopyImage = async () => {
+    if (!currentImage) return;
+
+    const blob = await fetch(currentImage.url).then((res) => res.blob());
+    const data = [new ClipboardItem({ [blob.type]: blob })];
+
+    await navigator.clipboard.write(data);
+
+    toast({
+      title: t('toast:imageCopied'),
+      status: 'success',
+      duration: 2500,
+      isClosable: true,
+    });
+  };
+
   const handleCopyImageLink = () => {
     navigator.clipboard
       .writeText(
@@ -401,6 +417,14 @@ const CurrentImageButtons = () => {
             >
               {t('parameters:sendToUnifiedCanvas')}
             </IAIButton>
+
+            <IAIButton
+              size={'sm'}
+              onClick={handleCopyImage}
+              leftIcon={<FaCopy />}
+            >
+              {t('parameters:copyImage')}
+            </IAIButton>
             <IAIButton
               size={'sm'}
               onClick={handleCopyImageLink}