From 1cace30971095479f53a6ae9cd7212a7a054aa58 Mon Sep 17 00:00:00 2001
From: Bnyro <bnyro@tutanota.com>
Date: Mon, 26 Jun 2023 12:05:16 +0200
Subject: [PATCH] Add support for extracting the comment count

---
 .../java/me/kavin/piped/server/handlers/StreamHandlers.java   | 4 ++--
 src/main/java/me/kavin/piped/utils/obj/CommentsPage.java      | 4 +++-
 2 files changed, 5 insertions(+), 3 deletions(-)

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<Comment> comments;
     public String nextpage;
     public boolean disabled;
+    public int commentCount;
 
-    public CommentsPage(List<Comment> comments, String nextpage, boolean disabled) {
+    public CommentsPage(List<Comment> comments, String nextpage, boolean disabled, int commentCount) {
         this.comments = comments;
         this.nextpage = nextpage;
         this.disabled = disabled;
+        this.commentCount = commentCount;
     }
 }