From 2c501557d09a38edc15b88dc4413b6edd6bd8c74 Mon Sep 17 00:00:00 2001
From: Stepan Usatiuk <usatiste@fit.cvut.cz>
Date: Wed, 27 Dec 2023 17:16:27 +0100
Subject: [PATCH] editing messages

---
 client/src/Chat.tsx                           |  8 +-
 client/src/Message.scss                       |  2 +-
 client/src/Message.tsx                        | 81 ++++++++++++++-----
 client/src/actions.ts                         | 22 +++--
 client/src/api/Message.ts                     | 26 ++++++
 .../server/controller/MessageController.java  | 23 ++++++
 6 files changed, 134 insertions(+), 28 deletions(-)

diff --git a/client/src/Chat.tsx b/client/src/Chat.tsx
index 1d5ebc9..c2d5d90 100644
--- a/client/src/Chat.tsx
+++ b/client/src/Chat.tsx
@@ -6,6 +6,7 @@ import "./Chat.scss";
 import "./PostForm.scss";
 import { Message } from "./Message";
 import { useEffect } from "react";
+import { getTokenUserUuid } from "./api/utils";
 
 export function Chat() {
     const loaderData = useLoaderData() as LoaderToType<typeof chatLoader>;
@@ -51,7 +52,12 @@ export function Chat() {
             </fetcher.Form>
             <div className={"messages"}>
                 {sortedMessages.map((m) => (
-                    <Message key={m.id} message={m} chat={chat} />
+                    <Message
+                        key={m.id}
+                        message={m}
+                        chat={chat}
+                        actions={m.authorUuid == getTokenUserUuid()}
+                    />
                 ))}
             </div>
         </div>
diff --git a/client/src/Message.scss b/client/src/Message.scss
index 79ccdd9..ccae94c 100644
--- a/client/src/Message.scss
+++ b/client/src/Message.scss
@@ -14,7 +14,7 @@
 
   &.messageEditing {
     padding: 0;
-    min-height: 6rem;
+    min-height: 3rem;
     border: 1px solid #D0D0D0;
 
     @include post-editor;
diff --git a/client/src/Message.tsx b/client/src/Message.tsx
index b41051e..a96f1f9 100644
--- a/client/src/Message.tsx
+++ b/client/src/Message.tsx
@@ -1,14 +1,53 @@
 import "./Message.scss";
 import { TChatTo, TMessageTo } from "./api/dto";
-import { Link } from "react-router-dom";
+import { Form, Link, useNavigation, useSubmit } from "react-router-dom";
+import { useState } from "react";
 
 export function Message({
     message,
     chat,
+    actions,
 }: {
     message: TMessageTo;
     chat: TChatTo;
+    actions: boolean;
 }) {
+    const [editing, setEditing] = useState(false);
+    const submit = useSubmit();
+    const navigation = useNavigation();
+    const busy = navigation.state === "submitting";
+
+    if (editing) {
+        return (
+            <div className={"message messageEditing"}>
+                <Form className={"postForm"} method="patch">
+                    <input
+                        hidden={true}
+                        name={"messageId"}
+                        value={message.id}
+                    />
+                    <textarea
+                        placeholder={"Write something!"}
+                        name="text"
+                        defaultValue={message.contents}
+                    />
+                    <button
+                        name="intent"
+                        value="updateMessage"
+                        type="submit"
+                        onClick={(e) => {
+                            setEditing(false);
+                            submit(e.currentTarget);
+                        }}
+                        disabled={busy}
+                    >
+                        Save
+                    </button>
+                </Form>
+            </div>
+        );
+    }
+
     return (
         <div className={"message"}>
             <span className={"text"}>{message.contents}</span>
@@ -24,26 +63,26 @@ export function Message({
                         by {message.authorUsername}
                     </Link>
                 </div>
-                {/*{actions && (*/}
-                {/*    <div className={"actions"}>*/}
-                {/*        {<button onClick={() => setEditing(true)}>edit</button>}*/}
-                {/*        <Form method={"delete"}>*/}
-                {/*            <input*/}
-                {/*                hidden={true}*/}
-                {/*                name={"postToDeleteId"}*/}
-                {/*                value={id}*/}
-                {/*            />*/}
-                {/*            <button*/}
-                {/*                name="intent"*/}
-                {/*                value="deletePost"*/}
-                {/*                type={"submit"}*/}
-                {/*                disabled={busy}*/}
-                {/*            >*/}
-                {/*                delete*/}
-                {/*            </button>*/}
-                {/*        </Form>*/}
-                {/*    </div>*/}
-                {/*)}*/}
+                {actions && (
+                    <div className={"actions"}>
+                        {<button onClick={() => setEditing(true)}>edit</button>}
+                        <Form method={"delete"}>
+                            <input
+                                hidden={true}
+                                name={"messageToDeleteId"}
+                                value={message.id}
+                            />
+                            <button
+                                name="intent"
+                                value="deleteMessage"
+                                type={"submit"}
+                                disabled={busy}
+                            >
+                                delete
+                            </button>
+                        </Form>
+                    </div>
+                )}
             </div>
         </div>
     );
diff --git a/client/src/actions.ts b/client/src/actions.ts
index e7ff8a4..5635352 100644
--- a/client/src/actions.ts
+++ b/client/src/actions.ts
@@ -11,7 +11,7 @@ import { isError } from "./api/dto";
 import { deleteToken, getTokenUserUuid, setToken } from "./api/utils";
 import { createPost, deletePost, updatePost } from "./api/Post";
 import { createChat } from "./api/Chat";
-import { addMessagesToChat } from "./api/Message";
+import { addMessagesToChat, deleteMessage, editMessage } from "./api/Message";
 
 export type ActionToType<T extends (...args: any) => any> =
     | Exclude<Awaited<ReturnType<T>>, Response>
@@ -117,8 +117,20 @@ export async function newChatAction({ request }: ActionFunctionArgs) {
 
 export async function chatAction({ request }: ActionFunctionArgs) {
     const formData = await request.formData();
-    return await addMessagesToChat(
-        Number(formData.get("chatId")!.toString()),
-        formData.get("text")!.toString(),
-    );
+    const intent = formData.get("intent")!.toString();
+    if (intent == "addMessage") {
+        return await addMessagesToChat(
+            Number(formData.get("chatId")!.toString()),
+            formData.get("text")!.toString(),
+        );
+    } else if (intent == "deleteMessage") {
+        return await deleteMessage(
+            Number(formData.get("messageToDeleteId")!.toString()),
+        );
+    } else if (intent == "updateMessage") {
+        return await editMessage(
+            Number(formData.get("messageId")!.toString()),
+            formData.get("text")!.toString(),
+        );
+    }
 }
diff --git a/client/src/api/Message.ts b/client/src/api/Message.ts
index bb9a562..b01c998 100644
--- a/client/src/api/Message.ts
+++ b/client/src/api/Message.ts
@@ -1,8 +1,10 @@
 import {
     MessagesToResp,
     MessageToResp,
+    NoContentToResp,
     TMessagesToResp,
     TMessageToResp,
+    TNoContentToResp,
 } from "./dto";
 import { fetchJSONAuth } from "./utils";
 
@@ -20,3 +22,27 @@ export async function addMessagesToChat(
         contents: messageContents,
     });
 }
+
+export async function editMessage(
+    messageId: number,
+    messageContents: string,
+): Promise<TMessageToResp> {
+    return fetchJSONAuth(
+        "/message/by-id/" + messageId,
+        "PATCH",
+        MessageToResp,
+        {
+            contents: messageContents,
+        },
+    );
+}
+
+export async function deleteMessage(
+    messageId: number,
+): Promise<TNoContentToResp> {
+    return fetchJSONAuth(
+        "/message/by-id/" + messageId,
+        "DELETE",
+        NoContentToResp,
+    );
+}
diff --git a/server/src/main/java/com/usatiuk/tjv/y/server/controller/MessageController.java b/server/src/main/java/com/usatiuk/tjv/y/server/controller/MessageController.java
index 21dd5de..ff9375a 100644
--- a/server/src/main/java/com/usatiuk/tjv/y/server/controller/MessageController.java
+++ b/server/src/main/java/com/usatiuk/tjv/y/server/controller/MessageController.java
@@ -14,6 +14,7 @@ import org.springframework.web.bind.annotation.*;
 import org.springframework.web.server.ResponseStatusException;
 
 import java.security.Principal;
+import java.util.Objects;
 import java.util.stream.Stream;
 
 @RestController
@@ -52,4 +53,26 @@ public class MessageController {
         messageService.create(message);
         return messageMapper.makeDto(message);
     }
+
+    @PatchMapping(path = "/by-id/{id}")
+    public MessageTo update(Principal principal, @PathVariable long id, @RequestBody MessageCreateTo messageCreateTo) {
+        var message = messageService.readById(id).orElseThrow(() -> new ResponseStatusException(HttpStatus.NOT_FOUND));
+        if (!Objects.equals(message.getAuthor().getUuid(), principal.getName()))
+            throw new ResponseStatusException(HttpStatus.FORBIDDEN);
+        message.setContents(messageCreateTo.contents());
+        messageService.update(message);
+        return messageMapper.makeDto(message);
+    }
+
+    @DeleteMapping(path = "/by-id/{id}")
+    @ResponseStatus(HttpStatus.NO_CONTENT)
+    public void delete(Principal principal, @PathVariable long id) {
+        var read = messageService.readById(id);
+        if (read.isEmpty()) return;
+        if (!Objects.equals(read.get().getAuthor().getId(), principal.getName())) {
+            throw new ResponseStatusException(HttpStatus.FORBIDDEN);
+        }
+        messageService.deleteById(id);
+    }
+
 }