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
130
131
132
133
134
135

use std::{fmt, io::Read, path::Path, sync::Arc};

use bstr::ByteSlice;
use indexmap::IndexMap;
use logix_vfs::LogixVfs;

use crate::{error::ParseError, parser::LogixParser, token::Token, type_trait::LogixType};

#[derive(Clone, PartialEq, Eq)]
pub(crate) struct CachedFile {
    path: Arc<Path>,
    data: Arc<[u8]>,
}

impl CachedFile {
    pub(crate) fn empty() -> CachedFile {
        Self {
            path: Arc::from(Path::new("")),
            data: Arc::from(b"".as_slice()),
        }
    }

    #[cfg(test)]
    pub(crate) fn from_slice(path: impl AsRef<Path>, data: &[u8]) -> CachedFile {
        Self {
            path: Arc::from(Path::new(path.as_ref())),
            data: Arc::from(data),
        }
    }

    pub fn path(&self) -> &Path {
        &self.path
    }

    pub fn lines(&self) -> bstr::Lines {
        self.data.lines()
    }

    pub fn data(&self) -> &[u8] {
        &self.data
    }
}

impl fmt::Debug for CachedFile {
    fn fmt(&self, f: &mut fmt::Formatter) -> fmt::Result {
        f.debug_tuple("CachedFile").field(&self.path).finish()
    }
}

#[derive(Debug)]
pub struct LogixLoader<FS: LogixVfs> {
    fs: FS,
    files: IndexMap<Arc<Path>, Arc<[u8]>>,
    tmp: Vec<u8>,
}

impl<FS: LogixVfs> LogixLoader<FS> {
    pub fn new(fs: FS) -> Self {
        Self {
            fs,
            files: IndexMap::new(),
            tmp: Vec::with_capacity(0x10000),
        }
    }

    pub(crate) fn get_file(&self, path: impl AsRef<Path>) -> Option<CachedFile> {
        let (key, value) = self.files.get_key_value(path.as_ref())?;

        Some(CachedFile {
            path: key.clone(),
            data: value.clone(),
        })
    }

    pub(crate) fn open_file(&mut self, path: impl AsRef<Path>) -> Result<CachedFile, ParseError> {
        match self
            .files
            .entry(Arc::<Path>::from(self.fs.canonicalize_path(path.as_ref())?))
        {
            indexmap::map::Entry::Vacant(entry) => {
                let path = entry.key().clone();
                self.tmp.clear();
                let mut r = self.fs.open_file(entry.key())?;
                r.read_to_end(&mut self.tmp)
                    .map_err(logix_vfs::Error::from)?;
                let data = entry.insert(Arc::from(self.tmp.as_slice())).clone();
                Ok(CachedFile { path, data })
            }
            indexmap::map::Entry::Occupied(entry) => Ok(CachedFile {
                path: entry.key().clone(),
                data: entry.get().clone(),
            }),
        }
    }

    pub fn load_file<T: LogixType>(&mut self, path: impl AsRef<Path>) -> Result<T, ParseError> {
        let file = self.open_file(path)?;
        let mut p = LogixParser::new(self, &file);

        let ret = T::logix_parse(&mut p)?;

        // This will either skip any newlines and comments, or return EOF
        p.req_newline(T::descriptor().name)?;

        // From now on EOF should always be returned
        p.req_token(T::descriptor().name, Token::Newline(true))?;

        Ok(ret.value)
    }
}

#[cfg(test)]
mod tests {
    use logix_vfs::RelFs;

    use super::*;

    #[test]
    fn basics() {
        let tmp = tempfile::tempdir().unwrap();
        std::fs::write(tmp.path().join("test.logix"), b"10").unwrap();

        format!(
            "{:?}",
            CachedFile {
                path: Arc::from(Path::new("a")),
                data: Arc::from(b"".as_slice()),
            }
        );

        let mut loader = LogixLoader::new(RelFs::new(tmp.path()));
        loader.load_file::<u32>("test.logix").unwrap();
        loader.load_file::<u32>("test.logix").unwrap(); // Twice to test cache
    }
}