From 25a6b9f3930e381a6516d922dec8d9d86c665ef1 Mon Sep 17 00:00:00 2001
From: Chenlei Hu <hcl@comfy.org>
Date: Fri, 16 May 2025 17:03:55 -0400
Subject: [PATCH] Copy to clipboard

---
 .../sidebar/tabs/queue/ResultText.vue         | 27 ++++++++++---------
 src/stores/queueStore.ts                      | 17 +++++++++---
 2 files changed, 27 insertions(+), 17 deletions(-)

diff --git a/src/components/sidebar/tabs/queue/ResultText.vue b/src/components/sidebar/tabs/queue/ResultText.vue
index 1e0e4b6001..c259150e60 100644
--- a/src/components/sidebar/tabs/queue/ResultText.vue
+++ b/src/components/sidebar/tabs/queue/ResultText.vue
@@ -1,28 +1,29 @@
 <template>
-  <div class="result-text-container" @click="copyToClipboard">
+  <div class="result-text-container">
     <div class="text-content">
-      {{ result }}
-    </div>
-    <div class="copy-button">
-      <i class="pi pi-copy" />
+      {{ result.text }}
     </div>
+
+    <Button
+      class="copy-button"
+      icon="pi pi-copy"
+      text
+      @click.stop="copyToClipboard(result.text ?? '')"
+    />
   </div>
 </template>
 
 <script setup lang="ts">
+import Button from 'primevue/button'
+
+import { useCopyToClipboard } from '@/composables/useCopyToClipboard'
 import { ResultItemImpl } from '@/stores/queueStore'
 
-const props = defineProps<{
+defineProps<{
   result: ResultItemImpl
 }>()
 
-const copyToClipboard = async () => {
-  try {
-    await navigator.clipboard.writeText(props.result.text)
-  } catch (err) {
-    console.error('Failed to copy text:', err)
-  }
-}
+const { copyToClipboard } = useCopyToClipboard()
 </script>
 
 <style scoped>
diff --git a/src/stores/queueStore.ts b/src/stores/queueStore.ts
index 2d76d7b900..52c747eaf6 100644
--- a/src/stores/queueStore.ts
+++ b/src/stores/queueStore.ts
@@ -30,6 +30,7 @@ export class ResultItemImpl {
   filename: string
   subfolder: string
   type: string
+  text?: string
 
   nodeId: NodeId
   // 'audio' | 'images' | ...
@@ -43,6 +44,7 @@ export class ResultItemImpl {
     this.filename = obj.filename ?? ''
     this.subfolder = obj.subfolder ?? ''
     this.type = obj.type ?? ''
+    this.text = obj.text
 
     this.nodeId = obj.nodeId
     this.mediaType = obj.mediaType
@@ -237,14 +239,21 @@ export class TaskItemImpl {
     }
     return Object.entries(this.outputs).flatMap(([nodeId, nodeOutputs]) =>
       Object.entries(nodeOutputs).flatMap(([mediaType, items]) =>
-        (items as ResultItem[]).map(
-          (item: ResultItem) =>
-            new ResultItemImpl({
+        (items as (ResultItem | string)[]).map((item: ResultItem | string) => {
+          if (typeof item === 'string') {
+            return new ResultItemImpl({
+              text: item,
+              nodeId,
+              mediaType
+            })
+          } else {
+            return new ResultItemImpl({
               ...item,
               nodeId,
               mediaType
             })
-        )
+          }
+        })
       )
     )
   }