extract cover

1 files changed, 35 insertions, 9 deletions

diff --git a/import/src/main.rs b/import/src/main.rs
index 806ebfb..bf90eb2 100644
--- a/import/src/main.rs
+++ b/import/src/main.rs
@@ -10,10 +10,10 @@ use clap::{Parser, Subcommand};
 use jellycommon::{AssetLocation, MediaInfo, MediaSource, Node, NodeKind, NodePrivate, NodePublic};
 use jellymatroska::read::EbmlReader;
 use jellyremuxer::import::{import_metadata, seek_index::import_seek_index};
-use log::info;
+use log::{info, warn};
 use std::{
     fs::{remove_file, File},
-    io::stdin,
+    io::{stdin, Write},
     path::PathBuf,
     process::exit,
 };
@@ -154,21 +154,47 @@ fn main() -> anyhow::Result<()> {
             if !args.dry {
                 std::fs::create_dir_all(&path)?;
 
-                poster = tmdb_details
+                poster = file_meta
                     .as_ref()
-                    .map(|d| {
-                        d.poster_path
+                    .map(|m| {
+                        m.cover
                             .as_ref()
-                            .map(|p| {
-                                let pu = path.join("poster.jpeg");
+                            .map(|(mime, data)| {
+                                let pu = path.join(format!(
+                                    "cover.{}",
+                                    match mime.as_str() {
+                                        "image/webp" => "webp",
+                                        "image/jpeg" => "jpeg",
+                                        "image/png" => "png",
+                                        _ => {
+                                            warn!("unknown mime, just using webp");
+                                            "webp"
+                                        }
+                                    }
+                                ));
                                 let mut f = File::create(&pu)?;
-                                tmdb_image(&p, &mut f)?;
+                                f.write_all(&data)?;
                                 Ok::<_, anyhow::Error>(pu)
                             })
                             .transpose()
                     })
                     .transpose()?
-                    .flatten();
+                    .flatten()
+                    .or(tmdb_details
+                        .as_ref()
+                        .map(|d| {
+                            d.poster_path
+                                .as_ref()
+                                .map(|p| {
+                                    let pu = path.join("poster.jpeg");
+                                    let mut f = File::create(&pu)?;
+                                    tmdb_image(&p, &mut f)?;
+                                    Ok::<_, anyhow::Error>(pu)
+                                })
+                                .transpose()
+                        })
+                        .transpose()?
+                        .flatten());
 
                 backdrop = tmdb_details
                     .as_ref()