dynasty 1.4.1

Dynasty Reader's CLI downloader
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67

use std::{sync::Arc, time::Duration};

use anyhow::Result;
use tokio::sync::Semaphore;
use tryhard::{backoff_strategies::FixedBackoff, NoOnRetry, RetryFutureConfig};

use crate::{downloader, parser::ParsedArgumentValue, resolver, Cli};

pub type DefaultRetryConfig = RetryFutureConfig<FixedBackoff, NoOnRetry>;

#[non_exhaustive]
#[derive(Debug, Clone)]
pub struct RunnerConfig {
    pub semaphore: Arc<Semaphore>,
    pub retry_config: DefaultRetryConfig,
    #[cfg(feature = "archiver")]
    pub archiver: Option<crate::archiver::Archiver>,
}

impl From<&Cli> for RunnerConfig {
    fn from(cli: &Cli) -> Self {
        let semaphore = Arc::new(Semaphore::new(cli.max_concurrent_request as usize));
        let retry_config =
            RetryFutureConfig::new(cli.max_retry as u32).fixed_backoff(Duration::from_secs(5));

        #[cfg(feature = "archiver")]
        let archiver = cli.archive.map(|archive_kind| crate::archiver::Archiver {
            keep_archived: cli.keep_archived,
            kind: archive_kind,
        });

        RunnerConfig {
            retry_config,
            semaphore,
            #[cfg(feature = "archiver")]
            archiver,
        }
    }
}

pub async fn execute(cli: &Cli) -> Result<()> {
    let config = &cli.into();

    #[cfg(feature = "search")]
    if cli.targets.is_empty() {
        let crate::Commands::Search(value) = &cli.commands;

        run(
            ParsedArgumentValue::Search(value.clone().try_into()?),
            cli,
            config,
        )
        .await?;
    }

    for target in &cli.targets {
        run(target.clone(), cli, config).await?;
    }

    Ok(())
}

async fn run(target: ParsedArgumentValue, cli: &Cli, config: &RunnerConfig) -> Result<()> {
    let (downloadable, client) = resolver::resolve(target, cli, config).await?;

    downloader::download(downloadable, client, cli, config).await
}