From 5fc4b8d6edb031cd4a65f788f15832eb659fd2bf Mon Sep 17 00:00:00 2001 From: Miguel Ángel Moreno Date: Sat, 31 Dec 2022 11:57:10 +0100 Subject: feat: Add support for comment replies --- src/backend/tau/api/comments.clj | 24 ++++++++++++++++-------- 1 file changed, 16 insertions(+), 8 deletions(-) (limited to 'src/backend/tau/api') diff --git a/src/backend/tau/api/comments.clj b/src/backend/tau/api/comments.clj index 477c72d..c0891ea 100644 --- a/src/backend/tau/api/comments.clj +++ b/src/backend/tau/api/comments.clj @@ -5,33 +5,41 @@ (:import org.schabi.newpipe.extractor.NewPipe org.schabi.newpipe.extractor.Page + org.schabi.newpipe.extractor.ListExtractor org.schabi.newpipe.extractor.comments.CommentsInfoItem org.schabi.newpipe.extractor.comments.CommentsInfo)) (defn get-comment-item - [item] + [item extractor] {:id (.getCommentId item) :text (.getCommentText item) :uploader-name (.getUploaderName item) :uploader-avatar (.getUploaderAvatarUrl item) :uploader-url (.getUploaderUrl item) :uploader-verified? (.isUploaderVerified item) - :upload-date (.getTextualUploadDate item) + :upload-date (.getTextualUploadDate item) :like-count (when-not (= (.getLikeCount item) -1) (.getLikeCount item)) + :reply-count (when-not (= (.getReplyCount item) -1) (.getReplyCount item)) :hearted-by-uploader? (.isHeartedByUploader item) :pinned? (.isPinned item) + :stream-position (when-not (= (.getStreamPosition item) -1) (.getStreamPosition item)) :replies (when (.getReplies item) - (j/from-java (.getReplies item)))}) + (if extractor + (let [comments-page (.getPage extractor (.getReplies item))] + {:next-page (when (.hasNextPage comments-page) (j/from-java (.getNextPage comments-page))) + :items (map #(get-comment-item % extractor) (.getItems comments-page))}) + (j/from-java (.getReplies item))))}) -(defn get-comment +(defn get-comments ([url] - (let [info (CommentsInfo/getInfo (url-decode url))] - {:comments (map get-comment-item (.getRelatedItems info)) + (let [info (CommentsInfo/getInfo (url-decode url)) + extractor (.getCommentsExtractor info)] + {:comments (map #(get-comment-item % extractor) (.getRelatedItems info)) :next-page (j/from-java (.getNextPage info)) :disabled? (.isCommentsDisabled info)})) ([url page-url] (let [service (NewPipe/getServiceByUrl (url-decode url)) - info (CommentsInfo/getMoreItems service url (Page. (url-decode page-url)))] - {:comments (map get-comment-item (.getItems info)) + info (CommentsInfo/getMoreItems service (url-decode url) (Page. (url-decode page-url)))] + {:comments (map #(get-comment-item % nil) (.getItems info)) :next-page (j/from-java (.getNextPage info)) :disabled? false}))) -- cgit v1.2.3