ontv 0.1.3

A rich desktop application for tracking tv shows
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

use std::collections::HashSet;

/// A bucket of tokens.
pub(crate) struct Tokens {
    tokens: HashSet<String>,
}

impl Tokens {
    /// Construct a new set of tokens.
    pub(crate) fn new(string: &str) -> Self {
        let mut tokens = HashSet::new();
        tokenize(string, &mut tokens);
        Self { tokens }
    }

    /// Test if the tokenization is empty.
    pub(crate) fn is_empty(&self) -> bool {
        self.tokens.is_empty()
    }

    /// Test if tokens matches the given string.
    pub(crate) fn matches(&self, string: &str) -> bool {
        let mut search = HashSet::new();
        tokenize_search(string, &mut search);
        self.tokens.iter().all(|t| search.contains(t.as_str()))
    }
}

/// Tokenize a string for filtering.
fn tokenize(input: &str, output: &mut HashSet<String>) {
    let mut string = String::new();

    for part in input.split_whitespace() {
        string.clear();

        for c in part.chars().filter(|c| c.is_alphanumeric()) {
            string.extend(c.to_lowercase());
        }

        output.insert(string.clone());
    }
}

/// Tokenize a string for searching which includes prefixes.
fn tokenize_search(input: &str, output: &mut HashSet<String>) {
    let mut string = String::new();

    for part in input.split_whitespace() {
        string.clear();

        for c in part.chars().filter(|c| c.is_alphanumeric()) {
            string.extend(c.to_lowercase());
            output.insert(string.clone());
        }
    }
}