Sebastian Thielmake fmt251b6df4 years agoHistory
107 lines · 3.6 KB · Rust
RawDownload
use std::path::{Path, PathBuf}; use crate::fs::Stack; impl Stack {    pub fn root(&self) -> &Path {        &self.root    }     pub fn current(&self) -> &Path {        &self.current    }     pub fn current_relative(&self) -> &Path {        &self.current_relative    }} pub trait Delegate {    fn push_directory(&mut self, stack: &Stack) -> std::io::Result<()>;    fn push(&mut self, is_last_component: bool, stack: &Stack) -> std::io::Result<()>;    fn pop_directory(&mut self);} impl Stack {    /// Create a new instance with `root` being the base for all future paths we handle, assuming it to be valid which includes    /// symbolic links to be included in it as well.    pub fn new(root: impl Into<PathBuf>) -> Self {        let root = root.into();        Stack {            current: root.clone(),            current_relative: PathBuf::with_capacity(128),            valid_components: 0,            root,            current_is_directory: true,        }    }     /// Set the current stack to point to the `relative` path and call `push_comp()` each time a new path component is popped    /// along with the stacks state for inspection to perform an operation that produces some data.    ///    /// The full path to `relative` will be returned along with the data returned by push_comp.    /// Note that this only works correctly for the delegate's `push_directory()` and `pop_directory()` methods if    /// `relative` paths are terminal, so point to their designated file or directory.    pub fn make_relative_path_current(        &mut self,        relative: impl AsRef<Path>,        delegate: &mut impl Delegate,    ) -> std::io::Result<()> {        let relative = relative.as_ref();        debug_assert!(            relative.is_relative(),            "only index paths are handled correctly here, must be relative"        );        debug_assert!(!relative.to_string_lossy().is_empty(), "empty paths are not allowed");         if self.valid_components == 0 {            delegate.push_directory(self)?;        }         let mut components = relative.components().peekable();        let mut existing_components = self.current_relative.components();        let mut matching_components = 0;        while let (Some(existing_comp), Some(new_comp)) = (existing_components.next(), components.peek()) {            if existing_comp == *new_comp {                components.next();                matching_components += 1;            } else {                break;            }        }         for _ in 0..self.valid_components - matching_components {            self.current.pop();            self.current_relative.pop();            if self.current_is_directory {                delegate.pop_directory();            }            self.current_is_directory = true;        }        self.valid_components = matching_components;         if !self.current_is_directory && components.peek().is_some() {            delegate.push_directory(self)?;        }         while let Some(comp) = components.next() {            let is_last_component = components.peek().is_none();            self.current_is_directory = !is_last_component;            self.current.push(comp);            self.current_relative.push(comp);            self.valid_components += 1;            let res = delegate.push(is_last_component, self);            if self.current_is_directory {                delegate.push_directory(self)?;            }             if let Err(err) = res {                self.current.pop();                self.current_relative.pop();                self.valid_components -= 1;                return Err(err);            }        }        Ok(())    }}