diff --git a/src/main/java/me/kavin/piped/server/handlers/StreamHandlers.java b/src/main/java/me/kavin/piped/server/handlers/StreamHandlers.java index 9b93784..6eb6744 100644 --- a/src/main/java/me/kavin/piped/server/handlers/StreamHandlers.java +++ b/src/main/java/me/kavin/piped/server/handlers/StreamHandlers.java @@ -357,7 +357,7 @@ public class StreamHandlers { nextpage = mapper.writeValueAsString(page); } - CommentsPage commentsItem = new CommentsPage(comments, nextpage, info.isCommentsDisabled()); + CommentsPage commentsItem = new CommentsPage(comments, nextpage, info.isCommentsDisabled(), info.getCommentsCount()); return mapper.writeValueAsBytes(commentsItem); @@ -395,7 +395,7 @@ public class StreamHandlers { nextpage = mapper.writeValueAsString(page); } - CommentsPage commentsItem = new CommentsPage(comments, nextpage, false); + CommentsPage commentsItem = new CommentsPage(comments, nextpage, false, -1); return mapper.writeValueAsBytes(commentsItem); diff --git a/src/main/java/me/kavin/piped/utils/obj/CommentsPage.java b/src/main/java/me/kavin/piped/utils/obj/CommentsPage.java index af8ae14..7622dc7 100644 --- a/src/main/java/me/kavin/piped/utils/obj/CommentsPage.java +++ b/src/main/java/me/kavin/piped/utils/obj/CommentsPage.java @@ -7,10 +7,12 @@ public class CommentsPage { public List comments; public String nextpage; public boolean disabled; + public int commentCount; - public CommentsPage(List comments, String nextpage, boolean disabled) { + public CommentsPage(List comments, String nextpage, boolean disabled, int commentCount) { this.comments = comments; this.nextpage = nextpage; this.disabled = disabled; + this.commentCount = commentCount; } }