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

use std::fmt;
use std::path::PathBuf;

#[derive(Eq, Debug, Clone, Hash)]
pub enum URI {
    #[cfg(test)]
    Test,

    Exact(String),
    File(PathBuf),
    Stdlib(PathBuf),
    Stdin,
    REPLLine(usize),

    Main,
}

impl URI {
    pub fn neighboring_file(&self, name: &str) -> Option<URI> {
        match self {
            URI::Exact(s) => {
                let mut segments: Vec<_> = s.split("/").collect();
                segments.pop();
                segments.push(name.as_ref());
                Some(URI::Exact(segments.join("/")))
            }
            URI::File(path) | URI::Stdlib(path) => {
                let mut path = path.clone();
                path.pop();
                Some(URI::File(path))
            }
            _ => None,
        }
    }

    pub fn basename(&self) -> Option<String> {
        match self {
            URI::Exact(s) => {
                let mut segments: Vec<_> = s.split("/").collect();
                segments.pop().map(|s| s.into())
            }
            URI::File(path) | URI::Stdlib(path) => {
                path.file_name().map(|os| os.to_string_lossy().to_string())
            }
            _ => None,
        }
    }

    pub fn matches_basename(&self, basename: &str) -> bool {
        self.basename() == Some(basename.into())
    }

    pub fn is_stdlib(&self) -> bool {
        match self {
            URI::Stdlib(_) => true,
            _ => false,
        }
    }
}

impl PartialEq for URI {
    fn eq(&self, other: &Self) -> bool {
        format!("{}", self) == format!("{}", other)
    }
}

impl fmt::Display for URI {
    fn fmt(&self, f: &mut fmt::Formatter) -> fmt::Result {
        match self {
            #[cfg(test)]
            URI::Test => write!(f, "test:"),

            URI::Exact(s) => write!(f, "{}", s),
            URI::File(path) | URI::Stdlib(path) => write!(f, "file://{}", path.display()),
            URI::Stdin => write!(f, "stdin:"),
            URI::REPLLine(n) => write!(f, "repl:{}", n),

            URI::Main => write!(f, "main"),
        }
    }
}