whatbpm 1.0.2

💓 Today's Trending Values for EDM Production
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

use scraper::{Html, Selector};
use std::collections::HashSet;

const GENRE_SELECTOR: &str = ".genre-drop-list__genre";
const NAME_ATTR: &str = "data-name";
const URL_ATTR: &str = "href";
const IGNORED_GENRES: &[&str] = &["DJ Tools"];

lazy_static! {
    static ref IGNORED_GENRE_SET: HashSet<&'static str> = IGNORED_GENRES.iter().copied().collect();
}

pub struct Genre {
    pub name: String,
    pub url: String,
}

pub fn get_genres(html: &str) -> Vec<Genre> {
    let document = Html::parse_document(html);

    // I use .unwrap() because of the
    // https://github.com/causal-agent/scraper/issues/77
    let selector = Selector::parse(GENRE_SELECTOR).unwrap();

    let mut unique_names: HashSet<&str> = HashSet::new();
    let mut genres = vec![];

    for element in document.select(&selector) {
        let name = element.value().attr(NAME_ATTR);
        let url = element.value().attr(URL_ATTR);
        if let (Some(name), Some(url)) = (name, url) {
            {
                if IGNORED_GENRE_SET.contains(name) {
                    continue;
                }
                if unique_names.contains(name) {
                    continue;
                }
                unique_names.insert(name);
                genres.push(Genre {
                    name: name.to_string(),
                    url: url.to_string(),
                })
            }
        }
    }

    genres
}