1
 2
 3
 4
 5
 6
 7
 8
 9
10
11
12
13
14
15
16
17

use std::collections::HashMap;
use std::path::Path;
use crate::lang::{OutputHandler, GeneratorError};


/// An output handler that operates on an in-memory file system.
pub struct MemoryOutputHandler {
    pub files: HashMap<String, Vec<u8>>,
}

impl <'output> OutputHandler<'output> for MemoryOutputHandler {
    type FileHandle = &'output mut Vec<u8>;
    fn create_file<P: AsRef<Path>>(&'output mut self, path: P) -> Result<Self::FileHandle, GeneratorError> {
        let filename = path.as_ref().to_str().expect("Invalid filename").to_string();
        Ok(self.files.entry(filename).or_insert_with(Vec::new))
    }
}