2015-01-28 16:55:34 +00:00
|
|
|
use std::old_io::{fs, IoResult};
|
2015-03-22 19:46:45 +00:00
|
|
|
use std::old_path::GenericPath;
|
|
|
|
use std::old_path::posix::Path;
|
2015-01-27 15:01:17 +00:00
|
|
|
|
2015-03-26 00:37:12 +00:00
|
|
|
use feature::Git;
|
|
|
|
use file::{File, GREY};
|
2014-06-16 23:27:05 +00:00
|
|
|
|
2015-01-24 12:38:05 +00:00
|
|
|
/// A **Dir** provides a cached list of the file paths in a directory that's
|
|
|
|
/// being listed.
|
|
|
|
///
|
|
|
|
/// This object gets passed to the Files themselves, in order for them to
|
|
|
|
/// check the existence of surrounding files, then highlight themselves
|
|
|
|
/// accordingly. (See `File#get_source_files`)
|
2014-11-26 07:40:52 +00:00
|
|
|
pub struct Dir {
|
2015-01-26 17:26:11 +00:00
|
|
|
contents: Vec<Path>,
|
|
|
|
path: Path,
|
2015-01-27 15:01:17 +00:00
|
|
|
git: Option<Git>,
|
2014-06-16 23:27:05 +00:00
|
|
|
}
|
|
|
|
|
2014-11-26 07:40:52 +00:00
|
|
|
impl Dir {
|
2015-03-26 00:37:12 +00:00
|
|
|
|
2015-01-24 12:38:05 +00:00
|
|
|
/// Create a new Dir object filled with all the files in the directory
|
|
|
|
/// pointed to by the given path. Fails if the directory can't be read, or
|
|
|
|
/// isn't actually a directory.
|
2015-02-01 02:14:31 +00:00
|
|
|
pub fn readdir(path: &Path) -> IoResult<Dir> {
|
|
|
|
fs::readdir(path).map(|paths| Dir {
|
2014-06-20 20:07:53 +00:00
|
|
|
contents: paths,
|
2014-06-21 18:39:27 +00:00
|
|
|
path: path.clone(),
|
2015-02-01 02:14:31 +00:00
|
|
|
git: Git::scan(path).ok(),
|
2014-06-20 20:07:53 +00:00
|
|
|
})
|
2014-06-16 23:27:05 +00:00
|
|
|
}
|
|
|
|
|
2015-01-24 12:38:05 +00:00
|
|
|
/// Produce a vector of File objects from an initialised directory,
|
|
|
|
/// printing out an error if any of the Files fail to be created.
|
2015-02-03 13:27:23 +00:00
|
|
|
///
|
|
|
|
/// Passing in `recurse` means that any directories will be scanned for
|
|
|
|
/// their contents, as well.
|
|
|
|
pub fn files(&self, recurse: bool) -> Vec<File> {
|
2014-06-20 20:07:53 +00:00
|
|
|
let mut files = vec![];
|
2014-12-12 12:08:14 +00:00
|
|
|
|
2014-06-20 20:07:53 +00:00
|
|
|
for path in self.contents.iter() {
|
2015-02-03 13:27:23 +00:00
|
|
|
match File::from_path(path, Some(self), recurse) {
|
2014-11-24 02:12:52 +00:00
|
|
|
Ok(file) => files.push(file),
|
|
|
|
Err(e) => println!("{}: {}", path.display(), e),
|
2014-06-20 20:07:53 +00:00
|
|
|
}
|
|
|
|
}
|
2014-12-12 12:08:14 +00:00
|
|
|
|
2014-06-20 20:07:53 +00:00
|
|
|
files
|
2014-06-16 23:27:05 +00:00
|
|
|
}
|
|
|
|
|
2015-01-24 12:38:05 +00:00
|
|
|
/// Whether this directory contains a file with the given path.
|
2014-06-16 23:27:05 +00:00
|
|
|
pub fn contains(&self, path: &Path) -> bool {
|
|
|
|
self.contents.contains(path)
|
|
|
|
}
|
2015-01-26 17:26:11 +00:00
|
|
|
|
|
|
|
/// Append a path onto the path specified by this directory.
|
|
|
|
pub fn join(&self, child: Path) -> Path {
|
|
|
|
self.path.join(child)
|
|
|
|
}
|
2015-01-27 15:01:17 +00:00
|
|
|
|
|
|
|
/// Return whether there's a Git repository on or above this directory.
|
|
|
|
pub fn has_git_repo(&self) -> bool {
|
|
|
|
self.git.is_some()
|
|
|
|
}
|
|
|
|
|
|
|
|
/// Get a string describing the Git status of the given file.
|
2015-01-28 10:43:19 +00:00
|
|
|
pub fn git_status(&self, path: &Path, prefix_lookup: bool) -> String {
|
|
|
|
match (&self.git, prefix_lookup) {
|
2015-03-26 00:37:12 +00:00
|
|
|
(&Some(ref git), false) => git.status(path),
|
|
|
|
(&Some(ref git), true) => git.dir_status(path),
|
|
|
|
(&None, _) => GREY.paint("--").to_string(),
|
2015-01-27 15:01:17 +00:00
|
|
|
}
|
|
|
|
}
|
2014-06-16 23:27:05 +00:00
|
|
|
}
|