[Youtube] Add linked album for ytmusic searches

extractor/src/main/java/org/schabi/newpipe/extractor/services/youtube/extractors/YoutubeMusicSongOrVideoInfoItemExtractor.java

@@ -174,4 +174,26 @@ public List<Image> getThumbnails() throws ParsingException {
             throw new ParsingException("Could not get thumbnails", e);
         }
     }
+
+    @Override
+    public String getPlaylistId() throws ParsingException {
+        if (searchType.equals(MUSIC_SONGS)) {
+            for (final Object item : descriptionElements) {
+                final JsonObject browseEndpoint = ((JsonObject) item)
+                        .getObject("navigationEndpoint")
+                        .getObject("browseEndpoint");
+
+                final String type = browseEndpoint
+                        .getObject("browseEndpointContextSupportedConfigs")
+                        .getObject("browseEndpointContextMusicConfig")
+                        .getString("pageType");
+
+                if (type != null && type.equals("MUSIC_PAGE_TYPE_ALBUM")) {
+                    return browseEndpoint.getString("browseId");
+                }
+            }
+        }
+        // handles singles, video, and others which may not belong in playlist
+        return "";
+    }
 }

extractor/src/main/java/org/schabi/newpipe/extractor/stream/StreamInfoItem.java

@@ -37,6 +37,7 @@ public class StreamInfoItem extends InfoItem {
     private String uploaderName;
     private String shortDescription;
     private String textualUploadDate;
+    private String playlistId;
     @Nullable
     private DateWrapper uploadDate;
     private long viewCount = -1;
@@ -118,6 +119,14 @@ public void setTextualUploadDate(final String textualUploadDate) {
         this.textualUploadDate = textualUploadDate;
     }
 
+    public String getPlaylistId() {
+        return playlistId;
+    }
+
+    public void setPlaylistId(final String playlistId) {
+        this.playlistId = playlistId;
+    }
+
     @Nullable
     public DateWrapper getUploadDate() {
         return uploadDate;

extractor/src/main/java/org/schabi/newpipe/extractor/stream/StreamInfoItemExtractor.java

@@ -147,4 +147,14 @@ default String getShortDescription() throws ParsingException {
     default boolean isShortFormContent() throws ParsingException {
         return false;
     }
+
+    /**
+     * Gets the playlist id of the stream item.
+     *
+     * @return the playlist id of the stream item.
+     * @throws ParsingException if there is an error in the extraction
+     */
+    default String getPlaylistId() throws ParsingException {
+        return "";
+    }
 }

extractor/src/main/java/org/schabi/newpipe/extractor/stream/StreamInfoItemsCollector.java

@@ -103,6 +103,11 @@ public StreamInfoItem extract(final StreamInfoItemExtractor extractor) throws Pa
         } catch (final Exception e) {
             addError(e);
         }
+        try {
+            resultItem.setPlaylistId(extractor.getPlaylistId());
+        } catch (final Exception e) {
+            addError(e);
+        }
 
         return resultItem;
     }