MediaItem uri fix

app/src/main/kotlin/org/mjdev/tvapp/ui/screens/IPTVScreen.kt

@@ -29,7 +29,6 @@ import org.mjdev.tvlib.screen.Screen
 import org.mjdev.tvlib.ui.components.media.MediaPlayerContainer
 import org.mjdev.tvlib.extensions.MediaItemExt.uri
 import org.mjdev.tvlib.extensions.ListExt.indexOf
-import org.mjdev.tvlib.extensions.MediaItemExt.mediaItem
 import org.mjdev.tvlib.ui.components.media.MediaPlayerState.Companion.rememberMediaPlayerState
 
 @androidx.annotation.OptIn(androidx.media3.common.util.UnstableApi::class)
@@ -57,7 +56,7 @@ class IPTVScreen : Screen() {
             IPTVViewModel.mock(context)
         }
 
-        val data: MediaItem = args[data].mediaItem
+        val data: Any? = args[data]
 
         val dataList: List<MediaItem> = remember(data) {
             viewModel.mediaItemsFor(data)

app/src/main/kotlin/org/mjdev/tvapp/viewmodel/IPTVViewModel.kt

@@ -57,7 +57,11 @@ class IPTVViewModel @Inject constructor() : BaseViewModel() {
         is ItemAudio -> getCachedList<ItemAudio> { localAudioCursor }
         is ItemVideo -> getCachedList<ItemVideo> { localVideoCursor }
         is ItemPhoto -> getCachedList<ItemPhoto> { localPhotoCursor }
-        is Movie -> getCachedList<Movie> { mediaItems }
+        is Movie -> getCachedList<Movie> {
+            mediaItems.filter { listItem ->
+                (listItem as? Movie?)?.category == data.category
+            }
+        }
         else -> listOf(data.mediaItem)
     }