tv-renamer 0.3.3

TV renaming application with TVDB titles and templating support
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

use std::iter;

/// A trait that adds the ability for numbers to find their digit count and to convert them to padded strings.
pub trait Digits {
    /// Counts the number of digits in a number. **Example:** {{0 = 0}, {1 = 1}, {10 = 2}, {100 = 3}}
    fn digits(&self) -> Self;

    /// Converts a number into a padded String, using `pad` as the character to pad with and `limit` as the size.
    fn to_padded_string(&self, pad: char, limit: usize) -> String;
}

impl Digits for u16 {
    fn digits(&self) -> u16 {
        let mut digits = if *self == 1 || *self % 10 == 0 { 1 } else { 0 };
        let mut temp = 1;
        while temp < *self {
            digits += 1;
            temp = (temp << 3) + (temp << 1);
        }
        digits
    }

    fn to_padded_string(&self, pad: char, limit: usize) -> String {
        if self.digits() < limit as u16 {
            let mut output = String::with_capacity(limit);
            output.push_str(&iter::repeat(pad).take(limit - self.digits() as usize).collect::<String>());
            output.push_str(&self.to_string());
            output
        } else {
            self.to_string()
        }
    }
}

impl Digits for u32 {
    fn digits(&self) -> u32 {
        let mut digits = if *self == 1 || *self % 10 == 0 { 1 } else { 0 };
        let mut temp = 1;
        while temp < *self {
            digits += 1;
            temp = (temp << 3) + (temp << 1);
        }
        digits
    }

    fn to_padded_string(&self, pad: char, limit: usize) -> String {
        if self.digits() < limit as u32 {
            let mut output = String::with_capacity(limit);
            output.push_str(&iter::repeat(pad).take(limit - self.digits() as usize).collect::<String>());
            output.push_str(&self.to_string());
            output
        } else {
            self.to_string()
        }
    }
}

#[test]
fn test_digits() {
    assert_eq!(1u16.digits(), 1);
    assert_eq!(9u16.digits(), 1);
    assert_eq!(10u16.digits(), 2);
    assert_eq!(100u16.digits(), 3);
}
#[test]
fn test_padded_digits() {
    assert_eq!(5u16.to_padded_string('0', 2).as_str(), "05");
    assert_eq!(10u16.to_padded_string('0', 2).as_str(), "10");
    assert_eq!(100u16.to_padded_string('0', 2).as_str(), "100");
    assert_eq!(10u16.to_padded_string('0', 3).as_str(), "010");
    assert_eq!(5u16.to_padded_string('0', 3).as_str(), "005");
}