1 files changed, 12 insertions, 8 deletions

diff --git a/src/archive.rs b/src/archive.rs
index e635007..4a70867 100644
--- a/src/archive.rs
+++ b/src/archive.rs
@@ -1,5 +1,5 @@
 use crate::driver::{ArchiveRead, ArchiveWrite, Driver};
-use crate::ArchiveResult;
+use crate::{ArchiveFile, ArchiveResult};
 use std::fs::File;
 use std::io::{Read, Write};
 use std::path::Path;
@@ -10,9 +10,9 @@ pub struct Archive<D: Driver> {
 
 impl<D: ArchiveRead> Archive<D>
 where
-    D::IO: std::io::Read,
+    D::Io: std::io::Read,
 {
-    pub fn read(io: D::IO) -> ArchiveResult<Self, D::Error> {
+    pub fn read(io: D::Io) -> ArchiveResult<Self, D::Error> {
         Ok(Self {
             driver: D::read(io)?,
         })
@@ -20,18 +20,22 @@ where
 
     pub fn read_from_file(path: impl AsRef<Path>) -> ArchiveResult<Self, D::Error>
     where
-        D: ArchiveRead<IO = File>,
+        D: ArchiveRead<Io = File>,
     {
         Self::read(File::open(path)?)
     }
 
-    pub fn files(&self) -> Vec<&D::File> {
+    pub fn files(&self) -> Vec<&D::FileInfo> {
         self.driver.files()
     }
 
-    pub fn get_file(&self, name: &str) -> Option<&D::File> {
-        self.driver.get_file(name)
+    pub fn get_file_info(&self, name: &str) -> Option<&D::FileInfo> {
+        self.driver.get_file_info(name)
+    }
+
+    pub fn get_file_reader<'d>(&'d mut self, name: &str) -> Option<ArchiveFile<D::FileDriver<'d>>> {
+        Some(ArchiveFile::new(self.driver.get_file_reader(name)?))
     }
 }
 
-impl<D: ArchiveWrite> Archive<D> where D::IO: Read + Write {}
+impl<D: ArchiveWrite> Archive<D> where D::Io: Read + Write {}