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
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115

use owo_colors::OwoColorize;
use std::{fmt, result::Result as StdResult};
use thiserror::Error;

pub mod nibl;
pub use nibl::Nibl;

/////////////////////////////////////////////////
//                    Error                    //
/////////////////////////////////////////////////
#[derive(Debug, Error)]
pub enum Error {
    #[error("Request failed: {0}")]
    Reqwest(#[from] reqwest::Error),
    #[error("The {api} API returned and error: {message}")]
    APIError { api: &'static str, message: String },
}

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

//////////////////////////////////////////////////////
//                    FindResult                    //
//////////////////////////////////////////////////////
pub struct FindResult {
    pub irc_config: crate::downloader::irc::Config,
    pub entries: Vec<Entry>,
}

//////////////////////////////////////////////////
//                    Finder                    //
//////////////////////////////////////////////////
pub trait Finder {
    fn find(&self, query: &Query) -> Result<FindResult>;
}

/////////////////////////////////////////////////////////
//                    EpisodeNumber                    //
/////////////////////////////////////////////////////////
#[derive(Debug, Clone, Copy, Default, PartialEq, Hash)]
pub enum EpisodeNumber {
    #[default]
    All,
    Latest,
    Number(i32),
}

impl std::str::FromStr for EpisodeNumber {
    type Err = String;

    fn from_str(s: &str) -> StdResult<Self, Self::Err> {
        match s {
            "latest" => Ok(Self::Latest),
            "all" => Ok(Self::All),
            _ => s
                .parse::<i32>()
                .map(Self::Number)
                .map_err(|_| format!("Invalid episode number {}", s)),
        }
    }
}

impl fmt::Display for EpisodeNumber {
    fn fmt(&self, f: &mut fmt::Formatter<'_>) -> StdResult<(), fmt::Error> {
        match self {
            Self::Number(n) => write!(f, "{}", n),
            Self::Latest => write!(f, "latest"),
            Self::All => write!(f, "all"),
        }
    }
}

/////////////////////////////////////////////////
//                    Query                    //
/////////////////////////////////////////////////
#[derive(Debug, Default, Clone, PartialEq, Hash)]
pub struct Query {
    pub search: String,
    pub resolution: Option<String>,
    pub episode: EpisodeNumber,
}

impl Query {
    pub fn new(search: String, resolution: Option<String>, episode: EpisodeNumber) -> Self {
        Self {
            search,
            resolution,
            episode,
        }
    }

    pub fn find<F: Finder>(&self, finder: &F) -> Result<FindResult> {
        finder.find(self)
    }
}

#[derive(Debug, Clone, PartialEq, Hash)]
pub struct Entry {
    pub package_number: i32,
    pub bot_id: i64,
    pub bot_name: String,
    pub name: String,
    pub size: String,
}

impl fmt::Display for Entry {
    fn fmt(&self, f: &mut fmt::Formatter<'_>) -> StdResult<(), fmt::Error> {
        write!(
            f,
            "{} [{}] ({})",
            self.name,
            self.bot_name.yellow(),
            self.size,
        )
    }
}