Bugfix for deleting a comment (#157)

cisagov · Jun 23, 2023 · b4a7fb7 · b4a7fb7
1 parent 354ddf1
commit b4a7fb7
Show file tree

Hide file tree

Showing 3 changed files with 8 additions and 1 deletion.
diff --git a/applications/client/src/views/Campaign/Explore/Panels/Comment/CommentGroup.tsx b/applications/client/src/views/Campaign/Explore/Panels/Comment/CommentGroup.tsx
@@ -16,6 +16,7 @@ export type CommentGroupProps = ComponentProps<'div'> & {
 	hideCommands?: boolean;
 	expandedCommandIDs?: string[];
 	removeExpandedCommandID?: (commandId: string) => void;
+	refetchComments?: () => void;
 };
 export const CommentGroup = observer<CommentGroupProps>(
 	({
@@ -26,6 +27,7 @@ export const CommentGroup = observer<CommentGroupProps>(
 		hideCommands,
 		expandedCommandIDs = [],
 		removeExpandedCommandID,
+		refetchComments,
 		...props
 	}) => {
 		const store = useStore();
@@ -59,6 +61,7 @@ export const CommentGroup = observer<CommentGroupProps>(
 							annotation={annotation?.maybeCurrent}
 							commandGroup={commandGroup}
 							isFullList
+							refetchComments={refetchComments}
 						/>
 					))}
 					{newComment === commandGroup?.id && (

diff --git a/applications/client/src/views/Campaign/Explore/Panels/Comment/Comments.tsx b/applications/client/src/views/Campaign/Explore/Panels/Comment/Comments.tsx
@@ -49,7 +49,7 @@ export const Comments = observer<CommentsProps>(({ sort }) => {
 		},
 	});
 
-	const { data: commandGroupIdData } = useQuery(
+	const { data: commandGroupIdData, refetch: refetchComments } = useQuery(
 		[
 			'commandGroups',
 			store.campaign?.id,
@@ -186,6 +186,7 @@ export const Comments = observer<CommentsProps>(({ sort }) => {
 						newComment={state.newComment}
 						expandedCommandIDs={state.expandedCommandIDs}
 						removeExpandedCommandID={state.removeExpandedCommandID}
+						refetchComments={refetchComments}
 					/>
 				))
 			)}

diff --git a/applications/client/src/views/Campaign/Explore/components/Comment/CommentBox.tsx b/applications/client/src/views/Campaign/Explore/components/Comment/CommentBox.tsx
@@ -39,6 +39,7 @@ type CommentBoxProps = ComponentProps<'div'> & {
 	reply?: () => void;
 	isAddingCommandToComment?: boolean;
 	isFullList?: boolean;
+	refetchComments?: () => void;
 };
 
 export const CommentBox = observer<CommentBoxProps>(
@@ -55,6 +56,7 @@ export const CommentBox = observer<CommentBoxProps>(
 		popoverRef,
 		isAddingCommandToComment = false,
 		isFullList = false,
+		refetchComments,
 		...props
 	}) => {
 		const store = useStore();
@@ -67,6 +69,7 @@ export const CommentBox = observer<CommentBoxProps>(
 				}),
 			{
 				onSuccess: (annotationData) => {
+					refetchComments?.();
 					state.refetch();
 					if (annotationData) {
 						store.graphqlStore.annotations.delete(annotationData.deleteAnnotation.id);