nyaa 0.6.0

A tui tool for browsing and downloading torrents from nyaa.si
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::error::Error;

use serde::{Deserialize, Serialize};

use crate::{
    app::{App, LoadType, Widgets},
    widget::EnumIter,
};

use self::{
    nyaa_html::{Item, NyaaHtmlSource},
    nyaa_rss::NyaaRssSource,
};

pub mod nyaa_html;
pub mod nyaa_rss;

#[derive(Clone, Copy, Serialize, Deserialize)]
pub enum Sources {
    NyaaHtml,
    NyaaRss,
}

impl EnumIter<Sources> for Sources {
    fn iter() -> std::slice::Iter<'static, Sources> {
        static SORTS: &[Sources] = &[Sources::NyaaHtml, Sources::NyaaRss];
        SORTS.iter()
    }
}

impl ToString for Sources {
    fn to_string(&self) -> String {
        match self {
            Sources::NyaaHtml => "Nyaa HTML".to_owned(),
            Sources::NyaaRss => "Nyaa RSS".to_owned(),
        }
    }
}

pub trait Source {
    async fn search(app: &mut App, w: &Widgets) -> Result<Vec<Item>, Box<dyn Error>>;
    async fn sort(app: &mut App, w: &Widgets) -> Result<Vec<Item>, Box<dyn Error>>;
    async fn filter(app: &mut App, w: &Widgets) -> Result<Vec<Item>, Box<dyn Error>>;
    async fn categorize(app: &mut App, w: &Widgets) -> Result<Vec<Item>, Box<dyn Error>>;
}

pub async fn load(
    src: Sources,
    load_type: LoadType,
    app: &mut App,
    w: &Widgets,
) -> Result<Vec<Item>, Box<dyn Error>> {
    match src {
        Sources::NyaaHtml => match load_type {
            LoadType::Searching => NyaaHtmlSource::search(app, w).await,
            LoadType::Sorting => NyaaHtmlSource::sort(app, w).await,
            LoadType::Filtering => NyaaHtmlSource::filter(app, w).await,
            LoadType::Categorizing => NyaaHtmlSource::categorize(app, w).await,
        },
        Sources::NyaaRss => match load_type {
            LoadType::Searching => NyaaRssSource::search(app, w).await,
            LoadType::Sorting => NyaaRssSource::sort(app, w).await,
            LoadType::Filtering => NyaaRssSource::filter(app, w).await,
            LoadType::Categorizing => NyaaRssSource::categorize(app, w).await,
        },
    }
}