From 59f3fd85488c2d97227cfba4c5fb455450c2faff Mon Sep 17 00:00:00 2001
From: Haggai Eran <haggai.eran@gmail.com>
Date: Sat, 4 Feb 2023 18:09:52 +0200
Subject: [PATCH] MediaBrowser: expose search history

---
 .../mediabrowser/MediaBrowserConnector.java   | 49 +++++++++++++++++--
 1 file changed, 46 insertions(+), 3 deletions(-)

diff --git a/app/src/main/java/org/schabi/newpipe/player/mediabrowser/MediaBrowserConnector.java b/app/src/main/java/org/schabi/newpipe/player/mediabrowser/MediaBrowserConnector.java
index ad724076f0f..91c5580949c 100644
--- a/app/src/main/java/org/schabi/newpipe/player/mediabrowser/MediaBrowserConnector.java
+++ b/app/src/main/java/org/schabi/newpipe/player/mediabrowser/MediaBrowserConnector.java
@@ -21,6 +21,7 @@
 import org.schabi.newpipe.NewPipeDatabase;
 import org.schabi.newpipe.R;
 import org.schabi.newpipe.database.AppDatabase;
+import org.schabi.newpipe.database.history.model.StreamHistoryEntry;
 import org.schabi.newpipe.database.playlist.PlaylistMetadataEntry;
 import org.schabi.newpipe.database.playlist.PlaylistStreamEntry;
 import org.schabi.newpipe.local.playlist.LocalPlaylistManager;
@@ -70,6 +71,10 @@ public void release() {
     private static final String ID_ROOT = "//${BuildConfig.APPLICATION_ID}/r";
     @NonNull
     private static final String ID_BOOKMARKS = ID_ROOT + "/playlists";
+    @NonNull
+    private static final String ID_HISTORY = ID_ROOT + "/history";
+    @NonNull
+    private static final String ID_STREAM = ID_ROOT + "/stream";
 
     private MediaItem createRootMediaItem(final String mediaId, final String folderName) {
         final var builder = new MediaDescriptionCompat.Builder();
@@ -135,7 +140,9 @@ public Single<List<MediaItem>> onLoadChildren(@NonNull final String parentId) {
             mediaItems.add(
                     createRootMediaItem(ID_BOOKMARKS,
                             playerService.getResources().getString(R.string.tab_bookmarks)));
-
+            mediaItems.add(
+                    createRootMediaItem(ID_HISTORY,
+                            playerService.getResources().getString(R.string.action_history)));
         } else if (parentId.startsWith(ID_BOOKMARKS)) {
             final var path = parentIdUri.getPathSegments();
             if (path.size() == 2) {
@@ -146,16 +153,38 @@ public Single<List<MediaItem>> onLoadChildren(@NonNull final String parentId) {
             } else {
                 Log.w(TAG, "Unknown playlist uri " + parentId);
             }
+        } else if (parentId.equals(ID_HISTORY)) {
+            return populateHistory();
         }
         return Single.just(mediaItems);
     }
 
-    private LocalPlaylistManager getPlaylistManager() {
+    private Single<List<MediaItem>> populateHistory() {
+        final var streamHistory = getDatabase().streamHistoryDAO();
+        final var history = streamHistory.getHistory().firstOrError();
+        return history.map(items ->
+                items.stream().map(this::createHistoryMediaItem).collect(Collectors.toList()));
+    }
+
+    private MediaItem createHistoryMediaItem(@NonNull final StreamHistoryEntry streamHistoryEntry) {
+        final var builder = new MediaDescriptionCompat.Builder();
+        builder.setMediaId(ID_STREAM + '/' + streamHistoryEntry.getStreamId())
+            .setTitle(streamHistoryEntry.getStreamEntity().getTitle())
+            .setIconUri(Uri.parse(streamHistoryEntry.getStreamEntity().getThumbnailUrl()));
+
+        return new MediaItem(builder.build(), MediaItem.FLAG_PLAYABLE);
+    }
+
+    private AppDatabase getDatabase() {
         if (database == null) {
             database = NewPipeDatabase.getInstance(playerService);
         }
+        return database;
+    }
+
+    private LocalPlaylistManager getPlaylistManager() {
         if (localPlaylistManager == null) {
-            localPlaylistManager = new LocalPlaylistManager(database);
+            localPlaylistManager = new LocalPlaylistManager(getDatabase());
         }
         return localPlaylistManager;
     }
@@ -205,6 +234,20 @@ private Single<PlayQueue> extractPlayQueueFromMediaId(final String mediaId) {
                             return new SinglePlayQueue(infoItems, index);
                         });
             }
+        } else if (mediaId.startsWith(ID_STREAM)) {
+            final var path = mediaIdUri.getPathSegments();
+            if (path.size() == 3) {
+                final long streamId = Long.parseLong(path.get(2));
+                return getDatabase().streamHistoryDAO().getHistory()
+                        .firstOrError()
+                        .map(items -> {
+                            final var infoItems = items.stream()
+                                    .filter(it -> it.getStreamId() == streamId)
+                                    .map(StreamHistoryEntry::toStreamInfoItem)
+                                    .collect(Collectors.toList());
+                            return new SinglePlayQueue(infoItems, 0);
+                        });
+            }
         }
 
         return Single.error(new NullPointerException());