From 29e01721c94e036916c374556a172fd1b52c1f91 Mon Sep 17 00:00:00 2001
From: Ahmad Hassan <76843311+ah7255703@users.noreply.github.com>
Date: Sun, 5 Nov 2023 15:18:30 +0200
Subject: [PATCH] fix: Fix error handling in ChatScreen

---
 copilot-widget/lib/components/Messages.tsx | 48 +++++++++++++++-------
 copilot-widget/lib/screens/ChatScreen.tsx  |  8 ++--
 2 files changed, 39 insertions(+), 17 deletions(-)

diff --git a/copilot-widget/lib/components/Messages.tsx b/copilot-widget/lib/components/Messages.tsx
index 5a2369d1e..2f8846072 100644
--- a/copilot-widget/lib/components/Messages.tsx
+++ b/copilot-widget/lib/components/Messages.tsx
@@ -7,6 +7,10 @@ import { FaRegUserCircle } from "react-icons/fa";
 import { format } from "timeago.js";
 import cn from "../utils/cn";
 import formatTimeFromTimestamp from "../utils/formatTime";
+import { useCopyToClipboard } from "@lib/hooks/useCopy";
+import { HiOutlineClipboard, HiOutlineClipboardCheck } from "react-icons/hi";
+import { FailedMessage, useChat } from "@lib/contexts/Controller";
+import { getLast } from "@lib/utils/utils";
 function BotIcon({ error }: { error?: boolean }) {
   return (
     <img
@@ -33,20 +37,24 @@ function UserIcon() {
 export function BotTextMessage({
   message,
   timestamp,
+  id,
 }: {
   message: string;
   timestamp?: number;
+  id?: string;
 }) {
   const { displayText } = useTypeWriter({
     text: message,
     every: 0.0001,
     shouldStart: true,
   });
-
+  const [copied, copy] = useCopyToClipboard();
+  const { messages } = useChat();
+  const isLast = getLast(messages)?.id === id;
   return (
-    <div className="opencopilot-p-2 opencopilot-w-full">
+    <div className="opencopilot-p-2 group opencopilot-w-full">
       <div
-        className="opencopilot-flex opencopilot-items-start opencopilot-gap-3 opencopilot-w-full"
+        className="opencopilot-flex opencopilot-select-none opencopilot-items-start opencopilot-gap-3 opencopilot-w-full"
         dir="auto"
       >
         <BotIcon />
@@ -63,15 +71,27 @@ export function BotTextMessage({
           </div>
         </div>
       </div>
-      <div className="opencopilot-w-full opencopilot-ps-[52px]">
-        <div>
-          {timestamp && (
-            <span className="opencopilot-text-xs opencopilot-m-0 group-last-of-type:opencopilot-inline opencopilot-hidden">
-              Bot · {format(timestamp)}
-            </span>
-          )}
+      {isLast && (
+        <div className="opencopilot-w-full opencopilot-ps-10 opencopilot-flex opencopilot-items-center opencopilot-justify-between">
+          <div>
+            {timestamp && (
+              <span className="opencopilot-text-xs opencopilot-m-0">
+                Bot · {format(timestamp)}
+              </span>
+            )}
+          </div>
+          <button
+            className="opencopilot-text-lg opencopilot-justify-self-end"
+            onClick={() => copy(displayText)}
+          >
+            {copied ? (
+              <HiOutlineClipboardCheck className="opencopilot-text-emerald-500" />
+            ) : (
+              <HiOutlineClipboard />
+            )}
+          </button>
         </div>
-      </div>
+      )}
     </div>
   );
 }
@@ -105,11 +125,12 @@ export function UserMessage({
 }: {
   content: string;
   timestamp?: number;
+  id?: string;
 }) {
   return (
     <div
       dir="auto"
-      className="opencopilot-w-full last-of-type:opencopilot-mb-10 opencopilot-bg-accent opencopilot-p-2 opencopilot-flex opencopilot-gap-3 opencopilot-items-center"
+      className="opencopilot-w-full opencopilot-overflow-x-auto opencopilot-max-w-full last-of-type:opencopilot-mb-10 opencopilot-bg-accent opencopilot-p-2 opencopilot-flex opencopilot-gap-3 opencopilot-items-center"
     >
       <UserIcon />
       <div>
@@ -128,12 +149,11 @@ export function UserMessage({
   );
 }
 
-export function BotMessageError() {
+export function BotMessageError({ message }: { message?: FailedMessage }) {
   const { displayText } = useTypeWriter({
     text: "Error sending the message.",
     every: 0.001,
   });
-
   return (
     <div className="opencopilot-clear-both opencopilot-w-full opencopilot-p-2">
       <div className="opencopilot-flex opencopilot-items-center opencopilot-gap-3 opencopilot-w-full">
diff --git a/copilot-widget/lib/screens/ChatScreen.tsx b/copilot-widget/lib/screens/ChatScreen.tsx
index fb456ccfe..57cfca29a 100644
--- a/copilot-widget/lib/screens/ChatScreen.tsx
+++ b/copilot-widget/lib/screens/ChatScreen.tsx
@@ -15,7 +15,7 @@ import { Map } from "../utils/Map";
 function ChatScreen() {
   const scrollElementRef = useRef(null);
   const [setPos] = useScrollToPercentage(scrollElementRef);
-  const { messages, loading, error } = useChat();
+  const { messages, loading, failedMessage } = useChat();
   const config = useConfigData();
   const initialMessage = config?.initialMessage;
   useEffect(() => {
@@ -32,7 +32,7 @@ function ChatScreen() {
       <ChatHeader />
       <main
         ref={scrollElementRef}
-        className="opencopilot-flex-1 opencopilot-w-full opencopilot-shrink-0 opencopilot-overflow-auto opencopilot-scrollbar-thin opencopilot-scroll-smooth"
+        className="opencopilot-flex-1 opencopilot-w-full opencopilot-overflow-x-hidden opencopilot-shrink-0 opencopilot-overflow-auto opencopilot-scrollbar-thin opencopilot-scroll-smooth"
       >
         <div className="opencopilot-flex opencopilot-h-fit opencopilot-mt-auto opencopilot-flex-col opencopilot-py-2 opencopilot-max-h-full opencopilot-items-center opencopilot-gap-1 last:fade-in-right">
           {
@@ -48,6 +48,7 @@ function ChatScreen() {
                   return (
                     <BotTextMessage
                       timestamp={message.timestamp}
+                      id={message.id}
                       key={index}
                       message={message.response.text}
                     />
@@ -56,6 +57,7 @@ function ChatScreen() {
                 return (
                   <UserMessage
                     key={index}
+                    id={message.id}
                     timestamp={message.timestamp}
                     content={message.content}
                   />
@@ -64,7 +66,7 @@ function ChatScreen() {
             }}
           />
           {loading && <BotMessageLoading />}
-          {error && <BotMessageError />}
+          {failedMessage && <BotMessageError message={failedMessage} />}
         </div>
       </main>
       <ChatInputFooter />