Update EditLabel.jsx

src/components/EditLabel.jsx

@@ -1,4 +1,4 @@
-import React from "react";
+import React, { useState } from "react";
 import { useComment, useEditedComment } from "@plebbit/plebbit-react-hooks";
 import { Link } from "react-router-dom";
 import OriginalCommentModal from "./modals/OriginalCommentModal";
@@ -6,33 +6,41 @@ import getDate from "../utils/getDate";
 
 
 const EditLabel = ({ commentCid, className }) => {
-  const [isOriginalCommentModalOpen, setIsOriginalCommentModalOpen] = React.useState(false);
+  const [isOriginalCommentModalOpen, setIsOriginalCommentModalOpen] = useState(false);
   const comment = useComment({commentCid});
   const timestamp = getDate(comment.edit?.timestamp);
   const {state: editedCommentState, editedComment} = useEditedComment({comment});
-  
+
   return (
     <>
       <OriginalCommentModal 
       isOpen={isOriginalCommentModalOpen}
       closeModal={() => setIsOriginalCommentModalOpen(false)}
       comment={comment}/> 
-      {editedComment && (
+      {editedComment || comment.edit ? (
         <>
           <br />
           <span className={className}>
-            {editedCommentState === 'succeeded' && (
+            {comment.removed ? (
               <>
-              (Edited at {timestamp}, <Link className="ttl-link" onClick={
-                () => setIsOriginalCommentModalOpen(true)
-              }>show original</Link>)
+                (This post has been removed)
               </>
-              )}
-            {editedCommentState === 'pending' && ("(Pending Edit)")}
-            {editedCommentState === 'failed' && ("(Failed Edit)")}
+            ) : (
+              <>
+                {comment.edit ? (
+                  <>
+                    (Edited at {timestamp}, <Link className="ttl-link" onClick={
+                      () => setIsOriginalCommentModalOpen(true)
+                    }>show original</Link>)
+                  </>
+                ) : null}
+                {editedCommentState === 'pending' && ("(Pending edit)")}
+                {editedCommentState === 'failed' && ("(Failed edit)")}
+              </>
+            )}
           </span>
         </>
-      )}
+      ) : null}
     </>
   );
 };