Fix #3124: Activity Feed: Add support for deleting a post

catalog-rest-service/src/main/java/org/openmetadata/catalog/jdbi3/FeedRepository.java

@@ -23,6 +23,7 @@
 import java.util.Comparator;
 import java.util.HashSet;
 import java.util.List;
+import java.util.Optional;
 import java.util.Set;
 import java.util.UUID;
 import java.util.concurrent.atomic.AtomicInteger;
@@ -38,17 +39,21 @@
 import org.openmetadata.catalog.api.feed.ThreadCount;
 import org.openmetadata.catalog.entity.feed.Thread;
 import org.openmetadata.catalog.entity.teams.User;
+import org.openmetadata.catalog.exception.CatalogExceptionMessage;
+import org.openmetadata.catalog.exception.EntityNotFoundException;
 import org.openmetadata.catalog.resources.feeds.FeedResource;
 import org.openmetadata.catalog.resources.feeds.FeedUtil;
 import org.openmetadata.catalog.resources.feeds.MessageParser;
 import org.openmetadata.catalog.resources.feeds.MessageParser.EntityLink;
+import org.openmetadata.catalog.security.AuthorizationException;
 import org.openmetadata.catalog.type.EntityReference;
 import org.openmetadata.catalog.type.Include;
 import org.openmetadata.catalog.type.Post;
 import org.openmetadata.catalog.type.Relationship;
 import org.openmetadata.catalog.util.EntityUtil;
 import org.openmetadata.catalog.util.JsonUtils;
 import org.openmetadata.catalog.util.RestUtil;
+import org.openmetadata.catalog.util.RestUtil.DeleteResponse;
 import org.openmetadata.catalog.util.RestUtil.PatchResponse;
 
 @Slf4j
@@ -144,12 +149,13 @@ private void storeMentions(Thread thread, String message) {
   }
 
   @Transaction
-  public Thread addPostToThread(String id, Post post) throws IOException {
+  public Thread addPostToThread(String id, Post post, String userName) throws IOException {
     // Query 1 - validate the user posting the message
     User fromUser = dao.userDAO().findEntityByName(post.getFrom());
 
     // Query 2 - Find the thread
     Thread thread = EntityUtil.validate(id, dao.feedDAO().findById(id), Thread.class);
+    thread.withUpdatedBy(userName).withUpdatedAt(System.currentTimeMillis());
     FeedUtil.addPost(thread, post);
 
     // TODO is rewriting entire json okay?
@@ -181,6 +187,38 @@ public Thread addPostToThread(String id, Post post) throws IOException {
     return thread;
   }
 
+  public Optional<Post> getPostById(Thread thread, String postId) {
+    return thread.getPosts().stream().filter(p -> p.getId().equals(UUID.fromString(postId))).findAny();
+  }
+
+  @Transaction
+  public DeleteResponse<Post> deletePost(String threadId, String postId, String userName) throws IOException {
+    // validate the thread
+    Thread thread = get(threadId);
+    Post post;
+
+    // validate post
+    Optional<Post> optionalPost = getPostById(thread, postId);
+    if (optionalPost.isPresent()) {
+      // delete post only if the author tries to delete it
+      post = optionalPost.get();
+      if (userName.equals(post.getFrom())) {
+        List<Post> posts = thread.getPosts();
+        // Remove the post to be deleted from the posts list
+        posts = posts.stream().filter(p -> !p.getId().equals(UUID.fromString(postId))).collect(Collectors.toList());
+        thread.withUpdatedAt(System.currentTimeMillis()).withUpdatedBy(userName).withPosts(posts);
+        // update the json document
+        dao.feedDAO().update(threadId, JsonUtils.pojoToJson(thread));
+      } else {
+        // TODO Should we also allow admins to delete a post?
+        throw new AuthorizationException(CatalogExceptionMessage.noPermission(userName));
+      }
+    } else {
+      throw EntityNotFoundException.byMessage(CatalogExceptionMessage.entityNotFound("Post", postId));
+    }
+    return new DeleteResponse<>(post, RestUtil.ENTITY_DELETED);
+  }
+
   @Transaction
   public ThreadCount getThreadsCount(String link, boolean isResolved) throws IOException {
     ThreadCount threadCount = new ThreadCount();

catalog-rest-service/src/main/java/org/openmetadata/catalog/resources/feeds/FeedResource.java

@@ -37,6 +37,7 @@
 import javax.validation.constraints.Max;
 import javax.validation.constraints.Min;
 import javax.ws.rs.Consumes;
+import javax.ws.rs.DELETE;
 import javax.ws.rs.DefaultValue;
 import javax.ws.rs.GET;
 import javax.ws.rs.PATCH;
@@ -50,6 +51,7 @@
 import javax.ws.rs.core.Response;
 import javax.ws.rs.core.SecurityContext;
 import javax.ws.rs.core.UriInfo;
+import org.openmetadata.catalog.api.feed.CreatePost;
 import org.openmetadata.catalog.api.feed.CreateThread;
 import org.openmetadata.catalog.api.feed.ThreadCount;
 import org.openmetadata.catalog.entity.feed.Thread;
@@ -259,14 +261,43 @@ public Response create(@Context UriInfo uriInfo, @Context SecurityContext securi
         @ApiResponse(
             responseCode = "200",
             description = "The post",
-            content = @Content(mediaType = "application/json", schema = @Schema(implementation = Post.class))),
+            content = @Content(mediaType = "application/json", schema = @Schema(implementation = CreatePost.class))),
         @ApiResponse(responseCode = "400", description = "Bad request")
       })
-  public Response addPost(@Context UriInfo uriInfo, @PathParam("id") String id, @Valid Post post) throws IOException {
-    Thread thread = addHref(uriInfo, dao.addPostToThread(id, post));
+  public Response addPost(
+      @Context SecurityContext securityContext,
+      @Context UriInfo uriInfo,
+      @PathParam("id") String id,
+      @Valid CreatePost createPost)
+      throws IOException {
+    Post post = getPost(createPost);
+    Thread thread = addHref(uriInfo, dao.addPostToThread(id, post, securityContext.getUserPrincipal().getName()));
     return Response.created(thread.getHref()).entity(thread).build();
   }
 
+  @DELETE
+  @Path("/{threadId}/posts/{postId}")
+  @Operation(
+      summary = "Delete a post from its thread",
+      tags = "feeds",
+      description = "Delete a post from an existing thread.",
+      responses = {
+        @ApiResponse(responseCode = "200", description = "OK"),
+        @ApiResponse(responseCode = "404", description = "post with {postId} is not found"),
+        @ApiResponse(responseCode = "400", description = "Bad request")
+      })
+  public Response deletePost(
+      @Context SecurityContext securityContext,
+      @Parameter(description = "ThreadId of the post to be deleted", schema = @Schema(type = "string"))
+          @PathParam("threadId")
+          String threadId,
+      @Parameter(description = "PostId of the post to be deleted", schema = @Schema(type = "string"))
+          @PathParam("postId")
+          String postId)
+      throws IOException {
+    return dao.deletePost(threadId, postId, securityContext.getUserPrincipal().getName()).toResponse();
+  }
+
   @GET
   @Path("/{id}/posts")
   @Operation(
@@ -294,4 +325,12 @@ private Thread getThread(SecurityContext securityContext, CreateThread create) {
         .withUpdatedBy(securityContext.getUserPrincipal().getName())
         .withUpdatedAt(System.currentTimeMillis());
   }
+
+  private Post getPost(CreatePost create) {
+    return new Post()
+        .withId(UUID.randomUUID())
+        .withMessage(create.getMessage())
+        .withFrom(create.getFrom())
+        .withPostTs(System.currentTimeMillis());
+  }
 }

catalog-rest-service/src/main/java/org/openmetadata/catalog/resources/feeds/FeedUtil.java

@@ -22,7 +22,6 @@ private FeedUtil() {}
 
   public static void addPost(Thread thread, Post post) {
     // Add new post to the thread
-    post.setPostTs(System.currentTimeMillis());
     thread.getPosts().add(post);
     thread.withPostsCount(thread.getPosts().size());
   }

catalog-rest-service/src/main/resources/json/schema/api/feed/createPost.json

@@ -0,0 +1,19 @@
+{
+  "$id": "https://github.com/open-metadata/OpenMetadata/blob/main/catalog-rest-service/src/main/resources/json/schema/api/feed/createPost.json",
+  "$schema": "http://json-schema.org/draft-07/schema#",
+  "title": "CreatePostRequest",
+  "description": "Create post request",
+  "type": "object",
+  "properties": {
+    "message": {
+      "description": "Message in markdown format. See markdown support for more details.",
+      "type": "string"
+    },
+    "from": {
+      "description": "Name of the User posting the message",
+      "type": "string"
+    }
+  },
+  "required": ["message", "from"],
+  "additionalProperties": false
+}

catalog-rest-service/src/main/resources/json/schema/entity/feed/thread.json

@@ -11,6 +11,10 @@
       "type": "object",
       "description": "Post within a feed.",
       "properties": {
+        "id": {
+          "description": "Unique identifier that identifies the post.",
+          "$ref": "../../type/basic.json#/definitions/uuid"
+        },
         "message": {
           "description": "Message in markdown format. See markdown support for more details.",
           "type": "string"
@@ -24,7 +28,7 @@
           "type": "string"
         }
       },
-      "required": ["message", "from"],
+      "required": ["id", "message", "from"],
       "additionalProperties": false
     }
   },
@@ -86,6 +90,6 @@
       }
     }
   },
-  "required": ["id", "about", "posts"],
+  "required": ["id", "about", "message"],
   "additionalProperties": false
 }