mirror of
https://github.com/TeamPiped/Piped-Backend.git
synced 2024-08-14 23:51:41 +00:00
Merge pull request #416 from TeamPiped/code-improvements
Minor code improvements
This commit is contained in:
commit
2ab8fde0bb
4 changed files with 68 additions and 64 deletions
|
@ -21,7 +21,6 @@ dependencies {
|
|||
implementation 'com.fasterxml.jackson.core:jackson-core:2.13.4'
|
||||
implementation 'com.fasterxml.jackson.core:jackson-annotations:2.13.4'
|
||||
implementation 'com.fasterxml.jackson.core:jackson-databind:2.13.4.2'
|
||||
implementation 'org.json:json:20220924'
|
||||
implementation 'com.github.ben-manes.caffeine:caffeine:3.1.1'
|
||||
implementation 'com.rometools:rome:1.18.0'
|
||||
implementation 'com.github.ipfs:java-ipfs-http-client:v1.3.3'
|
||||
|
|
|
@ -48,6 +48,9 @@ public class SearchHandlers {
|
|||
public static byte[] searchResponse(String q, String filter)
|
||||
throws IOException, ExtractionException {
|
||||
|
||||
if (StringUtils.isEmpty(q) || StringUtils.isEmpty(filter))
|
||||
ExceptionHandler.throwErrorResponse(new InvalidRequestResponse("query and filter are required parameters"));
|
||||
|
||||
Sentry.setExtra("query", q);
|
||||
|
||||
final SearchInfo info = SearchInfo.getInfo(YOUTUBE_SERVICE,
|
||||
|
@ -65,8 +68,8 @@ public class SearchHandlers {
|
|||
public static byte[] searchPageResponse(String q, String filter, String prevpageStr)
|
||||
throws IOException, ExtractionException {
|
||||
|
||||
if (StringUtils.isEmpty(prevpageStr))
|
||||
ExceptionHandler.throwErrorResponse(new InvalidRequestResponse("nextpage is a required parameter"));
|
||||
if (StringUtils.isEmpty(q) || StringUtils.isEmpty(prevpageStr))
|
||||
ExceptionHandler.throwErrorResponse(new InvalidRequestResponse("query and nextpage are required parameter"));
|
||||
|
||||
Page prevpage = mapper.readValue(prevpageStr, Page.class);
|
||||
|
||||
|
|
|
@ -119,16 +119,15 @@ public class FeedHandlers {
|
|||
)
|
||||
.orderBy(cb.desc(root.get("uploaded")));
|
||||
|
||||
List<StreamItem> feedItems = new ObjectArrayList<>();
|
||||
|
||||
for (Video video : s.createQuery(criteria).setTimeout(20).list()) {
|
||||
List<StreamItem> feedItems = s.createQuery(criteria).setTimeout(20).stream()
|
||||
.parallel().map(video -> {
|
||||
var channel = video.getChannel();
|
||||
|
||||
feedItems.add(new StreamItem("/watch?v=" + video.getId(), video.getTitle(),
|
||||
return new StreamItem("/watch?v=" + video.getId(), video.getTitle(),
|
||||
rewriteURL(video.getThumbnail()), channel.getUploader(), "/channel/" + channel.getUploaderId(),
|
||||
rewriteURL(channel.getUploaderAvatar()), null, null, video.getDuration(), video.getViews(),
|
||||
video.getUploaded(), channel.isVerified(), video.isShort()));
|
||||
}
|
||||
video.getUploaded(), channel.isVerified(), video.isShort());
|
||||
}).toList();
|
||||
|
||||
return mapper.writeValueAsBytes(feedItems);
|
||||
}
|
||||
|
@ -173,14 +172,11 @@ public class FeedHandlers {
|
|||
)
|
||||
.orderBy(cb.desc(root.get("uploaded")));
|
||||
|
||||
List<Video> videos = s.createQuery(criteria)
|
||||
final List<SyndEntry> entries = s.createQuery(criteria)
|
||||
.setTimeout(20)
|
||||
.setMaxResults(100)
|
||||
.list();
|
||||
|
||||
final List<SyndEntry> entries = new ObjectArrayList<>();
|
||||
|
||||
for (Video video : videos) {
|
||||
.stream()
|
||||
.map(video -> {
|
||||
var channel = video.getChannel();
|
||||
SyndEntry entry = new SyndEntryImpl();
|
||||
|
||||
|
@ -194,8 +190,9 @@ public class FeedHandlers {
|
|||
entry.setUri(Constants.FRONTEND_URL + "/watch?v=" + video.getId());
|
||||
entry.setTitle(video.getTitle());
|
||||
entry.setPublishedDate(new Date(video.getUploaded()));
|
||||
entries.add(entry);
|
||||
}
|
||||
|
||||
return entry;
|
||||
}).toList();
|
||||
|
||||
feed.setEntries(entries);
|
||||
|
||||
|
@ -232,16 +229,15 @@ public class FeedHandlers {
|
|||
))
|
||||
.orderBy(cb.desc(root.get("uploaded")));
|
||||
|
||||
List<StreamItem> feedItems = new ObjectArrayList<>();
|
||||
|
||||
for (Video video : s.createQuery(criteria).setTimeout(20).list()) {
|
||||
List<StreamItem> feedItems = s.createQuery(criteria).setTimeout(20).stream()
|
||||
.parallel().map(video -> {
|
||||
var channel = video.getChannel();
|
||||
|
||||
feedItems.add(new StreamItem("/watch?v=" + video.getId(), video.getTitle(),
|
||||
return new StreamItem("/watch?v=" + video.getId(), video.getTitle(),
|
||||
rewriteURL(video.getThumbnail()), channel.getUploader(), "/channel/" + channel.getUploaderId(),
|
||||
rewriteURL(channel.getUploaderAvatar()), null, null, video.getDuration(), video.getViews(),
|
||||
video.getUploaded(), channel.isVerified(), video.isShort()));
|
||||
}
|
||||
video.getUploaded(), channel.isVerified(), video.isShort());
|
||||
}).toList();
|
||||
|
||||
updateSubscribedTime(filtered);
|
||||
addMissingChannels(filtered);
|
||||
|
@ -451,10 +447,9 @@ public class FeedHandlers {
|
|||
query.select(root)
|
||||
.where(root.get("uploader_id").in(subquery));
|
||||
|
||||
var channels = s.createQuery(query).list();
|
||||
|
||||
List<SubscriptionChannel> subscriptionItems = channels
|
||||
List<SubscriptionChannel> subscriptionItems = s.createQuery(query)
|
||||
.stream().parallel()
|
||||
.filter(Objects::nonNull)
|
||||
.sorted(Comparator.comparing(me.kavin.piped.utils.obj.db.Channel::getUploader, String.CASE_INSENSITIVE_ORDER))
|
||||
.map(channel -> new SubscriptionChannel("/channel/" + channel.getUploaderId(),
|
||||
channel.getUploader(), rewriteURL(channel.getUploaderAvatar()), channel.isVerified()))
|
||||
|
@ -488,9 +483,7 @@ public class FeedHandlers {
|
|||
query.select(root);
|
||||
query.where(root.get("uploader_id").in(filtered));
|
||||
|
||||
var channels = s.createQuery(query).list();
|
||||
|
||||
List<SubscriptionChannel> subscriptionItems = channels
|
||||
List<SubscriptionChannel> subscriptionItems = s.createQuery(query)
|
||||
.stream().parallel()
|
||||
.sorted(Comparator.comparing(me.kavin.piped.utils.obj.db.Channel::getUploader, String.CASE_INSENSITIVE_ORDER))
|
||||
.map(channel -> new SubscriptionChannel("/channel/" + channel.getUploaderId(),
|
||||
|
|
|
@ -5,10 +5,14 @@ import okhttp3.MediaType;
|
|||
import okhttp3.Request;
|
||||
import okhttp3.RequestBody;
|
||||
import org.apache.commons.lang3.StringUtils;
|
||||
import org.json.JSONObject;
|
||||
|
||||
import java.io.IOException;
|
||||
|
||||
import static me.kavin.piped.consts.Constants.h2client;
|
||||
import static me.kavin.piped.consts.Constants.mapper;
|
||||
import static me.kavin.piped.utils.RequestUtils.sendGet;
|
||||
import static me.kavin.piped.utils.URLUtils.silentEncode;
|
||||
|
||||
public class LbryHelper {
|
||||
|
||||
public static String getLBRYId(String videoId) throws IOException {
|
||||
|
@ -16,9 +20,10 @@ public class LbryHelper {
|
|||
if (Constants.DISABLE_LBRY)
|
||||
return null;
|
||||
|
||||
return new JSONObject(
|
||||
RequestUtils.sendGet("https://api.lbry.com/yt/resolve?video_ids=" + URLUtils.silentEncode(videoId))
|
||||
).getJSONObject("data").getJSONObject("videos").optString(videoId, null);
|
||||
return mapper.readTree(sendGet("https://api.lbry.com/yt/resolve?video_ids=" + silentEncode(videoId)))
|
||||
.at("/data/videos")
|
||||
.path(videoId)
|
||||
.asText(null);
|
||||
}
|
||||
|
||||
public static String getLBRYStreamURL(String lbryId)
|
||||
|
@ -29,24 +34,28 @@ public class LbryHelper {
|
|||
|
||||
var request = new Request.Builder()
|
||||
.url("https://api.lbry.tv/api/v1/proxy?m=get")
|
||||
.post(RequestBody.create(String.valueOf(
|
||||
new JSONObject().put("id", System.currentTimeMillis())
|
||||
.post(RequestBody.create(mapper.writeValueAsBytes(
|
||||
mapper.createObjectNode()
|
||||
.put("id", System.currentTimeMillis())
|
||||
.put("id", System.currentTimeMillis())
|
||||
.put("jsonrpc", "2.0")
|
||||
.put("method", "get")
|
||||
.put("params",
|
||||
new JSONObject()
|
||||
.set("params",
|
||||
mapper.createObjectNode()
|
||||
.put("uri", "lbry://" + lbryId)
|
||||
.put("save_file", true)))
|
||||
, MediaType.get("application/json")))
|
||||
.put("save_file", true)
|
||||
)
|
||||
), MediaType.get("application/json")))
|
||||
.build();
|
||||
|
||||
var resp = Constants.h2client.newCall(request).execute();
|
||||
|
||||
var json = new JSONObject(resp.body().string());
|
||||
|
||||
resp.close();
|
||||
|
||||
return json.getJSONObject("result").getString("streaming_url");
|
||||
|
||||
try (var resp = h2client.newCall(request).execute()) {
|
||||
if (resp.isSuccessful()) {
|
||||
return mapper.readTree(resp.body().byteStream())
|
||||
.at("/result/streaming_url")
|
||||
.asText(null);
|
||||
}
|
||||
}
|
||||
|
||||
return null;
|
||||
}
|
||||
}
|
||||
|
|
Loading…
Reference in a new issue