Fix radio in home page

maxrave-dev · Dec 4, 2024 · 6bfee80 · 6bfee80
1 parent 08d81a8
commit 6bfee80
Showing 1 changed file with 14 additions and 6 deletions.
diff --git a/kotlinYtmusicScraper/src/main/java/com/maxrave/kotlinytmusicscraper/pages/RelatedPage.kt b/kotlinYtmusicScraper/src/main/java/com/maxrave/kotlinytmusicscraper/pages/RelatedPage.kt
@@ -101,15 +101,23 @@ data class RelatedPage(
                                 ?.musicItemThumbnailOverlayRenderer?.content
                                 ?.musicPlayButtonRenderer?.playNavigationEndpoint
                                 ?.watchPlaylistEndpoint ?: return null,
+                        // If the playlist is radio, shuffle is not available
                         shuffleEndpoint =
-                            renderer.menu?.menuRenderer?.items?.find {
-                                it.menuNavigationItemRenderer?.icon?.iconType == "MUSIC_SHUFFLE"
-                            }?.menuNavigationItemRenderer?.navigationEndpoint?.watchPlaylistEndpoint
-                                ?: return null,
+                        renderer.menu?.menuRenderer?.items?.find {
+                            it.menuNavigationItemRenderer?.icon?.iconType == "MUSIC_SHUFFLE"
+                        }?.menuNavigationItemRenderer?.navigationEndpoint?.watchPlaylistEndpoint
+                            ?: renderer.thumbnailOverlay
+                                .musicItemThumbnailOverlayRenderer.content
+                                .musicPlayButtonRenderer.playNavigationEndpoint
+                                .watchPlaylistEndpoint,
                         radioEndpoint =
-                            renderer.menu.menuRenderer.items.find {
+                            renderer.menu?.menuRenderer?.items?.find {
                                 it.menuNavigationItemRenderer?.icon?.iconType == "MIX"
-                            }?.menuNavigationItemRenderer?.navigationEndpoint?.watchPlaylistEndpoint,
+                            }?.menuNavigationItemRenderer?.navigationEndpoint?.watchPlaylistEndpoint
+                                ?: renderer.thumbnailOverlay
+                                    .musicItemThumbnailOverlayRenderer.content
+                                    .musicPlayButtonRenderer.playNavigationEndpoint
+                                    .watchPlaylistEndpoint,
                     )
 
                 renderer.isArtist -> {