eureka_manager_cli/commands/download/
cover.rs

1use std::{fs::File, io::BufReader, path::PathBuf, str::FromStr};
2
3use actix::Addr;
4use clap::Args;
5use eureka_mmanager::{
6    download::{
7        cover::CoverDownloadMessage, manga::MangaDownloadMessage, state::DownloadMessageState,
8    },
9    history::service::messages::is_in::IsInMessage,
10    prelude::*,
11};
12use indicatif::ProgressBar;
13use log::{info, trace};
14use mangadex_api::v5::schema::RelatedAttributes;
15use mangadex_api_types_rust::RelationshipType;
16use uuid::Uuid;
17
18use crate::commands::{AsyncRun, AsyncRunContext};
19
20#[derive(Debug, Args)]
21pub struct CoverDownloadArgs {
22    /// Manga ids
23    #[arg(long = "id")]
24    pub ids: Vec<Uuid>,
25    #[arg(long)]
26    pub id_text_file: Vec<PathBuf>,
27}
28
29impl CoverDownloadArgs {
30    pub fn get_ids(&self) -> Vec<Uuid> {
31        let mut ids = self.ids.clone();
32        self.id_text_file
33            .iter()
34            .map(|e| (e, File::open(e)))
35            .flat_map(|(path, res)| match res {
36                Ok(file) => Some(id_list_txt_reader::IdListTxtReader::new(BufReader::new(
37                    file,
38                ))),
39                Err(err) => {
40                    log::error!("Cannot open the {} file: {}", path.to_string_lossy(), err);
41                    None
42                }
43            })
44            .flat_map(|file| file.flat_map(|s| Uuid::from_str(&s)))
45            .for_each(|id| {
46                ids.push(id);
47            });
48        ids.dedup();
49        ids
50    }
51}
52
53impl AsyncRun for CoverDownloadArgs {
54    async fn run(&self, ctx: AsyncRunContext) -> anyhow::Result<()> {
55        let ids = self.get_ids();
56        let mut progress = ProgressBar::new(ids.len() as u64);
57        progress = ctx.progress.add(progress);
58        trace!(
59            "Downloading {} covers with their titles if missing",
60            ids.len()
61        );
62        for id in ids {
63            let manager = ctx.manager.clone();
64            let task = async move {
65                trace!("Downloading Cover {id}");
66                let dirs =
67                    <Addr<DownloadManager> as GetManagerStateData>::get_dir_options(&manager)
68                        .await?;
69                let manga = {
70                    let manga_manager =
71                        <Addr<DownloadManager> as GetManager<CoverDownloadManager>>::get(&manager)
72                            .await?;
73                    let mut task = manga_manager
74                        .send(
75                            CoverDownloadMessage::new(id).state(DownloadMessageState::Downloading),
76                        )
77                        .await?;
78                    let data = task.wait().await?.await?;
79                    info!(
80                        "downloaded cover {} = {:?}",
81                        data.id, data.attributes.file_name
82                    );
83                    data.find_first_relationships(RelationshipType::Manga)
84                        .ok_or(anyhow::Error::msg(format!(
85                            "Cannot find the title for cover art {}",
86                            id,
87                        )))?
88                        .clone()
89                };
90                if !dirs
91                    .send(IsInMessage(HistoryEntry::new(
92                        manga.id,
93                        RelationshipType::Manga,
94                    )))
95                    .await?
96                {
97                    trace!("Downloading title {}", manga.id);
98                    let manga_manager =
99                        <Addr<DownloadManager> as GetManager<MangaDownloadManager>>::get(&manager)
100                            .await?;
101                    let mut task = manga_manager
102                        .send(
103                            MangaDownloadMessage::new(manga.id)
104                                .state(DownloadMessageState::Downloading),
105                        )
106                        .await?;
107                    task.wait().await?.await?;
108                    info!(
109                        "downloaded title {} = {:?}",
110                        manga.id,
111                        manga.attributes.and_then(|attr| {
112                            let RelatedAttributes::Manga(manga) = attr else {
113                                return None;
114                            };
115                            manga.title.values().next().cloned()
116                        })
117                    );
118                }
119                Ok::<_, anyhow::Error>(())
120            };
121            if let Err(err) = task.await {
122                log::error!("{}", err);
123            }
124            progress.inc(1);
125        }
126        progress.finish();
127        ctx.progress.remove(&progress);
128        Ok(())
129    }
130}
eureka_manager_cli/commands/download/cover.rs

eureka_manager_cli/commands/download/
cover.rs
