ontv 0.1.0

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
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
116
117
118
119
120
121
122
123
124
125
126
127
128
129

use std::fmt;
use std::fmt::Write;

use serde::de;
use serde::ser;

#[derive(Clone, Copy, PartialEq, Eq, Hash, PartialOrd, Ord)]
pub(crate) struct Raw<const N: usize>([u8; N]);

impl<const N: usize> Raw<N> {
    /// Construct a raw identifier from a string.
    pub(crate) fn new<B>(bytes: &B) -> Option<Self>
    where
        B: ?Sized + AsRef<[u8]>,
    {
        let mut out = [0; N];
        let mut it = out.iter_mut();

        for b in bytes.as_ref() {
            *it.next()? = *b;
        }

        Some(Self(out))
    }
}

impl<const N: usize> fmt::Display for Raw<N> {
    #[inline]
    fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
        let bytes = trim_end(&self.0);

        for b in bytes {
            if b.is_ascii_graphic() {
                (*b as char).fmt(f)?;
            }
        }

        return Ok(());

        #[inline]
        fn trim_end(mut bytes: &[u8]) -> &[u8] {
            while let [head @ .., b] = bytes {
                if *b != 0 {
                    break;
                }

                bytes = head;
            }

            bytes
        }
    }
}

impl<const N: usize> fmt::Debug for Raw<N> {
    #[inline]
    fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
        f.write_char('"')?;
        fmt::Display::fmt(self, f)?;
        f.write_char('"')?;
        Ok(())
    }
}

impl<'de, const N: usize> de::Deserialize<'de> for Raw<N> {
    #[inline]
    fn deserialize<D>(deserializer: D) -> Result<Self, D::Error>
    where
        D: serde::Deserializer<'de>,
    {
        deserializer.deserialize_str(Visitor)
    }
}

impl<const N: usize> ser::Serialize for Raw<N> {
    #[inline]
    fn serialize<S>(&self, serializer: S) -> Result<S::Ok, S::Error>
    where
        S: serde::Serializer,
    {
        serializer.collect_str(self)
    }
}

struct Visitor<const N: usize>;

impl<'de, const N: usize> de::Visitor<'de> for Visitor<N> {
    type Value = Raw<N>;

    #[inline]
    fn expecting(&self, f: &mut fmt::Formatter) -> fmt::Result {
        write!(f, "an raw string")
    }

    #[inline]
    fn visit_str<E>(self, v: &str) -> Result<Self::Value, E>
    where
        E: de::Error,
    {
        match Raw::new(v) {
            Some(value) => Ok(value),
            None => Err(E::custom("value overflow")),
        }
    }

    #[inline]
    fn visit_bytes<E>(self, v: &[u8]) -> Result<Self::Value, E>
    where
        E: de::Error,
    {
        match Raw::new(v) {
            Some(value) => Ok(value),
            None => Err(E::custom("value overflow")),
        }
    }
}

#[cfg(test)]
mod tests {
    use super::Raw;

    #[test]
    fn test_raw16() {
        let id = Raw::<16>::new(b"foobarbaz").unwrap();
        assert_eq!(id.to_string(), "foobarbaz");
        let id = Raw::<16>::new("foobarbaz").unwrap();
        assert_eq!(id.to_string(), "foobarbaz");
    }
}