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

use std::fs::File;
use std::io::{Cursor, Read, Seek, SeekFrom};
use std::path::PathBuf;

#[derive(Debug)]
pub enum CodeSource {
    Memory {
        name: String,
        contents: Cursor<String>,
    },
    File {
        name: PathBuf,
        file: File,
    },
}

impl CodeSource {
    pub fn memory<S: Into<String>>(name: S, contents: String) -> Self {
        Self::Memory {
            name: name.into(),
            contents: Cursor::new(contents),
        }
    }

    pub fn file<S: Into<PathBuf>>(name: S, contents: File) -> Self {
        Self::File {
            name: name.into(),
            file: contents,
        }
    }

    #[must_use]
    pub fn name(&self) -> &str {
        match self {
            CodeSource::Memory { name, .. } => name,
            CodeSource::File { name, .. } => name.to_str().unwrap_or("<unknown location>"),
        }
    }
}

impl Read for CodeSource {
    fn read(&mut self, buf: &mut [u8]) -> std::io::Result<usize> {
        match self {
            CodeSource::Memory { contents, .. } => contents.read(buf),
            CodeSource::File { file, .. } => file.read(buf),
        }
    }
}

impl Seek for CodeSource {
    fn seek(&mut self, pos: SeekFrom) -> std::io::Result<u64> {
        match self {
            CodeSource::Memory { contents, .. } => contents.seek(pos),
            CodeSource::File { file, .. } => file.seek(pos),
        }
    }
}