From f66e77330eef492e8de9d071918a8074740bca05 Mon Sep 17 00:00:00 2001
From: FireMasterK <20838718+FireMasterK@users.noreply.github.com>
Date: Fri, 23 Jul 2021 01:34:21 +0530
Subject: [PATCH] Add route to see subscriptions.

---
 .../java/me/kavin/piped/ServerLauncher.java   |  7 ++++
 .../me/kavin/piped/utils/DatabaseHelper.java  |  9 +++++
 .../me/kavin/piped/utils/ResponseHelper.java  | 36 +++++++++++++++++++
 .../piped/utils/obj/SubscriptionChannel.java  | 15 ++++++++
 4 files changed, 67 insertions(+)
 create mode 100644 src/main/java/me/kavin/piped/utils/obj/SubscriptionChannel.java

diff --git a/src/main/java/me/kavin/piped/ServerLauncher.java b/src/main/java/me/kavin/piped/ServerLauncher.java
index 8f96933..4e81dfc 100644
--- a/src/main/java/me/kavin/piped/ServerLauncher.java
+++ b/src/main/java/me/kavin/piped/ServerLauncher.java
@@ -260,6 +260,13 @@ public class ServerLauncher extends MultithreadedHttpServerLauncher {
                     } catch (Exception e) {
                         return getErrorResponse(e);
                     }
+                })).map(GET, "/subscriptions", AsyncServlet.ofBlocking(executor, request -> {
+                    try {
+                        return getJsonResponse(ResponseHelper.subscriptionsResponse(request.getHeader(AUTHORIZATION)),
+                                "private");
+                    } catch (Exception e) {
+                        return getErrorResponse(e);
+                    }
                 }));
 
         return new CustomServletDecorator(router);
diff --git a/src/main/java/me/kavin/piped/utils/DatabaseHelper.java b/src/main/java/me/kavin/piped/utils/DatabaseHelper.java
index 7fb1ce7..f09047a 100644
--- a/src/main/java/me/kavin/piped/utils/DatabaseHelper.java
+++ b/src/main/java/me/kavin/piped/utils/DatabaseHelper.java
@@ -46,6 +46,15 @@ public class DatabaseHelper {
         return s.createQuery(cr).uniqueResult();
     }
 
+    public static final List<Channel> getChannelFromIds(Session s, List<String> id) {
+        CriteriaBuilder cb = s.getCriteriaBuilder();
+        CriteriaQuery<Channel> cr = cb.createQuery(Channel.class);
+        Root<Channel> root = cr.from(Channel.class);
+        cr.select(root).where(root.get("uploader_id").in(id));
+
+        return s.createQuery(cr).getResultList();
+    }
+
     public static final List<Video> getVideosFromChannelIds(Session s, List<String> id) {
         CriteriaBuilder cb = s.getCriteriaBuilder();
         CriteriaQuery<Video> cr = cb.createQuery(Video.class);
diff --git a/src/main/java/me/kavin/piped/utils/ResponseHelper.java b/src/main/java/me/kavin/piped/utils/ResponseHelper.java
index 657c606..f6d78ea 100644
--- a/src/main/java/me/kavin/piped/utils/ResponseHelper.java
+++ b/src/main/java/me/kavin/piped/utils/ResponseHelper.java
@@ -75,6 +75,7 @@ import me.kavin.piped.utils.obj.SearchResults;
 import me.kavin.piped.utils.obj.StreamItem;
 import me.kavin.piped.utils.obj.Streams;
 import me.kavin.piped.utils.obj.StreamsPage;
+import me.kavin.piped.utils.obj.SubscriptionChannel;
 import me.kavin.piped.utils.obj.Subtitle;
 import me.kavin.piped.utils.obj.db.PubSub;
 import me.kavin.piped.utils.obj.db.User;
@@ -864,6 +865,41 @@ public class ResponseHelper {
 
     }
 
+    public static final byte[] subscriptionsResponse(String session)
+            throws IOException, NoSuchAlgorithmException, InvalidKeySpecException {
+
+        Session s = DatabaseSessionFactory.createSession();
+
+        User user = DatabaseHelper.getUserFromSessionWithSubscribed(s, session);
+
+        if (user != null) {
+
+            List<SubscriptionChannel> subscriptionItems = new ObjectArrayList<>();
+
+            if (user.getSubscribed() != null && !user.getSubscribed().isEmpty()) {
+
+                List<me.kavin.piped.utils.obj.db.Channel> channels = DatabaseHelper.getChannelFromIds(s,
+                        user.getSubscribed());
+
+                channels.forEach(channel -> {
+                    subscriptionItems.add(new SubscriptionChannel("/channel/" + channel.getUploaderId(),
+                            channel.getUploader(), rewriteURL(channel.getUploaderAvatar()), channel.isVerified()));
+                });
+
+                Collections.sort(subscriptionItems, (a, b) -> (a.name.compareTo(b.name)));
+            }
+
+            s.close();
+
+            return Constants.mapper.writeValueAsBytes(subscriptionItems);
+        }
+
+        s.close();
+
+        return Constants.mapper.writeValueAsBytes(new AuthenticationFailureResponse());
+
+    }
+
     private static final String getLBRYStreamURL(String videoId) throws IOException, InterruptedException {
 
         String lbryId = new JSONObject(Constants.h2client.send(HttpRequest
diff --git a/src/main/java/me/kavin/piped/utils/obj/SubscriptionChannel.java b/src/main/java/me/kavin/piped/utils/obj/SubscriptionChannel.java
new file mode 100644
index 0000000..e5a7000
--- /dev/null
+++ b/src/main/java/me/kavin/piped/utils/obj/SubscriptionChannel.java
@@ -0,0 +1,15 @@
+package me.kavin.piped.utils.obj;
+
+public class SubscriptionChannel {
+
+    public String url, name, avatar;
+
+    public boolean verified;
+
+    public SubscriptionChannel(String url, String name, String avatar, boolean verified) {
+        this.url = url;
+        this.name = name;
+        this.avatar = avatar;
+        this.verified = verified;
+    }
+}