feat(stremio-core-protobuf): VideoDeepLinks for MetaDetails and Player

elpiel · elpiel · commit b73d1fd3fac1 · 2024-12-17T13:54:02.000+02:00
Signed-off-by: Lachezar Lechev &lt;lachezar@ambire.com&gt;
diff --git a/stremio-core-protobuf/proto/stremio/core/types/video.proto b/stremio-core-protobuf/proto/stremio/core/types/video.proto
@@ -18,9 +18,28 @@ message Video {
   required bool upcoming = 8;
   required bool watched = 9;
   required bool current_video = 10;
+  // Watch progress percentage
+  optional double progress = 11;
+  // scheduled: bool,
+
+  required VideoDeepLinks deep_links = 12;
 
   message SeriesInfo {
     required int64 season = 1;
     required int64 episode = 2;
   }
+}
+
+message VideoDeepLinks {
+  required string meta_details_videos = 1;
+  required string meta_details_streams = 2;
+  // pub player: Option<String>,
+  // pub external_player: Option<ExternalPlayerLink>,
+  optional string player = 3;
+  required ExternalPlayerLink external_player = 4;
+
+  message ExternalPlayerLink {
+    optional string download = 1;
+    optional string streaming = 2;
+  }
 }
diff --git a/stremio-core-protobuf/src/bridge/loadable.rs b/stremio-core-protobuf/src/bridge/loadable.rs
@@ -7,7 +7,7 @@ use stremio_core::{
         common::{Loadable, ResourceError},
         ctx::{Ctx, CtxError},
         link::LinkError,
-        streaming_server::PlaybackDevice,
+        streaming_server::{PlaybackDevice, StreamingServer},
     },
     runtime::EnvError,
     types::{
@@ -53,6 +53,8 @@ impl
     ToProtobuf<
         models::loadable_meta_item::Content,
         (
+            &Ctx,
+            &StreamingServer,
             Option<&LibraryItem>,
             Option<&WatchedBitField>,
             Option<&String>,
@@ -62,17 +64,26 @@ impl
 {
     fn to_protobuf<E: stremio_core::runtime::Env + 'static>(
         &self,
-        (library_item, watched, addon_name, meta_request): &(
+        (ctx, streaming_server, library_item, watched, addon_name, meta_request): &(
+            &Ctx,
+            &StreamingServer,
             Option<&LibraryItem>,
             Option<&WatchedBitField>,
             Option<&String>,
             &ResourceRequest,
         ),
     ) -> models::loadable_meta_item::Content {
         match &self {
-            Loadable::Ready(ready) => models::loadable_meta_item::Content::Ready(
-                ready.to_protobuf::<E>(&(*library_item, *watched, *addon_name, *meta_request)),
-            ),
+            Loadable::Ready(ready) => {
+                models::loadable_meta_item::Content::Ready(ready.to_protobuf::<E>(&(
+                    *ctx,
+                    *streaming_server,
+                    *library_item,
+                    *watched,
+                    *addon_name,
+                    *meta_request,
+                )))
+            }
             Loadable::Err(error) => models::loadable_meta_item::Content::Error(models::Error {
                 message: error.to_string(),
             }),
diff --git a/stremio-core-protobuf/src/bridge/resource_loadable.rs b/stremio-core-protobuf/src/bridge/resource_loadable.rs
@@ -6,6 +6,7 @@ use stremio_core::{
     models::{
         common::{DescriptorLoadable, ResourceLoadable},
         ctx::Ctx,
+        streaming_server::StreamingServer,
     },
     types::{
         addon::{Descriptor, ResourceRequest},
@@ -60,18 +61,33 @@ impl ToProtobuf<models::LoadablePage, Ctx> for ResourceLoadable<Vec<MetaItemPrev
     }
 }
 
-impl ToProtobuf<models::LoadableMetaItem, (&Ctx, Option<&LibraryItem>, Option<&WatchedBitField>)>
-    for &ResourceLoadable<MetaItem>
+impl
+    ToProtobuf<
+        models::LoadableMetaItem,
+        (
+            &Ctx,
+            &StreamingServer,
+            Option<&LibraryItem>,
+            Option<&WatchedBitField>,
+        ),
+    > for &ResourceLoadable<MetaItem>
 {
     fn to_protobuf<E: stremio_core::runtime::Env + 'static>(
         &self,
-        (ctx, library_item, watched): &(&Ctx, Option<&LibraryItem>, Option<&WatchedBitField>),
+        (ctx, streming_server, library_item, watched): &(
+            &Ctx,
+            &StreamingServer,
+            Option<&LibraryItem>,
+            Option<&WatchedBitField>,
+        ),
     ) -> models::LoadableMetaItem {
         let addon_name = get_addon_name(ctx, &self.request.base);
         models::LoadableMetaItem {
             title: addon_name.to_string(),
             request: self.request.to_protobuf::<E>(&()),
             content: self.content.to_protobuf::<E>(&(
+                *ctx,
+                *streming_server,
                 *library_item,
                 *watched,
                 Some(&addon_name),
diff --git a/stremio-core-protobuf/src/model/fields/meta_details.rs b/stremio-core-protobuf/src/model/fields/meta_details.rs
@@ -1,13 +1,19 @@
 use boolinator::Boolinator;
-use stremio_core::deep_links::MetaItemDeepLinks;
+
+use stremio_core::deep_links::{MetaItemDeepLinks, VideoDeepLinks};
 use stremio_core::models::ctx::Ctx;
-use stremio_core::models::meta_details::{MetaDetails, Selected};
+use stremio_core::models::{
+    meta_details::{MetaDetails, Selected},
+    streaming_server::StreamingServer,
+};
 use stremio_core::runtime::Env;
 use stremio_core::types::addon::ResourceRequest;
 use stremio_core::types::library::LibraryItem;
 use stremio_core::types::resource::{MetaItem, SeriesInfo, Video};
 use stremio_core::types::watched_bitfield::WatchedBitField;
 
+use url::Url;
+
 use crate::bridge::{FromProtobuf, ToProtobuf};
 use crate::protobuf::stremio::core::{models, types};
 
@@ -53,18 +59,25 @@ impl
     ToProtobuf<
         types::Video,
         (
+            &Ctx,
+            Option<&Url>,
+            // &StreamingServer,
             Option<&LibraryItem>,
             Option<&WatchedBitField>,
             Option<&String>,
+            &ResourceRequest,
         ),
     > for Video
 {
     fn to_protobuf<E: Env + 'static>(
         &self,
-        (library_item, watched, addon_name): &(
+        (ctx, streaming_server_url, library_item, watched, addon_name, request): &(
+            &Ctx,
+            Option<&Url>,
             Option<&LibraryItem>,
             Option<&WatchedBitField>,
             Option<&String>,
+            &ResourceRequest,
         ),
     ) -> types::Video {
         types::Video {
@@ -88,6 +101,23 @@ impl
                 .and_then(|library_item| library_item.state.video_id.to_owned())
                 .map(|current_video_id| current_video_id == self.id)
                 .unwrap_or_default(),
+            progress: library_item.and_then(|library_item| {
+                ctx.library
+                    .items
+                    .get(&library_item.id)
+                    .filter(|library_item| Some(self.id.to_owned()) == library_item.state.video_id)
+                    .map(|library_item| library_item.progress())
+            }),
+            // scheduled: meta_item.preview.behavior_hints.has_scheduled_videos,
+            deep_links: VideoDeepLinks::from((
+                self,
+                *request,
+                // necessary because Core doesn't use the
+                // preferred `Option<&Url>` format
+                &streaming_server_url.map(Clone::clone),
+                &ctx.profile.settings,
+            ))
+            .to_protobuf::<E>(&()),
         }
     }
 }
@@ -109,10 +139,32 @@ impl FromProtobuf<Video> for types::Video {
     }
 }
 
+impl ToProtobuf<types::VideoDeepLinks, ()> for VideoDeepLinks {
+    fn to_protobuf<E: Env + 'static>(&self, _args: &()) -> types::VideoDeepLinks {
+        types::VideoDeepLinks {
+            meta_details_videos: self.meta_details_videos.clone(),
+            meta_details_streams: self.meta_details_streams.clone(),
+            player: self.player.clone(),
+            external_player: types::video_deep_links::ExternalPlayerLink {
+                download: self
+                    .external_player
+                    .as_ref()
+                    .and_then(|ep| ep.download.clone()),
+                streaming: self
+                    .external_player
+                    .as_ref()
+                    .and_then(|ep| ep.streaming.clone()),
+            },
+        }
+    }
+}
+
 impl
     ToProtobuf<
         types::MetaItem,
         (
+            &Ctx,
+            &StreamingServer,
             Option<&LibraryItem>,
             Option<&WatchedBitField>,
             Option<&String>,
@@ -122,7 +174,9 @@ impl
 {
     fn to_protobuf<E: Env + 'static>(
         &self,
-        (library_item, watched, addon_name, meta_request): &(
+        (ctx, streaming_server, library_item, watched, addon_name, meta_request): &(
+            &Ctx,
+            &StreamingServer,
             Option<&LibraryItem>,
             Option<&WatchedBitField>,
             Option<&String>,
@@ -146,9 +200,15 @@ impl
                 .preview
                 .trailer_streams
                 .to_protobuf::<E>(&(None, None, None, None)),
-            videos: self
-                .videos
-                .to_protobuf::<E>(&(*library_item, *watched, *addon_name)),
+            videos: self.videos.to_protobuf::<E>(&(
+                *ctx,
+                streaming_server.base_url.as_ref(),
+                *library_item,
+                *watched,
+                *addon_name,
+                *meta_request,
+            )),
+
             behavior_hints: self.preview.behavior_hints.to_protobuf::<E>(&()),
             deep_links: MetaItemDeepLinks::from((self, *meta_request)).to_protobuf::<E>(&()),
             progress: library_item.and_then(|item| {
@@ -166,8 +226,11 @@ impl
     }
 }
 
-impl ToProtobuf<models::MetaDetails, Ctx> for MetaDetails {
-    fn to_protobuf<E: Env + 'static>(&self, ctx: &Ctx) -> models::MetaDetails {
+impl ToProtobuf<models::MetaDetails, (&Ctx, &StreamingServer)> for MetaDetails {
+    fn to_protobuf<E: Env + 'static>(
+        &self,
+        (ctx, streaming_server): &(&Ctx, &StreamingServer),
+    ) -> models::MetaDetails {
         let meta_item = self
             .meta_items
             .iter()
@@ -226,6 +289,7 @@ impl ToProtobuf<models::MetaDetails, Ctx> for MetaDetails {
             title,
             meta_item: meta_item.to_protobuf::<E>(&(
                 ctx,
+                streaming_server,
                 self.library_item.as_ref(),
                 self.watched.as_ref(),
             )),
diff --git a/stremio-core-protobuf/src/model/fields/player.rs b/stremio-core-protobuf/src/model/fields/player.rs
@@ -1,9 +1,16 @@
-use stremio_core::models::ctx::Ctx;
-use stremio_core::models::player::{Player, Selected, VideoParams};
-use stremio_core::types::streams::{AudioTrack, StreamItemState, SubtitleTrack};
+use stremio_core::{
+    models::{
+        ctx::Ctx,
+        player::{Player, Selected, VideoParams},
+        streaming_server::StreamingServer,
+    },
+    types::streams::{AudioTrack, StreamItemState, SubtitleTrack},
+};
 
-use crate::bridge::{FromProtobuf, ToProtobuf};
-use crate::protobuf::stremio::core::models;
+use crate::{
+    bridge::{FromProtobuf, ToProtobuf},
+    protobuf::stremio::core::models,
+};
 
 impl FromProtobuf<Selected> for models::player::Selected {
     fn from_protobuf(&self) -> Selected {
@@ -139,22 +146,35 @@ impl ToProtobuf<models::player::Selected, Ctx> for Selected {
     }
 }
 
-impl ToProtobuf<models::Player, Ctx> for Player {
-    fn to_protobuf<E: stremio_core::runtime::Env + 'static>(&self, ctx: &Ctx) -> models::Player {
+impl ToProtobuf<models::Player, (Ctx, StreamingServer)> for Player {
+    fn to_protobuf<E: stremio_core::runtime::Env + 'static>(
+        &self,
+        (ctx, streaming_server): &(Ctx, StreamingServer),
+    ) -> models::Player {
         models::Player {
             selected: self.selected.to_protobuf::<E>(ctx),
             video_params: self.video_params.to_protobuf::<E>(&()),
             meta_item: self.meta_item.as_ref().to_protobuf::<E>(&(
                 ctx,
+                streaming_server,
                 self.library_item.as_ref(),
                 self.watched.as_ref(),
             )),
             subtitles: self.subtitles.to_protobuf::<E>(ctx),
-            next_video: self.next_video.to_protobuf::<E>(&(
-                self.library_item.as_ref(),
-                self.watched.as_ref(),
-                None,
-            )),
+            next_video: self
+                .selected
+                .as_ref()
+                .and_then(|selected| selected.meta_request.as_ref())
+                .and_then(|meta_request| {
+                    self.next_video.to_protobuf::<E>(&(
+                        ctx,
+                        streaming_server.base_url.as_ref(),
+                        self.library_item.as_ref(),
+                        self.watched.as_ref(),
+                        None,
+                        meta_request,
+                    ))
+                }),
             series_info: self.series_info.to_protobuf::<E>(&()),
             library_item: self.library_item.to_protobuf::<E>(&(ctx, None)),
             stream_state: self.stream_state.to_protobuf::<E>(&()),
