Add isPlaylist to response JSON (lavalink-devs#91)

* Add isPlaylist to response JSON * Unmodifiable list * Expose selectedTrack and playlistName
freyacoded · Jun 2, 2018 · 458039c · 458039c
1 parent 4b893bd
commit 458039c
Show file tree

Hide file tree

Showing 3 changed files with 55 additions and 20 deletions.
diff --git a/IMPLEMENTATION.md b/IMPLEMENTATION.md
@@ -226,21 +226,24 @@ Authorization: youshallnotpass
 
 Response:
 ```json
-[
-  {
-    "track": "QAAAjQIAJVJpY2sgQXN0bGV5IC0gTmV2ZXIgR29ubmEgR2l2ZSBZb3UgVXAADlJpY2tBc3RsZXlWRVZPAAAAAAADPCAAC2RRdzR3OVdnWGNRAAEAK2h0dHBzOi8vd3d3LnlvdXR1YmUuY29tL3dhdGNoP3Y9ZFF3NHc5V2dYY1EAB3lvdXR1YmUAAAAAAAAAAA==",
-    "info": {
-      "identifier": "dQw4w9WgXcQ",
-      "isSeekable": true,
-      "author": "RickAstleyVEVO",
-      "length": 212000,
-      "isStream": false,
-      "position": 0,
-      "title": "Rick Astley - Never Gonna Give You Up",
-      "uri": "https://www.youtube.com/watch?v=dQw4w9WgXcQ"
+{
+  "isPlaylist": false,
+  "tracks": [
+    {
+      "track": "QAAAjQIAJVJpY2sgQXN0bGV5IC0gTmV2ZXIgR29ubmEgR2l2ZSBZb3UgVXAADlJpY2tBc3RsZXlWRVZPAAAAAAADPCAAC2RRdzR3OVdnWGNRAAEAK2h0dHBzOi8vd3d3LnlvdXR1YmUuY29tL3dhdGNoP3Y9ZFF3NHc5V2dYY1EAB3lvdXR1YmUAAAAAAAAAAA==",
+      "info": {
+        "identifier": "dQw4w9WgXcQ",
+        "isSeekable": true,
+        "author": "RickAstleyVEVO",
+        "length": 212000,
+        "isStream": false,
+        "position": 0,
+        "title": "Rick Astley - Never Gonna Give You Up",
+        "uri": "https://www.youtube.com/watch?v=dQw4w9WgXcQ"
+      }
     }
-  }
-]
+  ]
+}
 ```
 
 ### Special notes

diff --git a/LavalinkServer/src/main/java/lavalink/server/player/AudioLoader.java b/LavalinkServer/src/main/java/lavalink/server/player/AudioLoader.java
@@ -31,6 +31,7 @@
 import org.slf4j.LoggerFactory;
 
 import java.util.ArrayList;
+import java.util.Collections;
 import java.util.List;
 
 public class AudioLoader implements AudioLoadResultHandler {
@@ -39,13 +40,16 @@ public class AudioLoader implements AudioLoadResultHandler {
     private final AudioPlayerManager audioPlayerManager;
 
     private List<AudioTrack> loadedItems;
+    private boolean isPlaylist = false;
+    private String playlistName = null;
+    private Integer selectedTrack = null;
     private boolean used = false;
 
     public AudioLoader(AudioPlayerManager audioPlayerManager) {
         this.audioPlayerManager = audioPlayerManager;
     }
 
-    List<AudioTrack> loadSync(String identifier) throws InterruptedException {
+    LoadResult loadSync(String identifier) throws InterruptedException {
         if(used)
             throw new IllegalStateException("This loader can only be used once per instance");
 
@@ -57,7 +61,7 @@ List<AudioTrack> loadSync(String identifier) throws InterruptedException {
             this.wait();
         }
 
-        return loadedItems;
+        return new LoadResult(loadedItems, isPlaylist, playlistName, selectedTrack);
     }
 
     @Override
@@ -72,6 +76,12 @@ public void trackLoaded(AudioTrack audioTrack) {
 
     @Override
     public void playlistLoaded(AudioPlaylist audioPlaylist) {
+        if (!audioPlaylist.isSearchResult()) {
+            isPlaylist = true;
+            playlistName = audioPlaylist.getName();
+            selectedTrack = audioPlaylist.getTracks().indexOf(audioPlaylist.getSelectedTrack());
+        }
+
         log.info("Loaded playlist " + audioPlaylist.getName());
         loadedItems = audioPlaylist.getTracks();
         synchronized (this) {
@@ -98,3 +108,17 @@ public void loadFailed(FriendlyException e) {
     }
 
 }
+
+class LoadResult {
+    public List<AudioTrack> tracks;
+    public boolean isPlaylist;
+    public String playlistName;
+    public Integer selectedTrack;
+
+    public LoadResult(List<AudioTrack> tracks, boolean isPlaylist, String playlistName, Integer selectedTrack) {
+        this.tracks = Collections.unmodifiableList(tracks);
+        this.isPlaylist = isPlaylist;
+        this.playlistName = playlistName;
+        this.selectedTrack = selectedTrack;
+    }
+}
diff --git a/LavalinkServer/src/main/java/lavalink/server/player/AudioLoaderRestHandler.java b/LavalinkServer/src/main/java/lavalink/server/player/AudioLoaderRestHandler.java
@@ -41,7 +41,6 @@
 import javax.servlet.http.HttpServletRequest;
 import javax.servlet.http.HttpServletResponse;
 import java.io.IOException;
-import java.util.List;
 
 @Controller
 public class AudioLoaderRestHandler {
@@ -97,10 +96,12 @@ public String getLoadTracks(HttpServletRequest request, HttpServletResponse resp
         if (!isAuthorized(request, response))
             return "";
 
+        JSONObject json = new JSONObject();
+        JSONObject playlist = new JSONObject();
         JSONArray tracks = new JSONArray();
-        List<AudioTrack> list = new AudioLoader(audioPlayerManager).loadSync(identifier);
+        LoadResult result = new AudioLoader(audioPlayerManager).loadSync(identifier);
 
-        list.forEach(track -> {
+        result.tracks.forEach(track -> {
             JSONObject object = new JSONObject();
             object.put("info", trackToJSON(track));
 
@@ -113,7 +114,14 @@ public String getLoadTracks(HttpServletRequest request, HttpServletResponse resp
             }
         });
 
-        return tracks.toString();
+        playlist.put("name", result.playlistName);
+        playlist.put("selectedTrack", result.selectedTrack);
+
+        json.put("isPlaylist", result.isPlaylist);
+        json.put("playlistInfo", playlist);
+        json.put("tracks", tracks);
+
+        return json.toString();
     }
 
     @GetMapping(value = "/decodetrack", produces = "application/json")