aboutsummaryrefslogtreecommitdiffstats
path: root/src/drive.rs
diff options
context:
space:
mode:
authorBryan Newbold <bnewbold@robocracy.org>2017-11-06 10:19:54 -0800
committerBryan Newbold <bnewbold@robocracy.org>2017-11-06 10:19:54 -0800
commit7701c90fd67fe9598ba312cf0c0403c1d653bd75 (patch)
treeebab3147a84ec61dd7607fbd03af6b915aea537b /src/drive.rs
parent4a07ef7d2789b0a58a4b892a9098e1e5cdcbb807 (diff)
downloadgeniza-7701c90fd67fe9598ba312cf0c0403c1d653bd75.tar.gz
geniza-7701c90fd67fe9598ba312cf0c0403c1d653bd75.zip
work in progress on drive import/export
Diffstat (limited to 'src/drive.rs')
-rw-r--r--src/drive.rs34
1 files changed, 28 insertions, 6 deletions
diff --git a/src/drive.rs b/src/drive.rs
index 9aa6ea2..740e949 100644
--- a/src/drive.rs
+++ b/src/drive.rs
@@ -1,8 +1,8 @@
-use std::io::Read;
+use std::io::{Read, Write};
use std::path::{Path, PathBuf};
-use std::os::unix::fs::MetadataExt;
-use std::fs::File;
+use std::os::unix::fs::{MetadataExt, OpenOptionsExt};
+use std::fs::{File, OpenOptions};
use std::cmp::min;
use std::ffi::OsStr;
use protobuf::Message;
@@ -403,6 +403,7 @@ impl<'a> DatDrive {
Ok(children)
}
+ // XXX: needs test
/// Copies Stat metadata and all content from a file in the "real" filesystem into the
/// DatDrive.
pub fn import_file<P: AsRef<Path>, Q: AsRef<Path>>(&mut self, source: P, dest: Q) -> Result<()> {
@@ -418,15 +419,36 @@ impl<'a> DatDrive {
self.add_file(dest, &mut stat, in_file)
}
+ // XXX: needs test
/// Copies a file from the drive to the "real" filesystem, preserving Stat metadata.
- pub fn export_file<P: AsRef<Path>, Q: AsRef<Path>>(&mut self, _source: P, _dest: Q) -> Result<()> {
- unimplemented!()
+ pub fn export_file<P: AsRef<Path>, Q: AsRef<Path>>(&mut self, source: P, dest: Q) -> Result<()> {
+
+ let source = source.as_ref();
+ let de = self.get_file_entry(source)?;
+ if let Some(entry) = de {
+ let stat = entry.stat.unwrap();
+ let mut out_file = OpenOptions::new()
+ .create_new(true)
+ .write(true)
+ .mode(stat.get_mode())
+ .open(dest)?;
+ let offset = stat.get_offset();
+ let blocks = stat.get_blocks();
+ for i in offset..(offset+blocks) {
+ let chunk = self.content.get_data_entry(i)?;
+ out_file.write_all(&chunk)?;
+ }
+ // TODO: more outfile metadata (uid, guid, etc)
+ } else {
+ bail!("Couldn't find path: {}", source.display());
+ }
+
+ Ok(())
}
pub fn read_file_bytes<P: AsRef<Path>>(&mut self, path: P) -> Result<Vec<u8>> {
let de = self.get_file_entry(path.as_ref())?;
if let Some(entry) = de {
- // XXX: read and concatonate chunks
let stat = entry.stat.unwrap();
let mut buf = vec![];
let offset = stat.get_offset();