Function onSuccess

packages/shared/src/hooks/post/useMutateComment.ts:95–177

(comment: Comment)

  );

  const onSuccess = (comment: Comment) => {
    if (!comment) {
      return;
    }

    const updateQueryData = (data: PostCommentsData) => {
      if (!data) {
        return data;
      }
      const copy = structuredClone(data);

      if (!editCommentId) {
        const edge = generateCommentEdge(comment);

        if (!parentCommentId) {
          copy.postComments.edges.unshift(edge);
          return copy;
        }

        const index = copy.postComments.edges.findIndex(
          ({ node }) => node.id === parentCommentId,
        );
        copy.postComments.edges[index].node.children.edges.push(edge);

        return copy;
      }

      if (!parentCommentId) {
        const index = copy.postComments.edges.findIndex(
          ({ node }) => node.id === editCommentId,
        );
        copy.postComments.edges[index].node = {
          ...comment,
          children: copy.postComments.edges[index].node.children,
        };
        return copy;
      }

      const parent = copy.postComments.edges.find(
        ({ node }) => node.id === parentCommentId,
      );
      const index = parent.node.children.edges.findIndex(
        ({ node }) => node.id === editCommentId,
      );
      parent.node.children.edges[index].node = {
        ...comment,
        children: parent.node.children,
      };
      return copy;
    };

    const forInvalidation: QueryKey[] = [];

    getAllCommentsQuery(postId).forEach((queryKey) => {
      client.setQueryData<PostCommentsData>(queryKey, (data) => {
        if (!data) {
          forInvalidation.push(queryKey);
          return null;