idillon | 6847e25 | 2022-11-04 11:50:08 -0400 | [diff] [blame] | 1 | /* |
| 2 | * Copyright (C) 2022 Savoir-faire Linux Inc. |
| 3 | * |
| 4 | * This program is free software; you can redistribute it and/or modify |
| 5 | * it under the terms of the GNU Affero General Public License as |
| 6 | * published by the Free Software Foundation; either version 3 of the |
| 7 | * License, or (at your option) any later version. |
| 8 | * |
| 9 | * This program is distributed in the hope that it will be useful, |
| 10 | * but WITHOUT ANY WARRANTY; without even the implied warranty of |
| 11 | * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the |
| 12 | * GNU Affero General Public License for more details. |
| 13 | * |
| 14 | * You should have received a copy of the GNU Affero General Public |
| 15 | * License along with this program. If not, see |
| 16 | * <https://www.gnu.org/licenses/>. |
| 17 | */ |
idillon | cab81d7 | 2022-11-08 12:20:00 -0500 | [diff] [blame] | 18 | import { Box, Divider, Stack } from '@mui/material'; |
simon | f9d78f2 | 2022-11-25 15:47:15 -0500 | [diff] [blame] | 19 | import { ConversationMessage, Message, WebSocketMessageType } from 'jami-web-common'; |
idillon | 6847e25 | 2022-11-04 11:50:08 -0400 | [diff] [blame] | 20 | import { useCallback, useContext, useEffect, useState } from 'react'; |
idillon | cab81d7 | 2022-11-08 12:20:00 -0500 | [diff] [blame] | 21 | import { useDropzone } from 'react-dropzone'; |
idillon | 6847e25 | 2022-11-04 11:50:08 -0400 | [diff] [blame] | 22 | |
idillon | cab81d7 | 2022-11-08 12:20:00 -0500 | [diff] [blame] | 23 | import { FilePreviewRemovable } from '../components/FilePreview'; |
idillon | 6847e25 | 2022-11-04 11:50:08 -0400 | [diff] [blame] | 24 | import LoadingPage from '../components/Loading'; |
| 25 | import MessageList from '../components/MessageList'; |
| 26 | import SendMessageForm from '../components/SendMessageForm'; |
simon | f9d78f2 | 2022-11-25 15:47:15 -0500 | [diff] [blame] | 27 | import { ConversationContext } from '../contexts/ConversationProvider'; |
Issam E. Maghni | 0432cb7 | 2022-11-12 06:09:26 +0000 | [diff] [blame] | 28 | import { WebSocketContext } from '../contexts/WebSocketProvider'; |
idillon | 6847e25 | 2022-11-04 11:50:08 -0400 | [diff] [blame] | 29 | import { useMessagesQuery, useSendMessageMutation } from '../services/Conversation'; |
idillon | cab81d7 | 2022-11-08 12:20:00 -0500 | [diff] [blame] | 30 | import { FileHandler } from '../utils/files'; |
idillon | 6847e25 | 2022-11-04 11:50:08 -0400 | [diff] [blame] | 31 | |
simon | f9d78f2 | 2022-11-25 15:47:15 -0500 | [diff] [blame] | 32 | const ChatInterface = () => { |
Issam E. Maghni | 0432cb7 | 2022-11-12 06:09:26 +0000 | [diff] [blame] | 33 | const webSocket = useContext(WebSocketContext); |
simon | f9d78f2 | 2022-11-25 15:47:15 -0500 | [diff] [blame] | 34 | const { conversationId, conversation } = useContext(ConversationContext); |
idillon | 6847e25 | 2022-11-04 11:50:08 -0400 | [diff] [blame] | 35 | const [messages, setMessages] = useState<Message[]>([]); |
| 36 | const [isLoading, setIsLoading] = useState(true); |
| 37 | const [error, setError] = useState(false); |
| 38 | |
simon | 5da8ca6 | 2022-11-09 15:21:25 -0500 | [diff] [blame] | 39 | const messagesQuery = useMessagesQuery(conversationId); |
| 40 | const sendMessageMutation = useSendMessageMutation(conversationId); |
idillon | 6847e25 | 2022-11-04 11:50:08 -0400 | [diff] [blame] | 41 | |
idillon | cab81d7 | 2022-11-08 12:20:00 -0500 | [diff] [blame] | 42 | const [fileHandlers, setFileHandlers] = useState<FileHandler[]>([]); |
| 43 | |
| 44 | const onFilesDrop = useCallback( |
| 45 | (acceptedFiles: File[]) => { |
| 46 | const newFileHandlers = acceptedFiles.map((file) => new FileHandler(file)); |
| 47 | setFileHandlers((oldFileHandlers) => [...oldFileHandlers, ...newFileHandlers]); |
| 48 | }, |
| 49 | [setFileHandlers] |
| 50 | ); |
| 51 | |
| 52 | const removeFile = useCallback( |
| 53 | (fileId: string | number) => { |
| 54 | setFileHandlers((fileHandlers) => fileHandlers.filter((fileHandler) => fileHandler.id !== fileId)); |
| 55 | }, |
| 56 | [setFileHandlers] |
| 57 | ); |
| 58 | |
| 59 | const { |
| 60 | getRootProps, |
| 61 | getInputProps, |
| 62 | open: openFilePicker, |
| 63 | isDragActive, |
| 64 | } = useDropzone({ |
| 65 | onDrop: onFilesDrop, |
| 66 | noClick: true, |
| 67 | noKeyboard: true, |
| 68 | }); |
| 69 | |
idillon | 6847e25 | 2022-11-04 11:50:08 -0400 | [diff] [blame] | 70 | useEffect(() => { |
| 71 | if (messagesQuery.isSuccess) { |
| 72 | const sortedMessages = sortMessages(messagesQuery.data); |
| 73 | setMessages(sortedMessages); |
| 74 | } |
| 75 | }, [messagesQuery.isSuccess, messagesQuery.data]); |
| 76 | |
| 77 | useEffect(() => { |
| 78 | setIsLoading(messagesQuery.isLoading); |
| 79 | }, [messagesQuery.isLoading]); |
| 80 | |
| 81 | useEffect(() => { |
| 82 | setError(messagesQuery.isError); |
| 83 | }, [messagesQuery.isError]); |
| 84 | |
| 85 | const sendMessage = useCallback((message: string) => sendMessageMutation.mutate(message), [sendMessageMutation]); |
| 86 | |
| 87 | useEffect(() => { |
Issam E. Maghni | 0432cb7 | 2022-11-12 06:09:26 +0000 | [diff] [blame] | 88 | if (webSocket) { |
Misha Krieger-Raynauld | 20cf1c8 | 2022-11-23 20:26:50 -0500 | [diff] [blame] | 89 | const conversationMessageListener = (data: ConversationMessage) => { |
idillon | 6847e25 | 2022-11-04 11:50:08 -0400 | [diff] [blame] | 90 | console.log('newMessage'); |
Misha Krieger-Raynauld | 20cf1c8 | 2022-11-23 20:26:50 -0500 | [diff] [blame] | 91 | setMessages((messages) => addMessage(messages, data.message)); |
simon | a5c54ef | 2022-11-18 05:26:06 -0500 | [diff] [blame] | 92 | }; |
| 93 | |
| 94 | webSocket.bind(WebSocketMessageType.ConversationMessage, conversationMessageListener); |
Misha Krieger-Raynauld | 20cf1c8 | 2022-11-23 20:26:50 -0500 | [diff] [blame] | 95 | |
simon | a5c54ef | 2022-11-18 05:26:06 -0500 | [diff] [blame] | 96 | return () => { |
| 97 | webSocket.unbind(WebSocketMessageType.ConversationMessage, conversationMessageListener); |
| 98 | }; |
idillon | 6847e25 | 2022-11-04 11:50:08 -0400 | [diff] [blame] | 99 | } |
simon | a5c54ef | 2022-11-18 05:26:06 -0500 | [diff] [blame] | 100 | }, [webSocket]); |
idillon | 6847e25 | 2022-11-04 11:50:08 -0400 | [diff] [blame] | 101 | |
| 102 | if (isLoading) { |
| 103 | return <LoadingPage />; |
| 104 | } else if (error) { |
| 105 | return <div>Error loading {conversationId}</div>; |
| 106 | } |
| 107 | |
simon | f9d78f2 | 2022-11-25 15:47:15 -0500 | [diff] [blame] | 108 | const members = conversation.getMembers(); |
| 109 | |
idillon | 6847e25 | 2022-11-04 11:50:08 -0400 | [diff] [blame] | 110 | return ( |
idillon | cab81d7 | 2022-11-08 12:20:00 -0500 | [diff] [blame] | 111 | <Stack flex={1} overflow="hidden" {...getRootProps()} paddingBottom="16px"> |
| 112 | {isDragActive && ( |
| 113 | // dark overlay when the user is dragging a file |
| 114 | <Box |
| 115 | sx={{ |
| 116 | position: 'absolute', |
| 117 | width: '100%', |
| 118 | height: '100%', |
| 119 | backgroundColor: 'black', |
| 120 | opacity: '30%', |
| 121 | zIndex: 100, |
| 122 | }} |
| 123 | /> |
| 124 | )} |
| 125 | <input {...getInputProps()} /> |
simon | 5da8ca6 | 2022-11-09 15:21:25 -0500 | [diff] [blame] | 126 | <MessageList members={members} messages={messages} /> |
idillon | 6847e25 | 2022-11-04 11:50:08 -0400 | [diff] [blame] | 127 | <Divider |
| 128 | sx={{ |
| 129 | margin: '30px 16px 0px 16px', |
| 130 | borderTop: '1px solid #E5E5E5', |
| 131 | }} |
| 132 | /> |
simon | 5da8ca6 | 2022-11-09 15:21:25 -0500 | [diff] [blame] | 133 | <SendMessageForm members={members} onSend={sendMessage} openFilePicker={openFilePicker} /> |
idillon | cab81d7 | 2022-11-08 12:20:00 -0500 | [diff] [blame] | 134 | {fileHandlers.length > 0 && <FilePreviewsList fileHandlers={fileHandlers} removeFile={removeFile} />} |
| 135 | </Stack> |
| 136 | ); |
| 137 | }; |
| 138 | |
| 139 | interface FilePreviewsListProps { |
| 140 | fileHandlers: FileHandler[]; |
| 141 | removeFile: (fileId: string | number) => void; |
| 142 | } |
| 143 | |
| 144 | const FilePreviewsList = ({ fileHandlers, removeFile }: FilePreviewsListProps) => { |
| 145 | return ( |
| 146 | <Stack |
| 147 | direction="row" |
| 148 | flexWrap="wrap" |
| 149 | gap="16px" |
| 150 | overflow="auto" |
| 151 | maxHeight="30%" |
| 152 | paddingX="16px" |
| 153 | marginTop="12px" // spacing with the component on top |
| 154 | paddingTop="4px" // spacing so "RemoveButton" are not cut |
| 155 | > |
| 156 | {fileHandlers.map((fileHandler) => ( |
| 157 | <FilePreviewRemovable |
| 158 | key={fileHandler.id} |
| 159 | remove={() => removeFile(fileHandler.id)} |
| 160 | fileHandler={fileHandler} |
| 161 | borderColor={'#005699' /* Should be same color as message bubble */} |
| 162 | /> |
| 163 | ))} |
idillon | 6847e25 | 2022-11-04 11:50:08 -0400 | [diff] [blame] | 164 | </Stack> |
| 165 | ); |
| 166 | }; |
| 167 | |
| 168 | const addMessage = (sortedMessages: Message[], message: Message) => { |
| 169 | if (sortedMessages.length === 0) { |
| 170 | return [message]; |
| 171 | } else if (message.id === sortedMessages[sortedMessages.length - 1].linearizedParent) { |
| 172 | return [...sortedMessages, message]; |
| 173 | } else if (message.linearizedParent === sortedMessages[0].id) { |
| 174 | return [message, ...sortedMessages]; |
| 175 | } else { |
| 176 | console.error("Can't insert message " + message.id); |
| 177 | return sortedMessages; |
| 178 | } |
| 179 | }; |
| 180 | |
| 181 | const sortMessages = (messages: Message[]) => { |
| 182 | let sortedMessages: Message[] = []; |
| 183 | messages.forEach((message) => (sortedMessages = addMessage(sortedMessages, message))); |
| 184 | return sortedMessages; |
| 185 | }; |
| 186 | |
| 187 | export default ChatInterface; |