rojo/src/vfs.rs

use std::borrow::Borrow;
use std::collections::HashMap;
use std::fs::{self, File};
use std::io::Read;
use std::path::{Path, PathBuf};
use std::time::Instant;

use core::Config;
use plugin::PluginChain;

/// Represents a virtual layer over multiple parts of the filesystem.
///
/// Paths in this system are represented as slices of strings, and are always
/// relative to a partition, which is an absolute path into the real filesystem.
pub struct Vfs {
    /// Contains all of the partitions mounted by the Vfs.
    ///
    /// These must be absolute paths!
    pub partitions: HashMap<String, PathBuf>,

    /// When the Vfs was initialized; used for change tracking.
    pub start_time: Instant,

    /// A chronologically-sorted list of routes that changed since the Vfs was
    /// created, along with a timestamp denoting when.
    pub change_history: Vec<VfsChange>,

    plugin_chain: &'static PluginChain,

    config: Config,
}

#[derive(Debug, Serialize, Deserialize)]
#[serde(rename_all = "camelCase")]
pub struct VfsChange {
    timestamp: f64,
    route: Vec<String>,
}

#[derive(Debug, Serialize, Deserialize)]
#[serde(rename_all = "camelCase", tag = "type")]
pub enum VfsItem {
    File { name: String, contents: String },
    Dir { name: String, children: HashMap<String, VfsItem> },
}

impl VfsItem {
    pub fn name(&self) -> &String {
        match self {
            &VfsItem::File { ref name, .. } => name,
            &VfsItem::Dir { ref name, .. } => name,
        }
    }
}

impl Vfs {
    pub fn new(config: Config, plugin_chain: &'static PluginChain) -> Vfs {
        Vfs {
            partitions: HashMap::new(),
            start_time: Instant::now(),
            change_history: Vec::new(),
            plugin_chain,
            config,
        }
    }

    fn route_to_path<R: Borrow<str>>(&self, route: &[R]) -> Option<PathBuf> {
        let (partition_name, rest) = match route.split_first() {
            Some((first, rest)) => (first.borrow(), rest),
            None => return None,
        };

        let partition = match self.partitions.get(partition_name) {
            Some(v) => v,
            None => return None,
        };

        // It's possible that the partition points to a file if `rest` is empty.
        // Joining "" onto a path will put a trailing slash on, which causes
        // file reads to fail.
        let full_path = if rest.is_empty() {
            partition.clone()
        } else {
            let joined = rest.join("/");
            let relative = Path::new(&joined);

            partition.join(relative)
        };

        Some(full_path)
    }

    fn read_dir<P: AsRef<Path>>(&self, path: P) -> Result<VfsItem, ()> {
        let path = path.as_ref();
        let reader = match fs::read_dir(path) {
            Ok(v) => v,
            Err(_) => return Err(()),
        };

        let mut children = HashMap::new();

        for entry in reader {
            let entry = match entry {
                Ok(v) => v,
                Err(_) => return Err(()),
            };

            let path = entry.path();

            match self.read_path(&path) {
                Ok(child_item) => {
                    let name = path.file_name().unwrap().to_string_lossy().into_owned();

                    children.insert(name, child_item);
                },
                Err(_) => {},
            }
        }

        Ok(VfsItem::Dir {
            name: path.file_name().unwrap().to_string_lossy().into_owned(),
            children,
        })
    }

    fn read_file<P: AsRef<Path>>(&self, path: P) -> Result<VfsItem, ()> {
        let path = path.as_ref();
        let mut file = match File::open(path) {
            Ok(v) => v,
            Err(_) => return Err(()),
        };

        let mut contents = String::new();

        match file.read_to_string(&mut contents) {
            Ok(_) => {},
            Err(_) => return Err(()),
        }

        Ok(VfsItem::File {
            name: path.file_name().unwrap().to_string_lossy().into_owned(),
            contents,
        })
    }

    fn read_path<P: AsRef<Path>>(&self, path: P) -> Result<VfsItem, ()> {
        let path = path.as_ref();

        let metadata = match fs::metadata(path) {
            Ok(v) => v,
            Err(_) => return Err(()),
        };

        if metadata.is_dir() {
            self.read_dir(path)
        } else if metadata.is_file() {
            self.read_file(path)
        } else {
            Err(())
        }
    }

    pub fn current_time(&self) -> f64 {
        let elapsed = self.start_time.elapsed();

        elapsed.as_secs() as f64 + elapsed.subsec_nanos() as f64 * 1e-9
    }

    pub fn add_change(&mut self, timestamp: f64, route: Vec<String>) {
        if self.config.verbose {
            println!("Received change {:?}, running through plugins...", route);
        }

        match self.plugin_chain.handle_file_change(&route) {
            Some(routes) => {
                for route in routes {
                    self.change_history.push(VfsChange {
                        timestamp,
                        route,
                    });
                }
            },
            None => {}
        }
    }

    pub fn changes_since(&self, timestamp: f64) -> &[VfsChange] {
        let mut marker: Option<usize> = None;

        for (index, value) in self.change_history.iter().enumerate().rev() {
            if value.timestamp >= timestamp {
                marker = Some(index);
            } else {
                break;
            }
        }

        if let Some(index) = marker {
            &self.change_history[index..]
        } else {
            &self.change_history[..0]
        }
    }

    pub fn read<R: Borrow<str>>(&self, route: &[R]) -> Result<VfsItem, ()> {
        match self.route_to_path(route) {
            Some(path) => self.read_path(&path),
            None => Err(()),
        }
    }

    pub fn write<R: Borrow<str>>(&self, _route: &[R], _item: VfsItem) -> Result<(), ()> {
        unimplemented!()
    }

    pub fn delete<R: Borrow<str>>(&self, _route: &[R]) -> Result<(), ()> {
        unimplemented!()
    }
}