Merge pull request #782 from Bnyro/master

feat(rss feed): add optional query parameter 'filter' ('shorts' or 'videos')
This commit is contained in:
Bnyro 2024-03-15 14:52:22 +01:00 committed by GitHub
commit 77e8d3c5be
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
3 changed files with 20 additions and 5 deletions

View File

@ -315,7 +315,8 @@ public class ServerLauncher extends MultithreadedHttpServerLauncher {
}
})).map(GET, "/feed/rss", AsyncServlet.ofBlocking(executor, request -> {
try {
return getRawResponse(FeedHandlers.feedResponseRSS(request.getQueryParameter("authToken")),
return getRawResponse(FeedHandlers.feedResponseRSS(request.getQueryParameter("authToken"),
request.getQueryParameter("filter")),
"application/atom+xml", "public, s-maxage=120");
} catch (Exception e) {
return getErrorResponse(e, request.getPath());
@ -339,7 +340,8 @@ public class ServerLauncher extends MultithreadedHttpServerLauncher {
})).map(GET, "/feed/unauthenticated/rss", AsyncServlet.ofBlocking(executor, request -> {
try {
return getRawResponse(FeedHandlers.unauthenticatedFeedResponseRSS(
getArray(request.getQueryParameter("channels"))
getArray(request.getQueryParameter("channels")),
request.getQueryParameter("filter")
), "application/atom+xml", "public, s-maxage=120");
} catch (Exception e) {
return getErrorResponse(e, request.getPath());

View File

@ -21,6 +21,7 @@ import org.hibernate.Session;
import org.hibernate.StatelessSession;
import org.schabi.newpipe.extractor.channel.ChannelInfo;
import javax.annotation.Nullable;
import java.io.IOException;
import java.util.*;
import java.util.concurrent.TimeUnit;
@ -119,7 +120,7 @@ public class FeedHandlers {
return null;
}
public static byte[] feedResponseRSS(String session) throws FeedException {
public static byte[] feedResponseRSS(String session, @Nullable String filter) throws FeedException {
if (StringUtils.isBlank(session))
ExceptionHandler.throwErrorResponse(new InvalidRequestResponse("session is a required parameter"));
@ -131,6 +132,7 @@ public class FeedHandlers {
SyndFeed feed = FeedHelpers.createRssFeed(user.getUsername());
final List<SyndEntry> entries = FeedHelpers.generateAuthenticatedFeed(s, user.getId(), 100)
.filter(FeedHelpers.createFeedFilter(filter))
.map(video -> {
var channel = video.getChannel();
return ChannelHelpers.createEntry(video, channel);
@ -173,7 +175,7 @@ public class FeedHandlers {
}
}
public static byte[] unauthenticatedFeedResponseRSS(String[] channelIds) throws Exception {
public static byte[] unauthenticatedFeedResponseRSS(String[] channelIds, @Nullable String filter) throws Exception {
Set<String> filteredChannels = Arrays.stream(channelIds)
.filter(ChannelHelpers::isValidId)
@ -183,7 +185,9 @@ public class FeedHandlers {
ExceptionHandler.throwErrorResponse(new InvalidRequestResponse("No valid channel IDs provided"));
try (StatelessSession s = DatabaseSessionFactory.createStatelessSession()) {
List<Video> videos = FeedHelpers.generateUnauthenticatedFeed(s, filteredChannels, 100).toList();
List<Video> videos = FeedHelpers.generateUnauthenticatedFeed(s, filteredChannels, 100)
.filter(FeedHelpers.createFeedFilter(filter))
.toList();
List<SyndEntry> entries = videos.stream()
.map(video -> ChannelHelpers.createEntry(video, video.getChannel()))

View File

@ -16,6 +16,7 @@ import javax.annotation.Nullable;
import java.util.Comparator;
import java.util.Date;
import java.util.Set;
import java.util.function.Predicate;
import java.util.stream.Stream;
import static me.kavin.piped.utils.URLUtils.rewriteURL;
@ -79,6 +80,14 @@ public class FeedHelpers {
return feed;
}
public static Predicate<Video> createFeedFilter(@Nullable String filter) {
return video -> switch (filter) {
case "shorts" -> video.isShort();
case "videos" -> !video.isShort();
case null, default -> true;
};
}
public static Stream<SubscriptionChannel> generateSubscriptionsList(Stream<Channel> channels) {
return channels.parallel()
.filter(channel -> channel.getUploader() != null)