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

use super::*;

use std::str::FromStr;
use std::fmt;

#[derive(Debug, PartialEq, Eq, Clone, Copy)]
pub enum Vendor {
    Cgats,
    ColorBurst,
    Curve,
}

impl FromStr for Vendor {
    type Err = Error;

    fn from_str(s: &str) -> Result<Vendor> {
        use Vendor::*;
        let types: Vec<Vendor> = vec![Cgats, ColorBurst, Curve];

        for t in types.into_iter() {
            let tstring = &t.to_string().to_lowercase();
            if s.to_lowercase().contains(tstring) {
                return Ok(t);
            }
        }

       Err(Error::UnknownVendor)
    }
}

impl fmt::Display for Vendor {
    fn fmt(&self, f: &mut fmt::Formatter) -> fmt::Result {
        write!(f, "{:?}", &self)
    }
}

#[test]
fn from_str() {
    assert_eq!(Vendor::from_str("ColorBurst"), Ok(Vendor::ColorBurst));
    assert_eq!(Vendor::from_str("CGATS.17"), Ok(Vendor::Cgats));
    assert_eq!(Vendor::from_str("File Created by Curve3"), Ok(Vendor::Curve));
    assert_eq!(Vendor::from_str("derp"), Err(Error::UnknownVendor));
    assert_eq!(Vendor::from_str(""), Err(Error::UnknownVendor));
}