use std::{path::Path, str}; use anyhow::Context; use maplit::hashmap; use memofs::{IoResultExt, Vfs}; use crate::snapshot::{InstanceContext, InstanceMetadata, InstanceSnapshot}; use super::{meta_file::AdjacentMetadata, util::PathExt}; pub fn snapshot_txt( context: &InstanceContext, vfs: &Vfs, path: &Path, ) -> anyhow::Result> { let name = path.file_name_trim_end(".txt")?; let contents = vfs.read(path)?; let contents_str = str::from_utf8(&contents) .with_context(|| format!("File was not valid UTF-8: {}", path.display()))? .to_owned(); let properties = hashmap! { "Value".to_owned() => contents_str.into(), }; let meta_path = path.with_file_name(format!("{}.meta.json", name)); let mut snapshot = InstanceSnapshot::new() .name(name) .class_name("StringValue") .properties(properties) .metadata( InstanceMetadata::new() .instigating_source(path) .relevant_paths(vec![path.to_path_buf(), meta_path.clone()]) .context(context), ); if let Some(meta_contents) = vfs.read(&meta_path).with_not_found()? { let mut metadata = AdjacentMetadata::from_slice(&meta_contents, meta_path)?; metadata.apply_all(&mut snapshot)?; } Ok(Some(snapshot)) } #[cfg(test)] mod test { use super::*; use memofs::{InMemoryFs, VfsSnapshot}; #[test] fn instance_from_vfs() { let mut imfs = InMemoryFs::new(); imfs.load_snapshot("/foo.txt", VfsSnapshot::file("Hello there!")) .unwrap(); let mut vfs = Vfs::new(imfs.clone()); let instance_snapshot = snapshot_txt(&InstanceContext::default(), &mut vfs, Path::new("/foo.txt")) .unwrap() .unwrap(); insta::assert_yaml_snapshot!(instance_snapshot); } }