From 14fe29c03aeb6f737811055064803f71f3f731cd Mon Sep 17 00:00:00 2001
From: Jack Wu <jackwu833016@gmail.com>
Date: Sun, 26 Mar 2023 00:28:08 -0400
Subject: [PATCH] feat: Message copy button (#171)

* Add copy button

* Fix copy button not copying the entire message

* fix style

* remove prewrap

---------

Co-authored-by: Mckay Wrigley <mckaywrigley@gmail.com>
---
 components/Chat/ChatMessage.tsx | 97 ++++++++++++++++++++-------------
 components/Chat/CopyButton.tsx  | 24 ++++++++
 2 files changed, 84 insertions(+), 37 deletions(-)
 create mode 100644 components/Chat/CopyButton.tsx

diff --git a/components/Chat/ChatMessage.tsx b/components/Chat/ChatMessage.tsx
index c087f9d..126e31b 100644
--- a/components/Chat/ChatMessage.tsx
+++ b/components/Chat/ChatMessage.tsx
@@ -1,10 +1,11 @@
 import { Message } from "@/types";
 import { IconEdit } from "@tabler/icons-react";
-import { FC, useEffect, useRef, useState } from "react";
 import { useTranslation } from "next-i18next";
+import { FC, useEffect, useRef, useState } from "react";
 import ReactMarkdown from "react-markdown";
 import remarkGfm from "remark-gfm";
 import { CodeBlock } from "../Markdown/CodeBlock";
+import { CopyButton } from "./CopyButton";
 
 interface Props {
   message: Message;
@@ -14,10 +15,11 @@ interface Props {
 }
 
 export const ChatMessage: FC<Props> = ({ message, messageIndex, lightMode, onEditMessage }) => {
-  const { t } = useTranslation('chat');
+  const { t } = useTranslation("chat");
   const [isEditing, setIsEditing] = useState<boolean>(false);
   const [isHovering, setIsHovering] = useState<boolean>(false);
   const [messageContent, setMessageContent] = useState(message.content);
+  const [messagedCopied, setMessageCopied] = useState(false);
 
   const textareaRef = useRef<HTMLTextAreaElement>(null);
 
@@ -47,6 +49,17 @@ export const ChatMessage: FC<Props> = ({ message, messageIndex, lightMode, onEdi
     }
   };
 
+  const copyOnClick = () => {
+    if (!navigator.clipboard) return;
+
+    navigator.clipboard.writeText(message.content).then(() => {
+      setMessageCopied(true);
+      setTimeout(() => {
+        setMessageCopied(false);
+      }, 2000);
+    });
+  };
+
   useEffect(() => {
     if (textareaRef.current) {
       textareaRef.current.style.height = "inherit";
@@ -119,41 +132,51 @@ export const ChatMessage: FC<Props> = ({ message, messageIndex, lightMode, onEdi
               )}
             </div>
           ) : (
-            <ReactMarkdown
-              remarkPlugins={[remarkGfm]}
-              components={{
-                code({ node, inline, className, children, ...props }) {
-                  const match = /language-(\w+)/.exec(className || "");
-                  return !inline && match ? (
-                    <CodeBlock
-                      key={Math.random()}
-                      language={match[1]}
-                      value={String(children).replace(/\n$/, "")}
-                      lightMode={lightMode}
-                      {...props}
-                    />
-                  ) : (
-                    <code
-                      className={className}
-                      {...props}
-                    >
-                      {children}
-                    </code>
-                  );
-                },
-                table({ children }) {
-                  return <table className="border-collapse border border-black dark:border-white py-1 px-3">{children}</table>;
-                },
-                th({ children }) {
-                  return <th className="border border-black dark:border-white break-words py-1 px-3 bg-gray-500 text-white">{children}</th>;
-                },
-                td({ children }) {
-                  return <td className="border border-black dark:border-white break-words py-1 px-3">{children}</td>;
-                }
-              }}
-            >
-              {message.content}
-            </ReactMarkdown>
+            <>
+              <ReactMarkdown
+                className="prose dark:prose-invert"
+                remarkPlugins={[remarkGfm]}
+                components={{
+                  code({ node, inline, className, children, ...props }) {
+                    const match = /language-(\w+)/.exec(className || "");
+                    return !inline && match ? (
+                      <CodeBlock
+                        key={Math.random()}
+                        language={match[1]}
+                        value={String(children).replace(/\n$/, "")}
+                        lightMode={lightMode}
+                        {...props}
+                      />
+                    ) : (
+                      <code
+                        className={className}
+                        {...props}
+                      >
+                        {children}
+                      </code>
+                    );
+                  },
+                  table({ children }) {
+                    return <table className="border-collapse border border-black dark:border-white py-1 px-3">{children}</table>;
+                  },
+                  th({ children }) {
+                    return <th className="border border-black dark:border-white break-words py-1 px-3 bg-gray-500 text-white">{children}</th>;
+                  },
+                  td({ children }) {
+                    return <td className="border border-black dark:border-white break-words py-1 px-3">{children}</td>;
+                  }
+                }}
+              >
+                {message.content}
+              </ReactMarkdown>
+
+              {(isHovering || window.innerWidth < 640) && (
+                <CopyButton
+                  messagedCopied={messagedCopied}
+                  copyOnClick={copyOnClick}
+                />
+              )}
+            </>
           )}
         </div>
       </div>
diff --git a/components/Chat/CopyButton.tsx b/components/Chat/CopyButton.tsx
new file mode 100644
index 0000000..554bfb6
--- /dev/null
+++ b/components/Chat/CopyButton.tsx
@@ -0,0 +1,24 @@
+import { IconCheck, IconCopy } from "@tabler/icons-react";
+import { FC } from "react";
+
+type Props = {
+  messagedCopied: boolean;
+  copyOnClick: () => void;
+};
+
+export const CopyButton: FC<Props> = ({ messagedCopied, copyOnClick }) => (
+  <button className={`absolute ${window.innerWidth < 640 ? "right-3 bottom-1" : "right-[-20px] top-[26px] m-0"}`}>
+    {messagedCopied ? (
+      <IconCheck
+        size={20}
+        className="text-green-500 dark:text-green-400"
+      />
+    ) : (
+      <IconCopy
+        size={20}
+        className="text-gray-500 dark:text-gray-400 hover:text-gray-700 dark:hover:text-gray-300"
+        onClick={copyOnClick}
+      />
+    )}
+  </button>
+);