From abf60ddb6b7c6edd97046bfc9727c350ee2d94c3 Mon Sep 17 00:00:00 2001
From: metamuffin <metamuffin@disroot.org>
Date: Mon, 25 Dec 2023 13:53:07 +0100
Subject: import ratings

---
 import/src/lib.rs | 14 ++++++++++++--
 1 file changed, 12 insertions(+), 2 deletions(-)

(limited to 'import/src')

diff --git a/import/src/lib.rs b/import/src/lib.rs
index b37ba69..f804871 100644
--- a/import/src/lib.rs
+++ b/import/src/lib.rs
@@ -19,7 +19,7 @@ use jellybase::{
 use jellyclient::Session;
 use jellycommon::{
     AssetLocation, AssetRole, ImportOptions, ImportSource, MediaInfo, Node, NodeKind, NodePrivate,
-    NodePublic, TrackSource,
+    NodePublic, Rating, TrackSource,
 };
 use jellymatroska::read::EbmlReader;
 use jellyremuxer::import::import_metadata;
@@ -223,6 +223,9 @@ async fn process_source(
             node.public.tagline = details.tagline.clone();
             node.public.title = details.title.clone();
             node.public.description = Some(details.overview.clone());
+            node.public
+                .ratings
+                .insert(Rating::Tmdb, details.vote_average);
 
             insert_node(&id, node)?;
         }
@@ -292,7 +295,14 @@ async fn process_source(
                     node.public.title = Some(infojson.title);
                     node.public.description = Some(infojson.description);
                     node.public.tagline = Some(infojson.webpage_url);
-                    // TODO infojson
+                    node.public
+                        .ratings
+                        .insert(Rating::YoutubeViews, infojson.view_count as f64);
+                    node.public.ratings.extend(
+                        infojson
+                            .like_count
+                            .map(|l| (Rating::YoutubeLikes, l as f64)),
+                    );
                 }
             }
 
-- 
cgit v1.2.3-70-g09d2