mahou 0.1.0

Magically easy anime downloads
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
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84

use argh::FromArgs;
use mahou::{
    downloader,
    finder::{self, EpisodeNumber},
};
use std::error::Error;

type Result<T> = std::result::Result<T, Box<dyn Error>>;

/// Mahou -- magically easy anime downloader
///
/// Try not using any options for an interactive prompt :)
#[derive(Debug, FromArgs)]
struct Args {
    /// the query to search for
    #[argh(option, short = 's')]
    search: Option<String>,

    /// the episode to download
    #[argh(option, short = 'e')]
    episode: Option<EpisodeNumber>,

    /// download directory. Defaults to current directory
    #[argh(option, short = 'd', default = "\"./\".to_string()")]
    directory: String,

    /// preferred resolution.
    #[argh(option, short = 'r')]
    res: Option<String>,
}

fn prompt_search() -> Result<String> {
    Ok(inquire::Text::new("What show would you like to watch today?").prompt()?)
}

fn prompt_episode() -> Result<EpisodeNumber> {
    Ok(inquire::CustomType::<EpisodeNumber>::new("Which episode?")
        .with_default(EpisodeNumber::Latest)
        .with_help_message("Enter a number, 'latest', or 'all' to show all available episodes")
        .prompt()?)
}

fn main() -> Result<()> {
    let args: Args = argh::from_env();

    let search = match args.search {
        Some(search) => search,
        None => prompt_search()?,
    };

    let episode = match args.episode {
        Some(episode) => episode,
        None => prompt_episode()?,
    };

    let results = finder::Query::new(search, args.res, episode).find(&finder::Nibl::default())?;
    let finder::FindResult {
        irc_config,
        entries,
    } = results;

    if entries.is_empty() {
        println!("No results found :(");
        return Ok(());
    }

    // - Current user input, filter value
    // - Current option being evaluated, with type preserved
    // - String value of the current option
    // - Index of the current option in the original list
    let filter = &|input: &str, _: &finder::Entry, entry: &str, _: usize| {
        let entry = entry.to_lowercase();
        let input = input.to_lowercase();
        input.split_whitespace().all(|word| entry.contains(word))
    };

    let selected = inquire::Select::new("Pick an episode", entries)
        .with_filter(filter)
        .prompt()?;

    downloader::download(&selected, irc_config, args.directory)?;

    Ok(())
}